mirror of
https://github.com/kubevirt/containerized-data-importer.git
synced 2025-06-03 06:30:22 +00:00

* 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>
86 lines
2.9 KiB
Go
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)
|
|
}
|
|
}
|