invite-bean/yaml/data.go

package comcode

import (
	"active/constant"
	"errors"
	"fmt"
	"log"
	"strconv"
	"time"

	"git.jiaxianghudong.com/webs/pkg/rds"
	"gorm.io/gorm"
)

type Comcode struct {
	Id         int64  `json:"id" gorm:"column:id"`
	Code       string `json:"code" gorm:"column:code"`
	AwardsId   int    `json:"awards_id" gorm:"column:awards_id"`
	CreateTime string `json:"create_time" gorm:"column:create_time"`
	OpenTime   int64  `json:"open_time" gorm:"column:open_time"`
	EndTime    int64  `json:"end_time" gorm:"column:end_time"`
	UserId     int64  `json:"user_id" gorm:"column:user_id"`
	activeId   string `gorm:"-"`
}

func (c *Comcode) GetActiveId() string {
	return c.activeId
}

func (this *Comcode) SetActiveId(activeId string) {
	this.activeId = activeId
}

func (this Comcode) TableName() string {
	return fmt.Sprintf("common_code_%v", this.activeId)
}

func (this *Comcode) OneComcode(code string, db *gorm.DB) error {
	return db.Table(this.TableName()).Where("code = ?", code).First(this).Error
}

func (this *Comcode) CreateOne(db *gorm.DB) error {
	var count int64
	db.Table(this.TableName()).Where("awards_id = ? and user_id=?", this.AwardsId, this.UserId).Count(&count)
	if count > 0 {
		return errors.New("count>0")
	}
	db.Table(this.TableName()).Create(this)
	return nil
}

func (this *Comcode) UpdateUserId(db *gorm.DB) error {
	return db.Table(this.TableName()).Where("code = ?", this.Code).Updates(map[string]interface{}{
		"user_id":   this.UserId,
		"used_time": time.Now().Format("2006-01-02 15:04:05"),
	}).Error
}

func (this *Comcode) CountByUserid(db *gorm.DB, userid int64) (count int64) {
	db.Table(this.TableName()).Where("user_id = ? and awards_id = ?", userid, this.AwardsId).Count(&count)
	return count
}

func (this *Comcode) CountByUseridAndDate(db *gorm.DB, userid int64, date time.Time) (count int64) {
	db.Table(this.TableName()).Where("user_id = ? and awards_id = ? and DATE(used_time) = ?",
		userid, this.AwardsId, date.Format("2006-01-02")).Count(&count)
	return count
}

func GetLimitData(activeId string, userId string, tm string) int {
	data := rds.Redis.HGet(fmt.Sprintf(constant.ComCodeLimt, tm), fmt.Sprintf("%v%v", activeId, userId)).Val()
	if data == "" {
		return 0
	} else {
		number, err := strconv.Atoi(data)
		if err != nil {
			return 0
		}
		return number
	}
}

func SetLimitData(activeId string, userId string, tm string, data int) bool {
	rds.Redis.HSet(fmt.Sprintf(constant.ComCodeLimt, tm), fmt.Sprintf("%v%v", activeId, userId), fmt.Sprintf("%v", data))
	end, _ := time.Parse("2006-01-02", tm)
	expireAt := end.Add(25 * time.Hour)
	log.Printf("---更新领取兑换卷过期时间: expireAt:%v \n", expireAt)
	rds.Redis.ExpireAt(fmt.Sprintf(constant.ComCodeLimt, tm), expireAt)
	return true
}

func (this *Comcode) JudgeExp(timeUnix int64) bool {
	if this.OpenTime != 0 && timeUnix < this.OpenTime {
		return false
	}
	if this.EndTime != 0 && timeUnix > this.EndTime {
		return false
	}
	return true
}