wip v1.0.0

internal/health/factory.go

@@ -0,0 +1,56 @@
+package health
+
+import (
+	"fmt"
+
+	"github.com/loveuer/go-alived/pkg/config"
+	"github.com/loveuer/go-alived/pkg/logger"
+)
+
+func CreateChecker(cfg *config.HealthChecker) (Checker, error) {
+	configMap, ok := cfg.Config.(map[string]interface{})
+	if !ok {
+		return nil, fmt.Errorf("invalid config for checker %s", cfg.Name)
+	}
+
+	switch cfg.Type {
+	case "tcp":
+		return NewTCPChecker(cfg.Name, configMap)
+	case "http", "https":
+		return NewHTTPChecker(cfg.Name, configMap)
+	case "ping", "icmp":
+		return NewPingChecker(cfg.Name, configMap)
+	case "script":
+		return NewScriptChecker(cfg.Name, configMap)
+	default:
+		return nil, fmt.Errorf("unsupported checker type: %s", cfg.Type)
+	}
+}
+
+func LoadFromConfig(cfg *config.Config, log *logger.Logger) (*Manager, error) {
+	manager := NewManager(log)
+
+	for _, healthCfg := range cfg.Health {
+		checker, err := CreateChecker(&healthCfg)
+		if err != nil {
+			return nil, fmt.Errorf("failed to create checker %s: %w", healthCfg.Name, err)
+		}
+
+		monitorCfg := &CheckerConfig{
+			Name:     healthCfg.Name,
+			Type:     healthCfg.Type,
+			Interval: healthCfg.Interval,
+			Timeout:  healthCfg.Timeout,
+			Rise:     healthCfg.Rise,
+			Fall:     healthCfg.Fall,
+			Config:   healthCfg.Config.(map[string]interface{}),
+		}
+
+		monitor := NewMonitor(checker, monitorCfg, log)
+		manager.AddMonitor(monitor)
+
+		log.Info("loaded health checker: %s (type=%s)", healthCfg.Name, healthCfg.Type)
+	}
+
+	return manager, nil
+}