Use config

Author: sjmiller609

cmd/api/api/api_test.go

@@ -25,10 +25,10 @@ import (
 // newTestService creates an ApiService for testing with automatic cleanup
 func newTestService(t *testing.T) *ApiService {
 	cfg := &config.Config{
-		DataDir:     t.TempDir(),
-		BridgeName:  "vmbr0",
-		SubnetCIDR:  "10.100.0.0/16",
-		DNSServer:   "1.1.1.1",
+		DataDir:    t.TempDir(),
+		BridgeName: "vmbr0",
+		SubnetCIDR: "10.100.0.0/16",
+		DNSServer:  "1.1.1.1",
 	}
 
 	p := paths.New(cfg.DataDir)

cmd/api/config/config.go

@@ -201,8 +201,8 @@ func Load() *Config {
 		CloudflareApiToken: getEnv("CLOUDFLARE_API_TOKEN", ""),
 
 		// API ingress configuration
-		ApiHostname:     getEnv("API_HOSTNAME", ""),     // Empty = disabled
-		ApiTLS:          getEnvBool("API_TLS", true),    // Default to TLS enabled
+		ApiHostname:     getEnv("API_HOSTNAME", ""),  // Empty = disabled
+		ApiTLS:          getEnvBool("API_TLS", true), // Default to TLS enabled
 		ApiRedirectHTTP: getEnvBool("API_REDIRECT_HTTP", true),
 
 		// Build system configuration

cmd/api/main.go

@@ -18,7 +18,6 @@ import (
 	"github.com/ghodss/yaml"
 	"github.com/go-chi/chi/v5"
 	"github.com/go-chi/chi/v5/middleware"
-	nethttpmiddleware "github.com/oapi-codegen/nethttp-middleware"
 	"github.com/kernel/hypeman"
 	"github.com/kernel/hypeman/cmd/api/api"
 	"github.com/kernel/hypeman/cmd/api/config"
@@ -30,6 +29,7 @@ import (
 	"github.com/kernel/hypeman/lib/oapi"
 	"github.com/kernel/hypeman/lib/otel"
 	"github.com/kernel/hypeman/lib/vmm"
+	nethttpmiddleware "github.com/oapi-codegen/nethttp-middleware"
 	"github.com/riandyrn/otelchi"
 	"golang.org/x/sync/errgroup"
 )
@@ -51,6 +51,9 @@ func run() error {
 		return fmt.Errorf("invalid configuration: %w", err)
 	}
 
+	// Configure GPU profile cache TTL
+	devices.SetGPUProfileCacheTTL(cfg.GPUProfileCacheTTL)
+
 	// Initialize OpenTelemetry (before wire initialization)
 	otelCfg := otel.Config{
 		Enabled:           cfg.OtelEnabled,

lib/devices/mdev.go

@@ -35,22 +35,27 @@ type profileMetadata struct {
 }
 
 // cachedProfiles holds profile metadata with TTL-based expiry.
-// The cache TTL is configurable via GPU_PROFILE_CACHE_TTL environment variable.
 var (
 	cachedProfiles     []profileMetadata
 	cachedProfilesMu   sync.RWMutex
 	cachedProfilesTime time.Time
+	gpuProfileCacheTTL time.Duration = 30 * time.Minute // default
 )
 
-// getProfileCacheTTL returns the TTL for profile metadata cache.
-// Reads from GPU_PROFILE_CACHE_TTL env var, defaults to 30 minutes.
-func getProfileCacheTTL() time.Duration {
-	if ttl := os.Getenv("GPU_PROFILE_CACHE_TTL"); ttl != "" {
-		if d, err := time.ParseDuration(ttl); err == nil {
-			return d
-		}
+// SetGPUProfileCacheTTL sets the TTL for GPU profile metadata cache.
+// Should be called during application startup with the config value.
+func SetGPUProfileCacheTTL(ttl string) {
+	if ttl == "" {
+		return
+	}
+	if d, err := time.ParseDuration(ttl); err == nil {
+		gpuProfileCacheTTL = d
 	}
-	return 30 * time.Minute
+}
+
+// getProfileCacheTTL returns the configured TTL for profile metadata cache.
+func getProfileCacheTTL() time.Duration {
+	return gpuProfileCacheTTL
 }
 
 // getCachedProfiles returns cached profile metadata, refreshing if TTL has expired.