// Package models . package models import ( "time" "github.com/astaxie/beego/orm" "github.com/lifei6671/godoc/utils" "github.com/lifei6671/godoc/conf" "github.com/astaxie/beego/logs" "errors" "regexp" "strings" ) type Member struct { MemberId int `orm:"pk;auto;unique;column(member_id)" json:"member_id"` Account string `orm:"size(100);unique;column(account)" json:"account"` Password string `orm:"size(1000);column(password)" json:"-"` Description string `orm:"column(description);size(2000)" json:"description"` Email string `orm:"size(100);column(email);unique" json:"email"` Phone string `orm:"size(255);column(phone);null;default(null)" json:"phone"` Avatar string `orm:"size(1000);column(avatar)" json:"avatar"` //用户角色:0 超级管理员 /1 管理员/ 2 普通用户 . Role int `orm:"column(role);type(int);default(1);index" json:"role"` RoleName string `orm:"-" json:"role_name"` Status int `orm:"column(status);type(int);default(0)" json:"status"` //用户状态:0 正常/1 禁用 CreateTime time.Time `orm:"type(datetime);column(create_time);auto_now_add" json:"create_time"` CreateAt int `orm:"type(int);column(create_at)" json:"create_at"` LastLoginTime time.Time `orm:"type(datetime);column(last_login_time);null" json:"last_login_time"` } // TableName 获取对应数据库表名. func (m *Member) TableName() string { return "members" } // TableEngine 获取数据使用的引擎. func (m *Member) TableEngine() string { return "INNODB" } func (m *Member)TableNameWithPrefix() string { return conf.GetDatabasePrefix() + m.TableName() } func NewMember() *Member { return &Member{} } // Login 用户登录. func (m *Member) Login(account string,password string) (*Member,error) { o := orm.NewOrm() member := &Member{} err := o.QueryTable(m.TableNameWithPrefix()).Filter("account",account).Filter("status",0).One(member); if err != nil { logs.Error("用户登录 => ",err) return member,ErrMemberNoExist } ok,err := utils.PasswordVerify(member.Password,password) ; if ok && err == nil { m.ResolveRoleName() return member,nil } return member,ErrorMemberPasswordError } // Add 添加一个用户. func (m *Member) Add () (error) { o := orm.NewOrm() if ok,err := regexp.MatchString(conf.RegexpAccount,m.Account); m.Account == "" || !ok || err != nil { return errors.New("账号只能由英文字母数字组成,且在3-50个字符") } if m.Email == "" { return errors.New("邮箱不能为空") } if ok,err := regexp.MatchString(conf.RegexpEmail,m.Email); !ok || err != nil || m.Email == "" { return errors.New("邮箱格式不正确") } if l := strings.Count(m.Password,""); l < 6 || l >= 50{ return errors.New("密码不能为空且必须在6-50个字符之间") } if c,err := o.QueryTable(m.TableNameWithPrefix()).Filter("email",m.Email).Count(); err == nil && c > 0 { return errors.New("邮箱已被使用") } hash ,err := utils.PasswordHash(m.Password); if err != nil { return err } m.Password = hash _,err = o.Insert(m) if err != nil { return err } m.ResolveRoleName() return nil } // Update 更新用户信息. func (m *Member) Update(cols... string) (error) { o := orm.NewOrm() if m.Email == "" { return errors.New("邮箱不能为空") } if _,err := o.Update(m,cols...);err != nil { return err } return nil } func (m *Member) Find(id int) (*Member,error){ o := orm.NewOrm() m.MemberId = id if err := o.Read(m); err != nil { return m,err } m.ResolveRoleName() return m,nil } func (m *Member) ResolveRoleName (){ if m.Role == conf.MemberSuperRole { m.RoleName = "超级管理员" }else if m.Role == conf.MemberAdminRole { m.RoleName = "管理员" }else if m.Role == conf.MemberGeneralRole { m.RoleName = "普通用户" } } func (m *Member) FindByAccount (account string) (*Member,error) { o := orm.NewOrm() err := o.QueryTable(m.TableNameWithPrefix()).Filter("account",account).One(m) if err == nil { m.ResolveRoleName() } return m,err } func (m *Member) FindToPager(pageIndex, pageSize int) ([]*Member,int64,error) { o := orm.NewOrm() var members []*Member offset := (pageIndex - 1) * pageSize totalCount,err := o.QueryTable(m.TableNameWithPrefix()).Count() if err != nil { return members,0,err } _,err = o.QueryTable(m.TableNameWithPrefix()).OrderBy("-member_id").Offset(offset).Limit(pageSize).All(&members) if err != nil { return members,0,err } for _,m := range members { m.ResolveRoleName() } return members,totalCount,nil } func (c *Member) IsAdministrator() bool { if c == nil || c.MemberId <= 0{ return false } return c.Role == 0 || c.Role == 1 } func (m *Member) FindByFieldFirst(field string,value interface{}) (*Member,error) { o := orm.NewOrm() err := o.QueryTable(m.TableNameWithPrefix()).Filter(field,value).OrderBy("-member_id").One(m) return m,err }