71 lines
1.9 KiB
Go
71 lines
1.9 KiB
Go
package grpcserver
|
|
|
|
import (
|
|
"context"
|
|
"git.hlsq.asia/alex/service-common/log"
|
|
"git.hlsq.asia/alex/service-common/proto/rs/rspb"
|
|
"git.hlsq.asia/alex/service-common/proto/ss/sspb"
|
|
"git.hlsq.asia/alex/service-gateway/internal/global"
|
|
"git.hlsq.asia/alex/service-gateway/internal/handler/wshandler/client"
|
|
"google.golang.org/protobuf/proto"
|
|
"sync"
|
|
)
|
|
|
|
func (s *Server) ToClient(server rspb.Gateway_ToClientServer) error {
|
|
wg := &sync.WaitGroup{}
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
defer func() {
|
|
if err := recover(); err != nil {
|
|
log.Errorf("ToClient panic: %v", err)
|
|
}
|
|
}()
|
|
for {
|
|
if args, err := server.Recv(); err != nil {
|
|
return
|
|
} else {
|
|
if args.USN == -1 {
|
|
|
|
//utils.WorkerPool(ws_handler.UserMgr.GetAllInterface(), func(task interface{}) {
|
|
// client := task.(*ws_handler.Client)
|
|
// client.WriteBytes(ss_pb.MessageID(args.MessageID), args.Payload)
|
|
//})
|
|
|
|
data, err := proto.Marshal(&sspb.Message{
|
|
ID: sspb.MessageID(args.MessageID),
|
|
Payload: args.Payload,
|
|
})
|
|
if err != nil {
|
|
log.Errorf("ToClient proto.Marshal error: %v", err)
|
|
continue
|
|
}
|
|
dataLen := float64(len(data))
|
|
for _, cli := range client.UserMgr.GetAll() {
|
|
cli.WriteBytesPreMarshal(data)
|
|
global.FlowCounter.Add(dataLen)
|
|
}
|
|
|
|
//for _, client := range ws_handler.UserMgr.GetAll() {
|
|
// client.WriteBytes(ss_pb.MessageID(args.MessageID), args.Payload)
|
|
//}
|
|
} else {
|
|
if cli := client.UserMgr.GetByUSN(args.USN); cli != nil {
|
|
cli.WriteBytes(sspb.MessageID(args.MessageID), args.Payload)
|
|
global.FlowCounter.Add(float64(len(args.Payload)))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}()
|
|
wg.Wait()
|
|
return server.SendAndClose(&rspb.ToClientResp{})
|
|
}
|
|
|
|
func (s *Server) KickUser(ctx context.Context, req *rspb.KickUserReq) (*rspb.KickUserResp, error) {
|
|
if cli := client.UserMgr.GetByUSN(req.USN); cli != nil {
|
|
cli.CloseClient()
|
|
}
|
|
return &rspb.KickUserResp{}, nil
|
|
}
|