hongze_task/services/roadshow/report.go

package roadshow

import (
	"context"
	"errors"
	"fmt"
	"hongze/hongze_task/models"
	"hongze/hongze_task/models/roadshow"
	"hongze/hongze_task/utils"
	"strconv"
	"strings"
	"time"
)

func AddReportRecord(cont context.Context) (err error) {
	defer func() {
		if err != nil {
			fmt.Println("AddReportRecord Err:" + err.Error())
		}
	}()
	adminAll, err := models.GetAdminList()
	if err != nil {
		return
	}
	adminMap := make(map[int]*models.AdminView)
	mapRaiSllerId := make(map[int]bool) // 是否为权益销售
	for _, v := range adminAll {
		adminMap[v.AdminId] = v
		if v.RoleTypeCode == utils.ROLE_TYPE_CODE_RAI_SELLER || v.RoleTypeCode == utils.ROLE_TYPE_CODE_RAI_GROUP || v.RoleTypeCode == utils.ROLE_TYPE_CODE_RAI_ADMIN {
			mapRaiSllerId[v.AdminId] = true
		}
	}

	companyList, err := roadshow.RsCompanyList()
	if err != nil {
		return
	}

	companyListRai, err := roadshow.RsCompanyListRai() //权益客户信息
	if err != nil {
		return
	}
	companyMap := make(map[int]*roadshow.CompanySearchView)
	companyMapRai := make(map[int]*roadshow.CompanySearchView)
	for _, v := range companyList {
		companyMap[v.CompanyId] = v
	}

	for _, v := range companyListRai {
		companyMapRai[v.CompanyId] = v
	}

	endDate := time.Now().AddDate(-1, 0, 0).Format(utils.FormatDate)

	existList, err := roadshow.GetExistRsReportRecord(endDate)
	if err != nil {
		return
	}
	existMap := make(map[string]*roadshow.RsReportRecord)
	var companyIds []int
	for _, v := range existList {
		key := strconv.Itoa(v.RsCalendarId) + "_" + strconv.Itoa(v.RsCalendarResearcherId)
		existMap[key] = v
		if v.CompanyId > 0 {
			companyIds = append(companyIds, v.CompanyId)
		}
	}
	mapPermissionNameFicc := make(map[int][]string)
	mapPermissionNameRai := make(map[int][]string)

	if len(companyIds) > 0 {
		listPermissionNameFicc, errFicc := models.GetCompanyProductReportPermissionList(companyIds, 1) // FICC权限信息
		if errFicc != nil {
			err = errFicc
			return
		}
		mapPermissionArrFicc := make(map[string]bool)
		for _, v := range listPermissionNameFicc {
			if mapPermissionArrFicc[fmt.Sprint(v.CompanyId, v.PermissionName)] || v.PermissionName == "" {
				continue
			}
			mapPermissionNameFicc[v.CompanyId] = append(mapPermissionNameFicc[v.CompanyId], v.PermissionName)
			mapPermissionArrFicc[fmt.Sprint(v.CompanyId, v.PermissionName)] = true
		}

		listPermissionNameRai, errRai := models.GetCompanyProductReportPermissionList(companyIds, 2) // 权益权限信息
		if errRai != nil {
			err = errRai
			return
		}

		mapPermissionArrRai := make(map[string]bool)
		for _, v := range listPermissionNameRai {
			if mapPermissionArrRai[fmt.Sprint(v.CompanyId, v.PermissionName)] || v.PermissionName == "" {
				continue
			}
			mapPermissionNameRai[v.CompanyId] = append(mapPermissionNameRai[v.CompanyId], v.PermissionName)
			mapPermissionArrRai[fmt.Sprint(v.CompanyId, v.PermissionName)] = true
		}
	}

	list, err := roadshow.GetRsCalendarDetail(endDate)
	if err != nil {
		return
	}
	for _, v := range list {
		company := new(roadshow.CompanySearchView)
		if v.CompanyId > 0 {
			if mapRaiSllerId[v.SysUserId] {
				company, _ = companyMapRai[v.CompanyId] // 如果是权益销售添加的信息，就用权益相关的客户信息
				if len(mapPermissionNameRai[v.CompanyId]) > 0 {
					company.PermissionName = strings.Join(mapPermissionNameRai[v.CompanyId], "/")
				}
			} else {
				company, _ = companyMap[v.CompanyId]
				if len(mapPermissionNameFicc[v.CompanyId]) > 0 {
					company.PermissionName = strings.Join(mapPermissionNameFicc[v.CompanyId], "/")
				}
			}
		}

		researcher, ok := adminMap[v.ResearcherId]
		if !ok {
			err = errors.New("研究员信息不存在:" + strconv.Itoa(v.ResearcherId))
			//continue
		}

		seller, ok := adminMap[v.SysUserId]
		if !ok {
			err = errors.New("销售信息不存在:" + strconv.Itoa(v.SysUserId))
			//return
			//continue
		}

		key := strconv.Itoa(v.RsCalendarId) + "_" + strconv.Itoa(v.RsCalendarResearcherId)
		if _, ok := existMap[key]; ok { //修改
			utils.FileLog.Info("update:" + strconv.Itoa(v.RsCalendarId))
			reportWhereParams := make(map[string]interface{})
			reportWhereParams["rs_calendar_id"] = v.RsCalendarId
			reportWhereParams["rs_calendar_researcher_id"] = v.RsCalendarResearcherId
			reportUpdateParams := make(map[string]interface{})
			if company != nil {
				reportUpdateParams["company_id"] = company.CompanyId
				reportUpdateParams["company_name"] = company.CompanyName
				reportUpdateParams["credit_code"] = company.CreditCode
				reportUpdateParams["company_status"] = company.Status
				reportUpdateParams["permission_name"] = company.PermissionName
			}
			reportUpdateParams["activity_type"] = v.ActivityType
			if researcher != nil {
				reportUpdateParams["researcher_id"] = researcher.AdminId
				reportUpdateParams["researcher_name"] = researcher.RealName
				reportUpdateParams["researcher_group_id"] = researcher.GroupId
				reportUpdateParams["researcher_group_name"] = researcher.GroupName
			} else {
				reportUpdateParams["researcher_id"] = v.ResearcherId
				reportUpdateParams["researcher_name"] = v.ResearcherName
				reportUpdateParams["researcher_group_id"] = 0
				reportUpdateParams["researcher_group_name"] = ""
			}

			if seller != nil {
				reportUpdateParams["seller_id"] = seller.AdminId
				reportUpdateParams["seller_name"] = seller.RealName
				reportUpdateParams["seller_group_id"] = seller.GroupId
				reportUpdateParams["seller_group_name"] = seller.GroupName
			} else {
				reportUpdateParams["seller_id"] = v.SysUserId
				reportUpdateParams["seller_name"] = v.SysUserRealName
				reportUpdateParams["seller_group_id"] = 0
				reportUpdateParams["seller_group_name"] = ""
			}
			reportUpdateParams["start_date"] = v.StartDate
			reportUpdateParams["end_date"] = v.EndDate
			reportUpdateParams["start_time"] = v.StartTime
			reportUpdateParams["end_time"] = v.EndTime
			reportUpdateParams["start_week"] = v.StartWeek
			reportUpdateParams["end_week"] = v.EndWeek
			reportUpdateParams["rs_calendar_researcher_status"] = v.Status
			reportUpdateParams["activity_type"] = v.ActivityType
			reportUpdateParams["roadshow_type"] = v.RoadshowType
			reportUpdateParams["modify_time"] = time.Now()
			err = roadshow.UpdateRsReportRecord(reportWhereParams, reportUpdateParams)
		} else { //新增
			item := new(roadshow.RsReportRecord)
			if company != nil {
				item.CompanyId = company.CompanyId
				item.CompanyName = company.CompanyName
				item.CreditCode = company.CreditCode
				item.CompanyStatus = company.Status
				item.PermissionName = company.PermissionName
			}
			item.ActivityType = v.ActivityType
			item.ResearcherId = v.ResearcherId
			item.ResearcherName = v.ResearcherName
			if researcher != nil {
				item.ResearcherGroupId = researcher.GroupId
				item.ResearcherGroupName = researcher.GroupName
			} else {
				item.ResearcherGroupId = 0
				item.ResearcherGroupName = ""
			}

			if seller != nil {
				item.SellerId = seller.AdminId
				item.SellerName = seller.RealName
				item.SellerGroupId = seller.GroupId
				item.SellerGroupName = seller.GroupName
			} else {
				item.SellerId = v.SysUserId
				item.SellerName = v.SysUserRealName
				item.SellerGroupId = 0
				item.SellerGroupName = ""
			}
			utils.FileLog.Info("add:" + strconv.Itoa(v.RsCalendarId))
			item.RsCalendarId = v.RsCalendarId
			item.RsCalendarResearcherId = v.RsCalendarResearcherId
			item.StartDate = v.StartDate
			item.EndDate = v.EndDate
			item.StartTime = v.StartTime
			item.EndTime = v.EndTime
			item.StartWeek = v.StartWeek
			item.EndWeek = v.EndWeek
			item.CreateTime = time.Now()
			item.ModifyTime = time.Now()
			item.RsCalendarResearcherStatus = v.Status
			item.RoadshowType = v.RoadshowType
			_, err = roadshow.AddRsReportRecord(item)
			if err != nil {
				return
			}
		}
	}
	return err
}