feat:增加映射

mapping/bidcode.go

@@ -0,0 +1,99 @@
+package mapping
+
+import (
+	"log"
+
+	util "app.yhyue.com/moapp/jybase/common"
+	"app.yhyue.com/moapp/jybase/mysql"
+)
+
+var BidCodeMapping = &bidCodeMapping{}
+
+type bidCodeMapping struct {
+	Area          map[string]string //省
+	City          map[string]string //市
+	District      map[string]string //区
+	Toptype       map[string]string //一级信息类型
+	Subtype       map[string]string //二级信息类型
+	Buyerclass    map[string]string //采购单位行业
+	Subscopeclass map[string]string //信息行业
+}
+
+func (b *bidCodeMapping) Init(Mysql *mysql.Mysql) {
+	b.Buyerclass = map[string]string{}
+	codeBuyerclass := Mysql.SelectBySql("select code,name from global_common_data.code_buyerclass where level=1")
+	if codeBuyerclass != nil && len(*codeBuyerclass) > 0 {
+		for _, v := range *codeBuyerclass {
+			code, _ := v["code"].(string)
+			name, _ := v["name"].(string)
+			b.Buyerclass[name] = code
+		}
+	}
+	if len(b.Buyerclass) == 0 {
+		log.Fatalln("PushMapping Buyerclass Init Error")
+	}
+	//
+	b.Toptype = map[string]string{}
+	b.Subtype = map[string]string{}
+	codeBidTopSubtype := Mysql.SelectBySql("select code,name,level from global_common_data.code_bidtopsubtype")
+	if codeBidTopSubtype != nil && len(*codeBidTopSubtype) > 0 {
+		for _, v := range *codeBidTopSubtype {
+			code, _ := v["code"].(string)
+			name, _ := v["name"].(string)
+			level := util.IntAll(v["level"])
+			if level == 1 {
+				b.Toptype[name] = code
+			} else if level == 2 {
+				b.Subtype[name] = code
+			}
+		}
+	}
+	if len(b.Toptype) == 0 {
+		log.Fatalln("PushMapping Toptype Init Error")
+	}
+	if len(b.Subtype) == 0 {
+		log.Fatalln("PushMapping Subtype Init Error")
+	}
+	//
+	b.Subscopeclass = map[string]string{}
+	codeBidScope := Mysql.SelectBySql("SELECT b.code,CONCAT(a.name,'_',b.name) AS NAME FROM code_bidscope a INNER JOIN code_bidscope b ON (a.level=1 AND b.level=2 AND a.code=b.pcode)")
+	if codeBidScope != nil && len(*codeBidScope) > 0 {
+		for _, v := range *codeBuyerclass {
+			code, _ := v["code"].(string)
+			name, _ := v["name"].(string)
+			b.Subscopeclass[name] = code
+		}
+	}
+	if len(b.Subscopeclass) == 0 {
+		log.Fatalln("PushMapping Subscopeclass Init Error")
+	}
+	//
+	b.Area = map[string]string{}
+	b.City = map[string]string{}
+	b.District = map[string]string{}
+	codeArea := Mysql.SelectBySql("select code,area,city,district from global_common_data.code_area")
+	if codeArea != nil && len(*codeArea) > 0 {
+		for _, v := range *codeArea {
+			code, _ := v["code"].(string)
+			area, _ := v["area"].(string)
+			city, _ := v["city"].(string)
+			district, _ := v["district"].(string)
+			if area != "" && city == "" && district == "" {
+				b.Area[area] = code
+			} else if area != "" && city != "" && district == "" {
+				b.City[city] = code
+			} else if area != "" && city != "" && district != "" {
+				b.District[district] = code
+			}
+		}
+	}
+	if len(b.Area) == 0 {
+		log.Fatalln("BidCodeMapping Area Init Error")
+	}
+	if len(b.City) == 0 {
+		log.Fatalln("BidCodeMapping City Init Error")
+	}
+	if len(b.District) == 0 {
+		log.Fatalln("BidCodeMapping District Init Error")
+	}
+}