hldrCenter/server/main.go

package main

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"

	"github.com/JACKYMYPERSON/hldrCenter/config"
	"github.com/JACKYMYPERSON/hldrCenter/init/database/cache"
	"github.com/JACKYMYPERSON/hldrCenter/middleware/cors"
	"github.com/JACKYMYPERSON/hldrCenter/router"
)

func main() {
	// 目标：开发环境（go run）找项目根目录的 config，生产环境（打包后）找可执行文件同级的 config
	var configPath string

	// 1. 先获取当前程序的路径（开发时是临时路径，生产时是打包后的路径）
	exePath, err := os.Executable()
	if err != nil {
		fmt.Printf("获取程序路径失败：%v\n", err)
		return
	}
	exeDir := filepath.Dir(exePath)

	// 2. 开发环境兼容：判断是否在 GoLand 的临时目录（或项目根目录）
	// 逻辑：如果 exeDir 包含 "tmp"（临时目录特征），则去上级目录找项目根的 config
	// 可根据你的项目结构调整判断逻辑（比如项目名包含 "myproject"，也可以判断 strings.Contains(exeDir, "myproject")）
	if strings.Contains(strings.ToLower(exeDir), "tmp") {
		// 开发环境：从临时目录向上回溯，找到项目根目录（根据实际项目层级调整 ../ 的数量）
		// 示例：临时目录 → 项目根目录（假设临时目录在项目根下的 tmp 子目录，需回溯1级）
		projectRoot := filepath.Join(exeDir, "..") // 若回溯不够，可改成 "../.."，直到找到项目根
		configPath = filepath.Join(projectRoot, "config", "config.yaml")

		// 验证：如果没找到，再尝试用当前工作目录（go run 的工作目录默认是项目根）
		if _, err := os.Stat(configPath); err != nil {
			configPath = filepath.Join(".", "config", "config.yaml")
		}
	} else {
		// 生产环境（打包后）：用可执行文件同级的 config 文件夹
		configPath = filepath.Join(exeDir, "config", "config.yaml")
	}

	// 3. 验证路径是否存在（可选，方便调试）
	if _, err := os.Stat(configPath); err != nil {
		fmt.Printf("配置文件不存在：%s\n", configPath)
		return
	}
	cfg, err := config.LoadConfig(configPath)
	if err != nil {
		fmt.Printf("加载配置失败：%v\n", err)
		return
	}

	cache.InitCache()
	defer func() {
		err := cache.CloseCache()
		if err != nil {
			return
		}
	}()

	// 设置路由
	r := router.SetupRouter(cfg)

	// 应用跨域中间件
	r.Use(cors.CorsMiddleware(&cfg.Server))

	// 启动服务
	addr := fmt.Sprintf(":%s", cfg.Server.Port)
	fmt.Printf("后端服务启动成功，地址：http://localhost%s\n", addr)
	if err := r.Run(addr); err != nil {
		fmt.Printf("服务启动失败：%v\n", err)
	}
}