fix

model/gmodel/fs_merchant_category_logic.go

@@ -1,8 +1,15 @@
 package gmodel
 
-import "context"
+import (
+	"context"
+	"fusenapi/utils/handlers"
+	"reflect"
+
+	"gorm.io/gorm"
+)
 
 // TODO: 使用model的属性做你想做的
+
 func (m *FsMerchantCategoryModel) FindOne(ctx context.Context, id int64) (resp *FsMerchantCategory, err error) {
 	err = m.db.WithContext(ctx).Model(&FsMerchantCategory{}).Where("id = ? and status = ?", id, 1).Take(&resp).Error
 	return resp, err
@@ -11,3 +18,37 @@ func (m *FsMerchantCategoryModel) FindRandOne(ctx context.Context) (resp *FsMerc
 	err = m.db.WithContext(ctx).Model(&FsMerchantCategory{}).Where("status = ?", 1).Order("RAND()").Take(&resp).Error
 	return resp, err
 }
+
+func (m *FsMerchantCategoryModel) BuilderDB(ctx context.Context, selectData []string) *gorm.DB {
+	if selectData != nil {
+		return m.db.WithContext(ctx).Select(selectData)
+	} else {
+		return m.db.WithContext(ctx).Select("*")
+	}
+}
+
+func (m *FsMerchantCategoryModel) FindAll(gormDB *gorm.DB, filterMap map[string]string, orderBy string) ([]*FsMerchantCategory, error) {
+	var resp []*FsMerchantCategory
+
+	// 过滤
+	if filterMap != nil {
+		gormDB = gormDB.Scopes(handlers.FilterData(filterMap))
+	}
+
+	// 排序
+	if orderBy != "" {
+		var fieldsMap = make(map[string]struct{})
+		s := reflect.TypeOf(&FsOrder{}).Elem() //通过反射获取type定义
+		for i := 0; i < s.NumField(); i++ {
+			fieldsMap[s.Field(i).Tag.Get("json")] = struct{}{}
+		}
+		gormDB = gormDB.Scopes(handlers.OrderCheck(orderBy, fieldsMap))
+	}
+
+	result := gormDB.Find(&resp)
+	if result.Error != nil {
+		return nil, result.Error
+	} else {
+		return resp, nil
+	}
+}

model/gmodel/fs_user_info_logic.go

@@ -1,2 +1,44 @@
 package gmodel
-// TODO: 使用model的属性做你想做的
+
+// TODO: 使用model的属性做你想做的
+
+import (
+	"context"
+	"fusenapi/utils/handlers"
+
+	"gorm.io/gorm"
+)
+
+// TODO: 使用model的属性做你想做的
+
+func (m *FsUserInfoModel) BuilderDB(ctx context.Context, selectData []string) *gorm.DB {
+	if selectData != nil {
+		return m.db.WithContext(ctx).Select(selectData)
+	} else {
+		return m.db.WithContext(ctx).Select("*")
+	}
+}
+
+func (m *FsUserInfoModel) FindOne(gormDB *gorm.DB, filterMap map[string]string) (*FsUserInfo, error) {
+	var resp FsUserInfo
+
+	if filterMap != nil {
+		gormDB = gormDB.Scopes(handlers.FilterData(filterMap))
+	}
+
+	result := gormDB.Limit(1).Find(&resp)
+	if result.Error != nil {
+		return nil, result.Error
+	} else {
+		return &resp, nil
+	}
+}
+
+func (p *FsUserInfoModel) CreateOrUpdate(gormDB *gorm.DB, req *FsUserInfo) (resp *FsUserInfo, err error) {
+	if req.Id > 0 {
+		err = gormDB.Save(req).Error
+	} else {
+		err = gormDB.Create(req).Error
+	}
+	return req, err
+}