Remove InternalCCMSFlag

internal/memorystore/api.go

@@ -7,6 +7,7 @@ package memorystore
 
 import (
 	"errors"
+	"fmt"
 	"math"
 
 	"github.com/ClusterCockpit/cc-lib/schema"
@@ -124,6 +125,10 @@ func FetchData(req APIQueryRequest) (*APIQueryResponse, error) {
 
 	req.WithData = true
 	ms := GetMemoryStore()
+	if ms == nil {
+		return nil, fmt.Errorf("memorystore not initialized")
+	}
+
 
 	response := APIQueryResponse{
 		Results: make([][]APIMetricData, 0, len(req.Queries)),

internal/memorystore/config.go

@@ -19,8 +19,6 @@ const (
 	DefaultAvroCheckpointInterval = time.Minute
 )
 
-var InternalCCMSFlag bool = false
-
 type MetricStoreConfig struct {
 	// Number of concurrent workers for checkpoint and archive operations.
 	// If not set or 0, defaults to min(runtime.NumCPU()/2+1, 10)

internal/memorystore/memorystore.go

@@ -177,13 +177,19 @@ func InitMetrics(metrics map[string]MetricConfig) {
 
 func GetMemoryStore() *MemoryStore {
 	if msInstance == nil {
-		cclog.Fatalf("[METRICSTORE]> MemoryStore not initialized!")
+		return nil
 	}
 
 	return msInstance
 }
 
 func Shutdown() {
+	// Check if memorystore was initialized
+	if msInstance == nil {
+		cclog.Debug("[METRICSTORE]> MemoryStore not initialized, skipping shutdown")
+		return
+	}
+
 	// Cancel the context to signal all background goroutines to stop
 	if shutdownFunc != nil {
 		shutdownFunc()