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
}

// 根据id数组获取多条会员信息卡片
func GetrelationInfoDetailCardById(ids []uint) (cards []*pb.RelationItem, err error) {
	err = impl.DBService.Table("relation_extend").Select(Filed).Where("relation_id in ?", ids).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
		ids          []uint
		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 {
		ids = append(ids, item.FromID)
		last_msg_ids = append(last_msg_ids, item.LastMessageID)
	}

	result, err := GetrelationInfoDetailCardById(ids)
	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
		ids     []uint

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

		friendInTag    = make([]models.FriendInTag, 0)
		mapFriendInTag = make(map[string][]string)
	)
	reply = &pb.FriendsReply{}
	impl.DBService.Where("relation_id=?", id).Order("id desc").Find(&relationFriend)

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

	impl.DBService.Where("relation_id=?", id).Find(&friendInTag)

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

	err = impl.DBService.Model(models.FriendTag{}).Where("relation_id=?", id).Order("id asc").Scan(&reply.Tags).Error
	if err != nil {
		return
	}

	reply.Friends, err = GetrelationInfoDetailCardById(ids)
	for idx, item := range reply.Friends {
		reply.Friends[idx].RemarkName = item.RemarkName
		reply.Friends[idx].Popular = item.Popular
		if _, ok := mapFriendInTag[item.Identity]; ok {
			reply.Friends[idx].Tags = mapFriendInTag[item.Identity]
		}
	}

	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.name,
			pe.avatar,
			pe.birthday,
			pe.sex,
			pe.province,
			pe.city,
			pe.area,
			pe.sign,
			rm.status as foreign_status 
			From relation_match as rm,relation_extend as pe 
			Left join pe on pe.relation_identity=rm.recommend_identity
			Where rm.relation_identity=?
			Offset ?
			Limit ?
			Order by rm.id Desc`
	err = impl.DBService.Raw(sql, relation_identity, offset, limit).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
}