diff --git a/src/dao/user_dao.go b/src/dao/user_dao.go index 133b114..f8e2e9b 100644 --- a/src/dao/user_dao.go +++ b/src/dao/user_dao.go @@ -1,10 +1,14 @@ package dao -import "code.fnuoos.com/EggPlanet/egg_models.git/src/model" +import ( + "code.fnuoos.com/EggPlanet/egg_models.git/src/model" +) type UserDao interface { UserGetOneByParams(params map[string]interface{}) (*model.User, error) UserFindByParams(params map[string]interface{}) ([]model.User, error) UserUpdate(id interface{}, user *model.User, forceColumns ...string) (int64, error) UserFindAndCount(uid int64, phone string, nickname string, level int, page, limit int) (*[]model.User, int64, error) + UserFindByInviteCodes(codes ...string) (*[]model.User, error) + UserFindByCustomInviteCodes(codes ...string) (*[]model.User, error) } diff --git a/src/implement/sys_cfg_implement.go b/src/implement/sys_cfg_implement.go index ba57b2f..43feeed 100644 --- a/src/implement/sys_cfg_implement.go +++ b/src/implement/sys_cfg_implement.go @@ -36,13 +36,10 @@ func (s SysCfgDb) SysCfgGetAll() (*[]model.SysCfg, error) { func (s SysCfgDb) SysCfgGetOneNoDataNoErr(key string) (*model.SysCfg, error) { var cfgList model.SysCfg - has, err := s.Db.Where("`key`=?", key).Get(&cfgList) + _, err := s.Db.Where("`key`=?", key).Get(&cfgList) if err != nil { return nil, zhios_order_relate_logx.Error(err) } - if !has { - return nil, nil - } return &cfgList, nil } diff --git a/src/implement/user_implement.go b/src/implement/user_implement.go index f9fb997..83cafd3 100644 --- a/src/implement/user_implement.go +++ b/src/implement/user_implement.go @@ -18,6 +18,22 @@ type UserDb struct { Db *xorm.Engine } +func (u UserDb) UserFindByInviteCodes(codes ...string) (*[]model.User, error) { + var m []model.User + if err := u.Db.In("custom_invite_code", codes).Find(&m); err != nil { + return nil, zhios_order_relate_logx.Warn(err) + } + return &m, nil +} + +func (u UserDb) UserFindByCustomInviteCodes(codes ...string) (*[]model.User, error) { + var m []model.User + if err := u.Db.In("system_invite_code", codes).Find(&m); err != nil { + return nil, zhios_order_relate_logx.Warn(err) + } + return &m, nil +} + func (u UserDb) UserGetOneByParams(params map[string]interface{}) (*model.User, error) { var m model.User var query = fmt.Sprintf("%s = ?", params["key"])