resourceCenter/public/service/surplus.go

package service

import (
	"errors"
	"time"

	. "app.yhyue.com/moapp/jybase/date"
	. "bp.jydev.jianyu360.cn/BaseService/resourceCenter/public/entity"
)

/*
 * 获取资源剩余情况
 * @param appid
 * @param function_code 功能代码
 * @param user_id 用户id
 * @param ent_id 企业id
 * @return 状态 0：失败 1：成功 -1：不在有效期内 -2：数量不足 -3：没有授权
 * @return 总数
 * @return 剩余数
 * @return 错误信息
 */
func Surplus(appid, function_code string, user_id, ent_id, ent_user_id int64) (int64, int64, int64, error) {
	function := Base_function.FindByCode(appid, function_code)
	if function == nil {
		return 0, 0, 0, errors.New("功能原始表中没有找到该功能")
	}
	myPowers := Base_power.FindMyPower(appid, function_code, user_id, ent_id)
	if myPowers == nil || len(*myPowers) == 0 {
		return -1, 0, 0, errors.New("不在有效期内")
	}
	if function.Power_rule == 1 { //只对周期进行校验
		return 1, 0, 0, nil
	}
	if function.Power_rule == 2 { //周期+数量校验
		use_count, surplus_count := int64(0), int64(0)
		for _, v := range *myPowers {
			if v.Power_type == 2 && !Base_ent_empower.HasEmpower(appid, function_code, ent_id, ent_user_id) {
				return -3, 0, 0, errors.New("没有对该用户进行授权")
			}
			use_count += v.Use_count
			surplus_count += v.Surplus_count
		}
		if surplus_count > 0 {
			return 1, use_count, surplus_count, nil
		} else {
			return -2, use_count, surplus_count, nil
		}
	} else if function.Power_rule == 3 { //周期+频率+数量校验
		for _, v := range *myPowers {
			if v.Power_type == 2 && !Base_ent_empower.HasEmpower(appid, function_code, ent_id, ent_user_id) {
				return -3, 0, 0, errors.New("没有对该用户进行授权")
			}
			use, err := Base_resource_use.FindLastOne(appid, function_code, user_id, ent_id, v.Power_type)
			if err != nil {
				return 0, 0, 0, errors.New("查询资源使用记录失败")
			} else if use == nil {
				return 1, v.Strategy_count, v.Strategy_count, nil
			}
			create_time, err := time.ParseInLocation(Date_Full_Layout, use.Create_time, time.Local)
			if err != nil {
				return 0, 0, 0, errors.New("资源充值/消耗表创建时间错误，" + err.Error())
			}
			now := time.Now()
			if (*myPowers)[0].Limit_strategy == "1d" {
				if create_time.Year() == now.Year() && create_time.Month() == now.Month() && create_time.Day() == now.Day() {
					if use.Surplus_count > 0 {
						return 1, v.Strategy_count, use.Surplus_count, nil
					} else {
						return -2, v.Strategy_count, use.Surplus_count, nil
					}
				} else {
					return 1, v.Strategy_count, v.Strategy_count, nil
				}
			} else if (*myPowers)[0].Limit_strategy == "1m" {
				if create_time.Year() == now.Year() && create_time.Month() == now.Month() {
					if use.Surplus_count > 0 {
						return 1, v.Strategy_count, use.Surplus_count, nil
					} else {
						return -2, v.Strategy_count, use.Surplus_count, nil
					}
				} else {
					return 1, v.Strategy_count, v.Strategy_count, nil
				}
			} else {
				return 0, 0, 0, errors.New("limit_strategy格式错误")
			}
		}
	}
	return 0, 0, 0, nil
}