update config

Author: idoubi

cmd/api.go

@@ -1,12 +1,10 @@
 package cmd
 
 import (
-	"fmt"
 	"log"
 
 	"github.com/chatmcp/mcprouter/router"
 	"github.com/chatmcp/mcprouter/service/api"
-	"github.com/chatmcp/mcprouter/util"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
 )
@@ -27,29 +25,11 @@ var apiCmd = &cobra.Command{
 	Short: "start api server",
 	Long:  `start api server`,
 	Run: func(cmd *cobra.Command, args []string) {
-		if err := util.InitConfigWithFile(apiConfigFile); err != nil {
-			fmt.Printf("init config failed with file: %s, %v\n", apiConfigFile, err)
+		if err := Init(); err != nil {
+			log.Printf("init failed with error: %v", err)
 			return
 		}
 
-		log.Println("config initialized")
-
-		if viper.GetBool("app.use_db") && viper.GetString("app.db_name") != "" {
-			if err := util.InitDBWithName(viper.GetString("app.db_name")); err != nil {
-				fmt.Printf("init db failed with name: %s, %v\n", viper.GetString("app.db_name"), err)
-				return
-			}
-			log.Println("db initialized")
-		}
-
-		if viper.GetBool("app.use_cache") && viper.GetString("app.cache_name") == "redis" {
-			if err := util.InitRedisWithName(viper.GetString("app.cache_name")); err != nil {
-				fmt.Printf("init redis failed with name: %s, %v\n", viper.GetString("app.cache_name"), err)
-				return
-			}
-			log.Println("redis initialized")
-		}
-
 		port := viper.GetInt("api_server.port")
 		if port == 0 {
 			port = 8027

cmd/proxy.go

@@ -1,12 +1,10 @@
 package cmd
 
 import (
-	"fmt"
 	"log"
 
 	"github.com/chatmcp/mcprouter/router"
 	"github.com/chatmcp/mcprouter/service/proxy"
-	"github.com/chatmcp/mcprouter/util"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
 )
@@ -27,29 +25,11 @@ var proxyCmd = &cobra.Command{
 	Short: "start proxy server",
 	Long:  `start proxy server`,
 	Run: func(cmd *cobra.Command, args []string) {
-		if err := util.InitConfigWithFile(proxyConfigFile); err != nil {
-			fmt.Printf("init config failed with file: %s, %v\n", proxyConfigFile, err)
+		if err := Init(); err != nil {
+			log.Printf("init failed with error: %v", err)
 			return
 		}
 
-		log.Println("config initialized")
-
-		if viper.GetBool("app.use_db") && viper.GetString("app.db_name") != "" {
-			if err := util.InitDBWithName(viper.GetString("app.db_name")); err != nil {
-				fmt.Printf("init db failed with name: %s, %v\n", viper.GetString("app.db_name"), err)
-				return
-			}
-			log.Println("db initialized")
-		}
-
-		if viper.GetBool("app.use_cache") && viper.GetString("app.cache_name") == "redis" {
-			if err := util.InitRedisWithName(viper.GetString("app.cache_name")); err != nil {
-				fmt.Printf("init redis failed with name: %s, %v\n", viper.GetString("app.cache_name"), err)
-				return
-			}
-			log.Println("redis initialized")
-		}
-
 		port := viper.GetInt("proxy_server.port")
 		if port == 0 {
 			port = 8025

cmd/root.go

@@ -1,9 +1,13 @@
 package cmd
 
 import (
+	"fmt"
+	"log"
 	"os"
 
+	"github.com/chatmcp/mcprouter/util"
 	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
 )
 
 // rootCmd represents the base command when called without any subcommands
@@ -25,6 +29,38 @@ func Execute() {
 	}
 }
 
+func Init() error {
+	if err := util.InitConfigWithFile(proxyConfigFile); err != nil {
+		fmt.Printf("init config failed with file: %s, %v\n", proxyConfigFile, err)
+		return err
+	}
+
+	log.Println("config initialized")
+
+	dbs := []string{viper.GetString("app.web_db_name"), viper.GetString("app.api_db_name")}
+	if viper.GetBool("app.use_db") {
+		for _, db := range dbs {
+			if db != "" {
+				if err := util.InitDBWithName(db); err != nil {
+					fmt.Printf("init db failed with name: %s, %v\n", db, err)
+					return err
+				}
+				log.Printf("db %s initialized", db)
+			}
+		}
+	}
+
+	if viper.GetBool("app.use_cache") && viper.GetString("app.cache_name") == "redis" {
+		if err := util.InitRedisWithName(viper.GetString("app.cache_name")); err != nil {
+			fmt.Printf("init redis failed with name: %s, %v\n", viper.GetString("app.cache_name"), err)
+			return err
+		}
+		log.Printf("redis %s initialized", viper.GetString("app.cache_name"))
+	}
+
+	return nil
+}
+
 func init() {
 	rootCmd.PersistentFlags().StringP("version", "v", "0.0.1", "version")
 }

handler/proxy/mcp.go

@@ -185,6 +185,13 @@ func MCP(c echo.Context) error {
 
 	proxyInfoB, _ := json.Marshal(proxyInfo)
 
+	if proxyInfo.RequestMethod == "tools/call" {
+		// if err := model.CreateServerLog(proxyInfo.ToServerLog()); err != nil {
+		// 	log.Printf("save server log failed: %v\n", err)
+		// } else {
+		// 	log.Printf("save server log ok: %s\n", string(proxyInfoB))
+		// }
+	}
 	log.Printf("proxyInfo: %s\n", string(proxyInfoB))
 
 	// notification

model/base.go

@@ -7,7 +7,13 @@ import (
 )
 
 func db() *gorm.DB {
-	name := viper.GetString("app.db_name")
+	name := viper.GetString("app.web_db_name")
+
+	return util.GetClient(name)
+}
+
+func adb() *gorm.DB {
+	name := viper.GetString("app.api_db_name")
 
 	return util.GetClient(name)
 }

model/serverlog.go

@@ -1,8 +1,8 @@
 package model
 
 import (
-	"encoding/json"
 	"errors"
+	"fmt"
 	"time"
 )
 
@@ -46,13 +46,7 @@ func CreateServerLog(sl *ServerLog) error {
 		return errors.New("invalid server log")
 	}
 
-	if sl.RequestParams != nil {
-		sl.RequestParams, _ = json.Marshal(sl.RequestParams)
-	}
-
-	if sl.ResponseResult != nil {
-		sl.ResponseResult, _ = json.Marshal(sl.ResponseResult)
-	}
+	fmt.Printf("save server log: %+v\n", sl)
 
-	return db().Table(sl.TableName()).Create(sl).Error
+	return adb().Table(sl.TableName()).Create(sl).Error
 }