质量

data_quality/src/config.json

@@ -1,3 +1,32 @@
 {
+  "udpport": ":17007",
+  "mongodb": {
+    "addrName": "192.168.3.207:27092",
+    "dbName": "extract_kf",
+    "collName": "zk_move",
+    "pool": 5
+  },
+  "core_element":["buyer","projectname","title","winner","agency","projectcode"],
+  "other_element":["area","city"],
+  "score_standard": {
+    "total_score": 100,
+    "core_max": 70,
+    "core_each": 10,
+    "other_max": 30,
+    "other_each": 5
+  },
+  "qy_mongodb": {
+    "qy_addrName": "192.168.3.207:27092",
+    "qy_dbName": "extract_kf",
+    "qy_collName": "zk",
+    "pool": 5
+  },
 
+  "jkmail": {
+    "to": "zhengkun@topnet.net.cn",
+    "api": "http://10.171.112.160:19281/_send/_mail"
+  },
+  "nextNode": [
+
+  ]
 }

data_quality/src/main.go

@@ -1,18 +1,202 @@
 package main
 
 import (
-
+	"encoding/json"
+	"log"
+	"net"
+	"os"
+	"qfw/common/src/qfw/util"
+	qu "qfw/util"
 	"time"
+	mu "mfw/util"
+)
+
+
+var (
+	Sysconfig    map[string]interface{} //配置文件
+	mgo          *MongodbSim            //mongodb操作对象
+	qy_mgo		 *MongodbSim
+	udpclient    mu.UdpClient             //udp对象
+	nextNode     []map[string]interface{} //下节点数组
+
+	coll_name,qy_coll_name	string
+	core_element,other_element	[]string	//要素
+	total_score,core_max,core_each,other_max,other_each int
 )
 
 func init() {
+	//加载配置文件
+	qu.ReadConfig(&Sysconfig)
+
+	mconf := Sysconfig["mongodb"].(map[string]interface{})
+	mgo = &MongodbSim{
+		MongodbAddr: mconf["addrName"].(string),
+		DbName:      mconf["dbName"].(string),
+		Size:        qu.IntAllDef(mconf["pool"], 10),
+	}
+	mgo.InitPool()
+
+
+	qy_mconf := Sysconfig["qy_mongodb"].(map[string]interface{})
+	qy_mgo = &MongodbSim{
+		MongodbAddr: qy_mconf["qy_addrName"].(string),
+		DbName:      qy_mconf["qy_dbName"].(string),
+		Size:        qu.IntAllDef(qy_mconf["pool"], 10),
+	}
+	qy_mgo.InitPool()
 
+
+	coll_name = mconf["collName"].(string)
+	qy_coll_name = qy_mconf["qy_collName"].(string)
+
+	core_element = qu.ObjArrToStringArr(Sysconfig["core_element"].([]interface{}))
+	other_element = qu.ObjArrToStringArr(Sysconfig["other_element"].([]interface{}))
+	score_standard := Sysconfig["score_standard"].(map[string]interface{})
+	total_score = qu.IntAll(score_standard["total_score"])
+	core_max = qu.IntAll(score_standard["core_max"])
+	core_each = qu.IntAll(score_standard["core_each"])
+	other_max = qu.IntAll(score_standard["other_max"])
+	other_each = qu.IntAll(score_standard["other_each"])
+
+	log.Println("采用udp模式")
 }
 
 
+func mainT() {
+	go checkMapJob()
+	updport := Sysconfig["udpport"].(string)
+	udpclient = mu.UdpClient{Local: updport, BufSize: 1024}
+	udpclient.Listen(processUdpMsg)
+	log.Println("Udp服务监听", updport)
+	time.Sleep(99999 * time.Hour)
+}
+
+//快速测试使用
 func main() {
 
+	sid := "4f16936d52c1d9fbf843c60e"
+	eid := "6f16936d52c1d9fbf843c60e"
+	log.Println(sid, "---", eid)
+	mapinfo := map[string]interface{}{}
+	if sid == "" || eid == "" {
+		log.Println("sid,eid参数不能为空")
+		os.Exit(0)
+	}
+	mapinfo["gtid"] = sid
+	mapinfo["lteid"] = eid
+	mapinfo["stop"] = "true"
+	startTask([]byte{}, mapinfo)
+	time.Sleep(99999 * time.Hour)
+
+}
 
 
-	time.Sleep(99999 * time.Hour)
+func processUdpMsg(act byte, data []byte, ra *net.UDPAddr) {
+	switch act {
+	case mu.OP_TYPE_DATA: //上个节点的数据
+		//从表中开始处理
+		var mapInfo map[string]interface{}
+		err := json.Unmarshal(data, &mapInfo)
+		log.Println("err:", err, "mapInfo:", mapInfo)
+		if err != nil {
+			udpclient.WriteUdp([]byte("err:"+err.Error()), mu.OP_NOOP, ra)
+		} else if mapInfo != nil {
+			taskType := qu.ObjToString(mapInfo["stype"])
+			if taskType == "pingfen" {
+				go startTask(data, mapInfo)
+			} else {
+				log.Println("类别异常... ...")
+			}
+			key, _ := mapInfo["key"].(string)
+			if key == "" {
+				key = "udpok"
+			}
+			udpclient.WriteUdp([]byte(key), mu.OP_NOOP, ra)
+		}
+	case mu.OP_NOOP: //下个节点回应
+		ok := string(data)
+		if ok != "" {
+			log.Println("ok:", ok)
+			udptaskmap.Delete(ok)
+		}
+	}
+}
+
+
+func startTask(data []byte, mapInfo map[string]interface{}) {
+
+	//遍历数据
+	log.Println("开始评分流程")
+	defer qu.Catch()
+	//区间id
+	q := map[string]interface{}{
+		"_id": map[string]interface{}{
+			"$gt":  StringTOBsonId(mapInfo["gtid"].(string)),
+			"$lte": StringTOBsonId(mapInfo["lteid"].(string)),
+		},
+	}
+	sess := mgo.GetMgoConn()
+	defer mgo.DestoryMongoConn(sess)
+	it := sess.DB(mgo.DbName).C(coll_name).Find(&q).Iter()
+	updateExtract := [][]map[string]interface{}{}//更新需要
+	index:=0
+	for tmp := make(map[string]interface{}); it.Next(&tmp); index++ {
+		if index%10000 == 0 {
+			log.Println("current:", index, tmp["_id"])
+		}
+
+		element_score:=dealWithElementRate(tmp)
+		error_score,abnormal_score:=dealWithErrorRate(tmp)
+		log.Println("元素分:",element_score,"错误分:",error_score,"异常分:",abnormal_score)
+
+		updateExtract = append(updateExtract, []map[string]interface{}{
+			map[string]interface{}{
+				"_id": tmp["_id"],
+			},
+			map[string]interface{}{
+				"$set": map[string]interface{}{
+					"element_score": element_score,
+					"error_score":error_score,
+					"abnormal_score": abnormal_score,
+				},
+			},
+		})
+		if len(updateExtract) >= 200 {
+			mgo.UpSertBulk(coll_name, updateExtract...)
+			updateExtract = [][]map[string]interface{}{}
+		}
+
+		tmp = make(map[string]interface{})
+	}
+
+
+	if len(updateExtract) >0 {
+		mgo.UpSertBulk(coll_name, updateExtract...)
+	}
+
+	time.Sleep(60 * time.Second)
+
+	//任务完成，开始发送广播通知下面节点
+	if mapInfo["stop"] == nil {
+		log.Println("评分统计完成-发送udp")
+		for _, to := range nextNode {
+			sid, _ := mapInfo["gtid"].(string)
+			eid, _ := mapInfo["lteid"].(string)
+			key := sid + "-" + eid + "-" + util.ObjToString(to["stype"])
+			by, _ := json.Marshal(map[string]interface{}{
+				"gtid":  sid,
+				"lteid": eid,
+				"stype": util.ObjToString(to["stype"]),
+				"key":   key,
+			})
+			addr := &net.UDPAddr{
+				IP:   net.ParseIP(to["addr"].(string)),
+				Port: util.IntAll(to["port"]),
+			}
+			node := &udpNode{by, addr, time.Now().Unix(), 0}
+			udptaskmap.Store(key, node)
+			udpclient.WriteUdp(by, mu.OP_TYPE_DATA, addr)
+		}
+	}
+
 }

data_quality/src/mark

@@ -0,0 +1,45 @@
+备份数据使用
+{
+  "udpport": ":17007",
+  "mongodb": {
+    "addrName": "192.168.3.207:27092",
+    "dbName": "extract_kf",
+    "collName": "zk_move",
+    "pool": 5
+  },
+  "core_element":["buyer","projectname","title","winner","agency","projectcode"],
+  "other_element":["area","city"],
+  "score_standard": {
+    "total_score": 100,
+    "core_max": 70,
+    "core_each": 10,
+    "other_max": 30,
+    "other_each": 5
+  },
+  "qy_mongodb": {
+    "qy_addrName": "192.168.3.207:27092",
+    "qy_dbName": "extract_kf",
+    "qy_collName": "zk",
+    "pool": 5
+  },
+
+  "jkmail": {
+    "to": "zhengkun@topnet.net.cn",
+    "api": "http://10.171.112.160:19281/_send/_mail"
+  },
+  "nextNode": [
+
+  ]
+}
+
+
+
+
+
+
+
+
+
+
+
+

data_quality/src/mgo.go

@@ -0,0 +1,315 @@
+package main
+
+import (
+	"context"
+	"log"
+	"time"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+type MgoSess struct {
+	Db     string
+	Coll   string
+	Query  interface{}
+	Sorts  []string
+	fields interface{}
+	limit  int64
+	skip   int64
+	M      *MongodbSim
+}
+
+type MgoIter struct {
+	Cursor *mongo.Cursor
+}
+
+func (mt *MgoIter) Next(result interface{}) bool {
+	if mt.Cursor != nil {
+		if mt.Cursor.Next(nil) {
+			err := mt.Cursor.Decode(result)
+			if err != nil {
+				log.Println("mgo cur err", err.Error())
+				mt.Cursor.Close(nil)
+				return false
+			}
+			return true
+		} else {
+			mt.Cursor.Close(nil)
+			return false
+		}
+	} else {
+		return false
+	}
+
+}
+
+func (ms *MgoSess) DB(name string) *MgoSess {
+	ms.Db = name
+	return ms
+}
+
+func (ms *MgoSess) C(name string) *MgoSess {
+	ms.Coll = name
+	return ms
+}
+
+func (ms *MgoSess) Find(q interface{}) *MgoSess {
+	ms.Query = q
+	return ms
+}
+
+func (ms *MgoSess) Select(fields interface{}) *MgoSess {
+	ms.fields = fields
+	return ms
+}
+
+func (ms *MgoSess) Limit(limit int64) *MgoSess {
+	ms.limit = limit
+	return ms
+}
+func (ms *MgoSess) Skip(skip int64) *MgoSess {
+	ms.skip = skip
+	return ms
+}
+
+func (ms *MgoSess) Sort(sorts ...string) *MgoSess {
+	ms.Sorts = sorts
+	return ms
+}
+
+func (ms *MgoSess) Iter() *MgoIter {
+	it := &MgoIter{}
+	find := options.Find()
+	if ms.skip > 0 {
+		find.SetSkip(ms.skip)
+	}
+	if ms.limit > 0 {
+		find.SetLimit(ms.limit)
+	}
+	find.SetBatchSize(100)
+	if len(ms.Sorts) > 0 {
+		sort := bson.M{}
+		for _, k := range ms.Sorts {
+			switch k[:1] {
+			case "-":
+				sort[k[1:]] = -1
+			case "+":
+				sort[k[1:]] = 1
+			default:
+				sort[k] = 1
+			}
+		}
+		find.SetSort(sort)
+	}
+	if ms.fields != nil {
+		find.SetProjection(ms.fields)
+	}
+	cur, err := ms.M.C.Database(ms.Db).Collection(ms.Coll).Find(ms.M.Ctx, ms.Query, find)
+	if err != nil {
+		log.Println("mgo find err", err.Error())
+	} else {
+		it.Cursor = cur
+	}
+	return it
+}
+
+type MongodbSim struct {
+	MongodbAddr string
+	Size        int
+	//	MinSize     int
+	DbName   string
+	C        *mongo.Client
+	Ctx      context.Context
+	ShortCtx context.Context
+	pool     chan bool
+}
+
+func (m *MongodbSim) GetMgoConn() *MgoSess {
+	//m.Open()
+	ms := &MgoSess{}
+	ms.M = m
+	return ms
+}
+
+func (m *MongodbSim) DestoryMongoConn(ms *MgoSess) {
+	//m.Close()
+	ms.M = nil
+	ms = nil
+}
+
+func (m *MongodbSim) InitPool() {
+	opts := options.Client()
+	opts.SetConnectTimeout(3 * time.Second)
+	opts.ApplyURI("mongodb://" + m.MongodbAddr)
+	opts.SetMaxPoolSize(uint64(m.Size))
+	m.pool = make(chan bool, m.Size)
+	opts.SetMaxConnIdleTime(2 * time.Hour)
+	m.Ctx, _ = context.WithTimeout(context.Background(), 99999*time.Hour)
+	m.ShortCtx, _ = context.WithTimeout(context.Background(), 1*time.Minute)
+	client, err := mongo.Connect(m.ShortCtx, opts)
+	if err != nil {
+		log.Println("mgo init error:", err.Error())
+	} else {
+		m.C = client
+		log.Println("init success")
+	}
+}
+
+func (m *MongodbSim) Open() {
+	m.pool <- true
+}
+func (m *MongodbSim) Close() {
+	<-m.pool
+}
+
+//批量插入
+func (m *MongodbSim) UpSertBulk(c string, doc ...[]map[string]interface{}) (map[int64]interface{}, bool) {
+	m.Open()
+	defer m.Close()
+	coll := m.C.Database(m.DbName).Collection(c)
+	var writes []mongo.WriteModel
+	for _, d := range doc {
+		write := mongo.NewUpdateOneModel()
+		write.SetFilter(d[0])
+		write.SetUpdate(d[1])
+		write.SetUpsert(true)
+		writes = append(writes, write)
+	}
+	r, e := coll.BulkWrite(m.Ctx, writes)
+	if e != nil {
+		log.Println("mgo upsert error:", e.Error())
+		return nil, false
+	}
+	//	else {
+	//		if r.UpsertedCount != int64(len(doc)) {
+	//			log.Println("mgo upsert uncomplete:uc/dc", r.UpsertedCount, len(doc))
+	//		}
+	//		return true
+	//	}
+	return r.UpsertedIDs, true
+}
+
+//批量插入
+func (m *MongodbSim) SaveBulk(c string, doc ...map[string]interface{}) bool {
+	m.Open()
+	defer m.Close()
+	coll := m.C.Database(m.DbName).Collection(c)
+	var writes []mongo.WriteModel
+	for _, d := range doc {
+		write := mongo.NewInsertOneModel()
+		write.SetDocument(d)
+		writes = append(writes, write)
+	}
+	_, e := coll.BulkWrite(m.Ctx, writes)
+	if e != nil {
+		log.Println("mgo savebulk error:", e.Error())
+		return false
+	}
+	return true
+}
+
+//保存
+func (m *MongodbSim) Save(c string, doc map[string]interface{}) interface{} {
+	m.Open()
+	defer m.Close()
+	coll := m.C.Database(m.DbName).Collection(c)
+	r, err := coll.InsertOne(m.Ctx, doc)
+	if err != nil {
+		return nil
+	}
+	return r.InsertedID
+}
+
+//更新by Id
+func (m *MongodbSim) UpdateById(c, id string, doc map[string]interface{}) bool {
+	m.Open()
+	defer m.Close()
+	coll := m.C.Database(m.DbName).Collection(c)
+	_, err := coll.UpdateOne(m.Ctx, map[string]interface{}{"_id": StringTOBsonId(id)}, doc)
+	if err != nil {
+		return false
+	}
+	return true
+}
+
+//删除by id
+func (m *MongodbSim) DeleteById(c, id string) int64 {
+	m.Open()
+	defer m.Close()
+	coll := m.C.Database(m.DbName).Collection(c)
+	r, err := coll.DeleteOne(m.Ctx, map[string]interface{}{"_id": StringTOBsonId(id)})
+	if err != nil {
+		return 0
+	}
+	return r.DeletedCount
+}
+
+//通过条件删除
+func (m *MongodbSim) Delete(c string, query map[string]interface{}) int64 {
+	m.Open()
+	defer m.Close()
+	coll := m.C.Database(m.DbName).Collection(c)
+	r, err := coll.DeleteMany(m.Ctx, query)
+	if err != nil {
+		return 0
+	}
+	return r.DeletedCount
+}
+
+//findbyid
+func (m *MongodbSim) FindById(c, id string) map[string]interface{} {
+	m.Open()
+	defer m.Close()
+	coll := m.C.Database(m.DbName).Collection(c)
+	r := coll.FindOne(m.Ctx, map[string]interface{}{"_id": StringTOBsonId(id)})
+	v := map[string]interface{}{}
+	r.Decode(&v)
+	return v
+}
+
+//findone
+func (m *MongodbSim) FindOne(c string, query map[string]interface{}) map[string]interface{} {
+	m.Open()
+	defer m.Close()
+	coll := m.C.Database(m.DbName).Collection(c)
+	r := coll.FindOne(m.Ctx, query)
+	v := map[string]interface{}{}
+	r.Decode(&v)
+	return v
+}
+
+//find
+func (m *MongodbSim) Find(c string, query map[string]interface{}, sort, fields interface{}) ([]map[string]interface{}, error) {
+	m.Open()
+	defer m.Close()
+	coll := m.C.Database(m.DbName).Collection(c)
+	op := options.Find()
+	r, err := coll.Find(m.Ctx, query, op.SetSort(sort), op.SetProjection(fields))
+	if err != nil {
+		log.Fatal(err)
+		return nil, err
+	}
+	var results []map[string]interface{}
+	if err = r.All(m.Ctx, &results); err != nil {
+		log.Fatal(err)
+		return nil, err
+	}
+	return results, nil
+}
+
+//创建_id
+func NewObjectId() primitive.ObjectID {
+	return primitive.NewObjectID()
+}
+
+func StringTOBsonId(id string) primitive.ObjectID {
+	objectId, _ := primitive.ObjectIDFromHex(id)
+	return objectId
+}
+
+func BsonTOStringId(id interface{}) string {
+	return id.(primitive.ObjectID).Hex()
+}

data_quality/src/scoreExpError.go

@@ -0,0 +1,96 @@
+package main
+
+import (
+	qu "qfw/util"
+)
+
+func dealWithErrorRate(tmp map[string]interface{}) (int , int) {
+	//错误 ， 异常error_score  abnormal_score,
+	err_num,ab_num:=0,0
+
+	//金额类
+	budget:=qu.Float64All(tmp["budget"])
+	bidamount:=qu.Float64All(tmp["bidamount"])
+	if tmp["budget"]==nil&&tmp["bidamount"]==nil {
+		//均不存在
+		ab_num++
+	}else if tmp["budget"]!=nil&&tmp["bidamount"]!=nil&&budget!=0&&bidamount!=0 {
+		//均存在
+		err,ab:=amountAnalysis(budget,bidamount)
+		err_num = err_num+err
+		ab_num = err_num+ab
+	}else {
+		//二者存在一个
+		if budget==0 && qu.ObjToString(tmp["toptype"])=="招标"{
+			ab_num++
+		}
+		if bidamount==0 && qu.ObjToString(tmp["toptype"])=="结果"{
+			ab_num++
+		}
+	}
+
+	//采购单位
+	if tmp["buyer"]==nil {
+		ab_num++
+	}else {
+		if !buyerAnalysis(qu.ObjToString(tmp["buyer"])) {
+			ab_num++
+		}
+	}
+	//中标单位
+	if tmp["winner"]==nil && qu.ObjToString(tmp["toptype"])=="结果"{
+		ab_num++
+	}else {
+		if !winnerAnalysis(qu.ObjToString(tmp["winner"])) {
+			ab_num++
+		}
+	}
+	return err_num,ab_num
+}
+
+func amountAnalysis(budget float64,bidamount float64) (int ,int) {
+	err_num,ab_num:=0,0
+	proportion := bidamount/budget
+	if proportion>=0.1&&proportion<=10 {
+
+	}else if (proportion>=0.01&&proportion<0.1)||(proportion>10&&proportion<=100){
+		ab_num++
+	}else {
+		err_num++
+	}
+
+	return err_num,ab_num
+}
+
+
+func buyerAnalysis(buyer string) bool{
+
+	if buyer=="" {
+		return false
+	}
+	q := map[string]interface{}{
+		"name": buyer,
+	}
+	data,_:=mgo.Find(qy_coll_name,q,nil,map[string]interface{}{"name":1})
+	if data==nil {
+		return false
+	}
+	return true
+}
+
+func winnerAnalysis(winner string) bool {
+
+	if winner=="" {
+		return false
+	}
+	q := map[string]interface{}{
+		"name": winner,
+	}
+	data,_:=mgo.Find(qy_coll_name,q,nil,map[string]interface{}{"name":1})
+	if data==nil {
+		return false
+	}
+	return true
+}
+
+

data_quality/src/scoreLogic.go

@@ -0,0 +1,45 @@
+package main
+
+func dealWithElementRate(tmp map[string]interface{}) int {
+
+	//score_standard 打分标准
+	m,n:=0,0
+	for _,v:=range core_element{
+		if tmp[v]!=nil {
+			m++
+		}
+	}
+	for _,v:=range other_element{
+		if tmp[v]!=nil {
+			n++
+		}
+	}
+	return calculateScore(m,n)
+}
+
+func calculateScore(core_num int,other_num int) int  {
+	/*
+
+    "total_score": 100,
+    "core_max": 70,
+    "core_each": 10,
+    "other_max": 30,
+    "other_each": 5
+	*/
+	m :=core_each*core_num
+	if m>core_max {
+		m = core_max
+	}
+
+	n :=other_each*other_num
+	if n > other_max {
+		n = other_max
+	}
+
+	t :=m+n
+	if t > total_score {
+		t=total_score
+	}
+
+	return t
+}

data_quality/src/sendmail.go

@@ -0,0 +1,59 @@
+package main
+
+import (
+	"fmt"
+	"io/ioutil"
+	"log"
+	mu "mfw/util"
+	"net"
+	"net/http"
+	"sync"
+	"time"
+)
+
+var udptaskmap = &sync.Map{}
+var tomail string
+var api string
+
+type udpNode struct {
+	data      []byte
+	addr      *net.UDPAddr
+	timestamp int64
+	retry     int
+}
+
+func checkMapJob() {
+	//阿里云内网无法发送邮件
+	jkmail, _ := Sysconfig["jkmail"].(map[string]interface{})
+	if jkmail != nil {
+		tomail, _ = jkmail["to"].(string)
+		api, _ = jkmail["api"].(string)
+	}
+	log.Println("start checkMapJob", tomail, Sysconfig["jkmail"])
+	for {
+		udptaskmap.Range(func(k, v interface{}) bool {
+			now := time.Now().Unix()
+			node, _ := v.(*udpNode)
+			if now-node.timestamp > 120 {
+				node.retry++
+				if node.retry > 5 {
+					log.Println("udp重试失败", k)
+					udptaskmap.Delete(k)
+					res, err := http.Get(fmt.Sprintf("%s?to=%s&title=%s&body=%s", api, tomail, "extract-send-fail", k.(string)))
+					if err == nil {
+						defer res.Body.Close()
+						read, err := ioutil.ReadAll(res.Body)
+						log.Println("邮件发发送：", string(read), err)
+					}
+				} else {
+					log.Println("udp重发", k)
+					udpclient.WriteUdp(node.data, mu.OP_TYPE_DATA, node.addr)
+				}
+			} else if now-node.timestamp > 10 {
+				log.Println("udp任务超时中..", k)
+			}
+			return true
+		})
+		time.Sleep(60 * time.Second)
+	}
+}

udpfilterdup/src/main.go

@@ -483,6 +483,7 @@ func historyTaskDay() {
 						},
 					})
 					if len(updateExtract) > 50 {
+
 						mgo.UpSertBulk(extract, updateExtract...)
 
 						updateExtract = [][]map[string]interface{}{}