From b0d1fe4e0acb0fa40c684069c0e9d77d7a65ace3 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期二, 17 十月 2023 15:53:53 +0800
Subject: [PATCH] fix
---
model/user.go | 17 ++++++++++++++++-
1 files changed, 16 insertions(+), 1 deletions(-)
diff --git a/model/user.go b/model/user.go
index f0d074d..925d6d7 100644
--- a/model/user.go
+++ b/model/user.go
@@ -70,6 +70,11 @@
return slf
}
+func (slf *UserSearch) SetUserId(userId int) *UserSearch {
+ slf.ID = userId
+ return slf
+}
+
func (slf *UserSearch) SetUserName(username string) *UserSearch {
slf.Username = username
return slf
@@ -84,7 +89,7 @@
var db = slf.Orm.Model(&User{}).Preload("Menus")
if slf.UUID != "" {
- db = db.Where("id = ?", slf.UUID)
+ db = db.Where("uuid = ?", slf.UUID)
}
if slf.Username != "" {
@@ -271,3 +276,13 @@
return records, nil
}
+
+func (slf *UserSearch) FirstOrCreate(record User) error {
+ var db = slf.build()
+
+ if err := db.FirstOrCreate(&User{}, record).Error; err != nil {
+ return fmt.Errorf("first or create err: %v, record: %+v", err, record)
+ }
+
+ return nil
+}
--
Gitblit v1.8.0