mindoc/models/member.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
}