85 lines
1.8 KiB
Go
85 lines
1.8 KiB
Go
package stream_client
|
|
|
|
import (
|
|
"context"
|
|
"git.hlsq.asia/mmorpg/service-common/log"
|
|
"git.hlsq.asia/mmorpg/service-common/net/grpc/service"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/protobuf/proto"
|
|
"strconv"
|
|
"sync"
|
|
)
|
|
|
|
type SceneFun int
|
|
|
|
const (
|
|
FunAction SceneFun = iota
|
|
)
|
|
|
|
var sceneServer sync.Map // map[string]*sceneStream
|
|
|
|
type sceneStream struct {
|
|
mu sync.Mutex
|
|
stream grpc.ClientStream
|
|
}
|
|
|
|
func findSceneBySID(sid string, fun SceneFun) (*sceneStream, error) {
|
|
key := sceneKey(sid, fun)
|
|
|
|
if v, ok := sceneServer.Load(key); ok {
|
|
return v.(*sceneStream), nil
|
|
}
|
|
|
|
client, err := service.SceneNewClient(sid)
|
|
if err != nil {
|
|
log.Errorf("findSceneBySID cannot find client: %v", err)
|
|
return nil, err
|
|
}
|
|
var stream grpc.ClientStream
|
|
switch fun {
|
|
case FunAction:
|
|
stream, err = client.Action(context.Background())
|
|
}
|
|
if err != nil {
|
|
log.Errorf("findSceneBySID %v err: %v, sid: %v", fun, err, sid)
|
|
return nil, err
|
|
}
|
|
|
|
ss := &sceneStream{stream: stream}
|
|
if actual, loaded := sceneServer.LoadOrStore(key, ss); loaded {
|
|
go func() { _ = stream.CloseSend() }()
|
|
return actual.(*sceneStream), nil
|
|
}
|
|
|
|
return ss, nil
|
|
}
|
|
|
|
func SendMessageToScene(sid string, fun SceneFun, msg proto.Message, re ...bool) error {
|
|
ss, err := findSceneBySID(sid, fun)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
ss.mu.Lock()
|
|
err = ss.stream.SendMsg(msg)
|
|
ss.mu.Unlock()
|
|
|
|
if err != nil {
|
|
key := sceneKey(sid, fun)
|
|
if v, ok := sceneServer.Load(key); ok && v == ss {
|
|
sceneServer.Delete(key)
|
|
_ = ss.stream.CloseSend()
|
|
}
|
|
// 如果没有标识本次是重试的，就重试一次（默认重试）
|
|
if re == nil || !re[0] {
|
|
return SendMessageToScene(sid, fun, msg, true)
|
|
}
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func sceneKey(sid string, fun SceneFun) string {
|
|
return sid + "-" + strconv.Itoa(int(fun))
|
|
}
|