Method setOverridesToBucket

pkg/overrides/api.go:204–259

setOverridesToBucket writes overrides for a specific tenant to the runtime config file

(ctx context.Context, userID string, overrides map[string]any)

// setOverridesToBucket writes overrides for a specific tenant to the runtime config file
func (a *API) setOverridesToBucket(ctx context.Context, userID string, overrides map[string]any) error {
	var config runtimeconfig.RuntimeConfigValues
	reader, err := a.bucketClient.Get(ctx, a.runtimeConfigPath)
	if err != nil {
		return fmt.Errorf("failed to get runtime config: %w", err)
	}
	defer func() {
		if reader != nil {
			reader.Close()
		}
	}()
	if err := yaml.NewDecoder(reader).Decode(&config); err != nil {
		return fmt.Errorf("%s: %w", ErrRuntimeConfig, err)
	}

	if config.TenantLimits == nil {
		config.TenantLimits = make(map[string]*validation.Limits)
	}

	// Get existing limits for the user
	var existingLimitsMap map[string]any
	if existingLimits, exists := config.TenantLimits[userID]; exists && existingLimits != nil {
		// Convert existing limits to map
		yamlData, err := yaml.Marshal(existingLimits)
		if err != nil {
			return fmt.Errorf("failed to marshal existing limits: %w", err)
		}

		if err := yaml.Unmarshal(yamlData, &existingLimitsMap); err != nil {
			return fmt.Errorf("failed to unmarshal existing limits: %w", err)
		}
	}

	// Merge existing limits with new overrides
	mergedLimits := mergeLimits(existingLimitsMap, overrides)

	// Convert merged limits back to validation.Limits
	yamlData, err := yaml.Marshal(mergedLimits)
	if err != nil {
		return fmt.Errorf("failed to marshal overrides: %w", err)
	}

	var limits validation.Limits
	if err := yaml.Unmarshal(yamlData, &limits); err != nil {
		return fmt.Errorf("invalid overrides format: %w", err)
	}

	config.TenantLimits[userID] = &limits

	data, err := yaml.Marshal(config)
	if err != nil {
		return fmt.Errorf("%s: %w", ErrRuntimeConfig, err)
	}

	return a.bucketClient.Upload(ctx, a.runtimeConfigPath, bytes.NewReader(data))
}

// deleteOverridesFromBucket removes overrides for a specific tenant from the runtime config file