goaidb/main.go

package main

import (
	"fmt"
	"net"
	"os"
	"os/signal"
	"path/filepath"

	"syscall"

	"git.pyer.club/kingecg/goaidb/config"
	"git.pyer.club/kingecg/goaidb/log"
	"git.pyer.club/kingecg/goaidb/network"
	"git.pyer.club/kingecg/goaidb/storage"
)

func getExeDir() string {
	exePath, err := os.Executable()
	if err != nil {
		panic(err)
	}
	exeDir := filepath.Dir(exePath)
	return exeDir
}

// 主程序入口
func main() {
	// 加载配置文件（优先级：命令行参数 > 环境变量 > 默认配置）
	configPath := getExeDir() + "/config.yaml"
	if len(os.Args) > 1 {
		configPath = os.Args[1]
	} else if envPath := os.Getenv("GOAIDB_CONFIG"); envPath != "" {
		configPath = envPath
	}

	// 解析配置文件
	cfg, err := config.ParseConfig(configPath)
	if err != nil {
		// fmt.Fprintf(os.Stderr, "配置加载失败: %v\n", err)
		log.Error("配置加载失败", err)
		os.Exit(1)
	}

	// 设置全局配置
	config.SetGlobalConfig(cfg)

	// 启动配置文件监视
	err = config.WatchConfig(configPath)
	if err != nil {
		fmt.Fprintf(os.Stderr, "配置监控启动失败: %v\n", err)
		log.Error("配置监控启动失败", err)
		os.Exit(1)
	}

	// 设置信号处理（优雅关闭）
	sigChan := make(chan os.Signal, 1)
	signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)

	go func() {
		<-sigChan
		log.Info("正在关闭...")
		config.CloseWatcher()
		os.Exit(0)
	}()

	// 初始化存储引擎（默认使用内存引擎）
	storageEngine, err := storage.NewMemoryEngine()
	if err != nil {
		log.Error("Failed to initialize storage engine: %v\n", err)
		return
	}

	// 创建网络服务器
	server := network.NewServer(storageEngine)
	conf := config.GetConfig()
	// 启动服务
	listener, err := net.Listen("tcp", conf.Server.Host+":"+fmt.Sprintf("%d", conf.Server.Port))
	if err != nil {
		log.Error("Failed to start server: %v\n", err)
		return
	}

	log.Info("GoAIDB started on port", conf.Server.Port)
	server.Serve(listener)
}