feat 初始化

net/grpc/grpcclient/stream_gateway.go

@@ -3,7 +3,11 @@ package grpcclient
 import (
 	"context"
 	"fmt"
+	"git.hlsq.asia/alex/proto/proto/rs/rspb"
+	"git.hlsq.asia/alex/service-common/discover"
+	"git.hlsq.asia/alex/service-common/discover/common"
 	"git.hlsq.asia/alex/service-common/log"
+	"git.hlsq.asia/alex/service-common/utils/locator"
 	"google.golang.org/grpc"
 	"google.golang.org/protobuf/proto"
 	"sync"
@@ -13,6 +17,8 @@ type GatewayFun int
 
 const (
 	FunToClient GatewayFun = iota
+	FunBroadcastClient
+	FunBroadcastChannel
 )
 
 var gatewayServer sync.Map // map[string]*gatewayStream
@@ -37,7 +43,11 @@ func findGatewayBySID(sid int64, fun GatewayFun) (*gatewayStream, error) {
 	var stream grpc.ClientStream
 	switch fun {
 	case FunToClient:
-		stream, err = client.ToClient(context.Background())
+		stream, err = client.InnerToClient(context.Background())
+	case FunBroadcastClient:
+		stream, err = client.InnerBroadcastClient(context.Background())
+	case FunBroadcastChannel:
+		stream, err = client.InnerBroadcastChannel(context.Background())
 	}
 	if err != nil {
 		log.Errorf("findGatewayBySID %v err: %v, sid: %v", fun, err, sid)
@@ -53,7 +63,7 @@ func findGatewayBySID(sid int64, fun GatewayFun) (*gatewayStream, error) {
 	return ss, nil
 }
 
-func SendMessageToGateway(sid int64, fun GatewayFun, msg proto.Message, re ...bool) error {
+func sendMessageToGateway(sid int64, fun GatewayFun, msg proto.Message, re ...bool) error {
 	ss, err := findGatewayBySID(sid, fun)
 	if err != nil {
 		return err
@@ -71,13 +81,46 @@ func SendMessageToGateway(sid int64, fun GatewayFun, msg proto.Message, re ...bo
 		}
 		// 如果没有标识本次是重试的，就重试一次（默认重试）
 		if re == nil || !re[0] {
-			return SendMessageToGateway(sid, fun, msg, true)
+			return sendMessageToGateway(sid, fun, msg, true)
 		}
 		return err
 	}
 	return nil
 }
 
+// SendMessageToClient 单播消息给客户端
+func SendMessageToClient(ctx context.Context, usn int64, msg proto.Message, platform ...rspb.PlatformType) error {
+	if platform == nil || len(platform) == 0 {
+		return nil
+	}
+	for _, pf := range platform {
+		if gatewaySID := locator.NewLocator(ctx, usn, int32(pf)).LocateGateway(); gatewaySID > 0 {
+			return sendMessageToGateway(gatewaySID, FunToClient, msg)
+		}
+	}
+	return nil
+}
+
+// BroadcastMessageToClient 广播消息给客户端
+func BroadcastMessageToClient(msg *rspb.InnerBroadcastClientReq) error {
+	return broadcastMessageToGateway(FunBroadcastClient, msg)
+}
+
+// BroadcastMessageToChannel 广播消息到频道（加入频道的能收到，不区分平台）
+func BroadcastMessageToChannel(msg *rspb.InnerBroadcastChannelReq) error {
+	return broadcastMessageToGateway(FunBroadcastChannel, msg)
+}
+
+func broadcastMessageToGateway(fun GatewayFun, msg proto.Message) error {
+	sidArr := discover.FindServerAllSid(common.KeyDiscoverGateway)
+	for _, sid := range sidArr {
+		if err := sendMessageToGateway(sid, fun, msg); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
 func gatewayKey(sid int64, fun GatewayFun) string {
 	return fmt.Sprintf("%v-%v", sid, fun)
 }