mirror of https://github.com/mindoc-org/mindoc.git
152 lines
3.0 KiB
Go
152 lines
3.0 KiB
Go
// Package models .
|
||
package models
|
||
|
||
import (
|
||
"time"
|
||
"github.com/astaxie/beego/orm"
|
||
"github.com/lifei6671/godoc/utils"
|
||
"github.com/lifei6671/godoc/conf"
|
||
)
|
||
|
||
type Member struct {
|
||
MemberId int `orm:"pk;auto;unique;column(member_id)"`
|
||
Account string `orm:"size(100);unique;column(account)"`
|
||
Password string `orm:"size(1000);column(password)"`
|
||
Description string `orm:"column(description);size(2000)" json:"description"`
|
||
Email string `orm:"size(255);column(email);null;default(null)"`
|
||
Phone string `orm:"size(255);column(phone);null;default(null)"`
|
||
Avatar string `orm:"size(1000);column(avatar)"`
|
||
Role int `orm:"column(role);type(int);default(1);index"` //用户角色：0 管理员/ 1 普通用户
|
||
Status int `orm:"column(status);type(int);default(0)"` //用户状态：0 正常/1 禁用
|
||
CreateTime time.Time `orm:"type(datetime);column(create_time);auto_now_add"`
|
||
CreateAt int `orm:"type(int);column(create_at)"`
|
||
LastLoginTime time.Time `orm:"type(datetime);column(last_login_time);null"`
|
||
}
|
||
|
||
// 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("md_" + m.TableName()).Filter("account",account).Filter("status",0).One(member);
|
||
|
||
if err != nil {
|
||
return member,ErrMemberNoExist
|
||
}
|
||
|
||
ok,err := utils.PasswordVerify(member.Password,password) ;
|
||
|
||
if ok && err == nil {
|
||
return member,nil
|
||
}
|
||
|
||
return member,ErrorMemberPasswordError
|
||
}
|
||
|
||
// Add 添加一个用户.
|
||
func (member *Member) Add () (error) {
|
||
o := orm.NewOrm()
|
||
|
||
hash ,err := utils.PasswordHash(member.Password);
|
||
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
member.Password = hash
|
||
|
||
_,err = o.Insert(member)
|
||
|
||
if err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// Update 更新用户信息.
|
||
func (m *Member) Update(cols... string) (error) {
|
||
o := orm.NewOrm()
|
||
|
||
if _,err := o.Update(m,cols...);err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (m *Member) Find(id int) error{
|
||
o := orm.NewOrm()
|
||
|
||
m.MemberId = id
|
||
if err := o.Read(m); err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (m *Member) FindByAccount (account string) error {
|
||
o := orm.NewOrm()
|
||
|
||
err := o.QueryTable(m.TableNameWithPrefix()).Filter("account",account).One(m)
|
||
|
||
return err
|
||
}
|
||
|
||
func (m *Member) FindToPager(pageIndex, pageSize int) ([]Member,int,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()).Offset(offset).Limit(pageSize).All(&members)
|
||
|
||
if err != nil {
|
||
return members,0,err
|
||
}
|
||
|
||
return members,totalCount,nil
|
||
}
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|