full/module/social/relation/internal/logic/common/const.go

package common

import (
	"encoding/json"
	"fmt"
	"hash/crc64"
	"sort"

	"bsm/full/module/social/relation/internal/impl"
	"bsm/full/module/social/relation/internal/models"
	pb "bsm/full/module/social/relation/pb"
)

var (
	Filed string = "relation_identity as identity,nickname,avatar,sex,province,city,area,sign"
)

// 根据identity获取会员信息卡片
func GetrelationInfoDetailCard(identity string) (card *pb.RelationItem, err error) {
	err = impl.DBService.Table("relation_extend").Select(Filed).Where("relation_identity = ?", identity).First(&card).Error
	return
}

// 根据identity数组获取多条会员信息卡片
func GetrelationInfoDetailCardById(identities []string) (cards []*pb.RelationItem, err error) {
	err = impl.DBService.Table("relation_extend").Select(Filed).Where("relation_identity in ?", identities).Find(&cards).Error
	return
}

// 实现将2个字符串，不论先后顺序输入，输出得到的唯一值都是一样的。
func UniqueSessionID(s1, s2 string) string {
	strs := []string{s1, s2}
	sort.Strings(strs)
	table := crc64.MakeTable(crc64.ECMA)
	return fmt.Sprintf("%x", crc64.Checksum([]byte(strs[0]+strs[1]), table))
}

// 根据用户ID，找到用户的申请好友信息
func CollectionFriendApply(id uint) (reply *pb.ApplyFetchReply, err error) {
	var (
		version      int64
		identities   []string
		last_msg_ids []uint
		maxSize      int = 50
		applys           = make([]*models.FriendApply, 0)
		replyItems       = make([]*pb.ApplyItem, 0)
		mapCards         = make(map[string]*pb.RelationItem)

		messages    = make([]*models.FriendApplyMessage, 0)
		mapMessages = make(map[uint]*pb.MessageItem)
	)
	reply = &pb.ApplyFetchReply{}

	err = impl.DBService.Where("to_id=? ", id).Order("id desc").Limit(maxSize).Find(&applys).Error
	if err != nil {
		return nil, err
	}

	for _, item := range applys {
		identities = append(identities, item.FromIdentity)
		last_msg_ids = append(last_msg_ids, item.LastMessageID)
	}

	// 按申请人身份取资料卡
	result, err := GetrelationInfoDetailCardById(identities)
	if err != nil {
		return nil, err
	}
	for _, item := range result {
		mapCards[item.Identity] = item
	}

	err = impl.DBService.Where("id in ?", last_msg_ids).Find(&messages).Error
	if err != nil {
		return nil, err
	}
	for _, item := range messages {
		mapMessages[item.ID] = &pb.MessageItem{
			RelationIdentity: item.PassportIdentity,
			Body:             item.Body,
			CreatedAt:        item.CreatedAt.Local().Format("2006-01-02 15:04:05"),
		}
	}

	for _, item := range applys {
		var applyItem = &pb.ApplyItem{}
		applyItem.Identity = item.Identity
		applyItem.CreatedAt = item.CreatedAt.Local().Format("2006-01-02 15:04:05")
		applyItem.Status = int32(item.Status)

		if _, ok := mapCards[item.FromIdentity]; ok {
			applyItem.From = mapCards[item.FromIdentity]
		}

		if _, ok := mapMessages[item.LastMessageID]; ok {
			applyItem.Message = mapMessages[item.LastMessageID]
		}

		replyItems = append(replyItems, applyItem)
		version = int64(item.ID)
	}

	reply.Applys = replyItems
	reply.Version = version
	return
}

// 根据用户ID，找到用户的好友信息和分组信息
func CollectionFriendData(id uint) (reply *pb.FriendsReply, err error) {
	var (
		version    int64
		identities []string

		relationFriend   = make([]models.RelationFriend, 0)
		mapRelationFrend = make(map[string]models.RelationFriend)

		friendInTag    = make([]models.FriendInTag, 0)
		mapFriendInTag = make(map[string][]string)
		mapTagTotal    = make(map[string]int64)
	)
	reply = &pb.FriendsReply{}
	// 好友条目归属列是 passport_id，好友身份是 friend_relation_identity
	err = impl.DBService.Where("passport_id=?", id).Order("id desc").Find(&relationFriend).Error
	if err != nil {
		return nil, err
	}

	for _, item := range relationFriend {
		identities = append(identities, item.FriendrelationIdentity)
		mapRelationFrend[item.FriendrelationIdentity] = item
		version = int64(item.ID)
	}

	err = impl.DBService.Where("passport_id=?", id).Find(&friendInTag).Error
	if err != nil {
		return nil, err
	}

	for _, item := range friendInTag {
		mapFriendInTag[item.FriendIdentity] = append(mapFriendInTag[item.FriendIdentity], item.TagIdentity)
		mapTagTotal[item.TagIdentity]++
	}

	// 标签列表按归属过滤并补齐标签名
	err = impl.DBService.Model(models.FriendTag{}).Select("id, identity, name as tag_name").Where("passport_id=?", id).Order("id asc").Scan(&reply.Tags).Error
	if err != nil {
		return
	}
	// 补齐每个标签的成员数
	for idx := range reply.Tags {
		reply.Tags[idx].FriendTotal = mapTagTotal[reply.Tags[idx].Identity]
	}

	// 按好友身份装配资料卡，并回填备注与置顶
	reply.Friends, err = GetrelationInfoDetailCardById(identities)
	if err != nil {
		return
	}
	for idx := range reply.Friends {
		if item, ok := mapRelationFrend[reply.Friends[idx].Identity]; ok {
			reply.Friends[idx].RemarkName = item.RemarkName
			reply.Friends[idx].Popular = int32(item.Popular)
		}
		if tags, ok := mapFriendInTag[reply.Friends[idx].Identity]; ok {
			reply.Friends[idx].Tags = tags
		}
	}

	reply.Version = version

	return
}

// 根据Match表获取多条会员信息卡片
func GetrelationInfoDetailCardByMatch(relation_identity string, offset, limit int) (cards []*pb.RelationItem, err error) {
	sql := `Select 
			pe.relation_identity as identity, 
			pe.nickname,
			pe.avatar,
			pe.birthday,
			pe.sex,
			pe.province,
			pe.city,
			pe.area,
			pe.sign,
			rm.status as foreign_status 
			From relation_match as rm 
			Join relation_extend as pe on pe.relation_identity=rm.recommend_identity
			Where rm.relation_identity=?
			Order by rm.id Desc
			Limit ?
			Offset ?`
	err = impl.DBService.Raw(sql, relation_identity, limit, offset).Scan(&cards).Error
	return
}

func SetCache(identity string, collectionName string, in interface{}) error {
	key := "/RF/" + identity + "/" + collectionName
	val, err := json.Marshal(in)
	if err != nil {
		return err
	}
	err = impl.RedisService.Client.Set(impl.RedisService.Ctx, key, string(val), 0).Err()
	if err != nil {
		return err
	}

	return nil
}

func GetCache(identity, collectionName string) (reply *pb.ApplyFetchReply, err error) {
	key := "/RF/" + identity + "/" + collectionName
	collection, err := impl.RedisService.Client.Get(impl.RedisService.Ctx, key).Result()
	if err != nil {
		return
	}
	err = json.Unmarshal([]byte(collection), &reply)
	if err != nil {
		return
	}

	return
}