always add host to index if keepalive is running

luke-lombardi · luke-lombardi · commit c25248d6adef · 2024-12-29T16:11:37.000-05:00
diff --git a/pkg/metadata.go b/pkg/metadata.go
@@ -341,6 +341,11 @@ func (m *BlobCacheMetadata) SetHostKeepAlive(ctx context.Context, host *BlobCach
 		return err
 	}
 
+	err = m.AddHostToIndex(ctx, host)
+	if err != nil {
+		return err
+	}
+
 	return m.rdb.Set(ctx, MetadataKeys.MetadataHostKeepAlive(host.Addr), hostBytes, time.Duration(defaultHostKeepAliveTimeoutS)*time.Second).Err()
 }
 
diff --git a/pkg/server.go b/pkg/server.go
@@ -101,7 +101,7 @@ func NewCacheService(ctx context.Context, cfg BlobCacheConfig) (*CacheService, e
 func (cs *CacheService) HostKeepAlive() {
 	err := cs.metadata.AddHostToIndex(cs.ctx, cs.cas.currentHost)
 	if err != nil {
-		Logger.Fatalf("Failed to add host to index? %v", err)
+		Logger.Fatalf("Failed to add host to index: %v", err)
 	}
 
 	ticker := time.NewTicker(time.Duration(defaultHostKeepAliveIntervalS) * time.Second)

Original file line number	Diff line number	Diff line change
`@@ -341,6 +341,11 @@ func (m BlobCacheMetadata) SetHostKeepAlive(ctx context.Context, host BlobCach`
`341`	`341`	`return err`
`342`	`342`	`}`
`343`	`343`
	`344`	`+ err = m.AddHostToIndex(ctx, host)`
	`345`	`+ if err != nil {`
	`346`	`+ return err`
	`347`	`+ }`
	`348`	`+`
`344`	`349`	`return m.rdb.Set(ctx, MetadataKeys.MetadataHostKeepAlive(host.Addr), hostBytes, time.Duration(defaultHostKeepAliveTimeoutS)*time.Second).Err()`
`345`	`350`	`}`
`346`	`351`
Original file line number	Diff line number	Diff line change
`@@ -101,7 +101,7 @@ func NewCacheService(ctx context.Context, cfg BlobCacheConfig) (*CacheService, e`
`101`	`101`	`func (cs *CacheService) HostKeepAlive() {`
`102`	`102`	`err := cs.metadata.AddHostToIndex(cs.ctx, cs.cas.currentHost)`
`103`	`103`	`if err != nil {`
`104`		`- Logger.Fatalf("Failed to add host to index? %v", err)`
	`104`	`+ Logger.Fatalf("Failed to add host to index: %v", err)`
`105`	`105`	`}`
`106`	`106`
`107`	`107`	`ticker := time.NewTicker(time.Duration(defaultHostKeepAliveIntervalS) * time.Second)`