From bae4af47f77a195a12a0437584d667465e826e12 Mon Sep 17 00:00:00 2001 From: liuxiaolong <liuxiaolong@aiotlink.com> Date: 星期一, 17 八月 2020 16:09:28 +0800 Subject: [PATCH] add log --- models/user.go | 41 ++++++++++++++++++++++++++++++++--------- 1 files changed, 32 insertions(+), 9 deletions(-) diff --git a/models/user.go b/models/user.go index 5e2e8f0..a697d9b 100644 --- a/models/user.go +++ b/models/user.go @@ -23,12 +23,6 @@ return "sys_user" } -func (u *User) TableUnique() [][]string { - return [][]string { - []string{"phoneNum"}, - } -} - func (u *User) Insert() (int64,error) { o := orm.NewOrm() return o.Insert(u) @@ -36,28 +30,48 @@ func (u *User) SelectById(uid string) error { o := orm.NewOrm() - err := o.Raw("select * from ? where id=?", u.TableName(), uid).QueryRow(u) + err := o.QueryTable(u.TableName()).Filter("id", uid).One(u) return err } func (u *User) SelectByPhoneNum(phoneNum string) error { o := orm.NewOrm() - err := o.Raw("select * from ? where phoneNum=?", u.TableName(), phoneNum).QueryRow(u) + err := o.QueryTable(u.TableName()).Filter("phoneNum", phoneNum).One(u) return err } func (u *User) GetAllUsers() (all []User,err error) { o := orm.NewOrm() - _, err = o.Raw("select * from ?", u.TableName()).QueryRows(&all) + _, err = o.QueryTable(u.TableName()).All(&all) if err != nil { return nil, err } return all,nil } +func (u *User) GetAllMap() (map[string]User) { + idM := make(map[string]User) + all, _ := u.GetAllUsers() + if all !=nil { + for _,p := range all { + idM[p.Id] = p + } + } + return idM +} + func (u *User) Update() (int64, error) { o := orm.NewOrm() return o.Update(u) +} + +func (u *User) UpdatePhoneNum(phoneNum string, hikPersonId string) (int64, error) { + o := orm.NewOrm() + res, err := o.Raw("update sys_user set phoneNum=? where id=?", phoneNum, hikPersonId).Exec() + if err != nil { + return 0, err + } + return res.RowsAffected() } func Login(username, password string) bool { @@ -69,3 +83,12 @@ o := orm.NewOrm() return o.Update(u, "isDelete") } + +func (u *User) Delete(id string) (int64, error) { + o := orm.NewOrm() + res, err := o.Raw("delete from sys_user where id=?", id).Exec() + if err != nil { + return 0, err + } + return res.RowsAffected() +} \ No newline at end of file -- Gitblit v1.8.0