feat: start the configuration path specified with -c

This commit is contained in:
FENG 2024-08-03 19:28:54 +08:00
parent 0685e8c153
commit 5c98f03240
1 changed files with 11 additions and 3 deletions

14
main.go
View File

@ -5,6 +5,7 @@ import (
"context"
"encoding/json"
"errors"
"flag"
"fmt"
"github.com/gin-gonic/gin"
"github.com/tidwall/gjson"
@ -48,8 +49,8 @@ type config struct {
AuthToken string `json:"auth_token"`
}
func readConfig() *config {
content, err := os.ReadFile("config.json")
func readConfig(path string) *config {
content, err := os.ReadFile(path)
if nil != err {
log.Fatal(err)
}
@ -525,8 +526,13 @@ func constructWithChatModel(body []byte, messages interface{}) []byte {
return []byte(jsonStr)
}
var configPath string
func main() {
cfg := readConfig()
flag.StringVar(&configPath, "c", "config.json", "Path to the configuration file")
flag.Parse()
cfg := readConfig(configPath)
gin.SetMode(gin.ReleaseMode)
r := gin.Default()
@ -539,6 +545,8 @@ func main() {
proxyService.InitRoutes(r)
log.Printf("\nstart server on %s\nconfig path: %s\n", cfg.Bind, configPath)
err = r.Run(cfg.Bind)
if nil != err {
log.Fatal(err)