|
@@ -0,0 +1,426 @@
|
|
|
+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 := "商品猪出栏价"
|
|
|
+
|
|
|
+ //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]
|
|
|
+
|
|
|
+ if !okIndex {
|
|
|
+ // 新增指标
|
|
|
+ indexItem = new(models.YongyiExcelIndex)
|
|
|
+ indexItem.IndexName = fullIndexName
|
|
|
+ indexItem.ClassifyName = classifyName
|
|
|
+ // todo 处理indexCode
|
|
|
+ indexItem.IndexCode = fullIndexName
|
|
|
+ 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)
|
|
|
+ }
|
|
|
+ // todo 整理请求入参,并加入,改成一个指标请求一次还是多个指标请求一次
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+// HandleYongyiExcelDaily2 日度-商品猪标肥价差
|
|
|
+func HandleYongyiExcelDaily2(sheet *xlsx.Sheet) (indexList []*models.YongyiExcelIndex, err error) {
|
|
|
+ classifyName := "日度-商品猪标肥价差"
|
|
|
+ frequency := "日度"
|
|
|
+ unit := "元/公斤"
|
|
|
+
|
|
|
+ // 遍历行读取
|
|
|
+ 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
|
|
|
+ }
|
|
|
+ if name == "150公斤左右较标猪" {
|
|
|
+ name = "150公斤较标猪价差"
|
|
|
+ } else if name == "175公斤左右较标猪" {
|
|
|
+ name = "175公斤较标猪价差"
|
|
|
+ }
|
|
|
+ fullIndexName := fmt.Sprintf("%s/%s", name, province)
|
|
|
+
|
|
|
+ indexItem, okIndex := indexMap[fullIndexName]
|
|
|
+
|
|
|
+ if !okIndex {
|
|
|
+ // 新增指标
|
|
|
+ indexItem = new(models.YongyiExcelIndex)
|
|
|
+ indexItem.IndexName = fullIndexName
|
|
|
+ indexItem.ClassifyName = classifyName
|
|
|
+ // todo 处理indexCode
|
|
|
+ indexItem.IndexCode = fullIndexName
|
|
|
+ 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)
|
|
|
+ }
|
|
|
+ // todo 整理请求入参,并加入,改成一个指标请求一次还是多个指标请求一次
|
|
|
+ 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
|
|
|
+ }
|
|
|
+ if name == "全国均价" {
|
|
|
+ name = "商品猪出拦价/全国/均价"
|
|
|
+ unit = "元/公斤"
|
|
|
+ } else if name == "日屠宰量" {
|
|
|
+ name = "商品猪日屠宰量/全国"
|
|
|
+ unit = "头/日"
|
|
|
+ }
|
|
|
+ fullIndexName := name
|
|
|
+
|
|
|
+ indexItem, okIndex := indexMap[fullIndexName]
|
|
|
+
|
|
|
+ if !okIndex {
|
|
|
+ // 新增指标
|
|
|
+ indexItem = new(models.YongyiExcelIndex)
|
|
|
+ indexItem.IndexName = fullIndexName
|
|
|
+ indexItem.ClassifyName = classifyName
|
|
|
+ // todo 处理indexCode
|
|
|
+ indexItem.IndexCode = fullIndexName
|
|
|
+ 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)
|
|
|
+ }
|
|
|
+ // todo 整理请求入参,并加入,改成一个指标请求一次还是多个指标请求一次
|
|
|
+ return
|
|
|
+}
|
|
|
+
|
|
|
+// HandleYongyiExcelDaily4 日度-屠宰企业屠宰量
|
|
|
+func HandleYongyiExcelDaily4(sheet *xlsx.Sheet) (indexList []*models.YongyiExcelIndex, err error) {
|
|
|
+ classifyName := "日度-屠宰企业屠宰量"
|
|
|
+ frequency := "日度"
|
|
|
+ unit := "头/日"
|
|
|
+ namePrefix := "商品猪日屠宰量"
|
|
|
+ namePrefixPingin := utils.GetFirstPingYin(namePrefix)
|
|
|
+ // 遍历行读取
|
|
|
+ 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", 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
|
|
|
+}
|