package main

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

var (
	yangMap      = make(map[string]bool) //存储98家央企
	yangChildMap = make(map[string]bool) //存储央企 下属子公司
	Mgo181       *mongodb.MongodbSim     //
)

func InitMgo() {
	//181 凭安库
	Mgo181 = &mongodb.MongodbSim{
		//MongodbAddr: "172.17.4.181:27001",
		MongodbAddr: "127.0.0.1:27001",
		DbName:      "mixdata",
		Size:        10,
		UserName:    "",
		Password:    "",
		Direct:      true,
	}
	Mgo181.InitPool()
}

// readXlsx 读取央企
func readXlsx() {
	filePath := "央企.xlsx"
	// 1. 读取 Excel（获取 A 列数据）
	f, err := excelize.OpenFile(filePath)
	if err != nil {
		log.Fatal("❌ 无法打开 Excel 文件:", err)
	}
	defer f.Close()

	//读取央企
	rows, err := f.GetRows("Sheet1")
	if err != nil {
		log.Fatal("❌ 无法读取 Sheet1:", err)
	}
	for i := 1; i < len(rows); i++ {
		name := rows[i][0]
		if name != "" {
			yangMap[name] = true
		}
	}
	// 央企下属
	rows2, err := f.GetRows("Sheet2")
	if err != nil {
		log.Fatal("❌ 无法读取 Sheet2:", err)
	}
	for i := 1; i < len(rows2); i++ {
		name := rows2[i][1]
		if name != "" {
			yangChildMap[name] = true
		}
	}
}