pushpkg/p/struct.go

package p

import (
	"encoding/json"
	"sort"
	"strings"

	util "app.yhyue.com/moapp/jybase/common"
	. "app.yhyue.com/moapp/jybase/mongodb"
	"bp.jydev.jianyu360.cn/BaseService/pushpkg/dfa"
	"github.com/donnie4w/go-logger/logger"
)

//
type MgoConf struct {
	Address          string
	Size             int
	DbName           string
	ReplSet          string
	UserName         string
	Password         string
	Collection       string
	ExperienceDbName string
}

//
type RedisConf struct {
	Address string
}

//
type MysqlConf struct {
	DbName       string
	Address      string
	UserName     string
	PassWord     string
	MaxOpenConns int
	MaxIdleConns int
}

//
type EsConf struct {
	Address string
	Size    int
}

//
type MemberService struct {
	IsBuy    bool
	Services map[int]*memberService
}

//
type memberService struct {
	Id        int
	StartTime string
}

//关键词
type KeySet struct {
	Item       string   `json:"item"`      //分类名称
	Keys       []string `json:"key"`       //关键词
	NotKeys    []string `json:"notkey"`    //排除词
	SubTypes   []string `json:"infotype"`  //信息类型
	Areas      []string `json:"area"`      //地区
	AppendKeys []string `json:"appendkey"` //附加词
	MatchWay   int      `json:"matchway"`  //匹配模式 0:精准匹配 1:模糊匹配
}

//解析订阅词
type SubSet struct {
	MatchWay              int                        //匹配方式 1-标题 2-正文
	Matchbuyerclass_other int                        //是否开启其他按钮
	ProjectMatch          int                        //项目关联推送
	Items                 []string                   //关键词分类
	Keys                  []string                   //过滤后的关键词
	Notkeys               []string                   //排除词
	Key_item              map[string]string          //关键词对应的分类名称
	Key_notkey            map[string]map[string]bool //关键词所对应的排除词
	Key_area              map[string]map[string]bool //关键词所对应的信息范围
	Key_subtype           map[string]map[string]bool //关键词所对应的信息类型
	OriginalKeys          []string                   //原始关键词
	Areas                 []string                   //区域-省份
	Area                  map[string]interface{}     //区域-省份
	Subtype               []interface{}              //信息类型-二级分类
	Subtypes              []string                   //信息类型-二级分类
	Buyerclass            []interface{}              //采购单位行业
	Buyerclasss           []string                   //采购单位行业
	IsUpgrade             bool                       //免费用户是否订阅升级
	MaxPushSize           int                        //推送最大条数
	MaxMailSize           int                        //邮件最大条数
}

//推送设置
type PushSet struct {
	Email              string        //邮箱
	SubSet             *PushSetChild //订阅推送设置
	WeekReport         *PushSetChild //周报推送设置
	MonthReport        *PushSetChild //月报推送设置
	NewprojectForecast *PushSetChild //潜在项目预测推送设置
	EntInfo            *PushSetChild //企业情报监控-企业工商变动推送设置
	FollowProject      *PushSetChild //项目进度监控推送设置
	FollowEnt          *PushSetChild //企业情报监控-企业中标动态推送设置
}

func (p *PushSet) ResetSubSet() {
	if p.SubSet == nil {
		return
	}
	if p.SubSet.RateMode == 0 || p.SubSet.RateMode == 5 {
		p.SubSet.RateMode = 2
		if len(p.SubSet.Times) == 0 {
			p.SubSet.Times = append(p.SubSet.Times, "09:00", "14:00")
		}
	} else if p.SubSet.RateMode == 2 && len(p.SubSet.Times) == 0 {
		p.SubSet.Times = append(p.SubSet.Times, "09:00")
	}
	sort.Strings(p.SubSet.Times)
}

type PushSetChild struct {
	RateMode int      //推送时间
	Times    []string //自定义推送时间
	WxPush   int      //是否开启微信推送
	AppPush  int      //是否开启app推送
	MailPush int      //是否开启邮箱推送
}

//用户基本信息
type UserInfo struct {
	Id               string //mongoid
	BaseUserId       int64  //mysql用户id
	S_m_openid       string //公众号openid
	A_m_openid       string //app微信登录openid
	Phone            string //app手机号登录
	Jpushid          string //极光推送id
	Opushid          string //厂商推送id
	AppPhoneType     string //手机型号
	Subscribe        int    //是否关注
	VipStatus        int    //超级订阅 1--试用 2--正式
	MemberStatus     int    //大会员
	NicheStatus      int    //商机管理
	IsMainAccount    bool   //是否是主账号
	SonAccountStatus int    //子账号状态 0-禁用 1-启用
	MemberMainid     string //主账号id
	Entniche         *Entniche
	WxTplMsg         *WxTplMsg
	SubSet           *SubSet
	PushSet          *PushSet
	Extend           *UserInfoExtend
}

/*
 *获取用户信息
 *temp 用户数据
 *tp 1-免费 2-超级订阅 3-大会员
 */
func NewUserInfo(temp map[string]interface{}, tp int) *UserInfo {
	ui := &UserInfo{
		Id:           BsonIdToSId(temp["_id"]),
		BaseUserId:   util.Int64All(temp["base_user_id"]),
		S_m_openid:   util.ObjToString(temp["s_m_openid"]),
		A_m_openid:   util.ObjToString(temp["a_m_openid"]),
		Phone:        GetPhone(temp),
		Jpushid:      util.ObjToString(temp["s_jpushid"]),
		Opushid:      util.ObjToString(temp["s_opushid"]),
		AppPhoneType: util.ObjToString(temp["s_appponetype"]),
		Subscribe:    util.IntAllDef(temp["i_ispush"], 1),
		VipStatus:    util.IntAll(temp["i_vip_status"]),
		MemberStatus: util.IntAll(temp["i_member_status"]),
	}
	if ui.MemberStatus > 0 {
		ui.IsMainAccount = util.IntAll(temp["i_mainaccount"]) == 1
		ui.SonAccountStatus = util.IntAllDef(temp["i_member_sub_status"], -1)
		ui.MemberMainid = util.ObjToString(temp["s_member_mainid"])
	}
	if tp == 3 {
		obj, _ := temp["o_member_jy"].(map[string]interface{})
		ui.GetSubSet(false, ui.Id, obj)
	} else if tp == 2 {
		obj, _ := temp["o_vipjy"].(map[string]interface{})
		ui.GetSubSet(false, ui.Id, obj)
	} else if tp == 1 {
		obj, _ := temp["o_jy"].(map[string]interface{})
		ui.GetSubSet(true, ui.Id, obj)
	}
	o_pushset, _ := temp["o_pushset"].(map[string]interface{})
	ui.GetPushSet(o_pushset)
	ui.PushSetFilter()
	return ui
}

//解析用户的推送设置
func (u *UserInfo) GetPushSet(obj map[string]interface{}) {
	subSet, _ := obj["o_subset"].(map[string]interface{})
	times, _ := subSet["a_times"].([]interface{})
	weekReport, _ := obj["o_week_report"].(map[string]interface{})
	monthReport, _ := obj["o_month_report"].(map[string]interface{})
	newprojectForecast, _ := obj["o_newproject_forecast"].(map[string]interface{})
	entInfo, _ := obj["o_entinfo"].(map[string]interface{})
	followProject, _ := obj["o_follow_project"].(map[string]interface{})
	followEnt, _ := obj["o_follow_ent"].(map[string]interface{})
	u.PushSet = &PushSet{
		Email: strings.TrimSpace(util.ObjToString(obj["s_email"])),
		SubSet: &PushSetChild{
			WxPush:   util.IntAllDef(subSet["i_wxpush"], 1),
			AppPush:  util.IntAllDef(subSet["i_apppush"], 1),
			MailPush: util.IntAll(subSet["i_mailpush"]),
			RateMode: util.IntAll(subSet["i_ratemode"]),
			Times:    util.ObjArrToStringArr(times),
		},
		WeekReport: &PushSetChild{
			WxPush:   util.IntAllDef(weekReport["i_wxpush"], 1),
			AppPush:  util.IntAllDef(weekReport["i_apppush"], 1),
			MailPush: util.IntAll(weekReport["i_mailpush"]),
		},
		MonthReport: &PushSetChild{
			WxPush:   util.IntAllDef(monthReport["i_wxpush"], 1),
			AppPush:  util.IntAllDef(monthReport["i_apppush"], 1),
			MailPush: util.IntAll(monthReport["i_mailpush"]),
		},
		NewprojectForecast: &PushSetChild{
			WxPush:   util.IntAllDef(newprojectForecast["i_wxpush"], 1),
			AppPush:  util.IntAllDef(newprojectForecast["i_apppush"], 1),
			MailPush: util.IntAll(newprojectForecast["i_mailpush"]),
		},
		EntInfo: &PushSetChild{
			WxPush:   util.IntAllDef(entInfo["i_wxpush"], 1),
			AppPush:  util.IntAllDef(entInfo["i_apppush"], 1),
			MailPush: util.IntAll(entInfo["i_mailpush"]),
		},
		FollowProject: &PushSetChild{
			WxPush:   util.IntAllDef(followProject["i_wxpush"], 1),
			AppPush:  util.IntAllDef(followProject["i_apppush"], 1),
			MailPush: util.IntAll(followProject["i_mailpush"]),
		},
		FollowEnt: &PushSetChild{
			WxPush:   util.IntAllDef(followEnt["i_wxpush"], 1),
			AppPush:  util.IntAllDef(followEnt["i_apppush"], 1),
			MailPush: util.IntAll(followEnt["i_mailpush"]),
		},
	}
	u.PushSet.ResetSubSet()
}

//解析用户的推送设置
func (u *UserInfo) PushSetFilter() {
	if u.Subscribe == 0 || u.S_m_openid == "" {
		u.PushSet.SubSet.WxPush = -1
		u.PushSet.WeekReport.WxPush = -1
		u.PushSet.MonthReport.WxPush = -1
		u.PushSet.NewprojectForecast.WxPush = -1
		u.PushSet.EntInfo.WxPush = -1
		u.PushSet.FollowProject.WxPush = -1
		u.PushSet.FollowEnt.WxPush = -1
	}
	if u.Jpushid == "" && u.Opushid == "" {
		u.PushSet.SubSet.AppPush = -1
		u.PushSet.WeekReport.AppPush = -1
		u.PushSet.MonthReport.AppPush = -1
		u.PushSet.NewprojectForecast.AppPush = -1
		u.PushSet.EntInfo.AppPush = -1
		u.PushSet.FollowProject.AppPush = -1
		u.PushSet.FollowEnt.AppPush = -1
	}
	mailPush := true
	if (u.S_m_openid == "" || (u.S_m_openid != "" && u.Subscribe == 0)) && u.Phone == "" && u.A_m_openid == "" {
		mailPush = false
	} else if (u.S_m_openid == "" || (u.S_m_openid != "" && u.Subscribe == 0)) && u.Phone == "" && u.A_m_openid != "" && u.Jpushid == "" && u.Opushid == "" {
		mailPush = false
	} else if !MailReg.MatchString(u.PushSet.Email) {
		mailPush = false
	}
	if !mailPush {
		u.PushSet.SubSet.MailPush = -1
		u.PushSet.WeekReport.MailPush = -1
		u.PushSet.MonthReport.MailPush = -1
		u.PushSet.NewprojectForecast.MailPush = -1
		u.PushSet.EntInfo.MailPush = -1
		u.PushSet.FollowProject.MailPush = -1
		u.PushSet.FollowEnt.MailPush = -1
	}
}

//解析用户的订阅设置
func (u *UserInfo) GetSubSet(isFreeUser bool, userId string, obj map[string]interface{}) {
	subSet := &SubSet{
		Keys:                  []string{},
		Notkeys:               []string{},
		Key_item:              map[string]string{},
		Key_notkey:            map[string]map[string]bool{},
		Key_area:              map[string]map[string]bool{},
		Key_subtype:           map[string]map[string]bool{},
		OriginalKeys:          []string{},
		Areas:                 []string{},
		Subtypes:              []string{},
		Buyerclasss:           []string{},
		MatchWay:              util.IntAllDef(obj["i_matchway"], 1),
		Matchbuyerclass_other: util.IntAllDef(obj["i_matchbuyerclass_other"], 1),
		ProjectMatch:          util.IntAll(obj["i_projectmatch"]),
		MaxPushSize:           util.IntAll(obj["i_maxpushsize"]),
		MaxMailSize:           util.IntAll(obj["i_maxmailsize"]),
	}
	var keySets []*KeySet
	if isFreeUser {
		var err error
		keySets, err = u.GetKeySets(obj["a_key"])
		if err != nil {
			logger.Error("获取用户关键词错误！", userId, obj["a_key"], err)
			return
		}
		area := obj["o_area"]
		if ppStatus := util.IntAll(obj["i_ppstatus"]); ppStatus == 1 {
			area = obj["o_area_p"]
		}
		if newFree := util.IntAll(obj["i_newfree"]); newFree == 1 && area != nil {
			subSet.IsUpgrade = true
			subSet.Subtype, _ = obj["a_infotype"].([]interface{})
			subSet.Area, _ = area.(map[string]interface{})
		}
	} else {
		items, _ := obj["a_items"].([]interface{})
		for _, v := range items {
			item, _ := v.(map[string]interface{})
			itemName := strings.TrimSpace(util.ObjToString(item["s_item"]))
			subSet.Items = append(subSet.Items, itemName)
			kss, err := u.GetKeySets(item["a_key"])
			if err != nil {
				logger.Error("获取用户关键词错误！", userId, item["a_key"], err)
				continue
			}
			for _, vv := range kss {
				if vv == nil {
					continue
				}
				if vv.MatchWay == 1 {
					for _, vvv := range vv.Keys {
						keySets = append(keySets, &KeySet{
							Item:    itemName,
							Keys:    []string{vvv},
							NotKeys: vv.NotKeys,
						})
					}
					for _, vvv := range vv.AppendKeys {
						keySets = append(keySets, &KeySet{
							Item:    itemName,
							Keys:    []string{vvv},
							NotKeys: vv.NotKeys,
						})
					}
				} else {
					vv.Item = itemName
					keySets = append(keySets, vv)
				}
			}
		}
		subSet.Buyerclass, _ = obj["a_buyerclass"].([]interface{})
		subSet.Subtype, _ = obj["a_infotype"].([]interface{})
		subSet.Area, _ = obj["o_area"].(map[string]interface{})
	}
	for _, v := range subSet.Buyerclass {
		s_v, _ := v.(string)
		if s_v == "" {
			continue
		}
		subSet.Buyerclasss = append(subSet.Buyerclasss, s_v)
	}
	for _, v := range subSet.Subtype {
		s_v, _ := v.(string)
		if s_v == "" {
			continue
		}
		subSet.Subtypes = append(subSet.Subtypes, s_v)
	}
	for area, _ := range subSet.Area {
		if area == "" {
			continue
		}
		subSet.Areas = append(subSet.Areas, area)
	}
	////////////////
	for _, vs := range keySets {
		if vs == nil {
			logger.Error(userId, "关键词设置异常")
			continue
		}
		var vs_keys []string
		for _, vs_v := range [][]string{vs.Keys, vs.AppendKeys} {
			for _, vs_vv := range vs_v {
				vs_vv = strings.TrimSpace(vs_vv)
				if vs_vv == "" {
					continue
				}
				vs_keys = append(vs_keys, vs_vv)
			}
		}
		if len(vs_keys) == 0 {
			continue
		}
		key := strings.Join(vs_keys, "+")
		subSet.OriginalKeys = append(subSet.OriginalKeys, key)
		if KeyFilterReg.MatchString(key) {
			continue
		} else if !KeyRetainReg.MatchString(key) {
			continue
		}
		subSet.Keys = append(subSet.Keys, key)
		subSet.Notkeys = append(subSet.Notkeys, vs.NotKeys...)
		//转大写
		upperKey := strings.ToUpper(key)
		subSet.Key_item[upperKey] = vs.Item
		//建立与排除词的对应关系
		for _, notkey := range vs.NotKeys {
			upperNotkey := strings.ToUpper(notkey)
			if subSet.Key_notkey[upperKey] == nil {
				subSet.Key_notkey[upperKey] = map[string]bool{}
			}
			subSet.Key_notkey[upperKey][upperNotkey] = true
		}
		//免费用户需要进行映射
		if isFreeUser {
			//建立与信息范围的对应关系
			for _, area := range vs.Areas {
				if subSet.Key_area[upperKey] == nil {
					subSet.Key_area[upperKey] = map[string]bool{}
				}
				subSet.Key_area[upperKey][area] = true
			}
			//建立与信息类型的对应关系
			for _, subtype := range vs.SubTypes {
				if subSet.Key_subtype[upperKey] == nil {
					subSet.Key_subtype[upperKey] = map[string]bool{}
				}
				subSet.Key_subtype[upperKey][subtype] = true
			}
		}
	}
	u.SubSet = subSet
}

//得到用户的关键词
func (u *UserInfo) GetKeySets(a_key interface{}) ([]*KeySet, error) {
	var keySets []*KeySet
	if a_key == nil {
		return keySets, nil
	}
	_bs, err := json.Marshal(a_key)
	if err == nil {
		err = json.Unmarshal(_bs, &keySets)
	}
	if err != nil {
		return keySets, err
	}
	for _, v := range keySets {
		if v == nil {
			continue
		}
		keys, appendKeys, notKeys := []string{}, []string{}, []string{}
		for _, vv := range v.Keys {
			keys = append(keys, SpaceReg.Split(strings.TrimSpace(vv), -1)...)
		}
		for _, vv := range v.AppendKeys {
			appendKeys = append(appendKeys, SpaceReg.Split(strings.TrimSpace(vv), -1)...)
		}
		for _, vv := range v.NotKeys {
			notKeys = append(notKeys, SpaceReg.Split(strings.TrimSpace(vv), -1)...)
		}
		v.Keys = keys
		v.AppendKeys = appendKeys
		v.NotKeys = notKeys
	}
	return keySets, err
}

//
type Entniche struct {
	EntId       int    //企业id
	EntName     string //企业名称
	DeptId      int    //部门id
	DisId       string //分发id
	UserId      int
	Unique      string
	OnlyPush    int
	IsNew       int
	PowerSource int    //权益来源
	ProductType string //产品类型
	Mail        string //邮箱
	IsDis       int    //是否是分发
	DisMember   int    //分发-超级订阅
	DisVip      int    //分发-大会员
}

//用户基本信息-扩展
type UserInfoExtend struct {
	CreateTime int64 //
	Size       int
}

func (u *UserInfo) Add(k string, m *map[string]map[*UserInfo]bool) {
	mk := (*m)[k]
	if mk == nil {
		mk = map[*UserInfo]bool{}
	}
	mk[u] = true
	(*m)[k] = mk
}

//添加信息类型映射关系
func (u *UserInfo) AddSbuype(m *map[string]map[*UserInfo]bool) {
	if len(u.SubSet.Subtypes) == 0 {
		u.Add("", m)
	} else {
		for _, v := range u.SubSet.Subtypes {
			u.Add(v, m)
		}
	}
}

//添加采购单位类型映射关系
func (u *UserInfo) AddBuyerclass(m *map[string]map[*UserInfo]bool) {
	//如果有关键词 有采购单位行业，行业加上“其它”
	if len(u.SubSet.Buyerclasss) == 0 {
		u.Add("", m)
	} else {
		if u.SubSet.Matchbuyerclass_other == 1 && len(u.SubSet.Keys) > 0 {
			u.Add("其它", m)
		}
		for _, v := range u.SubSet.Buyerclasss {
			u.Add(v, m)
		}
	}
}

//添加区域映射关系
func (u *UserInfo) AddArea(m *map[string]map[*UserInfo]bool) {
	if len(u.SubSet.Areas) == 0 {
		u.Add("", m)
	} else {
		for _, v := range u.SubSet.Areas {
			u.Add(v, m)
		}
	}
}

//添加区域城市映射关系
func (u *UserInfo) AddAreaCity(area, city *map[string]map[*UserInfo]bool) {
	if len(u.SubSet.Area) == 0 {
		u.Add("", area)
	} else {
		for k, v := range u.SubSet.Area {
			if k == "" {
				continue
			}
			vs, _ := v.([]interface{})
			if len(vs) == 0 {
				u.Add(k, area)
			} else {
				for _, vv := range vs {
					s_vv, _ := vv.(string)
					if s_vv == "" {
						continue
					}
					u.Add(s_vv, city)
				}
			}
		}
	}
}

//把用户挂在词下面
func (u *UserInfo) MakeKeyUser(keys []string, key_user *map[string]*[]*UserInfo) {
	mp := map[string]bool{}
	for _, key := range keys {
		v := strings.ToUpper(key)
		if v == "" || mp[v] {
			continue
		}
		mp[v] = true
		arr := (*key_user)[v]
		if arr == nil {
			arr = &[]*UserInfo{}
		}
		*arr = append(*arr, u)
		(*key_user)[v] = arr
	}
}

type WxTplMsg struct {
	Key  string
	Area string
}

type SortList []*MatchInfo

func (s SortList) Len() int {
	return len(s)
}

func (s SortList) Less(i, j int) bool {
	return util.Int64All((*s[i].Info)["publishtime"]) > util.Int64All((*s[j].Info)["publishtime"])
}

func (s SortList) Swap(i, j int) {
	s[i], s[j] = s[j], s[i]
}

type CSortList []map[string]interface{}

func (s CSortList) Len() int {
	return len(s)
}

func (s CSortList) Less(i, j int) bool {
	return util.Int64All(s[i]["publishtime"]) > util.Int64All(s[j]["publishtime"])
}

func (s CSortList) Swap(i, j int) {
	s[i], s[j] = s[j], s[i]
}

type MatchInfo struct {
	Info      *map[string]interface{}
	Keys      []string
	Items     []string
	MatchWays []string
	DisId     string
}

type KeyDfa struct {
	Key         *dfa.DFA
	NotKey      *dfa.DFA
	Key_user    *map[string]*[]*UserInfo
	Notkey_user *map[string]*[]*UserInfo
}

//所有用户的关键词和排除词
func (p *KeyDfa) CreateDaf() {
	//关键词
	p.Key = &dfa.DFA{}
	keys := make([]string, 0)
	for k, _ := range *p.Key_user {
		keys = append(keys, k)
	}
	p.Key.AddWord(keys...)
	//排除关键词
	p.NotKey = &dfa.DFA{}
	notKeys := make([]string, 0)
	for k, _ := range *p.Notkey_user {
		notKeys = append(notKeys, k)
	}
	p.NotKey.AddWord(notKeys...)
}

//
type MatchUser struct {
	Keys      []string
	MatchWays []string
	MatchWay  map[string]bool
}

//
type RelationProjectUser struct {
	TopTypes []string
}

//
type PushInfo struct {
	Ids  []interface{}
	Info map[string]interface{}
}

//推送返回结果
type PushResult struct {
	WxStatus   int
	AppStatus  int
	MailStatus int
	PushDate   int64
	Infos      *SortList
}

//
type PushParam struct {
	JpushTitle   string
	LastInfoDate int64
	TitleArray   []string
	Infos        *SortList
	InfosLength  int
	MailHtml     string
	PushDate     int64
	PushCount    int
	IsPush       bool
}

//
type BiddingInfo struct {
	Id              string
	Title           string
	ClearTitle      string
	HighlightTitle  string
	AreaTitle       string
	Area            string
	Publishtime     int64
	PublishtimeDiff string
	PublishtimeYMD  string
	Buyerclass      string
	Subscopeclass   string
	Bidamount       interface{}
	Budget          interface{}
	Acount          string
	Subtype         string
	Toptype         string
	Infotype        string
}

//身份信息
type IdentityInfo struct {
	Name         string
	PersonId     int64
	UserName     string
	AccountId    int64
	EntAccountId int64
	PositionId   int64
	PositionType int64
	EntId        int64
	EntUserId    int64
	EntUserName  string
	UserId       int64
}