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