128 lines
3.2 KiB
Go
128 lines
3.2 KiB
Go
package user
|
|
|
|
import (
|
|
"errors"
|
|
"go_dreamfactory/comm"
|
|
"go_dreamfactory/lego/core"
|
|
"go_dreamfactory/modules"
|
|
"go_dreamfactory/pb"
|
|
"go_dreamfactory/utils"
|
|
"time"
|
|
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
)
|
|
|
|
// 签到
|
|
type ModelSign struct {
|
|
modules.MCompModel
|
|
module *User
|
|
}
|
|
|
|
func (this *ModelSign) Init(service core.IService, module core.IModule, comp core.IModuleComp, options core.IModuleOptions) (err error) {
|
|
this.TableName = comm.TableSign
|
|
err = this.MCompModel.Init(service, module, comp, options)
|
|
this.module = module.(*User)
|
|
return
|
|
}
|
|
|
|
func (this *ModelSign) GetUserSign(uid string) (result *pb.DBSign, err error) {
|
|
result = &pb.DBSign{}
|
|
if err = this.module.modelSign.Get(uid, result); err != nil {
|
|
if mongo.ErrNoDocuments == err { // 创建一条新的数据
|
|
_data := this.module.configure.GetSignConf(1, 1)
|
|
if _data != nil {
|
|
result = &pb.DBSign{
|
|
Id: primitive.NewObjectID().Hex(),
|
|
Uid: uid,
|
|
SignTime: 0,
|
|
SignCount: 0,
|
|
Group: _data.Group,
|
|
Cid: 0,
|
|
RTime: 0,
|
|
}
|
|
this.Add(uid, result)
|
|
} else {
|
|
err = errors.New("conf err")
|
|
return nil, err
|
|
}
|
|
|
|
}
|
|
err = nil
|
|
return
|
|
}
|
|
err = nil
|
|
return result, err
|
|
}
|
|
|
|
//修改用户签到数据
|
|
func (this *ModelSign) ChangeUserSign(uid string, value map[string]interface{}) (err error) {
|
|
if len(value) == 0 {
|
|
return nil
|
|
}
|
|
return this.module.modelSign.Change(uid, value)
|
|
}
|
|
|
|
// 累计登录天数
|
|
func (this *ModelSign) updateSignData(uid string, sign *pb.DBSign) (err error) {
|
|
if !utils.IsToday(sign.SignTime) {
|
|
sign.SignCount += 1
|
|
count := sign.SignCount
|
|
update := map[string]interface{}{
|
|
"signCount": count,
|
|
"signTime": time.Now().UTC(),
|
|
}
|
|
this.ChangeUserSign(uid, update)
|
|
}
|
|
return
|
|
}
|
|
|
|
// 检测是否达到重置日期
|
|
func (this *ModelSign) checkResetSignData(session comm.IUserSession) {
|
|
var (
|
|
update map[string]interface{}
|
|
bReward bool
|
|
)
|
|
update = make(map[string]interface{}, 0)
|
|
if sign, err := this.module.modelSign.GetUserSign(session.GetUserId()); err == nil {
|
|
start, _ := utils.GetMonthStartEnd()
|
|
if sign.RTime < start { // 重置
|
|
sign.RTime = time.Now().Unix()
|
|
sign.SignTime = sign.RTime
|
|
sign.SignCount = 1
|
|
if newGroup := this.module.configure.GetSignResetConf(sign.Cid + 1); newGroup != -1 { // 获取当前的组id
|
|
sign.Cid += 1
|
|
sign.Group = newGroup
|
|
update["cid"] = sign.Cid
|
|
update["group"] = sign.Group
|
|
}
|
|
update["rTime"] = sign.RTime
|
|
update["signTime"] = sign.SignTime
|
|
update["signCount"] = sign.SignCount
|
|
this.Change(session.GetUserId(), update)
|
|
bReward = true
|
|
} else {
|
|
if !utils.IsToday(sign.SignTime) {
|
|
sign.SignCount += 1
|
|
update["signCount"] = sign.SignCount
|
|
sign.SignTime = time.Now().Unix()
|
|
update["signTime"] = sign.SignTime
|
|
sign.RTime = sign.SignTime
|
|
update["rTime"] = sign.RTime
|
|
this.Change(session.GetUserId(), update)
|
|
bReward = true
|
|
}
|
|
}
|
|
_data := this.module.configure.GetSignConf(sign.Cid, sign.Group)
|
|
if bReward && _data != nil { // 发奖
|
|
this.module.DispenseRes(session, _data.Loopgift, true)
|
|
}
|
|
session.SendMsg(string(this.module.GetType()), "sign", &pb.UserSignResp{
|
|
Data: sign,
|
|
Reward: bReward,
|
|
})
|
|
}
|
|
|
|
return
|
|
}
|