75 lines
1.9 KiB
Go
75 lines
1.9 KiB
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")
|
|
}
|