实现找回密码功能

models/member_token.go

@@ -0,0 +1,66 @@
+package models
+
+import (
+	"time"
+	"github.com/lifei6671/godoc/conf"
+	"github.com/astaxie/beego/orm"
+)
+
+type MemberToken struct {
+	TokenId int		`orm:"column(token_id);pk;auto;unique" json:"token_id"`
+	MemberId int		`orm:"column(member_id);type(int)" json:"member_id"`
+	Token string		`orm:"column(token);size(255);index" json:"token"`
+	Email string 		`orm:"column(email);size(255)" json:"email"`
+	IsValid bool		`orm:"column(is_valid)" json:"is_valid"`
+	ValidTime time.Time	`orm:"column(valid_time);null" json:"valid_time"`
+	SendTime time.Time	`orm:"column(send_time);auto_now_add;type(datetime)" json:"send_time"`
+}
+
+
+// TableName 获取对应数据库表名.
+func (m *MemberToken) TableName() string {
+	return "member_token"
+}
+// TableEngine 获取数据使用的引擎.
+func (m *MemberToken) TableEngine() string {
+	return "INNODB"
+}
+
+func (m *MemberToken)TableNameWithPrefix() string {
+	return conf.GetDatabasePrefix() +  m.TableName()
+}
+
+func NewMemberToken() *MemberToken {
+	return &MemberToken{}
+}
+
+func (m *MemberToken)  InsertOrUpdate() (*MemberToken,error){
+	o := orm.NewOrm()
+
+	if m.TokenId > 0 {
+		_,err := o.Update(m)
+		return m,err
+	}
+	_,err := o.Insert(m)
+
+	return m,err
+}
+
+func (m *MemberToken) FindByFieldFirst(field string,value interface{}) (*MemberToken,error) {
+	o := orm.NewOrm()
+
+	err := o.QueryTable(m.TableNameWithPrefix()).Filter(field,value).OrderBy("-token_id").One(m)
+
+	return m,err
+}
+
+func (m *MemberToken) FindSendCount(mail string,start_time time.Time,end_time time.Time) (int ,error) {
+	o := orm.NewOrm()
+
+	c,err := o.QueryTable(m.TableNameWithPrefix()).Filter("send_time__gte",start_time.Format("2006-01-02 15:04:05")).Filter("send_time__lte",end_time.Format("2006-01-02 15:04:05")).Count()
+
+	if err != nil {
+		return 0,err
+	}
+	return int(c),nil
+}