containerized-data-importer/tools/cdi-source-update-poller/main.go
Arnon Gilboa 50467a68be
Get rid of unnecessary DIC reconcile updates (#2294)
* Get rid of unnecessary DIC reconcile updates

Also fixed status.lastExecutionTimestamp to be the last polling time
as intended in the design, and not the last reconcile update time.

Signed-off-by: Arnon Gilboa <agilboa@redhat.com>

* Pass DIC last execution time by annotation

Signed-off-by: Arnon Gilboa <agilboa@redhat.com>
2022-05-27 15:19:38 +03:00

86 lines
2.9 KiB
Go

package main
import (
"context"
"flag"
"fmt"
"log"
"os"
"strconv"
"time"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/client-go/tools/clientcmd"
cdiClientset "kubevirt.io/containerized-data-importer/pkg/client/clientset/versioned"
"kubevirt.io/containerized-data-importer/pkg/common"
"kubevirt.io/containerized-data-importer/pkg/controller"
"kubevirt.io/containerized-data-importer/pkg/importer"
"kubevirt.io/containerized-data-importer/pkg/util"
)
var (
configPath string
kubeURL string
cronNamespace string
cronName string
url string
certDir string
accessKey string
secretKey string
insecureTLS bool
)
func init() {
flag.StringVar(&configPath, "kubeconfig", os.Getenv("KUBECONFIG"), "(Optional) Overrides $KUBECONFIG.")
flag.StringVar(&kubeURL, "server", "", "(Optional) URL address of a remote api server. Do not set for local clusters.")
flag.StringVar(&cronNamespace, "ns", "", "DataImportCron namespace.")
flag.StringVar(&cronName, "cron", "", "DataImportCron name.")
flag.StringVar(&url, "url", "", "registry source url.")
flag.StringVar(&certDir, "certdir", "", "registry certificates path.")
flag.Parse()
if url == "" || cronNamespace == "" || cronName == "" {
log.Fatalf("One or more mandatory parameters are missing")
}
accessKey, _ = util.ParseEnvVar(common.ImporterAccessKeyID, false)
secretKey, _ = util.ParseEnvVar(common.ImporterSecretKey, false)
insecureTLS, _ = strconv.ParseBool(os.Getenv(common.InsecureTLSVar))
}
func main() {
digest, err := importer.GetImageDigest(url, accessKey, secretKey, certDir, insecureTLS)
if err != nil {
os.Exit(1)
}
fmt.Println("Digest is", digest)
cfg, err := clientcmd.BuildConfigFromFlags(kubeURL, configPath)
if err != nil {
log.Fatalf("Failed BuildConfigFromFlags, kubeURL %s configPath %s: %v", kubeURL, configPath, err)
}
cdiClient, err := cdiClientset.NewForConfig(cfg)
if err != nil {
log.Fatalf("Failed NewForConfig: %v", err)
}
dataImportCron, err := cdiClient.CdiV1beta1().DataImportCrons(cronNamespace).Get(context.TODO(), cronName, metav1.GetOptions{})
if err != nil {
log.Fatalf("Failed getting DataImportCron %s/%s: %v", cronNamespace, cronName, err)
}
controller.AddAnnotation(dataImportCron, controller.AnnLastCronTime, time.Now().Format(time.RFC3339))
imports := dataImportCron.Status.CurrentImports
if digest != "" && (imports == nil || digest != imports[0].Digest) &&
digest != dataImportCron.Annotations[controller.AnnSourceDesiredDigest] {
controller.AddAnnotation(dataImportCron, controller.AnnSourceDesiredDigest, digest)
fmt.Println("Digest updated")
} else {
fmt.Println("No digest update")
}
_, err = cdiClient.CdiV1beta1().DataImportCrons(cronNamespace).Update(context.TODO(), dataImportCron, metav1.UpdateOptions{})
if err != nil {
log.Fatalf("Failed updating DataImportCron %s/%s: %v", cronNamespace, cronName, err)
}
}