60 lines
1.7 KiB
Go
60 lines
1.7 KiB
Go
package userDAO
|
|
|
|
import (
|
|
"PersonalMain/internal/model/do"
|
|
"context"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
)
|
|
|
|
// InsertUser
|
|
//
|
|
// 插入用户
|
|
func InsertUser(userDO do.UserDO) string {
|
|
// 检查数据库是否存在用户
|
|
result, err := g.Model("xf_user").Where("user_name = ? or email = ?", userDO.UserName, userDO.Email).One()
|
|
if err == nil {
|
|
if result.IsEmpty() {
|
|
_, err := g.Model("xf_user").Data(userDO).Insert()
|
|
if err != nil {
|
|
g.Log().Cat("Database").Cat("User").Error(context.Background(), err.Error())
|
|
return "DatabaseError"
|
|
} else {
|
|
g.Log().Cat("Database").Cat("User").Notice(context.Background(), "User", userDO.UserName, "创建成功")
|
|
return "Success"
|
|
}
|
|
} else {
|
|
g.Log().Cat("Database").Cat("User").Notice(context.Background(), "无法创建", userDO.UserName, "用户。原因：已存在此用户")
|
|
return "UserExist"
|
|
}
|
|
} else {
|
|
g.Log().Cat("Database").Cat("User").Error(context.Background(), err.Error())
|
|
return "DatabaseError"
|
|
}
|
|
}
|
|
|
|
// GetUser
|
|
//
|
|
// 获取用户信息
|
|
func GetUser(user string) *do.UserDO {
|
|
userDO := do.UserDO{}
|
|
result, err := g.Model("xf_user").Where("user_name = ? or email = ?", user, user).One()
|
|
if err == nil {
|
|
if result.IsEmpty() {
|
|
g.Log().Cat("Database").Cat("User").Notice(context.Background(), "无法获取", user, "用户。原因：不存在此用户")
|
|
return nil
|
|
} else {
|
|
err := result.Struct(&userDO)
|
|
if err != nil {
|
|
g.Log().Cat("Database").Cat("User").Error(context.Background(), err.Error())
|
|
return nil
|
|
} else {
|
|
g.Log().Cat("Database").Cat("User").Notice(context.Background(), "获取", user, "用户成功")
|
|
return &userDO
|
|
}
|
|
}
|
|
} else {
|
|
g.Log().Cat("Database").Cat("User").Error(context.Background(), err.Error())
|
|
return nil
|
|
}
|
|
}
|