eta_api/services/data/supply_analysis/variety.go

package supply_analysis

import (
	"eta_gn/eta_api/models/data_manage/supply_analysis"
	"eta_gn/eta_api/models/system"
	"eta_gn/eta_api/utils"
	"time"
)

func IsVarietySuperAdmin(sysUser *system.Admin) (isSuperAdmin bool) {
	if sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_ADMIN || sysUser.RoleTypeCode == utils.ROLE_TYPE_CODE_FICC_ADMIN {
		isSuperAdmin = true
	}
	return
}

func HasVarietyPlantPermission(sysUser *system.Admin, varietyId int) (hasPermission bool, err error) {
	hasPermission = IsVarietySuperAdmin(sysUser)
	if hasPermission {
		return
	}

	tmp, err := supply_analysis.GetVarietyPermissionByVarietyIdAndUserId(varietyId, sysUser.AdminId)
	if err != nil && !utils.IsErrNoRow(err) {
		return
	}
	if tmp != nil { //找到记录那就代表有操作权限
		hasPermission = true
	}
	return
}

func CalculateByDay(varietyInfo *supply_analysis.Variety) (dataMap map[time.Time]float64, startDate, lastDate time.Time, err error) {
	plantList, err := supply_analysis.GetAllVarietyPlantByVarietyId(varietyInfo.VarietyId)
	if err != nil {
		return
	}

	{
		currDate := time.Now()
		currHour := currDate.Hour()
		if currHour < 22 {
			lastDate, _ = time.ParseInLocation(utils.FormatDate, currDate.Format(utils.FormatDate), time.Local)
		} else {
			lastDate, _ = time.ParseInLocation(utils.FormatDate, currDate.AddDate(0, 0, 1).Format(utils.FormatDate), time.Local)
		}

		lastDate = lastDate.AddDate(0, 0, 365) //业务需要计算的数据日期往当前日期后面加上配置的日期
	}

	maintenanceDataMap := make(map[time.Time][]float64)

	for _, plantInfo := range plantList {
		if plantInfo.IsStop == 1 { // 如果是停产了，那么结束日期就是到 产生的数据的结束日期
			plantInfo.ResumptionDate = lastDate.Format(utils.FormatDate)
		}
		if plantInfo.MaintenanceDate == `` || plantInfo.ResumptionDate == `` {
			continue
		}
		if plantInfo.AverageDailyCapacityReduction == 0 {
			continue
		}
		maintenanceDate, _ := time.ParseInLocation(utils.FormatDate, plantInfo.MaintenanceDate, time.Local)
		if startDate.IsZero() || maintenanceDate.Before(startDate) {
			startDate = maintenanceDate
		}

		resumptionDate, _ := time.ParseInLocation(utils.FormatDate, plantInfo.ResumptionDate, time.Local)

		for tmpDate := maintenanceDate; !tmpDate.After(resumptionDate); tmpDate = tmpDate.AddDate(0, 0, 1) {
			dataList, ok := maintenanceDataMap[tmpDate]
			if !ok {
				dataList = make([]float64, 0)
			}
			dataList = append(dataList, plantInfo.AverageDailyCapacityReduction)
			maintenanceDataMap[tmpDate] = dataList
		}
	}

	if len(maintenanceDataMap) <= 0 {
		return
	}

	dataMap = make(map[time.Time]float64)
	for tmpDate := startDate; !tmpDate.After(lastDate); tmpDate = tmpDate.AddDate(0, 0, 1) {
		var calculateVal float64
		tmpDataList, ok := maintenanceDataMap[tmpDate]
		if !ok {
			calculateVal = 0
		} else {
			for _, tmpVal := range tmpDataList {
				calculateVal += tmpVal
			}
		}

		dataMap[tmpDate] = calculateVal
	}

	return
}