package gmodel // 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 }