pushpkg/matcher/matcher.go

package matcher

import (
	"sync"

	util "app.yhyue.com/moapp/jybase/common"
	"app.yhyue.com/moapp/jybase/logger"
	. "bp.jydev.jianyu360.cn/BaseService/pushpkg/p"
)

type Matcher interface {
	Match(poolSize int, datas *[]map[string]interface{}) (*map[*UserInfo]*SortList, *map[*UserInfo]*[]string)
	ToMatch(info map[string]interface{}) (*map[*UserInfo]*MatchUser, *[]*UserInfo)
}

//
func EachFoMatch(matcher Matcher, poolSize int, datas *[]map[string]interface{}) (*map[*UserInfo]*SortList, *map[*UserInfo]*[]string) {
	defer util.Catch()
	logger.Info("开始匹配数据。。。")
	userMap := map[*UserInfo]*SortList{}
	projectUserMap := map[*UserInfo]*[]string{}
	var index int
	if datas != nil {
		lock := &sync.Mutex{}
		matchPool := make(chan bool, poolSize)
		matchWaitGroup := &sync.WaitGroup{}
		for _, temp := range *datas {
			matchPool <- true
			matchWaitGroup.Add(1)
			go func(info map[string]interface{}) {
				defer util.Catch()
				defer func() {
					matchWaitGroup.Done()
					<-matchPool
				}()
				users, projectUsers := matcher.ToMatch(info)
				lock.Lock()
				defer lock.Unlock()
				if users != nil && len(*users) > 0 {
					for k, v := range *users {
						l := userMap[k]
						if l == nil {
							l = &SortList{}
						}
						*l = append(*l, &MatchInfo{
							Info:      info,
							Keys:      v.Keys,
							Items:     v.Items,
							MatchWays: v.MatchWays,
							Extend:    map[string]interface{}{},
						})
						userMap[k] = l
					}
				}
				if projectUsers != nil && len(*projectUsers) > 0 {
					for _, v := range *projectUsers {
						l := projectUserMap[v]
						if l == nil {
							l = &[]string{}
						}
						*l = append(*l, util.ObjToString(info["_id"]))
						projectUserMap[v] = l
					}
				}
			}(temp)
			index++
			if index%500 == 0 {
				logger.Info("匹配数据", index)
			}
		}
		matchWaitGroup.Wait()
	}
	logger.Info("匹配数据结束。。。", index)
	return &userMap, &projectUserMap
}