egg_system_rules/rule/user_upgrade.go

package rule

import (
	db "code.fnuoos.com/EggPlanet/egg_models.git/src"
	"code.fnuoos.com/EggPlanet/egg_models.git/src/implement"
	"code.fnuoos.com/EggPlanet/egg_models.git/src/model"
	utils "code.fnuoos.com/EggPlanet/egg_system_rules.git/utils"
	"errors"
	"fmt"
	"xorm.io/xorm"
)

/*
会员升级
uid 用户id
lv 要升级的等级
*/

func UserUpgradeTask(eg *xorm.Engine, uid, lv int) ([]map[string]string, error) {
	taskList := make([]map[string]string, 0)
	//获取任务
	NewUserLevelTaskDb := implement.NewUserLevelTaskDb(eg)
	levelTask, _ := NewUserLevelTaskDb.UserLevelTaskFindByParams(map[string]interface{}{
		"key":   "level_id",
		"value": lv,
	})
	if len(levelTask) == 0 {
		return taskList, errors.New("任务条件未设置")
	}
	NewUserLevelDb := implement.NewUserLevelDb(eg)
	level, _ := NewUserLevelDb.UserLevelByID(lv)
	if level == nil {
		return taskList, errors.New("等级不存在")
	}
	chooseNum := 0
	mustTaskErr := 0
	for _, v := range levelTask {
		isSuccess := 0
		var count int64 = 0
		switch v.TaskType {
		case 1: //直推会员人数
			count = ExtendUserCount(eg, uid)
			if count >= int64(v.FinishCount) {
				isSuccess = 1
			}
			break
		case 2: //直推会员活跃天数
			count = ExtendUserCountWithDay(eg, uid, v.WithinDays, v.ActiveDays)
			if count >= int64(v.FinishCount) {
				isSuccess = 1
			}
			break
		case 3: //直推会员等级
			count = ExtendUserCountWithLv(eg, uid, v.TaskTypeLevelId)
			if count >= int64(v.FinishCount) {
				isSuccess = 1
			}
			break
		}
		tmp := map[string]string{
			"id":         utils.IntToStr(v.Id),
			"count":      utils.Int64ToStr(count),
			"is_success": utils.IntToStr(isSuccess),
		}
		taskList = append(taskList, tmp)
		if isSuccess == 1 || v.IsMustTask == 0 { //记录选做任务数量
			chooseNum++
		}
		if isSuccess == 0 && v.IsMustTask == 1 { //必做任务未完成
			mustTaskErr = 1
		}
	}
	if mustTaskErr == 1 {
		return taskList, errors.New("必做任务未完成")
	}
	if level.ChoosableNum > chooseNum { //选做任务未完成
		return taskList, errors.New("选择任务未完成")
	}
	return taskList, nil
}

// 直推人数
func ExtendUserCount(eg *xorm.Engine, uid int) int64 {
	count, _ := eg.Where("parent_uid=?", uid).Count(&model.User{})
	return count
}

// 直推人数达到对应等级
func ExtendUserCountWithLv(eg *xorm.Engine, uid, lv int) int64 {
	count, _ := eg.Where("parent_uid=? and level=?", uid, lv).Count(&model.User{})
	return count
}

// X天直推X个用户活跃X天
func ExtendUserCountWithDay(eg *xorm.Engine, uid, day, activityDay int) int64 {
	var user []model.User
	today := utils.GetTimeRange("today")
	times := today["start"] - int64(day*86400)
	eg.Where("parent_uid=? and create_at>=?", uid, times).Find(&user)
	var num int64 = 0
	if len(user) > 0 {
		uids := ""
		for _, v := range user {
			if uids == "" {
				uids += utils.Int64ToStr(v.Id)
			} else {
				uids += "," + utils.Int64ToStr(v.Id)
			}
		}
		sql := `select count(*) as count from egg_energy_user_activity where uid in(%s) `
		sql = fmt.Sprintf(sql, uids)
		nativeString, _ := db.QueryNativeString(eg, sql)
		for _, v := range nativeString {
			if utils.StrToInt(v["count"]) >= activityDay {
				num++
			}
		}
	}
	return num
}