data_field_dispose/data_mgo_to_tidb/test.go

package main

import (
	"fmt"
	"go.mongodb.org/mongo-driver/bson"
	"go.uber.org/zap"
	util "jygit.jydev.jianyu360.cn/data_processing/common_utils"
	"jygit.jydev.jianyu360.cn/data_processing/common_utils/log"
	"jygit.jydev.jianyu360.cn/data_processing/common_utils/mongodb"
	"jygit.jydev.jianyu360.cn/data_processing/common_utils/redis"
	"sync"
	"time"
)

func taskMysql() {
	pool := make(chan bool, 5) //控制线程数
	wg := &sync.WaitGroup{}

	finalId := 0
	lastInfo := MysqlTool.SelectBySql(fmt.Sprintf("SELECT * FROM %s ORDER BY id DESC LIMIT 1", "dws_f_bpmc_relation"))
	//lastInfo := MysqlTool.SelectBySql(fmt.Sprintf("SELECT id, projectid, infoid, name_id, identity_type+0 FROM %s ORDER BY id DESC LIMIT 1", "dws_f_bpmc_relation_new"))
	if len(*lastInfo) > 0 {
		finalId = util.IntAll((*lastInfo)[0]["id"])
	}
	log.Info("查询最后id---", zap.Int("finally id: ", finalId))
	lastid, count := 0, 0
	for {
		log.Info("重新查询，lastid---", zap.Int("lastid: ", lastid))
		q := fmt.Sprintf("SELECT * FROM %s WHERE id > %d ORDER BY id ASC limit 100000", "dws_f_bpmc_relation", lastid)
		//q := fmt.Sprintf("SELECT id, projectid, infoid, name_id, identity_type+0 FROM %s WHERE id=61771536 ORDER BY id ASC limit 1000000", "dws_f_bpmc_relation")
		//q := fmt.Sprintf("SELECT id, name, name_id FROM %s WHERE id>%d ORDER BY id ASC limit 1000000", "dws_f_ent_baseinfo", lastid)
		rows, err := MysqlTool.DB.Query(q)
		if err != nil {
			log.Error("mysql query err ", zap.Error(err))
		}
		columns, err := rows.Columns()
		if finalId == lastid {
			log.Info("----finish-----", zap.Int("count: ", count))
			break
		}
		for rows.Next() {
			scanArgs := make([]interface{}, len(columns))
			values := make([]interface{}, len(columns))
			ret := make(map[string]interface{})
			for k := range values {
				scanArgs[k] = &values[k]
			}
			err = rows.Scan(scanArgs...)
			if err != nil {
				log.Error("mysql scan err ", zap.Error(err))
				break
			}
			for i, col := range values {
				if col == nil {
					ret[columns[i]] = nil
				} else {
					switch val := (*scanArgs[i].(*interface{})).(type) {
					case byte:
						ret[columns[i]] = val
						break
					case []byte:
						v := string(val)
						switch v {
						case "\x00": // 处理数据类型为bit的情况
							ret[columns[i]] = 0
						case "\x01": // 处理数据类型为bit的情况
							ret[columns[i]] = 1
						default:
							ret[columns[i]] = v
							break
						}
						break
					case time.Time:
						if val.IsZero() {
							ret[columns[i]] = nil
						} else {
							ret[columns[i]] = val.Format("2006-01-02 15:04:05")
						}
						break
					default:
						ret[columns[i]] = val
					}
				}
			}
			lastid = util.IntAll(ret["id"])
			count++
			if count%20000 == 0 {
				log.Info("current----", zap.Int("count: ", count), zap.Int("lastid: ", lastid))
			}
			pool <- true
			wg.Add(1)
			go func(tmp map[string]interface{}) {
				defer func() {
					<-pool
					wg.Done()
				}()
				//cid := util.Int64All(tmp["id"])
				//iid := util.ObjToString(tmp["infoid"])
				//name_id := util.ObjToString(tmp["name_id"])
				//identity_type := util.Int64All(tmp["identity_type+0"])
				//if name_id != "" {
				//	coll := "bidding"
				//	if iid > "5a862e7040d2d9bbe88e3b1f" {
				//		coll = "bidding"
				//	} else {
				//		coll = "bidding_back"
				//	}
				//	info, _ := MongoB.FindById(coll, iid, bson.M{"agencytel": 1, "agencyperson": 1, "buyertel": 1, "buyerperson": 1, "winnertel": 1, "winnerperson": 1})
				//	if len(*info) > 0 {
				//		if identity_type == 1 {
				//			if util.ObjToString((*info)["buyertel"]) != "" {
				//				q := make(map[string]interface{})
				//				q["name_id"] = name_id
				//				q["identity_type"] = identity_type
				//				q["contact_tel"] = util.ObjToString((*info)["buyertel"])
				//				if util.ObjToString((*info)["buyerperson"]) != "" {
				//					q["contact_name"] = util.ObjToString((*info)["buyerperson"])
				//				}
				//				cinfo := MysqlTool.FindOne("dws_f_ent_contact", q, "", "")
				//				if cinfo != nil && len(*cinfo) > 0 {
				//					MysqlTool.Update("dws_f_bpmc_relation_new", bson.M{"id": cid}, bson.M{"contact_id": (*cinfo)["id"]})
				//				}
				//			}
				//		} else if identity_type == 2 {
				//			if util.ObjToString((*info)["winnertel"]) != "" {
				//				q := make(map[string]interface{})
				//				q["name_id"] = name_id
				//				q["identity_type"] = identity_type
				//				q["contact_tel"] = util.ObjToString((*info)["winnertel"])
				//				if util.ObjToString((*info)["winnerperson"]) != "" {
				//					q["contact_name"] = util.ObjToString((*info)["winnerperson"])
				//				}
				//				cinfo := MysqlTool.FindOne("dws_f_ent_contact", q, "", "")
				//				if cinfo != nil && len(*cinfo) > 0 {
				//					MysqlTool.Update("dws_f_bpmc_relation_new", bson.M{"id": cid}, bson.M{"contact_id": (*cinfo)["id"]})
				//				}
				//			}
				//		} else if identity_type == 4 {
				//			if util.ObjToString((*info)["agencytel"]) != "" {
				//				q := make(map[string]interface{})
				//				q["name_id"] = name_id
				//				q["identity_type"] = identity_type
				//				q["contact_tel"] = util.ObjToString((*info)["agencytel"])
				//				if util.ObjToString((*info)["agencyperson"]) != "" {
				//					q["contact_name"] = util.ObjToString((*info)["agencyperson"])
				//				}
				//				cinfo := MysqlTool.FindOne("dws_f_ent_contact", q, "", "")
				//				if cinfo != nil && len(*cinfo) > 0 {
				//					MysqlTool.Update("dws_f_bpmc_relation_new", bson.M{"id": cid}, bson.M{"contact_id": (*cinfo)["id"]})
				//				}
				//			}
				//		}
				//	}
				//}

				//redis.PutCKV("qyxy_id", util.ObjToString(tmp["name"]), util.ObjToString(tmp["name_id"]))

				saveM := make(map[string]interface{})
				if util.ObjToString(tmp["name_id"]) != "" {
					saveM["name_id"] = util.ObjToString(tmp["name_id"])
				} else {
					return
				}
				if util.ObjToString(tmp["contact_id"]) != "" {
					saveM["contact_id"] = util.IntAll(tmp["contact_id"])
				} else {
					return
				}
				saveM["projectid"] = util.ObjToString(tmp["projectid"])
				saveM["infoid"] = util.ObjToString(tmp["infoid"])
				saveM["identity_type"] = tmp["identity_type"]
				saveRelationPool <- saveM
			}(ret)
			ret = make(map[string]interface{})
		}
		_ = rows.Close()
		wg.Wait()
	}
}

func taskMgo() {
	sess := MongoP.GetMgoConn()
	defer MongoP.DestoryMongoConn(sess)

	ch := make(chan bool, 3)
	wg := &sync.WaitGroup{}
	q := bson.M{"_id": bson.M{"$lte": mongodb.StringTOBsonId("63411488911e1eb3459fb87e")}}
	field := map[string]interface{}{"ids": 1}
	query := sess.DB("qfw").C("projectset_20220721").Find(q).Select(field).Iter()
	count := 0
	for tmp := make(map[string]interface{}); query.Next(tmp); count++ {
		if count%20000 == 0 {
			util.Debug("current ---", count, tmp["_id"])
		}
		ch <- true
		wg.Add(1)
		go func(tmp map[string]interface{}) {
			defer func() {
				<-ch
				wg.Done()
			}()
			id := mongodb.BsonIdToSId(tmp["_id"])
			for _, i := range util.ObjArrToStringArr(tmp["ids"].([]interface{})) {
				redis.PutCKV("s_id", i, id)
			}
		}(tmp)
		tmp = make(map[string]interface{})
	}
	wg.Wait()

	util.Debug("over ---", count)
}