main.go

package main

import (
	"github.com/gin-contrib/cors"
	"github.com/gin-gonic/gin"
	"log"
	"simple-one-api/pkg/mylog"

	//"log"
	"os"
	"simple-one-api/pkg/config"
	"simple-one-api/pkg/handler"
	"time"
)

func main() {
	log.SetFlags(log.LstdFlags | log.Lshortfile)
	log.Println("123")

	// 获取程序的第一个参数作为配置文件名
	var configName string
	if len(os.Args) > 1 {
		configName = os.Args[1]
	} else {
		configName = "config.json"
	}

	var err error
	// 初始化配置
	err = config.InitConfig(configName)
	if err != nil {
		mylog.Logger.Error(err.Error())
		mylog.Logger.Error("exit")
		return
	}

	if config.Debug == false {
		gin.SetMode(gin.ReleaseMode)
	}

	mylog.InitLog(config.LogLevel)
	defer mylog.Logger.Sync()

	// 创建一个 Gin 路由器实例
	r := gin.Default()

	// 配置 CORS 中间件
	r.Use(cors.New(cors.Config{
		AllowOrigins:     []string{"*"}, // 允许所有来源，如果需要限制来源，可以将 "*" 替换为具体的 URL
		AllowMethods:     []string{"GET", "POST", "PUT", "PATCH", "DELETE", "OPTIONS"},
		AllowHeaders:     []string{"Origin", "Content-Type", "Accept", "Authorization"},
		ExposeHeaders:    []string{"Content-Length"},
		AllowCredentials: true,
		MaxAge:           12 * time.Hour,
	}))

	// 添加POST请求方法处理
	r.POST("/v1/chat/completions", handler.OpenAIHandler)

	// 启动服务器，使用配置中的端口
	err = r.Run(config.ServerPort)
	if err != nil {
		mylog.Logger.Error(err.Error())
		return
	} // 使用配置文件中指定的端口号
}