Update cmd/jaeger/internal/integration/storagecleaner/extension.go

Co-authored-by: Yuri Shkuro <yurishkuro@users.noreply.github.com> Signed-off-by: Harshvir Potpose <122517264+akagami-harsh@users.noreply.github.com>
akagami-harsh · Apr 22, 2024 · 4ef7ad8 · 4ef7ad8
1 parent f35a6a4
commit 4ef7ad8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cmd/jaeger/internal/integration/storagecleaner/extension.go b/cmd/jaeger/internal/integration/storagecleaner/extension.go
@@ -45,7 +45,10 @@ func (c *storageCleaner) Start(ctx context.Context, host component.Host) error {
 	}
 
 	purgeStorage := func() error {
-		purger := storageFactory.(storage.Purger)
+		purger, ok := storageFactory.(storage.Purger)
+		if !ok {
+		  return fmt.Errorf("storage %s does not implement Purger interface", c.config.TraceStorage)
+		}
 		if err := purger.Purge(); err != nil {
 			return fmt.Errorf("error purging Badger storage: %w", err)
 		}