Merge pull request #90 from JaD1ng/feat/issue80

feat(prometheus_adapter): 新增prometheus_adapter实现指标查询功能

Author: Erickw87

cmd/prometheus_adapter/main.go

@@ -0,0 +1,91 @@
+package main
+
+import (
+	"context"
+	"os"
+	"os/signal"
+	"syscall"
+	"time"
+
+	"github.com/fox-gonic/fox"
+	"github.com/qiniu/zeroops/internal/config"
+	prometheusadapter "github.com/qiniu/zeroops/internal/prometheus_adapter"
+	"github.com/rs/zerolog"
+	"github.com/rs/zerolog/log"
+)
+
+func main() {
+	// 配置日志
+	log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr})
+
+	log.Info().Msg("Starting Prometheus Adapter server")
+
+	// 加载 Prometheus Adapter 配置
+	adapter, err := prometheusadapter.NewPrometheusAdapterServer(&config.Config{})
+	if err != nil {
+		log.Fatal().Err(err).Msg("Failed to create Prometheus Adapter server")
+	}
+
+	// 获取 Prometheus Adapter 内部配置的绑定地址
+	bindAddr := ":9999" // 默认端口
+	if adapter.GetBindAddr() != "" {
+		bindAddr = adapter.GetBindAddr()
+	}
+
+	// 如果有环境变量，优先使用环境变量的端口
+	if port := os.Getenv("ADAPTER_PORT"); port != "" {
+		bindAddr = ":" + port
+	}
+
+	// 更新配置（虽然已经创建了 adapter，但需要端口信息用于启动服务器）
+	cfg := &config.Config{
+		Server: config.ServerConfig{
+			BindAddr: bindAddr,
+		},
+	}
+
+	// 创建路由
+	router := fox.New()
+
+	// 启动 API
+	if err := adapter.UseApi(router); err != nil {
+		log.Fatal().Err(err).Msg("Failed to setup API routes")
+	}
+
+	// 设置信号处理
+	sigChan := make(chan os.Signal, 1)
+	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
+
+	// 创建一个用于优雅关闭的context
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	// 在goroutine中启动服务器
+	serverErr := make(chan error, 1)
+	go func() {
+		log.Info().Msgf("Starting Prometheus Adapter on %s", cfg.Server.BindAddr)
+		if err := router.Run(cfg.Server.BindAddr); err != nil {
+			serverErr <- err
+		}
+	}()
+
+	// 等待信号或服务器错误
+	select {
+	case sig := <-sigChan:
+		log.Info().Msgf("Received signal %s, shutting down...", sig)
+
+		// 创建超时context
+		shutdownCtx, shutdownCancel := context.WithTimeout(ctx, 10*time.Second)
+		defer shutdownCancel()
+
+		// 调用adapter的Shutdown方法
+		if err := adapter.Close(shutdownCtx); err != nil {
+			log.Error().Err(err).Msg("Error during shutdown")
+		}
+
+		log.Info().Msg("Shutdown complete")
+
+	case err := <-serverErr:
+		log.Fatal().Err(err).Msg("Server error")
+	}
+}