add config enabled boolean

beam-cloud · Jun 17, 2024 · 58e70c1 · 58e70c1
1 parent a78c68c
commit 58e70c1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/pkg/types/config.go b/pkg/types/config.go
@@ -93,6 +93,7 @@ type GatewayServiceConfig struct {
 type ImageServiceConfig struct {
 	CacheURL                       string                    `key:"cacheURL" json:"cache_url"`
 	CacheConfig                    blobcache.BlobCacheConfig `key:"cacheConfig" json:"cache_config"`
+	CachedEnabled                  bool                      `key:"cachedEnabled" json:"cache_enabled"`
 	RegistryStore                  string                    `key:"registryStore" json:"registry_store"`
 	RegistryCredentialProviderName string                    `key:"registryCredentialProvider" json:"registry_credential_provider_name"`
 	Registries                     ImageRegistriesConfig     `key:"registries" json:"registries"`

diff --git a/pkg/worker/image.go b/pkg/worker/image.go
@@ -85,9 +85,13 @@ func NewImageClient(config types.ImageServiceConfig, workerId string, workerRepo
 		return nil, err
 	}
 
-	client, err := blobcache.NewBlobCacheClient(context.TODO(), config.CacheConfig)
-	if err != nil {
-		return nil, err
+	var client *blobcache.BlobCacheClient = nil
+
+	if config.CachedEnabled {
+		client, err = blobcache.NewBlobCacheClient(context.TODO(), config.CacheConfig)
+		if err != nil {
+			return nil, err
+		}
 	}
 
 	c := &ImageClient{