123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248 |
- package Account
- import (
- "ColdVerify_server/lib"
- "ColdVerify_server/logs"
- "github.com/beego/beego/v2/adapter/orm"
- orm2 "github.com/beego/beego/v2/client/orm"
- _ "github.com/go-sql-driver/mysql"
- uuid "github.com/satori/go.uuid"
- "log"
- "time"
- )
- type User struct {
- Id int `orm:"column(ID);size(11);auto;pk"`
- T_uuid string `orm:"size(256);null"`
-
- T_name string `orm:"size(256);null"`
- T_pass string `orm:"size(256);null"`
- T_passstr string `orm:"size(256);null"`
- T_Show int `orm:"size(200);default(1)"`
- T_State int `orm:"size(200);default(1)"`
- CreateTime time.Time `orm:"column(create_time);type(timestamp);null;auto_now_add"`
- UpdateTime time.Time `orm:"column(update_time);type(timestamp);null;auto_now"`
- }
- type User_R struct {
- T_uuid string
-
- T_name string
- T_passstr string
- T_Show int
- T_State int
- }
- func (t *User) TableName() string {
- return "user"
- }
- func init() {
-
- orm.RegisterModel(new(User))
- }
- func UserToUser_R(T User) (T_r User_R) {
- T_r.T_uuid = T.T_uuid
-
- T_r.T_name = T.T_name
- T_r.T_passstr = T.T_passstr
- T_r.T_Show = T.T_Show
- T_r.T_State = T.T_State
-
- return T_r
- }
- func Read_User_ByT_uuid(T_uuid string) (e error, r User) {
- o := orm.NewOrm()
- qs := o.QueryTable(new(User))
- e = qs.Filter("T_uuid", T_uuid).Filter("T_State", 1).One(&r)
- return e, r
- }
- func Read_User_verification(T_user string, T_pass string) (error, User) {
- o := orm.NewOrm()
- r := User{T_name: T_user, T_pass: T_pass, T_State: 1}
- err := o.Read(&r, "T_name", "T_pass", "T_State")
- if err != nil {
- logs.Error(lib.FuncName(), err)
- }
- return err, r
- }
- func Read_User_ById(id int) (r User, is bool) {
- o := orm.NewOrm()
- r = User{Id: id, T_State: 1}
- err := o.Read(&r, "T_State")
- if err != nil {
- logs.Error(lib.FuncName(), err)
- return r, false
- }
- return r, true
- }
- func Read_User(T_uuid string) (r User, is bool) {
- o := orm.NewOrm()
- qs := o.QueryTable(new(User))
- err := qs.Filter("T_uuid", T_uuid).Filter("T_State", 1).One(&r)
- if err != nil {
- return r, false
- }
- return r, true
- }
- func Read_UserByT_name(T_name string) (r User, is bool) {
- o := orm.NewOrm()
- qs := o.QueryTable(new(User))
- err := qs.Filter("T_name", T_name).Filter("T_State", 1).One(&r)
- if err != nil {
- return r, false
- }
- return r, true
- }
- func Add_User(r User) (id int64, is bool) {
- o := orm.NewOrm()
-
- rand_x := 0
- for true {
- r.T_uuid = uuid.NewV4().String()
- err := o.Read(&r, "T_uuid")
- if err != nil {
- break
- }
- rand_x += 1
- }
- id, err := o.Insert(&r)
- if err != nil {
- logs.Error(lib.FuncName(), err)
- return 0, false
- }
- return id, true
- }
- func Delete_User(v User) bool {
- o := orm.NewOrm()
- if num, err := o.Delete(&v); err == nil {
- logs.Println("Number of records deleted in database:", num)
- } else {
- return false
- }
- return true
- }
- func Delete_User_(v User) bool {
- o := orm.NewOrm()
- v.T_State = 0
- if num, err := o.Update(&v, "T_State"); err == nil {
- log.Println("Number of records updated in database:", num)
- } else {
- return false
- }
- return true
- }
- func Update_User(m User, cols ...string) bool {
- o := orm.NewOrm()
- if num, err := o.Update(&m, cols...); err == nil {
- log.Println("Number of records updated in database:", num)
- return true
- }
- return false
- }
- func Read_User_List(T_name string, page int, page_z int) ([]User_R, int64) {
- o := orm.NewOrm()
-
- var r []User
- qs := o.QueryTable(new(User))
- var offset int64
- if page <= 1 {
- offset = 0
- } else {
- offset = int64((page - 1) * page_z)
- }
- cond := orm.NewCondition()
- cond1 := cond.And("T_State", 1).AndCond(cond.Or("T_name__icontains", T_name))
- qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("-Id").All(&r)
- cnt, _ := qs.SetCond((*orm2.Condition)(cond1)).Count()
-
- var User_r []User_R
- for _, v := range r {
- User_r = append(User_r, UserToUser_R(v))
- }
- return User_r, cnt
- }
- func Read_User_List_ALL() (maps []User) {
- o := orm.NewOrm()
- qs := o.QueryTable(new(Admin))
- qs.Filter("T_State", 1).OrderBy("Id").All(&maps)
- return maps
- }
- func Read_User_List_ALL_1() (maps []User) {
- o := orm.NewOrm()
- qs := o.QueryTable(new(User))
- qs.OrderBy("Id").All(&maps)
- return maps
- }
- func UserListToMap(T []User) map[string]string {
- maps := make(map[string]string, len(T))
- for _, v := range T {
- maps[v.T_uuid] = v.T_name
- }
- return maps
- }
- func Read_User_T_uuid_ListByT_name(T_name string) (list []string) {
- o := orm.NewOrm()
-
- var r []User
- qs := o.QueryTable(new(User))
- cond := orm.NewCondition()
- cond1 := cond.And("T_State", 1).AndCond(cond.Or("T_name__icontains", T_name))
- qs.SetCond((*orm2.Condition)(cond1)).All(&r)
-
- for _, v := range r {
- list = append(list, v.T_uuid)
- }
- return list
- }
|