pushpkg/cache/cache.go

package cache

import (
	"encoding/json"
	"fmt"
	"strings"

	. "app.yhyue.com/moapp/jybase/common"
	. "app.yhyue.com/moapp/jybase/encrypt"
	"app.yhyue.com/moapp/jybase/redis"
)

const (
	threeDay            = 172800
	oneDay              = 86400
	AllSubPushCacheSize = 200
)

type SubPushList struct {
	Id            string        `json:"_id"`
	Title         string        `json:"title"`
	Area          string        `json:"area"`
	BuyerClass    string        `json:"buyerClass"`
	Subtype       string        `json:"subtype"`
	Industry      string        `json:"industry"`
	PublishTime   int64         `json:"publishTime"`
	Ca_index      int64         `json:"ca_index"`
	Ca_date       int64         `json:"ca_date"`
	Ca_isvisit    int           `json:"ca_isvisit"`
	Ca_isvip      int           `json:"ca_isvip"`
	Ca_type       int           `json:"ca_type"`
	Ca_fileExists bool          `json:"ca_fileExists"`
	MatchKeys     []string      `json:"matchKeys"`
	Budget        interface{}   `json:"budget"`
	BidAmount     interface{}   `json:"bidAmount"`
	Collection    int           `json:"collection"`
	Buyer         string        `json:"buyer"`
	ProjectName   string        `json:"projectName"`
	Winner        string        `json:"winner"`
	BidOpenTime   int64         `json:"bidOpenTime"`
	Source        int64         `json:"source"`
	WinnerInfo    []*WinnerInfo `json:"winnerInfo"`
	Site          string        `json:"site"`
	BuyerTel      string        `json:"buyerTel"`
	BuyerPerson   string        `json:"buyerPerson"`
	Agency        string        `json:"agency"`
	AgencyPerson  string        `json:"agencyPerson"`
	AgencyTel     string        `json:"agencyTel"`
	SignendTime   int64         `json:"signendTime"`
	BidendTime    int64         `json:"bidendTime"`
}

//
type WinnerInfo struct {
	Winner       string `json:"winner"`
	WinnerTel    string `json:"winnerTel"`
	WinnerPerson string `json:"winnerPerson"`
	WinnerId     string `json:"winnerId"`
}

type SubPush struct {
	Date  string
	Datas []*SubPushList
	Count int64
}

type PushCa struct {
	Date       int64
	InfoId     string
	Visit      int
	Keys       []string
	Type       int
	Isvip      int
	FileExists bool
	Source     int64
	Info       *map[string]interface{}
}

type Cache struct {
	Prefix string
	UserId string
	Ymd    string
}

func NewEntnicheCache(userId, ymd string) *Cache {
	return &Cache{
		Prefix: "entnichepush",
		UserId: userId,
		Ymd:    ymd,
	}
}

func NewSubscribeCache(userId, ymd string) *Cache {
	return &Cache{
		Prefix: "subpush",
		UserId: userId,
		Ymd:    ymd,
	}
}

func NewMemberCache(userId, ymd string) *Cache {
	return &Cache{
		Prefix: "memberpush",
		UserId: userId,
		Ymd:    ymd,
	}
}

//
func NewSubPushList(pushCas []*PushCa) []*SubPushList {
	subPushList := []*SubPushList{}
	for _, p := range pushCas {
		info := p.Info
		area := ObjToString((*info)["area"])
		if area == "A" {
			area = "全国"
		}
		industry := ObjToString((*info)["s_subscopeclass"])
		scs := strings.Split(industry, ",")
		if len(scs) > 0 {
			industry = scs[0]
			if industry != "" {
				iss := strings.Split(industry, "_")
				if len(iss) > 0 {
					industry = iss[0]
				}
			}
		}
		infotype := ObjToString((*info)["subtype"])
		if infotype == "" {
			infotype = ObjToString((*info)["toptype"])
		}
		_id := p.InfoId
		if _id == "" {
			_id = ObjToString((*info)["_id"])
		}
		winner := ObjToString((*info)["s_winner"])
		winnerInfo := []*WinnerInfo{}
		if winner != "" && len(strings.Split(winner, ",")) > 0 {
			for wk, wv := range strings.Split(winner, ",") {
				var winnerId = ""
				eil, _ := (*info)["entidlist"].([]interface{})
				if entIdList := ObjArrToStringArr(eil); len(entIdList) > wk { //中标企业id集合
					winnerId = entIdList[wk]
				}
				winnerInfo = append(winnerInfo, &WinnerInfo{
					Winner:       wv,                                                                                               //中标企业 需要单独处理
					WinnerTel:    ObjToString((*info)["winnertel"]),                                                                //中标企业联系电话
					WinnerPerson: ObjToString((*info)["winnerperson"]),                                                             //中标企业联系人
					WinnerId:     If(winnerId != "" && len([]rune(winnerId)) > 12, EncodeArticleId2ByCheck(winnerId), "").(string), //中标企业加密id  存在winnerId 异常的情况
				})
			}
		}
		subPushList = append(subPushList, &SubPushList{
			Id:            EncodeArticleId2ByCheck(_id),
			Title:         ObjToString((*info)["title"]),
			Area:          area,
			BuyerClass:    ObjToString((*info)["buyerclass"]),
			Subtype:       infotype,
			Industry:      industry,
			PublishTime:   Int64All((*info)["publishtime"]),
			Ca_date:       p.Date,
			Ca_isvisit:    p.Visit,
			Ca_isvip:      p.Isvip,
			Ca_type:       p.Type,
			Ca_fileExists: p.FileExists,
			MatchKeys:     p.Keys,
			Budget:        (*info)["budget"],
			BidAmount:     (*info)["bidamount"],
			Buyer:         ObjToString((*info)["buyer"]),
			ProjectName:   ObjToString((*info)["projectname"]),
			Winner:        winner,
			BidOpenTime:   Int64All((*info)["bidopentime"]),
			Source:        p.Source,
			Site:          ObjToString((*info)["site"]),
			BuyerTel:      ObjToString((*info)["buyerTel"]),
			BuyerPerson:   ObjToString((*info)["buyerPerson"]),
			Agency:        ObjToString((*info)["agency"]),
			AgencyPerson:  ObjToString((*info)["agencyPerson"]),
			AgencyTel:     ObjToString((*info)["agencyTel"]),
			SignendTime:   Int64All((*info)["signendTime"]),
			BidendTime:    Int64All((*info)["bidendTime"]),
			WinnerInfo:    winnerInfo,
		})
	}
	return subPushList
}

//从pushcache_2_b中取
func (c *Cache) GetTodayCache(userId string) (*SubPush, error) {
	pc_a, err := redis.GetNewBytes("pushcache_2_b", c.todayKey(userId))
	if err != nil {
		return nil, err
	}
	if pc_a == nil {
		return nil, nil
	}
	var p *SubPush
	if err := json.Unmarshal(*pc_a, &p); err != nil {
		return nil, err
	}
	return p, nil
}

//往pushcache_2_b中放
func (c *Cache) PutTodayCache(userId string, pc_a *SubPush) {
	redis.Put("pushcache_2_b", c.todayKey(userId), pc_a, oneDay)
}

//获取redis key
func (c *Cache) todayKey(userId string) string {
	return fmt.Sprintf("%s_%s", c.Prefix, userId)
}
func (c *Cache) allKey(userId string) string {
	return fmt.Sprintf("all_%s_%s", c.Prefix, userId)
}

//查看全部列表缓存
func (c *Cache) PutAllCache(userId string, datas *SubPush) {
	redis.Put("pushcache_2_a", c.allKey(userId), datas, oneDay)
}

func (c *Cache) GetAllCache(userId string) (*SubPush, error) {
	return c.GetCache("pushcache_2_a", c.allKey(userId))
}

func (c *Cache) GetCache(code, key string) (*SubPush, error) {
	pc_a, err := redis.GetNewBytes(code, key)
	if err != nil {
		return nil, err
	}
	if pc_a == nil {
		return nil, nil
	}
	var p *SubPush
	if err := json.Unmarshal(*pc_a, &p); err != nil {
		return nil, err
	}
	return p, nil
}

func (c *Cache) DoCache(subPushList []*SubPushList) {
	subPush_datas := []*SubPushList{}
	for i := len(subPushList) - 1; i >= 0; i-- {
		subPush_datas = append(subPush_datas, subPushList[i])
	}
	//更新redis
	subPush, err := c.GetTodayCache(c.UserId)
	if err == nil && subPush != nil && len(subPush.Datas) > 0 {
		if subPush.Date != c.Ymd {
			subPush = &SubPush{
				Date:  c.Ymd,
				Datas: subPush_datas,
				Count: int64(len(subPush_datas)),
			}
		} else {
			subPush.Datas = append(subPush_datas, subPush.Datas...)
			subPush.Count = subPush.Count + int64(len(subPush_datas))
		}
		c.PutTodayCache(c.UserId, subPush)
	}
	//全部列表缓存
	allCache, err := c.GetAllCache(c.UserId)
	if err == nil && allCache != nil && len(allCache.Datas) > 0 {
		allCache.Datas = append(subPush_datas, allCache.Datas...)
		if len(allCache.Datas) > AllSubPushCacheSize {
			allCache.Datas = allCache.Datas[:AllSubPushCacheSize]
		}
		allCache.Count = allCache.Count + int64(len(subPush_datas))
		c.PutAllCache(c.UserId, allCache)
	}
}