复制项目

pkg/rpcli/relation.go

@@ -0,0 +1,24 @@
+package rpcli
+
+import (
+	"context"
+
+	"git.imall.cloud/openim/protocol/relation"
+	"google.golang.org/grpc"
+)
+
+func NewRelationClient(cc grpc.ClientConnInterface) *RelationClient {
+	return &RelationClient{relation.NewFriendClient(cc)}
+}
+
+type RelationClient struct {
+	relation.FriendClient
+}
+
+func (x *RelationClient) GetFriendsInfo(ctx context.Context, ownerUserID string, friendUserIDs []string) ([]*relation.FriendInfoOnly, error) {
+	if len(friendUserIDs) == 0 {
+		return nil, nil
+	}
+	req := &relation.GetFriendInfoReq{OwnerUserID: ownerUserID, FriendUserIDs: friendUserIDs}
+	return extractField(ctx, x.FriendClient.GetFriendInfo, req, (*relation.GetFriendInfoResp).GetFriendInfos)
+}