ljydlv
/
saw-go
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

添加mongodb使用，添加用户前端配置的上传及获取接口

This commit is contained in:
junleea 2025-05-20 19:36:46 +08:00
parent 0897d1e562
commit a265cd459b
3 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dao/user.go
View File

@ -6,6 +6,7 @@ import (
"fmt"
"go.mongodb.org/mongo-driver/bson"
"gorm.io/gorm"
"log"
)
type User struct {
@ -300,6 +301,7 @@ func CreateUserUIConfigInfo(config proto.UserUIConfigInfo) (string, error) {
}
func GetUserUIConfigInfo(userID int) (proto.UserUIConfigInfo, error) {
log.Println("get user ui config info database:", proto.Config.MONGO_DATABASE, " collection:", UserUIConfigCollection)
collection := mongoClient.Database(proto.Config.MONGO_DATABASE).Collection(UserUIConfigCollection)
var config proto.UserUIConfigInfo
err := collection.FindOne(context.TODO(), bson.M{"user_id": userID}).Decode(&config)

1
main.go
View File

@ -47,6 +47,7 @@ func main() {
handler.SetUpIMGroup(r) // IM
defer dao.Close()
defer worker.CloseRedis()
defer dao.CloseMongoDB()
//定时任务
c := cron.New(cron.WithSeconds())
// 添加每 10 秒执行一次的任务

18
proto/user.go
View File

@ -1,6 +1,7 @@
package proto
import (
"go.mongodb.org/mongo-driver/bson/primitive"
"time"
)
@ -125,15 +126,16 @@ type DeleteThirdPartyLoginReq struct {
// 用户对前端交互的配置信息结构,mongodb
type UserUIConfigInfo struct {
UserID int `json:"user_id" form:"user_id"` // 用户ID
Theme string `json:"theme" form:"theme"` // 主题
Language string `json:"language" form:"language"` // 语言
FontSize int `json:"font_size" form:"font_size"` // 字体大小
GenAIFunction UserUIFunctionConfigInfo `json:"gen_ai_function" form:"gen_ai_function"` // 生成AI功能配置
KBaseFunction UserUIFunctionConfigInfo `json:"k_base_function" form:"k_base_function"` // 知识库功能配置
ID primitive.ObjectID `bson:"_id,omitempty"` // 自动生成ID
UserID int `json:"user_id" form:"user_id" bson:"user_id"` // 用户ID
Theme string `json:"theme" form:"theme" bson:"theme"` // 主题
Language string `json:"language" form:"language" bson:"language"` // 语言
FontSize int `json:"font_size" form:"font_size" bson:"font_size"` // 字体大小
GenAIFunction UserUIFunctionConfigInfo `json:"gen_ai_function" form:"gen_ai_function" bson:"gen_ai_function"` // 生成AI功能配置
KBaseFunction UserUIFunctionConfigInfo `json:"k_base_function" form:"k_base_function" bson:"k_base_function"` // 知识库功能配置
}
type UserUIFunctionConfigInfo struct {
ModelID int `json:"model_id" form:"model_id"` // 模型ID,选择的模型
SessionID int `json:"session_id" form:"session_id"` // 会话ID,上一次的会话
ModelID int `json:"model_id" form:"model_id" bson:"model_id"` // 模型ID,选择的模型
SessionID int `json:"session_id" form:"session_id" bson:"session_id"` // 会话ID,上一次的会话
}