Exemplo n.º 1
0
func main() {

	models.InitDB()
	fmt.Println("Read Begin")
	filename := "../doc/ss_a_page1.txt"
	// filename := "E:\\GoProjects\\src\\github.com\\royburns\\goStockAnalyst\\doc\\ss_a_page1.txt"
	j := new(JsonData)
	err := ReadJson(filename, j)
	if err != nil {
		fmt.Println(err.Error())
	} else {
		fmt.Println(len(j.Result))
	}
	fmt.Println("Read End")

	fmt.Println("Insert Begin")
	sc := make([]models.StockCompany, 0)
	for i := 0; i < len(j.Result); i++ {
		var tmp models.StockCompany
		// tmp.Id = j.Result[i].ID
		number, _ := strconv.Atoi(j.Result[i].ProductID)
		tmp.Id = int64(number)
		tmp.Name = j.Result[i].ProductName
		tmp.FullName = j.Result[i].FullName

		tmp.ACodeID = j.Result[i].ProductID
		tmp.AName = j.Result[i].ProductName

		sc = append(sc, tmp)
	}
	orm := models.GetEngine()
	orm.Insert(sc)
	fmt.Println("Insert End")
}
Exemplo n.º 2
0
// 0 公司代码,1 公司简称,2 公司全称,3 英文名称,4 注册地址,
// 5 A股代码, 6 A股简称, 7 A股上市日期, 8 A股总股本, 9 A股流通股本,
// 10 B股代码, 11 B股简称, 12 B股上市日期, 13 B股总股本, 14 B股流通股本,
// 15 地区, 16 省份, 17 城市, 18 所属行业, 19 公司网址
func main() {
	models.InitDB()
	fmt.Println("Read Begin")
	f, err := os.Open("../doc/sz_上市公司列表.csv")
	if err != nil {
		fmt.Println("Error.")
	}
	defer f.Close()
	r := csv.NewReader(f)

	lines, err := r.ReadAll()
	fmt.Println("Read End")

	fmt.Println("Insert Begin")
	orm := models.GetEngine()
	// sc := make([]models.StockCompany, 0)
	fmt.Println(len(lines))
	for i := 1; i < len(lines); i++ {
		if lines[i][0] != "" {
			var tmp models.StockCompany

			// fmt.Println(lines[i][0])
			number, _ := strconv.Atoi(lines[i][0])
			tmp.Id = int64(number)
			tmp.Name = lines[i][1]
			tmp.FullName = lines[i][2]
			tmp.EnName = lines[i][3]
			tmp.Address = lines[i][4]

			if lines[i][5] != "" {
				tmp.ACodeID = fmt.Sprintf("%06s", lines[i][5])
			}
			tmp.AName = lines[i][6]
			tmp.AData = lines[i][7]
			number, _ = strconv.Atoi(strings.Join(strings.Split(lines[i][8], ","), ""))
			tmp.ACapital = int64(number)
			number, _ = strconv.Atoi(strings.Join(strings.Split(lines[i][9], ","), ""))
			tmp.AOutCapital = int64(number)

			if lines[i][10] != "" {
				tmp.BCodeID = fmt.Sprintf("%06s", lines[i][10])
			}
			tmp.BName = lines[i][11]
			tmp.BData = lines[i][12]
			number, _ = strconv.Atoi(strings.Join(strings.Split(lines[i][13], ","), ""))
			tmp.BCapital = int64(number)
			number, _ = strconv.Atoi(strings.Join(strings.Split(lines[i][14], ","), ""))
			tmp.BOutCapital = int64(number)

			tmp.Region = lines[i][15]
			tmp.Province = lines[i][16]
			tmp.City = lines[i][17]
			tmp.Industry = lines[i][18]
			tmp.Site = lines[i][19]

			// sc = append(sc, tmp)
			orm.Insert(tmp)
		}
	}

	fmt.Println("Insert End")
}