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 20:01:43 +08:00
parent a265cd459b
commit e983317f5b
2 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
dao/user.go
View File

@ -5,8 +5,8 @@ import (
"context"
"fmt"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"gorm.io/gorm"
"log"
)
type User struct {
@ -297,11 +297,16 @@ func CreateUserUIConfigInfo(config proto.UserUIConfigInfo) (string, error) {
return "", err
}
fmt.Println("Inserted a single document:", res.InsertedID)
return res.InsertedID.(string), nil
// 类型安全转换
insertedID, ok := res.InsertedID.(primitive.ObjectID)
if !ok {
return primitive.NilObjectID.String(), fmt.Errorf("意外的ID类型: %T", res.InsertedID)
}
return insertedID.String(), nil
}
func GetUserUIConfigInfo(userID int) (proto.UserUIConfigInfo, error) {
log.Println("get user ui config info database:", proto.Config.MONGO_DATABASE, " collection:", UserUIConfigCollection)
//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)

13
service/userService.go
View File

@ -464,7 +464,15 @@ func GetUserUIConfigInfo(userID int) proto.UserUIConfigInfo {
userConfig, err := dao.GetUserUIConfigInfo(userID)
if err != nil {
log.Println("GetUserUIConfigInfo error:", err)
return proto.UserUIConfigInfo{}
} else {
userConfig.UserID = userID
//没有则插入
id, err2 := dao.CreateUserUIConfigInfo(userConfig)
if err2 != nil {
log.Println("InsertUserUIConfigInfo error:", err.Error())
} else {
log.Println("InsertUserUIConfigInfo success, id:", id, "config:", userConfig)
}
}
return userConfig
}
@ -475,14 +483,13 @@ func SetUserUIConfigInfo(userID int, config proto.UserUIConfigInfo) error {
userConfig, err := dao.GetUserUIConfigInfo(userID)
if err != nil {
log.Println("SetUserUIConfigInfo error:", err)
return err
}
config.UserID = userID
if userConfig.UserID == 0 {
//没有则插入
id, err2 := dao.CreateUserUIConfigInfo(config)
if err2 != nil {
log.Println("InsertUserUIConfigInfo error:", err)
log.Println("InsertUserUIConfigInfo error:", err.Error())
return err
} else {
log.Println("InsertUserUIConfigInfo success, id:", id, "config:", config)