eta_data_analysis/services/base_from_yongyi/daily.go

package base_from_yongyi

import (
	"eta/eta_data_analysis/models"
	"eta/eta_data_analysis/utils"
	"fmt"
	"github.com/tealeg/xlsx"
	"strings"
	"time"
)

var yongyiSheetMap map[string]*models.YongyiSheet
var yongyiNotAddMap map[string][]string
var yongyiSpecialNameMap map[string]map[string]string
var yongyiSpecialUnitMap map[string]string

func init() {
	yongyiSheetMap = map[string]*models.YongyiSheet{
		"出栏价":             {"商品猪出栏价", "日度-商品猪出栏价", "日度", "元/公斤"},
		"标肥价差":           {"", "日度-商品猪标肥价差", "日度", "元/公斤"},
		"价格+宰量":          {"", "", "日度", "元/公斤"},
		"屠宰企业日度屠宰量": {"商品猪日屠宰量", "日度-屠宰企业屠宰量", "日度", "头/日"},
	}
	yongyiNotAddMap = map[string][]string{"出栏价": {"商品猪出栏价/全国/均价"}, "屠宰企业日度屠宰量": {"商品猪日屠宰量/合计（单位：头）"}}
	yongyiSpecialNameMap = map[string]map[string]string{"价格+宰量": {"全国均价": "商品猪出栏价/全国/均价", "日屠宰量": "商品猪日屠宰量/全国"}}
	yongyiSpecialUnitMap = map[string]string{"商品猪日屠宰量/全国": "头/日"}
}

// HandleYongyiExcelDaily1 日度-商品猪出栏价
func HandleYongyiExcelDaily1(sheet *xlsx.Sheet) (indexList []*models.YongyiExcelIndex, err error) {
	classifyName := "日度-商品猪出栏价"
	frequency := "日度"
	unit := "元/公斤"
	namePrefix := "商品猪出栏价"
	namePrefixPingin := "yyzxspzclj"
	//sheet := dailyPriceSheet
	// 遍历行读取
	indexList = make([]*models.YongyiExcelIndex, 0)
	dateMap := make(map[int]string)
	nameMap := make(map[int]string)
	maxRow := sheet.MaxRow
	fmt.Println("最大行")
	fmt.Println(maxRow)
	// 指标名称
	indexMap := make(map[string]*models.YongyiExcelIndex)
	for i := 0; i < maxRow; i++ {
		fmt.Printf("当前第%d行 \n", i)
		if i == 0 { // 首行，表示时间
			row := sheet.Row(i)
			cells := row.Cells
			for k, cell := range cells {
				text := cell.String()
				if k > 1 && text != "" {
					if cell.IsTime() {
						dateText, _ := cell.GetTime(false)
						text = dateText.Format(utils.FormatDate)
					}
					// 检查单元格是否为合并单元格
					if cell.HMerge > 0 {
						for j := 1; j <= cell.HMerge; j++ {
							dateMap[k+j] = text
						}
					}
					fmt.Printf("合并单元格开始列：%d \n", k)
					dateMap[k] = text
				}
			}
		} else if i == 1 { //表示表头
			// 处理 index指标表
			row := sheet.Row(i)
			cells := row.Cells
			for k, cell := range cells {
				text := cell.String()
				nameMap[k] = text
			}
		} else { //数据列
			row := sheet.Row(i)
			cells := row.Cells
			province := ""
			for k, cell := range cells {
				fmt.Printf("当前第%d列 \n", k)
				text := cell.String()
				if k == 0 {
					province = text
					continue
				} else if k == 1 {
					continue
				}

				date, ok1 := dateMap[k]
				if !ok1 {
					err = fmt.Errorf("找不到对应的日期，第%d行，第%d列", i, k)
					return
				}
				name, ok2 := nameMap[k]
				if !ok2 {
					err = fmt.Errorf("找不到对应的列名，第%d行，第%d列", i, k)
					return
				}
				fmt.Printf("当前第%d行第%d列, 当前省份%s \n", i, k, province)
				// 处理指标名称
				fullIndexName := fmt.Sprintf("%s/%s/%s", namePrefix, province, name)

				if name != "规模场" && name != "小散户" && name != "均价" {
					// 只处理以上三个类型，其余过滤
					continue
				}

				if fullIndexName == "商品猪出栏价/全国/均价" {
					continue
				}

				indexItem, okIndex := indexMap[fullIndexName]

				provincePingyin := utils.GetFullPingYin(province)
				namePingin := utils.GetFirstPingYin(name)
				fullIndexNamePingyin := fmt.Sprintf("%s%s%s", namePrefixPingin, provincePingyin, namePingin)

				if !okIndex {
					// 新增指标
					indexItem = new(models.YongyiExcelIndex)
					indexItem.IndexName = fullIndexName
					indexItem.ClassifyName = classifyName
					indexItem.IndexCode = fullIndexNamePingyin
					indexItem.Frequency = frequency
					indexItem.Unit = unit
					indexItem.ExcelDataMap = make(map[string]string)
				}
				fmt.Printf("indexItem%s", indexItem.IndexCode)

				indexItem.ExcelDataMap[date] = text
				indexMap[fullIndexName] = indexItem
				continue
			}
		}
	}
	for _, v := range indexMap {
		indexList = append(indexList, v)
	}
	return
}

// HandleYongyiExcelDaily2 日度-商品猪标肥价差
func HandleYongyiExcelDaily2(sheet *xlsx.Sheet) (indexList []*models.YongyiExcelIndex, err error) {
	classifyName := "日度-商品猪标肥价差"
	frequency := "日度"
	unit := "元/公斤"
	pingYinPrefix := "yyzx"
	// 遍历行读取
	indexList = make([]*models.YongyiExcelIndex, 0)
	dateMap := make(map[int]string)
	nameMap := make(map[int]string)
	maxRow := sheet.MaxRow
	fmt.Println("最大行")
	fmt.Println(maxRow)
	// 指标名称
	indexMap := make(map[string]*models.YongyiExcelIndex)
	for i := 0; i < maxRow; i++ {
		fmt.Printf("当前第%d行 \n", i)
		if i == 0 { // 首行，表示时间
			row := sheet.Row(i)
			cells := row.Cells
			for k, cell := range cells {
				text := cell.String()
				if text != "" {
					if cell.IsTime() {
						dateText, _ := cell.GetTime(false)
						text = dateText.Format(utils.FormatDate)
					}
					// 检查单元格是否为合并单元格
					if cell.HMerge > 0 {
						for j := 1; j <= cell.HMerge; j++ {
							dateMap[k+j] = text
						}
					}
					fmt.Printf("合并单元格开始列：%d \n", k)
					dateMap[k] = text
				}
			}
		} else if i == 1 { //表示表头
			// 处理 index指标表
			row := sheet.Row(i)
			cells := row.Cells
			for k, cell := range cells {
				text := cell.String()
				nameMap[k] = text
			}
		} else { //数据列
			row := sheet.Row(i)
			cells := row.Cells
			province := ""
			for k, cell := range cells {
				fmt.Printf("当前第%d列 \n", k)
				text := cell.String()
				if k == 0 {
					province = text
					continue
				}

				date, ok1 := dateMap[k]
				if !ok1 {
					err = fmt.Errorf("找不到对应的日期，第%d行，第%d列", i, k)
					return
				}
				name, ok2 := nameMap[k]
				if !ok2 {
					err = fmt.Errorf("找不到对应的列名，第%d行，第%d列", i, k)
					return
				}
				fmt.Printf("当前第%d行第%d列, 当前省份%s \n", i, k, province)
				// 处理指标名称

				// 过滤特殊的指标 放到对应的指标名称的下方

				if name != "市场标重猪价" && name != "150公斤左右较标猪" && name != "175公斤左右较标猪" {
					// 只处理以上三个类型，其余过滤
					continue
				}
				var namePrefixPingin string
				if name == "150公斤左右较标猪" {
					name = "150公斤较标猪价差"
					namePrefixPingin = "150gjjbzjc"
				} else if name == "175公斤左右较标猪" {
					name = "175公斤较标猪价差"
					namePrefixPingin = "175gjjbzjc"
				} else if name == "市场标重猪价" {
					namePrefixPingin = "scbzzj"
				}
				fullIndexName := fmt.Sprintf("%s/%s", name, province)
				indexItem, okIndex := indexMap[fullIndexName]

				provincePingyin := utils.GetFullPingYin(province)
				fullIndexNamePingyin := fmt.Sprintf("%s%s%s", pingYinPrefix, namePrefixPingin, provincePingyin)

				if !okIndex {
					// 新增指标
					indexItem = new(models.YongyiExcelIndex)
					indexItem.IndexName = fullIndexName
					indexItem.ClassifyName = classifyName
					indexItem.IndexCode = fullIndexNamePingyin
					indexItem.Frequency = frequency
					indexItem.Unit = unit
					indexItem.ExcelDataMap = make(map[string]string)
				}
				fmt.Printf("indexItem%s", indexItem.IndexCode)

				indexItem.ExcelDataMap[date] = text
				indexMap[fullIndexName] = indexItem
				continue
			}
		}
	}
	for _, v := range indexMap {
		indexList = append(indexList, v)
	}
	return
}

// HandleYongyiExcelDaily3 日度-商品猪全国均价和宰量
func HandleYongyiExcelDaily3(sheet *xlsx.Sheet) (indexList []*models.YongyiExcelIndex, err error) {
	classifyName := "日度-商品猪全国均价和宰量"
	frequency := "日度"
	unit := ""
	// 遍历行读取
	indexList = make([]*models.YongyiExcelIndex, 0)
	nameMap := make(map[int]string)
	maxRow := sheet.MaxRow
	fmt.Println("最大行")
	fmt.Println(maxRow)
	// 指标名称
	indexMap := make(map[string]*models.YongyiExcelIndex)
	for i := 0; i < maxRow; i++ {
		fmt.Printf("当前第%d行 \n", i)
		if i == 0 { // 首行，名称
			row := sheet.Row(i)
			cells := row.Cells
			for k, cell := range cells {
				text := cell.String()
				nameMap[k] = text
			}
		} else { //数据列
			row := sheet.Row(i)
			cells := row.Cells
			date := ""
			for k, cell := range cells {
				fmt.Printf("当前第%d列 \n", k)
				text := cell.String()
				if k == 0 {
					if cell.IsTime() {
						dateText, _ := cell.GetTime(false)
						text = dateText.Format(utils.FormatDate)
					}
					date = text
					continue
				}

				name, ok2 := nameMap[k]
				if !ok2 {
					err = fmt.Errorf("找不到对应的列名，第%d行，第%d列", i, k)
					return
				}
				if name != "全国均价" && name != "日屠宰量" {
					// 只处理以上三个类型，其余过滤
					continue
				}
				var fullIndexNamePingyin string
				if name == "全国均价" {
					name = "商品猪出拦价/全国/均价"
					fullIndexNamePingyin = "yyzxspzcljquanguojj"
					unit = "元/公斤"
				} else if name == "日屠宰量" {
					name = "商品猪日屠宰量/全国"
					fullIndexNamePingyin = "yyzxspzrtzlquanguo"
					unit = "头/日"
				}
				fullIndexName := name
				indexItem, okIndex := indexMap[fullIndexName]

				if !okIndex {
					// 新增指标
					indexItem = new(models.YongyiExcelIndex)
					indexItem.IndexName = fullIndexName
					indexItem.ClassifyName = classifyName
					indexItem.IndexCode = fullIndexNamePingyin
					indexItem.Frequency = frequency
					indexItem.Unit = unit
					indexItem.ExcelDataMap = make(map[string]string)
				}
				fmt.Printf("indexItem%s", indexItem.IndexCode)

				indexItem.ExcelDataMap[date] = text
				indexMap[fullIndexName] = indexItem
			}
		}
	}
	for _, v := range indexMap {
		indexList = append(indexList, v)
	}
	return
}

// HandleYongyiExcelDaily4 日度-屠宰企业屠宰量
func HandleYongyiExcelDaily4(sheet *xlsx.Sheet) (indexList []*models.YongyiExcelIndex, err error) {
	classifyName := "日度-屠宰企业屠宰量"
	frequency := "日度"
	unit := "头/日"
	namePrefix := "商品猪日屠宰量"
	pingYinPrefix := "yyzx"
	namePrefixPingin := "spzrtzl"
	// 遍历行读取
	indexList = make([]*models.YongyiExcelIndex, 0)
	dateMap := make(map[int]string)
	maxRow := sheet.MaxRow
	fmt.Println("最大行")
	fmt.Println(maxRow)
	// 指标名称
	indexMap := make(map[string]*models.YongyiExcelIndex)
	for i := 0; i < maxRow; i++ {
		fmt.Printf("当前第%d行 \n", i)
		if i == 0 { // 首行，表示时间
			row := sheet.Row(i)
			cells := row.Cells
			for k, cell := range cells {
				text := cell.String()
				if text != "" && text != "省份" {
					if cell.IsTime() {
						dateText, _ := cell.GetTime(false)
						text = dateText.Format(utils.FormatDate)
					} else {
						if strings.Contains(text, "（") {
							tmp := strings.Split(text, "（")
							text = tmp[0]
						}
						timeTmp, tErr := time.ParseInLocation(utils.FormatDate1, text, time.Local)
						if tErr != nil {
							err = fmt.Errorf("时间格式转换失败，Err:%s", tErr)
							return
						}
						text = timeTmp.Format(utils.FormatDate)
					}

					dateMap[k] = text
				}
			}
		} else { //数据列
			row := sheet.Row(i)
			cells := row.Cells
			province := ""
			for k, cell := range cells {
				fmt.Printf("当前第%d列 \n", k)
				text := cell.String()
				if k == 0 {
					province = text
					continue
				}

				date, ok1 := dateMap[k]
				if !ok1 {
					err = fmt.Errorf("找不到对应的日期，第%d行，第%d列", i, k)
					return
				}
				fmt.Printf("当前第%d行第%d列, 当前省份%s \n", i, k, province)
				// 处理指标名称

				// 过滤特殊的指标 放到对应的指标名称的下方
				fullIndexName := fmt.Sprintf("%s/%s", namePrefix, province)

				provincePingyin := utils.GetFullPingYin(province)
				fullIndexNamePingyin := fmt.Sprintf("%s%s%s", pingYinPrefix, namePrefixPingin, provincePingyin)

				indexItem, okIndex := indexMap[fullIndexName]

				if !okIndex {
					// 新增指标
					indexItem = new(models.YongyiExcelIndex)
					indexItem.IndexName = fullIndexName
					indexItem.ClassifyName = classifyName
					indexItem.IndexCode = fullIndexNamePingyin
					indexItem.Frequency = frequency
					indexItem.Unit = unit
					indexItem.ExcelDataMap = make(map[string]string)
				}
				fmt.Printf("indexItem%s", indexItem.IndexCode)

				indexItem.ExcelDataMap[date] = text
				indexMap[fullIndexName] = indexItem
			}
		}
	}
	for _, v := range indexMap {
		indexList = append(indexList, v)
	}
	return
}