mirror-seaweedfs/weed/admin/maintenance/config_verification.go

package maintenance

import (
	"fmt"

	"github.com/seaweedfs/seaweedfs/weed/pb/worker_pb"
)

// VerifyProtobufConfig demonstrates that the protobuf configuration system is working
func VerifyProtobufConfig() error {
	// Create configuration manager
	configManager := NewMaintenanceConfigManager()
	config := configManager.GetConfig()

	// Verify basic configuration
	if !config.Enabled {
		return fmt.Errorf("expected config to be enabled by default")
	}

	if config.ScanIntervalSeconds != 30*60 {
		return fmt.Errorf("expected scan interval to be 1800 seconds, got %d", config.ScanIntervalSeconds)
	}

	// Verify policy configuration
	if config.Policy == nil {
		return fmt.Errorf("expected policy to be configured")
	}

	if config.Policy.GlobalMaxConcurrent != 4 {
		return fmt.Errorf("expected global max concurrent to be 4, got %d", config.Policy.GlobalMaxConcurrent)
	}

	// Verify task policies
	vacuumPolicy := config.Policy.TaskPolicies["vacuum"]
	if vacuumPolicy == nil {
		return fmt.Errorf("expected vacuum policy to be configured")
	}

	if !vacuumPolicy.Enabled {
		return fmt.Errorf("expected vacuum policy to be enabled")
	}

	// Verify typed configuration access
	vacuumConfig := vacuumPolicy.GetVacuumConfig()
	if vacuumConfig == nil {
		return fmt.Errorf("expected vacuum config to be accessible")
	}

	if vacuumConfig.GarbageThreshold != 0.3 {
		return fmt.Errorf("expected garbage threshold to be 0.3, got %f", vacuumConfig.GarbageThreshold)
	}

	// Verify helper functions work
	if !IsTaskEnabled(config.Policy, "vacuum") {
		return fmt.Errorf("expected vacuum task to be enabled via helper function")
	}

	maxConcurrent := GetMaxConcurrent(config.Policy, "vacuum")
	if maxConcurrent != 2 {
		return fmt.Errorf("expected vacuum max concurrent to be 2, got %d", maxConcurrent)
	}

	// Verify erasure coding configuration
	ecPolicy := config.Policy.TaskPolicies["erasure_coding"]
	if ecPolicy == nil {
		return fmt.Errorf("expected EC policy to be configured")
	}

	ecConfig := ecPolicy.GetErasureCodingConfig()
	if ecConfig == nil {
		return fmt.Errorf("expected EC config to be accessible")
	}

	// Verify configurable EC fields only
	if ecConfig.FullnessRatio <= 0 || ecConfig.FullnessRatio > 1 {
		return fmt.Errorf("expected EC config to have valid fullness ratio (0-1), got %f", ecConfig.FullnessRatio)
	}

	return nil
}

// GetProtobufConfigSummary returns a summary of the current protobuf configuration
func GetProtobufConfigSummary() string {
	configManager := NewMaintenanceConfigManager()
	config := configManager.GetConfig()

	summary := fmt.Sprintf("SeaweedFS Protobuf Maintenance Configuration:\n")
	summary += fmt.Sprintf("  Enabled: %v\n", config.Enabled)
	summary += fmt.Sprintf("  Scan Interval: %d seconds\n", config.ScanIntervalSeconds)
	summary += fmt.Sprintf("  Max Retries: %d\n", config.MaxRetries)
	summary += fmt.Sprintf("  Global Max Concurrent: %d\n", config.Policy.GlobalMaxConcurrent)
	summary += fmt.Sprintf("  Task Policies: %d configured\n", len(config.Policy.TaskPolicies))

	for taskType, policy := range config.Policy.TaskPolicies {
		summary += fmt.Sprintf("    %s: enabled=%v, max_concurrent=%d\n",
			taskType, policy.Enabled, policy.MaxConcurrent)
	}

	return summary
}

// CreateCustomConfig demonstrates creating a custom protobuf configuration
func CreateCustomConfig() *worker_pb.MaintenanceConfig {
	return &worker_pb.MaintenanceConfig{
		Enabled:             true,
		ScanIntervalSeconds: 60 * 60, // 1 hour
		MaxRetries:          5,
		Policy: &worker_pb.MaintenancePolicy{
			GlobalMaxConcurrent: 8,
			TaskPolicies: map[string]*worker_pb.TaskPolicy{
				"custom_vacuum": {
					Enabled:       true,
					MaxConcurrent: 4,
					TaskConfig: &worker_pb.TaskPolicy_VacuumConfig{
						VacuumConfig: &worker_pb.VacuumTaskConfig{
							GarbageThreshold:  0.5,
							MinVolumeAgeHours: 48,
						},
					},
				},
			},
		},
	}
}