JCS-pub/coordinator/main.go

package main

// 主程序包，负责初始化和启动协调器服务器。
import (
	"fmt"
	"os"

	"gitlink.org.cn/cloudream/common/pkgs/logger"
	mydb "gitlink.org.cn/cloudream/storage/common/pkgs/db"
	coormq "gitlink.org.cn/cloudream/storage/common/pkgs/mq/coordinator"
	scmq "gitlink.org.cn/cloudream/storage/common/pkgs/mq/scanner"
	"gitlink.org.cn/cloudream/storage/coordinator/internal/config"
	"gitlink.org.cn/cloudream/storage/coordinator/internal/mq"
)

// 主函数，负责程序的初始化和启动。
func main() {
	// 初始化配置
	err := config.Init()
	if err != nil {
		fmt.Printf("init config failed, err: %s", err.Error())
		os.Exit(1)
	}

	// 初始化日志系统
	err = logger.Init(&config.Cfg().Logger)
	if err != nil {
		fmt.Printf("init logger failed, err: %s", err.Error())
		os.Exit(1)
	}

	// 初始化数据库连接
	db, err := mydb.NewDB(&config.Cfg().DB)
	if err != nil {
		logger.Fatalf("new db failed, err: %s", err.Error())
	}

	// 初始化扫描器客户端
	scanner, err := scmq.NewClient(&config.Cfg().RabbitMQ)
	if err != nil {
		logger.Fatalf("new scanner client failed, err: %s", err.Error())
	}

	// 初始化协调器服务器
	coorSvr, err := coormq.NewServer(mq.NewService(db, scanner), &config.Cfg().RabbitMQ)
	if err != nil {
		logger.Fatalf("new coordinator server failed, err: %s", err.Error())
	}

	// 设置协调器服务器错误处理
	coorSvr.OnError(func(err error) {
		logger.Warnf("coordinator server err: %s", err.Error())
	})

	// 启动协调器服务器为异步操作
	go serveCoorServer(coorSvr)

	// 永久等待，保持程序运行
	forever := make(chan bool)
	<-forever
}

// serveCoorServer 启动并运行协调器服务器。
func serveCoorServer(server *coormq.Server) {
	logger.Info("start serving command server")

	// 服务启动和错误处理
	err := server.Serve()
	if err != nil {
		logger.Errorf("command server stopped with error: %s", err.Error())
	}

	logger.Info("command server stopped")
}