package overseas_custom

import (
	"context"
	"fmt"
	"hongze/hongze_task/models/overseas_custom"
)

// 修改英文客户路演统计
func FixEnglishCompanyRoadShowTotal(cont context.Context) (err error) {
	obj := new(overseas_custom.Custom)
	list, err := obj.GetRsCompanyTotal()
	if err != nil {
		fmt.Println("GetRsCompanyTotal Err:" + err.Error())
		return err
	}

	companyList, err := obj.GetEnglishCompanyAll()
	if err != nil {
		fmt.Println("GetEnglishCompanyAll Err:" + err.Error())
		return err
	}

	companyMap := make(map[int]int)
	for _, v := range companyList {
		companyMap[v.CompanyId] = v.RoadShowTotal
	}

	for _, v := range list {
		if v.Total > 0 {
			if val, ok := companyMap[v.CompanyId]; ok {
				if val != v.Total {
					err = obj.UpdateRoadShowTotal(v.CompanyId, v.Total)
					if err != nil {
						fmt.Println("UpdateRoadShowTotal Err:" + err.Error())
						return err
					}
				}
			}

		}
	}
	return nil
}