Add locality label (#61)

* add locality label

* update call

Author: dleviminzi

pkg/metrics.go

@@ -16,7 +16,7 @@ type BlobcacheMetrics struct {
 	MemCacheUsagePct  *metrics.Histogram
 }
 
-func initMetrics(ctx context.Context, config BlobCacheMetricsConfig, currentHost *BlobCacheHost) BlobcacheMetrics {
+func initMetrics(ctx context.Context, config BlobCacheMetricsConfig, currentHost *BlobCacheHost, locality string) BlobcacheMetrics {
 	username := config.Username
 	password := config.Password
 	credentials := base64.StdEncoding.EncodeToString([]byte(username + ":" + password))
@@ -25,7 +25,7 @@ func initMetrics(ctx context.Context, config BlobCacheMetricsConfig, currentHost
 		Headers: []string{
 			fmt.Sprintf("Authorization: Basic %s", credentials),
 		},
-		ExtraLabels: "host=\"" + currentHost.HostId + "\"",
+		ExtraLabels: "host=\"" + currentHost.HostId + "\",locality=\"" + locality + "\"",
 	}
 
 	pushURL := config.URL

pkg/storage.go

@@ -49,7 +49,7 @@ func NewContentAddressableStorage(ctx context.Context, currentHost *BlobCacheHos
 		locality:     locality,
 		diskCacheDir: config.Server.DiskCacheDir,
 		mu:           sync.Mutex{},
-		metrics:      initMetrics(ctx, config.Metrics, currentHost),
+		metrics:      initMetrics(ctx, config.Metrics, currentHost, locality),
 	}
 
 	Logger.Infof("Disk cache directory located at: '%s'", cas.diskCacheDir)