powerCheckCenter/service/power.go

package service

import (
	"encoding/json"
	"fmt"
	"math/rand"
	"strings"
	"time"

	"app.yhyue.com/moapp/jybase/common"
	"app.yhyue.com/moapp/jybase/date"
	"app.yhyue.com/moapp/jybase/redis"
	"bp.jydev.jianyu360.cn/BaseService/powerCheckCenter/entity"
)

//权益
type PowerService struct {
	*entity.Conn
}

//
func NewPower(conn *entity.Conn) *PowerService {
	return &PowerService{
		conn,
	}
}

var level_map = map[int64]string{
	1: "专家版",
	2: "智慧版",
	3: "商机版",
	4: "试用版",
	5: "自定义",
	6: "商机版2.0",
	7: "专家版2.0",
}

/*
	权益相关
	userid  		//mongodb用户id
	baseUserId 		//base_user用户id
	accountId 		//账户id
	entId =5;		//企业id
	positionType    //职位类型 0个人 1企业
	positionId		//职位id
	切换到企业身份后，session中userId存的是企业雇员的职位id，个人身份，userId还是原来mgo库里的_id
*/
func (this *PowerService) Power(userid string, baseUserId, accountId, entId, positionType, positionId int64) *entity.Power {
	ent := &entity.Ent{}
	entniche := &entity.Entniche{}
	vip := &entity.Vip{}
	member := &entity.Member{}
	free := &entity.Free{}
	userPower := entity.Power{}
	if bytes, err := redis.GetBytes("newother", GetRedisName(positionId)); err == nil && bytes != nil {
		if err := json.Unmarshal(*bytes, &userPower); err == nil {
			return &userPower
		}
	}

	//获取用户本身的注册时间和邮箱、这个与个人、企业无关
	mgoUserFields := map[string]interface{}{
		"s_myemail":     1,
		"l_registedate": 1,
		"s_phone":       1,
		"s_m_phone":     1,
		"base_user_id":  1,
	}
	//大会员权益表查询条件
	bigmemberServiceUserQuery := map[string]interface{}{"s_userid": userid, "i_status": 0}
	//个人身份 取user表中权益
	if positionType == 0 {
		mgoUserFields["i_vip_status"] = 1
		mgoUserFields["l_vip_starttime"] = 1
		mgoUserFields["l_vip_endtime"] = 1
		mgoUserFields["o_vipjy"] = 1
		mgoUserFields["i_member_status"] = 1
		mgoUserFields["i_member_give"] = 1
		mgoUserFields["s_member_mainid"] = 1
		mgoUserFields["i_member_sub_status"] = 1
		mgoUserFields["i_member_trial"] = 1
		mgoUserFields["o_member_jy"] = 1
		mgoUserFields["o_jy"] = 1
		mgoUserFields["i_ts_guide"] = 1
		mgoUserFields["i_member_apppushunread"] = 1
		mgoUserFields["i_entniche_apppushunread"] = 1
		mgoUserFields["i_apppushunread"] = 1
	} else {
		bigmemberServiceUserQuery["s_userid"] = positionId
	}
	data, ok := this.Conn.MgoJy.FindById("user", userid, mgoUserFields)
	if ok && data != nil && len(*data) > 0 {
		//基本信息
		registeDate := common.Int64All((*data)["l_registedate"])                                            //注册时间
		phone, _ := common.If((*data)["s_phone"] != nil, (*data)["s_phone"], (*data)["s_m_phone"]).(string) //
		mail := common.ObjToString((*data)["s_myemail"])
		i_ts_guide := common.Int64All((*data)["i_ts_guide"])
		i_member_apppushunread := common.Int64All((*data)["i_member_apppushunread"])
		i_entniche_apppushunread := common.Int64All((*data)["i_entniche_apppushunread"])
		i_apppushunread := common.Int64All((*data)["i_apppushunread"])
		free.Mail = mail
		free.Registedate = registeDate
		free.Phone = phone
		free.TsGuide = i_ts_guide
		free.Apppushunread = i_apppushunread
		free.EntnicheApppushunread = i_entniche_apppushunread
		free.MemberApppushunread = i_member_apppushunread
		//个人权益
		if positionType == 0 {
			o_jy, _ := (*data)["o_jy"].(map[string]interface{})
			a_key, _ := o_jy["a_key"].([]interface{})
			free.FreeHasKey = len(a_key) > 0
			if common.IntAll((o_jy)["i_newfree"]) > 0 || entity.IsNewFreeTimeCell < registeDate {
				//IsNewFreeTimeCell dev3.6.4版本之前发了个紧急版本处理老用户订阅问题，i_newfree字段必须用户选择地区才能生成，不能作为判断是否是新用户得唯一标识，在此版本添加了常量：IsNewFreeTimeCell作为判断标准；--ws
				free.IsUpgrade = true //新免费用户
			}
			//超级订阅
			vipStatus := common.Int64All((*data)["i_vip_status"])
			vipStartTime := common.Int64All((*data)["l_vip_starttime"])
			vipEndTime := common.Int64All((*data)["l_vip_endtime"])
			ovipjy := common.ObjToMap((*data)["o_vipjy"])
			oBuyset := common.ObjToMap((*ovipjy)["o_buyset"])
			upgrade := common.Int64All((*oBuyset)["upgrade"])
			areacount := common.Int64All((*oBuyset)["areacount"])
			buyerclasscount := common.Int64All((*oBuyset)["buyerclasscount"])
			vip = &entity.Vip{
				Status:          vipStatus,
				StartTime:       vipStartTime,
				EndTime:         vipEndTime,
				Upgrade:         upgrade,
				Areacount:       areacount,
				Buyerclasscount: buyerclasscount,
				MaxKeyLength:    300,
				PowerType:       common.Int64All(common.If(vipStatus > 0, 1, 0)),
				//TODO 附件下载次数
			}
			if (*ovipjy)["a_items"] != nil {
				a_items := common.ObjArrToMapArr((*ovipjy)["a_items"].([]interface{}))
				vip.HasKey = HasKey(a_items)
			}
			//个人member
			i_member_sub_status := common.IntAllDef((*data)["i_member_sub_status"], 0)

			memberStatus := common.Int64All((*data)["i_member_status"])
			memberStarttime := common.Int64All((*data)["i_member_starttime"])
			memberEndtime := common.Int64All((*data)["i_member_endtime"])
			member = &entity.Member{
				Status:       memberStatus,
				StartTime:    memberStarttime,
				EndTime:      memberEndtime,
				MaxKeyLength: 300, //最大关键词数量限制
				PowerType:    common.Int64All(common.If(memberStatus > 0, 1, 0)),
			}
			omemberjy := common.ObjToMap((*data)["o_member_jy"])
			if (*omemberjy)["a_items"] != nil {
				if arr, ok := (*omemberjy)["a_items"].([]interface{}); ok {
					a_items := common.ObjArrToMapArr(arr)
					member.HasKey = HasKey(a_items)
				}
			}
			if memberStatus > 0 {
				member.MemberPower = 1 //是否分配了大会员；0：否 1：是
			}
			if (*data)["s_member_mainid"] != nil && common.ObjToString((*data)["s_member_mainid"]) != "" && i_member_sub_status > 0 {
				member.Pid = common.ObjToString((*data)["s_member_mainid"])
				member.IsSubCount = 1
			}
			if (member.Pid != "" && common.IntAllDef((*data)["i_member_sub_status"], 0) == 1) || this.Conn.MgoJy.Count("member", map[string]interface{}{"userid": userid}) > 0 {
				member.Used = true
			}
			if (*data)["i_member_trial"] != nil {
				member.IsMemberTrial = 1
			}
			//获取大会员版本名称
			member.MemberName = level_map[memberStatus]
		} else if positionType == 1 {
			entnicheUserId := this.GetEntnicheUserId(entId, phone)
			//免费
			fdata, ok := this.Conn.MgoJy.FindOne("entniche_rule", map[string]interface{}{
				"i_ent":    entId,
				"i_userid": entnicheUserId,
				"i_type":   2, //0：商机管理 1：超级订阅/大会员 2：免费
			})
			o_jy, _ := (*fdata)["o_entniche"].(map[string]interface{})
			a_key, _ := o_jy["a_key"].([]interface{})
			free.FreeHasKey = len(a_key) > 0
			if common.IntAll((o_jy)["i_newfree"]) > 0 || entity.IsNewFreeTimeCell < registeDate { //IsNewFreeTimeCell dev3.6.4版本之前发了个紧急版本处理老用户订阅问题，i_newfree字段必须用户选择地区才能生成，不能作为判断是否是新用户得唯一标识，在此版本添加了常量：IsNewFreeTimeCell作为判断标准；--ws
				free.IsUpgrade = true //新免费用户
			}
			//企业
			edata, ok := this.Conn.MgoJy.FindOne("entniche_rule", map[string]interface{}{
				"i_ent":    entId,
				"i_userid": entnicheUserId,
				"i_type":   1, //0：商机管理 1：超级订阅/大会员
				"l_endtime": map[string]interface{}{
					"$gt": time.Now().Unix(),
				},
			})
			if ok && edata != nil && len(*edata) > 0 {
				haskey := false
				status := common.Int64All((*edata)["i_status"])
				starttime := common.Int64All((*edata)["l_starttime"])
				endtime := common.Int64All((*edata)["l_endtime"])
				o_entniche := common.ObjToMap((*edata)["o_entniche"])
				if (*o_entniche)["a_items"] != nil {
					if arr, ok := (*o_entniche)["a_items"].([]interface{}); ok {
						a_items := common.ObjArrToMapArr(arr)
						haskey = HasKey(a_items)
					}
				}
				//获取用户是什么版本权益
				//用于判断是否含有企业主体的超级订阅、大会员服务
				epowerData := this.Conn.Mysql.SelectBySql(`SELECT c.product_type FROM entniche_user a INNER JOIN entniche_power b 
		 									INNER JOIN entniche_wait_empower c 
		 									ON a.id =b.ent_user_id  AND c.id = b.wait_empower_id WHERE a.phone =? AND b.ent_id=? AND c.end_time >?`, phone, entId, time.Now().Format(date.Date_Full_Layout))

				if epowerData != nil && len(*epowerData) > 0 {
					for _, v := range *epowerData {
						product_type := common.ObjToString(v["product_type"])
						if strings.Contains(product_type, entity.ProductType_vip) {
							oBuyset := common.ObjToMap((*o_entniche)["o_buyset"])
							upgrade := common.Int64All((*oBuyset)["oBuyset"])
							areacount := common.Int64All((*oBuyset)["areacount"])
							buyerclasscount := common.Int64All((*oBuyset)["buyerclasscount"])
							vip = &entity.Vip{
								Status:          status,
								StartTime:       starttime,
								EndTime:         endtime,
								Upgrade:         upgrade,
								Areacount:       areacount,
								Buyerclasscount: buyerclasscount,
								MaxKeyLength:    300,
								PowerType:       2,
								HasKey:          haskey,
							}
						} else if strings.Contains(product_type, entity.ProductType_member) {
							member.PowerType = 2
							member = &entity.Member{
								Status:     status,
								StartTime:  starttime,
								EndTime:    endtime,
								MemberName: level_map[status],
								PowerType:  2,
								HasKey:     haskey,
							}
							//是否试用
							if this.Conn.MgoJy.Count("member", map[string]interface{}{"userid": positionId}) > 0 {
								member.Used = true
							}
						}
					}
				}
			}
		}
		//大会员相关权益数量
		memberServiceMap := map[int64]bool{}
		serviceList := this.Conn.Mysql.Find("bigmember_service_user", bigmemberServiceUserQuery, "DISTINCT(s_serviceid),i_frequency", "", -1, -1)
		if serviceList != nil && len(*serviceList) != 0 {
			pCount, eCount, dailyNum, customers := 0, 0, 0, 10
			for _, item := range *serviceList {
				serviceid := common.Int64All(item["s_serviceid"])
				memberServiceMap[serviceid] = true
				member.MemberPowerList = append(member.MemberPowerList, serviceid)
				if serviceid == 14 { //项目数量
					pCount = common.IntAll(item["i_frequency"])
				} else if serviceid == 4 || serviceid == 12 || serviceid == 13 { //企业情报监控 企业中标动态
					tEcount := common.IntAll(item["i_frequency"])
					if tEcount > eCount {
						eCount = tEcount
					}
				} else if serviceid == 17 || serviceid == 18 { //每日数据包
					dailyNum = common.IntAll(item["i_frequency"])
				} else if serviceid == 7 { //潜在客户 关注客户
					customers = common.IntAll(item["i_frequency"])
				}
			}
			member.EntNum = int64(eCount)
			member.ProNum = int64(pCount)
			member.DailyNum = int64(dailyNum)
			member.Customers = int64(customers)
		}
		//
		if phone != "" {
			//查询是否是商机管理付费用户
			res := this.Conn.Mysql.SelectBySql(`SELECT i. STATUS AS status, i.isNew, i.power_source, r.role_id, u.power, i.name,i.id,i.startdate,i.enddate,i.auth_status,i.auth_reason,i.dept_subscribe FROM (entniche_user u LEFT JOIN entniche_user_role r ON r.user_id = u.id ) LEFT JOIN entniche_info i ON u.ent_id = i.id WHERE u.phone = ? and i.id = ? ORDER BY i. STATUS DESC, i.auth_status DESC`, phone, entId)
			if res != nil && len(*res) > 0 {
				for _, v := range *res {
					if common.IntAll(v["id"]) == 0 {
						continue
					}
					entniche.IsNew = common.Int64All(v["isNew"])
					entniche.Status = common.Int64All(v["status"])
					entniche.IsEntPower = common.Int64All(v["power"])
					entniche.PowerSource = common.Int64All(v["power_source"])
					entniche.StartTime = common.Int64All(v["startdate"])
					entniche.EndTime = common.Int64All(v["enddate"])
					//
					ent.Name = common.ObjToString(v["name"])
					ent.EntRoleId = common.Int64All(v["role_id"])
					ent.EntAuthStatus = common.Int64All(v["auth_status"])
					ent.EntAuthReason = common.ObjToString(v["auth_reason"])
					ent.DeptSubscribe = common.Int64All(v["dept_subscribe"])
					//判断企业是否购买企业版相关
					eweData := this.Conn.Mysql.SelectBySql(`select * from entniche_wait_empower where ent_id=? and end_time>?`, entId, time.Now().Format(date.Date_Full_Layout))
					if eweData != nil && len(*eweData) > 0 {
						for _, v := range *eweData {
							product_type := common.ObjToString(v["product_type"])
							if strings.Contains(product_type, entity.ProductType_member) {
								ent.BuyMember = 1
							} else if strings.Contains(product_type, entity.ProductType_vip) {
								ent.BuyVip = 1
							}
						}
					}
					//企业管理员
					if ent.EntRoleId == 1 && ((ent.BuyMember == 1 || ent.BuyVip == 1) || (entniche.Status == 1 && entniche.PowerSource == 0)) {
						ent.EntSubscribeManager = 1
					}
					//部门管理员
					if ent.EntRoleId == 2 && entniche.Status == 1 && entniche.PowerSource == 0 && ent.DeptSubscribe == 1 {
						ent.EntSubscribeManager = 1
					}
					//
					if ent.EntRoleId == 1 || ent.EntRoleId == 2 {
						if (ent.BuyMember == 1 || ent.BuyVip == 1) || (entniche.Status == 1 && entniche.PowerSource == 0) {
							ent.EntSubscribe = 1
						}
					}
				}

				//广东移动DICT 用户
				ent.PrivateGD = this.Conn.Mysql.CountBySql(`select count(1) from privatedata where phone = ?`, phone) > 0
			}
		}
		if vip.Status <= 0 && member.Status <= 0 && entniche.Status <= 0 {
			free.IsFree = true
		} else {
			free.IsFree = false
		}
	}
	userPower = entity.Power{
		Vip:      vip,
		Member:   member,
		Free:     free,
		Ent:      ent,
		Entniche: entniche,
	}
	//存储缓存
	go func() {
		if bytes, err := json.Marshal(userPower); err == nil && bytes != nil {
			oneDayMore := 60*60*24 + rand.Intn(60*60)
			_ = redis.PutBytes("newother", GetRedisName(positionId), &bytes, oneDayMore)
		}
	}()
	return &userPower
}

//免费用户体验会员功能权限
//免费用户在企业画像/采购单位画像/附件下载留资 留资成功后用户才有功能使用次数
func FreeExperience(userId string) (int, int, int) {
	return redis.GetInt(entity.PowerCacheDb, fmt.Sprintf(entity.PowerCacheEntPortKey, userId)),
		redis.GetInt(entity.PowerCacheDb, fmt.Sprintf(entity.PowerCacheBuyerPortKey, userId)),
		redis.GetInt(entity.PowerCacheDb, fmt.Sprintf(entity.PowerCacheFileKey, userId))
}

/*
判断是企业或个人
1.企业
 entniche_rule
 查询 vip/member  free ent entniche
2.个人
  查询vip member free
*/

func (this *PowerService) GetEntnicheUserId(entId int64, phone string) int64 {
	data := this.Conn.Mysql.SelectBySql(`select id from entniche_user where phone =? and ent_id =?`, phone, entId)
	if data != nil && len(*data) > 0 {
		return common.Int64All((*data)[0]["id"])
	}
	return -1
}

//是否有关键词
func HasKey(a_items []map[string]interface{}) bool {
	for _, v := range a_items {
		akey, _ := v["a_key"].([]interface{})
		if len(akey) > 0 {
			return true
		}
	}
	return false
}

//获取redis key
func GetRedisName(positionId int64) string {
	if positionId == 0 {
		return ""
	}
	return fmt.Sprintf("user_power_info_%v", positionId)
}

//清除redis缓存
func (this *PowerService) DelRedisPower(positionId int64) bool {
	if positionId == 0 {
		return false
	}
	return redis.Del("newother", GetRedisName(positionId))
}