jyps/updateBidding/lua_test.go

package main

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

func TestUpdateLuaConfig(t *testing.T) {
	//87 竞品
	MgoLua := &mongodb.MongodbSim{
		//MongodbAddr: "172.17.189.140:27080",
		MongodbAddr: "127.0.0.1:27081",
		Size:        10,
		DbName:      "editor",
		UserName:    "",
		Password:    "",
		Direct:      true,
	}
	MgoLua.InitPool()
	sess := MgoLua.GetMgoConn()
	defer MgoLua.DestoryMongoConn(sess)

	f, err := excelize.OpenFile("./luaconfig.xlsx")

	if err != nil {
		fmt.Println(err)
		return
	}
	defer func() {
		if err := f.Close(); err != nil {
			fmt.Println(err)
		}
	}()

	rows, err := f.GetRows("Sheet1")
	if err != nil {
		fmt.Println(err)
		return
	}
	/**
	1、刷至 golua平台 爬虫，state=0，platform=golua平台
	2、刷至通用平台爬虫，state=0，platform=通用平台，claimtype=1
	3、刷至jschrome平台爬虫，state=0，platform=jschrome
	4、以上所有爬虫均修改createuser、modifyuser、modifyuserid、createuserid、createuseremail、next字段。createuser、modifyuser 为user表s_name；modifyuserid、createuserid为user表_id；createuseremail、next为user表s_email
	5、爬虫表87/editor/luaconfig 用户表87/editor/user
	*/

	for i := 1; i < len(rows); i++ {
		row := rows[i]
		code := strings.TrimSpace(row[0])
		modifyuser := strings.TrimSpace(row[1])
		platform := strings.TrimSpace(row[2])

		//更新MongoDB
		updateWhere := map[string]interface{}{
			"code": code,
		}

		exists, _ := MgoLua.FindOne("luaconfig", updateWhere)
		if len(*exists) == 0 {
			log.Println("code 没有找到数据", code)
			continue
		}
		log.Println(code, modifyuser, platform)
		update := make(map[string]interface{})
		if platform == "golua平台" {
			update["state"] = 0
			update["platform"] = "golua平台"
		} else if platform == "通用平台" {
			update["state"] = 0
			update["platform"] = "通用平台"
			update["claimtype"] = 1
		} else if platform == "jschrome" {
			update["state"] = 0
			update["platform"] = "jschrome"
		}

		update["createuser"] = modifyuser
		update["modifyuser"] = modifyuser

		where := map[string]interface{}{
			"s_name": modifyuser,
		}

		user, _ := MgoLua.FindOne("user", where)
		if user == nil {
			log.Println("user 查询失败", where)
			return
		}

		update["modifyuserid"] = mongodb.BsonIdToSId((*user)["_id"])
		update["createuserid"] = mongodb.BsonIdToSId((*user)["_id"])
		update["createuseremail"] = (*user)["s_email"]
		update["next"] = (*user)["s_email"]

		MgoLua.Update("luaconfig", updateWhere, map[string]interface{}{"$set": update}, true, false)
	}
}