jy-data-extract/versioncomparison/main.go

package main

import (
	"fmt"
	"gopkg.in/mgo.v2"
	"gopkg.in/mgo.v2/bson"
	"jy/mongodbutil"
	"log"
	"qfw/common/src/github.com/tealeg/xlsx"
	"qfw/util"
	"strings"
)

var (
	SysConfig   map[string]interface{}
	Extractmgo  *mgo.Session      //抽取
	Previousmgo *mongodbutil.Pool //之前抽取
	Newmgo      *mongodbutil.Pool //最新抽取
)

/**
与上个抽取版本做比较
 */
func init() {
	util.ReadConfig(&SysConfig)
	if len(SysConfig) < 1 {
		log.Println("配置文件读取失败")
		return
	}
	session, e := mgo.Dial(util.ObjToString(SysConfig["extractmgo"]))
	if e != nil {
		log.Fatal(e)
	}
	Extractmgo = session
	Previousmgo = mongodbutil.MgoFactory(2, 5, 120, util.ObjToString(SysConfig["previousmgo"]), util.ObjToString(SysConfig["previousdb"]))
	Newmgo = mongodbutil.MgoFactory(2, 5, 120, util.ObjToString(SysConfig["newmgo"]), util.ObjToString(SysConfig["newdb"]))
}

type versionComparison struct {
	Id  interface{} `json:"_id"`
	Url string      `json:"url"`
}

func main() {
	Query(util.IntAll(SysConfig["queryNum"]), util.ObjToString(SysConfig["querySid"]))
}

func Query(num int, sid string) {
	xf, err := xlsx.OpenFile("抽取结果对比.xlsx")
	if err != nil {
		log.Println("读取文件", err)
		return
	}
	var projectcodenum, bidamountnum, winnernum, buyernum, budgetnum, projectnamenum int                               //不相等计数器
	var pcodeNotNilNumP, bidamountNotNilNumP, winnerNotNilNumP, buyerNotNilNumP, budgetNotNilNumP, pnameNotNilNumP int //不相等计数器
	var pcodeNotNilNumN, bidamountNotNilNumN, winnerNotNilNumN, buyerNotNilNumN, budgetNotNilNumN, pnameNotNilNumN int //不相等计数器

	log.Println(num, sid)
	if num < 1 {
		log.Println("查询数量应该大于0")
		return
	}
	sum := num //总量
	//if strings.TrimSpace(gteid) == "" {
	//	gteid = "386cd3000000000000000000"
	//}
	var iter *mgo.Iter
	if sid == "" {
		iter = Extractmgo.DB(util.ObjToString(SysConfig["extractdb"])).C(util.ObjToString(SysConfig["extractc"])).Find(nil).Select(bson.M{"_id": 1}).Iter()
	} else {
		iter = Extractmgo.DB(util.ObjToString(SysConfig["extractdb"])).C(util.ObjToString(SysConfig["extractc"])).Find(bson.M{"_id": bson.M{
			"$gte": bson.ObjectIdHex(sid)},
		}).Select(bson.M{"_id": 1}).Iter()
	}
	defer log.Println("关闭 iter:", iter.Close())
	var data map[string]bson.ObjectId
	getdata := make([]bson.ObjectId, 0)
	for iter.Next(&data) {
		if num == 0 {
			break
		}
		getdata = append(getdata, data["_id"])
		num--
	}
	log.Println(sum, "条数据加载完成")
	projectnames := make([]*Projectname, 0)
	buyers := make([]*Buyer, 0)
	projectcodes := make([]*Projectcode, 0)
	winners := make([]*Winner, 0)
	budgets := make([]*Budget, 0)
	bidamounts := make([]*Bidamount, 0)
	for _, gv := range getdata {
		log.Println(gv)
		gvid := gv.Hex()
		pdata, b := Previousmgo.FindById(util.ObjToString(SysConfig["previousc"]), gvid, SysConfig["keyfield"])
		if !b || len(*pdata) == 0 {
			log.Println("oldId不存在")
			continue
		}
		log.Println("pdata:", pdata)

		ndata, b := Newmgo.FindById(util.ObjToString(SysConfig["newc"]), gvid, SysConfig["keyfield"])
		if !b || len(*ndata) == 0 {
			log.Println("nweId不存在")
			continue
		}
		log.Println("ndata:", ndata)

		versioncomparison := new(versionComparison)
		versioncomparison.Id = gvid
		versioncomparison.Url = "https://www.jianyu360.com/article/content/" + util.CommonEncodeArticle("content", gvid) + ".html"
		for k := range SysConfig["keyfield"].(map[string]interface{}) {
			var pd interface{}
			var nd interface{}
			if k == "budget" || k == "bidamount" {
				pd = util.Float64All((*pdata)[k])
				nd = util.Float64All((*ndata)[k])
				if pd.(float64) > 0 {
					switch k {
					case "budget":
						budgetNotNilNumP++
					case "bidamount":
						bidamountNotNilNumP++
					}
				}
				if nd.(float64) > 0 {
					switch k {
					case "budget":
						budgetNotNilNumN++
					case "bidamount":
						bidamountNotNilNumN++
					}
				}
			} else {
				pd = strings.TrimSpace(util.ObjToString((*pdata)[k]))
				nd = strings.TrimSpace(util.ObjToString((*ndata)[k]))
				if strings.TrimSpace(pd.(string)) != "" {
					switch k {
					case "projectname":
						pnameNotNilNumP++
					case "buyer":
						buyerNotNilNumP++
					case "projectcode":
						pcodeNotNilNumP++
					case "winner":
						winnerNotNilNumP++
					}
				}
				if strings.TrimSpace(nd.(string)) != "" {
					switch k {
					case "projectname":
						pnameNotNilNumN++
					case "buyer":
						buyerNotNilNumN++
					case "projectcode":
						pcodeNotNilNumN++
					case "winner":
						winnerNotNilNumN++
					}
				}
			}
			if pd != nd {
				//log.Println(k)
				switch k {
				case "projectname":
					projectname := new(Projectname)
					projectname.versionComparison = *versioncomparison
					projectname.ProjectnameOld = fmt.Sprint(pd)
					projectname.ProjectnameNew = fmt.Sprint(nd)
					projectnames = append(projectnames, projectname)
					projectnamenum++
				case "buyer":
					buyer := new(Buyer)
					buyer.versionComparison = *versioncomparison
					buyer.BuyerOld = fmt.Sprint(pd)
					buyer.BuyerNew = fmt.Sprint(nd)
					buyers = append(buyers, buyer)
					buyernum++
				case "projectcode":
					projectcode := new(Projectcode)
					projectcode.ProjectcodeOld = fmt.Sprint(pd)
					projectcode.ProjectcodeNew = fmt.Sprint(nd)
					projectcode.versionComparison = *versioncomparison
					projectcodes = append(projectcodes, projectcode)
					projectcodenum++
				case "winner":
					winner := new(Winner)
					winner.WinnerOld = fmt.Sprint(pd)
					winner.WinnerNew = fmt.Sprint(nd)
					winner.versionComparison = *versioncomparison
					winners = append(winners, winner)
					winnernum++
				case "budget":
					budget := new(Budget)
					budget.BudgetOld = fmt.Sprint(pd)
					budget.BudgetNew = fmt.Sprint(nd)
					budget.versionComparison = *versioncomparison
					budgets = append(budgets, budget)
					budgetnum++
				case "bidamount":
					bidamount := new(Bidamount)
					bidamount.BidamountOld = fmt.Sprint(pd)
					bidamount.BidamountNew = fmt.Sprint(nd)
					bidamount.versionComparison = *versioncomparison
					bidamounts = append(bidamounts, bidamount)
					bidamountnum++
				}
			}
		}
		fmt.Println()
	}

	//log.Println(projectcodenum, bidamountnum, winnernum, buyernum, budgetnum, projectnamenum)
	for ins, ivs := range xf.Sheets {
		for inr, ivr := range ivs.Rows {
			for _, ivc := range ivr.Cells {
				//抽取对比
				if ins == 0 {
					if inr < 3 {
						continue
					}
					switch strings.TrimSpace(ivc.String()) {
					case "projectname":
						style := ivr.Cells[1].GetStyle()
						style.Font.Color = "000000"
						ivr.Cells[1].SetValue(pnameNotNilNumP)
						ivr.Cells[2].SetValue(pnameNotNilNumN)
						//结果相同数量
						ivr.Cells[3].SetValue(sum - projectnamenum)
						ivr.Cells[3].SetStyle(style)
						//结果不同数量
						ivr.Cells[4].SetValue(projectnamenum)
						ivr.Cells[4].SetStyle(style)
					case "buyer":
						style := ivr.Cells[1].GetStyle()
						style.Font.Color = "000000"
						ivr.Cells[1].SetValue(buyerNotNilNumP)
						ivr.Cells[2].SetValue(buyerNotNilNumN)
						//结果相同数量
						ivr.Cells[3].SetValue(sum - buyernum)
						ivr.Cells[3].SetStyle(style)
						//结果不同数量
						ivr.Cells[4].SetValue(buyernum)
						ivr.Cells[4].SetStyle(style)
					case "projectcode":
						style := ivr.Cells[1].GetStyle()
						style.Font.Color = "000000"
						ivr.Cells[1].SetValue(pcodeNotNilNumP)
						ivr.Cells[2].SetValue(pcodeNotNilNumN)
						//结果相同数量
						ivr.Cells[3].SetValue(sum - projectcodenum)
						ivr.Cells[3].SetStyle(style)
						//结果不同数量
						ivr.Cells[4].SetValue(projectcodenum)
						ivr.Cells[4].SetStyle(style)
					case "winner":
						style := ivr.Cells[1].GetStyle()
						style.Font.Color = "000000"
						ivr.Cells[1].SetValue(winnerNotNilNumP)
						ivr.Cells[2].SetValue(winnerNotNilNumN)
						//结果相同数量
						ivr.Cells[3].SetValue(sum - winnernum)
						ivr.Cells[3].SetStyle(style)
						//结果不同数量
						ivr.Cells[4].SetValue(winnernum)
						ivr.Cells[4].SetStyle(style)
					case "budget":
						style := ivr.Cells[1].GetStyle()
						style.Font.Color = "000000"
						ivr.Cells[1].SetValue(budgetNotNilNumP)
						ivr.Cells[2].SetValue(budgetNotNilNumN)
						//结果相同数量
						ivr.Cells[3].SetValue(sum - budgetnum)
						ivr.Cells[3].SetStyle(style)
						//结果不同数量
						ivr.Cells[4].SetValue(budgetnum)
						ivr.Cells[4].SetStyle(style)
					case "bidamount":
						style := ivr.Cells[1].GetStyle()
						style.Font.Color = "000000"
						ivr.Cells[1].SetValue(bidamountNotNilNumP)
						ivr.Cells[2].SetValue(bidamountNotNilNumN)
						//结果相同数量
						ivr.Cells[3].SetValue(sum - bidamountnum)
						ivr.Cells[3].SetStyle(style)
						//结果不同数量
						ivr.Cells[4].SetValue(bidamountnum)
						ivr.Cells[4].SetStyle(style)
					}
				}
			}
		}
		if ins > 0 {
			if len(ivs.Rows) == 0 {
				row := ivs.AddRow()
				row.AddCell().SetValue("ObjectId")
				row.AddCell().SetValue("dev3.1.2")
				row.AddCell().SetValue("dev3.2")
				row.AddCell().SetValue("URL")
			}
			//log.Println(ivs.Name)
			switch strings.TrimSpace(ivs.Name) {
			case "projectname":
				for _, v := range projectnames {
					row := ivs.AddRow()
					row.AddCell().SetValue(v.Id)
					row.AddCell().SetValue(v.ProjectnameOld)
					row.AddCell().SetValue(v.ProjectnameNew)
					row.AddCell().SetValue(v.Url)
				}
			case "buyer":
				for _, v := range buyers {
					row := ivs.AddRow()
					row.AddCell().SetValue(v.Id)
					row.AddCell().SetValue(v.BuyerOld)
					row.AddCell().SetValue(v.BuyerNew)
					row.AddCell().SetValue(v.Url)
				}
			case "projectcode":
				for _, v := range projectcodes {
					row := ivs.AddRow()
					row.AddCell().SetValue(v.Id)
					row.AddCell().SetValue(v.ProjectcodeOld)
					row.AddCell().SetValue(v.ProjectcodeNew)
					row.AddCell().SetValue(v.Url)
				}
			case "winner":
				for _, v := range winners {
					row := ivs.AddRow()
					row.AddCell().SetValue(v.Id)
					row.AddCell().SetValue(v.WinnerOld)
					row.AddCell().SetValue(v.WinnerNew)
					row.AddCell().SetValue(v.Url)
				}
			case "budget":
				for _, v := range budgets {
					row := ivs.AddRow()
					row.AddCell().SetValue(v.Id)
					row.AddCell().SetValue(v.BudgetOld)
					row.AddCell().SetValue(v.BudgetNew)
					row.AddCell().SetValue(v.Url)
				}
			case "bidamount":
				for _, v := range bidamounts {
					row := ivs.AddRow()
					row.AddCell().SetValue(v.Id)
					row.AddCell().SetValue(v.BidamountOld)
					row.AddCell().SetValue(v.BidamountNew)
					row.AddCell().SetValue(v.Url)
				}
			}
		}
	}
	err = xf.Save("resultdata.xlsx")
	if err != nil {
		log.Println("保存xlsx失败:", err)
		return
	}
	log.Println("xlsx保存成功")
}

type Projectname struct {
	versionComparison
	ProjectnameOld string
	ProjectnameNew string
}
type Buyer struct {
	versionComparison
	BuyerOld string
	BuyerNew string
}
type Projectcode struct {
	versionComparison
	ProjectcodeOld string
	ProjectcodeNew string
}
type Winner struct {
	versionComparison
	WinnerOld string
	WinnerNew string
}
type Budget struct {
	versionComparison
	BudgetOld string
	BudgetNew string
}
type Bidamount struct {
	versionComparison
	BidamountOld string
	BidamountNew string
}