update

internal/logic/domain/friend/friend_service.go

@@ -153,6 +153,15 @@ func (*friendService) AgreeAddFriend(ctx context.Context, userId, friendId int64
 
 // SendToFriend 消息发送至好友
 func (*friendService) SendToFriend(ctx context.Context, sender *pb.Sender, req *pb.SendMessageReq) (int64, error) {
+	//TODO::判断是否为好友
+	friend, err := FriendRepo.Get(sender.SenderId, req.ReceiverId)
+	if err != nil {
+		return 0, err
+	}
+	if friend == nil || friend.Status != FriendStatusAgree {
+		return 0, gerrors.ErrNotIsFriend
+	}
+
 	// 发给发送者
 	seq, err := proxy.MessageProxy.SendToUser(ctx, sender, sender.SenderId, req)
 	if err != nil {

pkg/gerrors/define.go

@@ -16,6 +16,7 @@ var (
 	ErrDeviceNotExist  = newError(10014, "设备不存在")
 	ErrAlreadyIsFriend = newError(10015, "对方已经是好友了")
 	ErrUserNotFound    = newError(10016, "用户找不到")
+	ErrNotIsFriend     = newError(10017, "对方暂不是好友")
 )
 
 func newError(code int, message string) error {