hd/controllers/pc/vote.go

package pc

import (
	"fmt"
	"hudongzhuanjia/controllers"
	"hudongzhuanjia/models"
	"hudongzhuanjia/utils"
	"hudongzhuanjia/utils/code"
	"hudongzhuanjia/utils/define"
	"strconv"
	"time"

	"github.com/ouxuanserver/osmanthuswine/src/core"
)

type VoteCtl struct {
	controllers.AuthorCtl
}

//二维码
func (t *VoteCtl) Qrcode() {
	qrcode, err := utils.GenH5Qrcode(define.H5Vote, map[string]interface{}{
		"activity_id":      t.MustGetInt64("activity_id"),
		"customer_id":      t.MustGetUID(),
		"vote_activity_id": t.MustGetInt64("vote_activity_id"),
	})
	t.CheckErr(err)

	//直接输出二维码
	t.JSON(map[string]interface{}{
		"qrcode": qrcode,
	})
}

// 装备数据
func (t *VoteCtl) ReadyVote() {
	voteId := t.MustGetInt64("vote_activity_id")
	vote := new(models.NewVoteActivity)
	exist, err := models.GetById(vote, voteId)
	t.CheckErr(err)
	t.Assert(exist, code.MSG_MODULE_NOT_EXIST, "投票活动不存在")
	if vote.VoteStatus != define.StatusNotBegin {
		t.ERROR(fmt.Sprintf("该活动%s", vote.VoteStatus), code.MSG_ERR)
	}
	_, err = vote.UpdateToStatusByAid(vote.ActivityId, define.StatusReady, define.StatusNotBegin)
	t.CheckErr(err)

	// 初始化某些数据， 例如投票的初始票数
	// 彩排结束恢复某些数据
	ladders := make([]*models.NewVoteActivityLadder, 0)
	err = core.GetXormAuto().Where("is_delete=0 and vote_activity_id=?", vote.Id).Find(&ladders)
	t.CheckErr(err)
	for _, ladder := range ladders {
		ladder.TotalNumber = ladder.VoteNumber
		_, err = core.GetXormAuto().Where("is_delete=0 and id=?", ladder.Id).
			Cols("total_number").Update(ladder)
		t.CheckErr(err)
	}

	_, err = vote.UpdateStatusById(vote.Id, define.StatusReady)
	t.CheckErr(err)
	t.SUCCESS("success")
}

//开始投票
func (t *VoteCtl) StartVote() {
	voteId := t.MustGetInt64("vote_activity_id")
	vote := new(models.NewVoteActivity)
	exist, err := models.GetById(vote, voteId)
	t.CheckErr(err)
	t.Assert(exist, code.MSG_MODULE_NOT_EXIST, "投票活动不存在")

	if vote.VoteStatus != define.StatusReady {
		t.ERROR(fmt.Sprintf("该活动%s", vote.VoteStatus), code.MSG_ERR)
	}
	vote.VoteStatus = define.StatusRunning
	vote.UpdatedAt = time.Now()
	if vote.VoteLastTime == "" {
		vote.VoteLastTime = "0"
	}
	duration, _ := strconv.ParseInt(vote.VoteLastTime, 10, 64)
	vote.VoteEndTime = time.Now().Unix() + duration
	_, err = core.GetXormAuto().Where("is_delete=0 and id=?", voteId).
		Cols("vote_status", "vote_end_time", "updated_at").Update(vote)
	t.CheckErr(err)
	t.STRING("操作成功")
}

//停止投票
func (t *VoteCtl) StopVote() {
	voteId := t.MustGetInt64("vote_activity_id")

	vote := new(models.NewVoteActivity)
	exist, err := models.GetById(vote, voteId)
	t.CheckErr(err)
	t.Assert(exist, code.MSG_MODULE_NOT_EXIST, "投票活动不存在")

	if vote.VoteStatus != define.StatusRunning {
		t.ERROR(fmt.Sprintf("该活动%s", vote.VoteStatus), code.MSG_ERR)
	}

	vote.VoteStatus = define.StatusEnding
	vote.UpdatedAt = time.Now()
	vote.VoteEndTime = 0
	_, err = core.GetXormAuto().Where("is_delete=0 and id=?", voteId).
		Cols("vote_status", "updated_at", "vote_end_time").Update(vote)
	t.CheckErr(err)

	t.STRING("操作成功")
}

//获取参与人数
func (t *VoteCtl) JoinTotal() {
	voteActivityId := t.MustGetInt64("vote_activity_id")
	rehearsalId := t.MustGetInt64("rehearsal_id")
	total, err := core.GetXormAuto().Distinct("user_id").Where("vote_activity_id=? and rehearsal_id=? and is_delete=0",
		voteActivityId, rehearsalId).Count(new(models.NewVoteActivityHistory))
	t.CheckErr(err)
	t.RAW(total)
}

//获取所有的投票活动
func (t *VoteCtl) List() {
	activityId := t.MustGetInt64("activity_id")

	list := make([]*models.NewVoteActivity, 0)
	total, err := core.GetXormAuto().Where("is_delete=0 and activity_id=?", activityId).
		Asc("created_at").FindAndCount(&list)
	t.CheckErr(err)

	ids := make([]int64, 0)
	for _, v := range list {
		ids = append(ids, v.Id)
	}
	ladders := make([]*models.NewVoteActivityLadder, 0)
	err = core.GetXormAuto().Where("is_delete=0").In("vote_activity_id", ids).Find(&ladders)
	t.CheckErr(err)

	for index := range list {
		for i := range ladders {
			if list[index].Id == ladders[i].VoteActivityId {
				list[index].VoteActivityLadders = append(list[index].VoteActivityLadders, ladders[i])
			}
		}
	}

	t.JSON(map[string]interface{}{
		"total": total,
		"list":  list,
	})
}

//获取投票的前几（头像、姓名、票数）
func (t *VoteCtl) History() {
	voteActivityId := t.MustGetInt64("vote_activity_id")
	//rehearsalId := t.MustGetInt64("rehearsal_id")
	total := t.MustGetInt("total")

	ladders := make([]*models.NewVoteActivityLadder, 0)
	err := core.GetXormAuto().Where("is_delete=0 and vote_activity_id=?", voteActivityId).
		Desc("total_number").Asc("updated_at").Limit(total).Find(&ladders)
	t.CheckErr(err)
	t.RAW(ladders)
}