jyps/export_employee/l2.go

package main

import (
	"fmt"
	"github.com/xuri/excelize/v2"
	"jygit.jydev.jianyu360.cn/data_processing/common_utils/mongodb"
	"log"
)

func dealL2() {
	f, err := excelize.OpenFile("L2.xlsx")
	if err != nil {
		fmt.Println(err)
		return
	}
	defer func() {
		f.Save()
		if err := f.Close(); err != nil {
			fmt.Println(err)
		}
	}()
	rows, err := f.Rows("Sheet1")

	mgo := &mongodb.MongodbSim{
		MongodbAddr: "192.168.3.166:27082",
		DbName:      "wcc",
		Size:        10,
		UserName:    "",
		Password:    "",
	}

	mgo.InitPool()

	//存储 l0 对应地下 的l2 数量总和
	//resa := make(map[string]map[string]interface{}, 0)

	i := 0
	for rows.Next() {
		i++
		if i == 1 {
			continue
		}
		if i%1000 == 0 {
			log.Println(i)
		}

		row, err := rows.Columns()
		if err != nil {
			fmt.Println(err)
		}
		//1.没有参保人数、企业人数
		where := map[string]interface{}{
			"company_name": row[1],
		}
		var insert = make(map[string]interface{}, 0)
		insert["l0"] = row[0]
		insert["l2"] = row[1]
		if len(row) == 2 {
			data, _ := mgo.FindOne("L2data", where)
			da := *data
			if len(da) > 0 {
				insert["insurance_amount"] = da["insurance_amount"]
				insert["employee"] = da["employee"]
			} else {
				insert["insurance_amount"] = ""
				insert["employee"] = ""
			}
		} else if len(row) == 3 {
			data, _ := mgo.FindOne("L2data", where)
			da := *data
			if len(da) > 0 {
				insert["insurance_amount"] = da["insurance_amount"]
				insert["employee"] = da["employee"]
			} else {
				insert["insurance_amount"] = row[2]
				insert["employee"] = ""
			}

		} else if len(row) == 4 {
			if (row[2] != "" && row[2] != "0") && (row[3] != "" && row[3] != "0") {
				insert["insurance_amount"] = row[2]
				insert["employee"] = row[3]
			} else {
				data, _ := mgo.FindOne("L2data", where)
				da := *data
				if len(da) > 0 {
					insert["insurance_amount"] = da["insurance_amount"]
					insert["employee"] = da["employee"]
				} else {
					insert["insurance_amount"] = row[2]
					insert["employee"] = row[3]
				}
			}
		}

		//
		mgo.Save("wcc_l2", insert)
	}
	if err = rows.Close(); err != nil {
		fmt.Println(err)
	}

	log.Println("over")
}