bilibili-go-common/app/job/main/growup/service/creative_activity.go

package service

import (
	"bytes"
	"context"
	"fmt"
	"sort"
	"strconv"
	"strings"
	"time"

	"go-common/app/job/main/growup/model"

	"go-common/library/log"
	xtime "go-common/library/time"
	"go-common/library/xstr"
)

const (
	_like  = "1"
	_share = "2"
	_play  = "3"
	_reply = "4"
	_dm    = "5"
)

// CreativeActivity creative activity job
func (s *Service) CreativeActivity(c context.Context, date time.Time) (err error) {
	activities, err := s.dao.GetCActivities(c)
	if err != nil {
		log.Error("s.dao.GetCActivities error(%v)", err)
		return
	}
	// get signed up
	ups, err := s.signed(c, int64(_dbLimit))
	if err != nil {
		log.Error("s.signed error(%v)", err)
		return
	}
	now := xtime.Time(date.Unix())
	// calculate activity within the statistical period
	for _, ac := range activities {
		if now >= ac.StatisticsStart && now <= ac.StatisticsEnd {
			log.Info("calculate ac: %d", ac.ID)
			err = s.handleActivity(c, ac, ups, date)
			if err != nil {
				log.Error("s.handleActivity error(%v)", err)
				return
			}
		}
	}
	return
}

func (s *Service) handleActivity(c context.Context, ac *model.CActivity, upInfo map[int64]*model.UpInfoVideo, date time.Time) (err error) {
	// 获取已报名并且在签约时间内的up主，
	signedUps, err := s.getSignedUps(c, ac, upInfo)
	if err != nil {
		log.Error("s.getSignedUps error(%v)", err)
		return
	}
	// 获取活动稿件点赞、分享、播放、评论、弹幕
	archiveInfo, err := s.getArchiveInfo(c, ac.ID)
	if err != nil {
		log.Error("s.getArchiveInfo error(%v)", err)
		return
	}
	if len(archiveInfo) == 0 {
		log.Info("activity(%d) get 0 archiveInfo", ac.ID)
		return
	}
	log.Info("get %d archiveInfo", len(archiveInfo))
	fmt.Printf("get %d archiveInfo\n", len(archiveInfo))
	// 获取在投稿时间内的稿件
	avs, err := s.getAvsUpload(c, signedUps)
	if err != nil {
		log.Error("s.getAvsByMID error(%v)", err)
		return
	}
	if len(avs) == 0 {
		log.Info("activity(%d) get 0 av", ac.ID)
		return
	}
	log.Info("get %d archives", len(avs))
	fmt.Printf("get %d archives\n", len(avs))

	// 筛选在签约时间内的稿件
	handleActivityAvs(signedUps, avs, ac)
	for _, u := range signedUps {
		// 获取up主对应稿件要求的和
		err = s.getUpStatisData(c, u, ac, date, archiveInfo)
		if err != nil {
			log.Error("s.getUpData error(%v)", err)
			return
		}
	}
	// 计算中奖up主
	upBonus, err := s.calUpWin(c, signedUps, ac)
	if err != nil {
		log.Error("s.calUpBonus error(%v)", err)
		return
	}

	err = s.updateUpActivity(c, ac.ID, upBonus)
	if err != nil {
		log.Error("s.updateUpActivity error(%v)", err)
	}
	return
}

func (s *Service) calUpWin(c context.Context, ups map[int64]*model.UpActivity, ac *model.CActivity) (bonusUp []*model.UpActivity, err error) {
	activityBonus, err := s.dao.GetActivityBonus(c, ac.ID)
	if err != nil {
		log.Error("s.dao.GetActivityBonus error(%v)", err)
		return
	}
	bonusUp = make([]*model.UpActivity, 0)
	var (
		upList = make([]*model.UpActivity, len(ups))
		index  int
	)
	for _, up := range ups {
		upList[index] = up
		index++
	}
	sort.Slice(upList, func(i, j int) bool {
		return upList[i].ItemVal > upList[j].ItemVal
	})

	// 1:达标型 2:排序型
	if ac.WinType == 1 {
		for _, up := range upList {
			// uplist经过ItemVal,所以如果有ItemVal<RequireValue,可以直接break
			if up.ItemVal < ac.RequireValue {
				break
			}
			bonusUp = append(bonusUp, up)
		}
	} else if ac.WinType == 2 {
		// 去掉item为0的up
		for i := 0; i < len(upList); i++ {
			if upList[i].ItemVal == 0 {
				upList = upList[:i]
				break
			}
		}
		if len(upList) > int(ac.RequireValue) {
			upList = upList[:int(ac.RequireValue)]
		}
		bonusUp = upList
	}
	if len(bonusUp) == 0 {
		return
	}
	// 计算up主奖金
	calUpBonus(bonusUp, activityBonus, ac.BonusType, ac.WinType)
	return
}

// cal ups bonus
func calUpBonus(bonusUp []*model.UpActivity, activityBonus map[int64]int64, bonusType, winType int) (err error) {
	// 中奖类型 1:达标型 2:排序型
	// 奖金类型 1:平分 2:各得
	if winType == 1 {
		money, ok := activityBonus[0]
		if !ok {
			err = fmt.Errorf("活动奖金设置错误:达标型未设置金额")
			return
		}
		if bonusType == 1 {
			money = money / int64(len(bonusUp))
		}
		for _, up := range bonusUp {
			up.Bonus = money
			up.Rank = 0
			if up.State < 2 {
				// 中奖
				up.State = 2
				up.SuccessTime = xtime.Time(time.Now().Unix())
			}
		}
	} else if winType == 2 {
		other := len(activityBonus)
		otherMoney, ok := activityBonus[int64(other)]
		if !ok {
			err = fmt.Errorf("活动奖金设置错误:排序型没有其他金额")
			return
		}
		for i := 0; i < len(bonusUp); i++ {
			var (
				rank  = i + 1
				money int64
				ok    bool
			)
			if rank >= other {
				money = otherMoney
			} else {
				money, ok = activityBonus[int64(rank)]
				if !ok {
					err = fmt.Errorf("活动奖金设置错误:没有名次金额")
					return
				}
			}
			if money == 0 {
				continue
			}
			bonusUp[i].Bonus = money
			bonusUp[i].Rank = i + 1
			if bonusUp[i].State < 2 {
				// 中奖
				bonusUp[i].State = 2
				bonusUp[i].SuccessTime = xtime.Time(time.Now().Unix())
			}
		}
	}
	return
}

func (s *Service) getUpStatisData(c context.Context, up *model.UpActivity, ac *model.CActivity, date time.Time, archiveInfo map[int64]map[int]*model.ArchiveStat) (err error) {
	if len(up.AIDs) == 0 {
		return
	}
	avItem := make([]*model.AvItem, 0)
	aIDs := up.AIDs
	for _, avID := range aIDs {
		// 稿件统计数据: 计算当天的和 - 统计开始前一天的和
		var itemSumStart, itemSumEnd int64
		itemSumEnd, err = s.getAvStatisState(c, avID, ac, archiveInfo, 2)
		if err != nil {
			log.Error("s.getAvStatisState error(%v)", err)
			return
		}
		itemSumStart, err = s.getAvStatisState(c, avID, ac, archiveInfo, 1)
		if err != nil {
			log.Error("s.getAvStatisState error(%v)", err)
			return
		}
		itemSum := itemSumEnd - itemSumStart
		if itemSum > 0 {
			avItem = append(avItem, &model.AvItem{
				AvID:  avID,
				Value: itemSum,
			})
		}
	}
	if len(avItem) == 0 {
		return
	}
	sort.Slice(avItem, func(i, j int) bool {
		return avItem[i].Value > avItem[j].Value
	})

	// 1:uid 2 avid
	if ac.Object == 1 {
		up.AIDs = make([]int64, 0)
		for _, av := range avItem {
			if len(up.AIDs) < 5 {
				up.AIDs = append(up.AIDs, av.AvID)
			}
			up.ItemVal += av.Value
		}
		up.AIDNum = int64(len(avItem))
	} else if ac.Object == 2 {
		up.AIDs = []int64{avItem[0].AvID}
		up.ItemVal = avItem[0].Value
		up.AIDNum = 1
	}
	return
}

func (s *Service) getAvStatisState(c context.Context, avID int64, ac *model.CActivity, archiveInfo map[int64]map[int]*model.ArchiveStat, state int) (sum int64, err error) {
	if _, ok := archiveInfo[avID]; !ok {
		return
	}
	stat, ok := archiveInfo[avID][state]
	if !ok {
		return
	}
	requireItems := strings.Split(ac.RequireItems, ",")
	for _, item := range requireItems {
		switch item {
		case _like:
			sum += stat.Like
		case _share:
			sum += stat.Share
		case _play:
			sum += stat.Play
		case _reply:
			sum += stat.Reply
		case _dm:
			sum += stat.Dm
		}
	}
	return
}

func handleActivityAvs(ups map[int64]*model.UpActivity, avs []*model.AvUpload, ac *model.CActivity) {
	for _, av := range avs {
		if !(av.UploadTime >= ac.UploadStart && av.UploadTime <= ac.UploadEnd) {
			continue
		}
		if _, ok := ups[av.MID]; ok {
			if len(ups[av.MID].AIDs) == 0 {
				ups[av.MID].AIDs = make([]int64, 0)
			}
			ups[av.MID].AIDs = append(ups[av.MID].AIDs, av.AvID)
		}
	}
}

func (s *Service) getAvsUpload(c context.Context, ups map[int64]*model.UpActivity) (avs []*model.AvUpload, err error) {
	avs = make([]*model.AvUpload, 0)
	var id int64
	for {
		var av []*model.AvUpload
		av, err = s.dao.GetAvUploadByMID(c, id, _dbLimit)
		if err != nil {
			return
		}
		avs = append(avs, av...)
		if len(av) < _dbLimit {
			break
		}
		id = av[len(av)-1].ID
	}
	return
}

func (s *Service) getArchiveInfo(c context.Context, activityID int64) (archives map[int64]map[int]*model.ArchiveStat, err error) {
	archives = make(map[int64]map[int]*model.ArchiveStat) // map[av_id][state]*model.ArchiveStat
	var id int64
	for {
		var arch []*model.ArchiveStat
		arch, err = s.dao.GetArchiveInfo(c, activityID, id, _dbLimit)
		if err != nil {
			return
		}
		for _, a := range arch {
			if _, ok := archives[a.AvID]; !ok {
				archives[a.AvID] = make(map[int]*model.ArchiveStat)
			}
			archives[a.AvID][a.State] = a
		}
		if len(arch) < _dbLimit {
			break
		}
		id = arch[len(arch)-1].ID
	}
	return
}

func (s *Service) getSignedUps(c context.Context, ac *model.CActivity, upInfo map[int64]*model.UpInfoVideo) (signedUps map[int64]*model.UpActivity, err error) {
	signedUps = make(map[int64]*model.UpActivity)
	// if need sign up
	if ac.SignUp == 1 {
		var ups []*model.UpActivity
		ups, err = s.dao.ListUpActivity(c, ac.ID)
		if err != nil {
			return
		}
		for _, up := range ups {
			if info, ok := upInfo[up.MID]; ok && info.SignedAt >= ac.SignedStart && info.SignedAt <= ac.SignedEnd {
				signedUps[up.MID] = up
			}
		}
	} else {
		for _, info := range upInfo {
			if info.SignedAt >= ac.SignedStart && info.SignedAt <= ac.SignedEnd {
				signedUps[info.MID] = &model.UpActivity{MID: info.MID, ActivityID: ac.ID, State: 1, Nickname: info.Nickname}
			}
		}
	}
	return
}

func (s *Service) updateUpActivity(c context.Context, id int64, upBonus []*model.UpActivity) (err error) {
	// 更新所有之前获奖up主为已报名
	// update state 2->1 将所有状态设置为已报名
	_, err = s.dao.UpdateUpActivityState(c, id, 2, 1)
	if err != nil {
		log.Error("s.dao.UpdateUpActivityState error(%v)", err)
		return
	}
	if len(upBonus) > 0 {
		_, err = s.insertUpActivityBatch(c, upBonus)
		if err != nil {
			log.Error("s.insertUpActivity error(%v)", err)
		}
	}
	return
}
func (s *Service) insertUpActivityBatch(c context.Context, ups []*model.UpActivity) (rows int64, err error) {
	insert := make([]*model.UpActivity, _dbBatchSize)
	insertIndex := 0
	for _, up := range ups {
		insert[insertIndex] = up
		insertIndex++
		if insertIndex >= _dbBatchSize {
			_, err = s.insertUpActivity(c, insert[:insertIndex])
			if err != nil {
				return
			}
			insertIndex = 0
		}
	}
	if insertIndex > 0 {
		_, err = s.insertUpActivity(c, insert[:insertIndex])
	}
	return
}

func assembleUpActivity(ups []*model.UpActivity) (vals string) {
	var buf bytes.Buffer
	for _, row := range ups {
		buf.WriteString("(")
		buf.WriteString(strconv.FormatInt(row.MID, 10))
		buf.WriteByte(',')
		buf.WriteString("\"" + row.Nickname + "\"")
		buf.WriteByte(',')
		buf.WriteString(strconv.FormatInt(row.ActivityID, 10))
		buf.WriteByte(',')
		buf.WriteString("'" + xstr.JoinInts(row.AIDs) + "'")
		buf.WriteByte(',')
		buf.WriteString(strconv.FormatInt(row.AIDNum, 10))
		buf.WriteByte(',')
		buf.WriteString(strconv.Itoa(row.Rank))
		buf.WriteByte(',')
		buf.WriteString(strconv.FormatInt(row.Bonus, 10))
		buf.WriteByte(',')
		buf.WriteString(strconv.Itoa(row.State))
		buf.WriteByte(',')
		buf.WriteString("'" + row.SuccessTime.Time().Format("2006-01-02 15:04:05") + "'")
		buf.WriteString(")")
		buf.WriteByte(',')
	}
	if buf.Len() > 0 {
		buf.Truncate(buf.Len() - 1)
	}
	vals = buf.String()
	buf.Reset()
	return
}

func (s *Service) insertUpActivity(c context.Context, ups []*model.UpActivity) (rows int64, err error) {
	vals := assembleUpActivity(ups)
	rows, err = s.dao.InsertUpActivityBatch(c, vals)
	return
}