package main import ( "log" "math/rand" "sync" "time" ) type weightInfo struct { maxLevel bool minLevel bool siteLevel int qualityScore int ranking int data map[string]interface{} } //一般数据判重 type weightDataMap struct { lock sync.Mutex //锁 data map[string]*weightInfo allids []string saveids []string templateid string //模板id } func NewWeightData(arr []string,templateid string) *weightDataMap { //测试-默认第一个 arr = []string{"5638baccaf53745d9a000994","5638baccaf53745d9a000995","5638baccaf53745d9a000998", "603717b8fc702705550b8df4","603717b8fc702705550b8df5","603717b8fc702705550b8df6"} weight := &weightDataMap{sync.Mutex{},map[string]*weightInfo{},[]string{},[]string{},templateid} data := make(map[string]*weightInfo,0) for _,v:=range arr { dict := mgo.FindById(coll_name,v) if dict!=nil && len(dict)>2{ data[v] = analyzeTheSoureData(dict) } } //测试模拟分数 //weight := &weightDataMap{sync.Mutex{},map[string]*weightInfo{},[]string{},[]string{},templateid} //data := make(map[string]*weightInfo,0) //max :=[]bool{false,false,false,false,false,false,false,false,false,false} //min :=[]bool{false,false,false,false,false,false,false,false,false,false} //site :=[]int{2,1,5,3,4,2,3,5,1,0} //qua :=[]int{15,11,11,11,22,19,22,44,22,66} //rank :=[]int{-1,-1,-1,-1,-1,-1,-1,-1,-1,-1} //for k,v:=range arr { // data[v] = &weightInfo{ // max[k], // min[k], // site[k], // qua[k], // rank[k], // } //} weight.data = data return weight } //分析源数据数据 func analyzeTheSoureData(tmp map[string]interface{}) *weightInfo { maxLevel := false minLevel := false ranking := -1 //默认 无排名 //分析站点方法 siteLevel := analyzeTheSite(tmp) //质量评分 qualityScore := analyzeTheElements(tmp) return &weightInfo{ maxLevel, minLevel, siteLevel, qualityScore, ranking, tmp, } } //分析模板数据-打标记构建数据结构 func (weight *weightDataMap) analyzeBuildStandardData() { //log.Print("分析前",weight.allids,weight.saveids,weight.templateid,len(weight.data)) weight.lock.Lock() //分析里面的打分，以及是否参与融合来决定 data:=weight.data //分析不同维度的数据-排列ranking,并调换顺序 //先构建 arrAllIds := make([]string,0) arrSaveIds := make([]string,0) arrMaxLevel := make([]bool,0) arrMinLevel := make([]bool,0) arrSiteLevel := make([]int,0) arrQualityScore := make([]int,0) arrRanking := make([]int,0) //主要排名 //无序 for k,v:=range data{ //log.Println(v) //涉及前置条件，哪些数据不需要融合 arrAllIds = append(arrAllIds,k) arrSaveIds = append(arrSaveIds,k) // arrMaxLevel = append(arrMaxLevel,v.maxLevel) arrMinLevel = append(arrMinLevel,v.minLevel) arrSiteLevel = append(arrSiteLevel,v.siteLevel) arrQualityScore = append(arrQualityScore,v.qualityScore) arrRanking = append(arrRanking,v.ranking) } log.Println("初始排名：",arrRanking) log.Println("初始质量：",arrQualityScore) log.Println("初始站点：",arrSiteLevel) //第一步，最大权重，重置排名 isMaxIndexArr := make([]int,0) //记录索引 isMaxIndexValueArr := make([]int,0) for k,v :=range arrMaxLevel { if v == true { arrRanking[k]=1 isMaxIndexArr = append(isMaxIndexArr,k) isMaxIndexValueArr = append(isMaxIndexValueArr,arrQualityScore[k]) } } rank_s :=1 if len(isMaxIndexArr)>=1 { log.Println("进行最大权重...") rankIndexArr := dealWithGroupScores(isMaxIndexArr,isMaxIndexValueArr,arrSiteLevel) //log.Println(rankIndexArr) //重新排名 for _,v:=range rankIndexArr { arrRanking[v] = rank_s rank_s++ } }else { log.Println("无最大权重-质量-站点排序") } log.Println("第一步：经过最高权重比较得出--",arrRanking) //第二步，最小权重，重置排名 isMinIndexArr := make([]int,0) isMinIndexValueArr := make([]int,0) for k,v :=range arrMinLevel { if v == true && arrMaxLevel[k]!=true { isMinIndexArr = append(isMinIndexArr,k) isMinIndexValueArr = append(isMinIndexValueArr,arrQualityScore[k]) } } if len(isMinIndexArr)>=1 { log.Println("进行最小权重...") rankIndexArr := dealWithGroupScores(isMinIndexArr,isMinIndexValueArr,arrSiteLevel) //重新排名 lastRank := len(arrSaveIds) //log.Println("最小排名分",lastRank,rankIndexArr) for i:=len(rankIndexArr)-1;i>=0;i-- { index:=rankIndexArr[i] arrRanking[index] = lastRank lastRank-- } }else { log.Println("无最小权重-质量-站点排序") } log.Println("第二步：经过最小权重比较得出--",arrRanking) //第三步，分析第一步没排名的数据 isQuaIndexArr := make([]int,0) isQuaIndexValueArr := make([]int,0) for k,v:=range arrRanking{ if v==-1 { isQuaIndexArr = append(isQuaIndexArr,k) isQuaIndexValueArr = append(isQuaIndexValueArr,arrQualityScore[k]) } } if len(isQuaIndexArr)>=1 { log.Println("进行质量-站点组合...") rankIndexArr := dealWithGroupScores(isQuaIndexArr,isQuaIndexValueArr,arrSiteLevel) //log.Println(rankIndexArr) //重新排名 for _,v:=range rankIndexArr { arrRanking[v] = rank_s rank_s++ } }else { log.Println("不需要进行质量-站点组合...") } log.Println("第三步：经过质量-站点权重比较得出--",arrRanking) template_id:="" //根据-排名-修改 for k,v:=range arrRanking { index:=arrSaveIds[k] data[index].ranking = v //log.Println("key：",index,"排名：",v) if v==1 { template_id = index } } weight.data = data weight.templateid = template_id weight.allids = arrAllIds weight.saveids = arrSaveIds weight.lock.Unlock() } func dealWithGroupScores(indexArr []int, scoreArr []int,siteArr []int) []int { //log.Println("下标组",indexArr,"质量分组",scoreArr,"整体站点组",siteArr) //处理分组 sort_scoreArr,sort_indexArr := sortGroupInt(scoreArr,indexArr) //log.Println("排序质量分：",sort_scoreArr,sort_indexArr) totalIndexArr:=make([][]int,0) lastTmp := -1 for k,v :=range sort_scoreArr { if v1 { //[6 3 4] arr_s :=make([]int,0) for _,v1:=range v{ arr_s = append(arr_s,siteArr[v1]) } _,b:=sortGroupInt(arr_s,v) for _,v2:=range b { finallyIndexArr = append(finallyIndexArr,v2) } }else { finallyIndexArr = append(finallyIndexArr,v[0]) } } return finallyIndexArr } //排序 正常排序 ，站点 func sortNormalInt(arrValue []int) ([]int){ for i := 0; i < len(arrValue); i++ { for j := i + 1; j < len(arrValue); j++ { if arrValue[i] < arrValue[j] { arrValue[i], arrValue[j] = arrValue[j], arrValue[i] } } } return arrValue } //排序 质量，分组 func sortGroupInt(arrValue []int,arrIndex []int) ([]int ,[]int){ for i := 0; i < len(arrValue); i++ { for j := i + 1; j < len(arrValue); j++ { if arrValue[i] < arrValue[j] { arrValue[i], arrValue[j] = arrValue[j], arrValue[i] arrIndex[i], arrIndex[j] = arrIndex[j], arrIndex[i] } } } return arrValue,arrIndex } //分析站点评分 func analyzeTheSite(tmp map[string]interface{}) int { /* 站点评分1-5级 */ //测试随机分 rand.Seed(time.Now().UnixNano()) //以当前系统时间作为种子参数 return rand.Intn(10) } //分析要素评分 func analyzeTheElements(tmp map[string]interface{}) int { /* 质量评分总分 */ //测试随机分 rand.Seed(time.Now().UnixNano()) //以当前系统时间作为种子参数 return rand.Intn(100) }