|
|
|
|
@ -18,13 +18,18 @@ import (
|
|
|
|
|
const ( |
|
|
|
|
PingInterval = 5 * time.Second |
|
|
|
|
PingWait = 10 * time.Second |
|
|
|
|
SessionUidKey = "uid" |
|
|
|
|
SessionGwsConnKey = "gws" |
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
type soRequest struct { |
|
|
|
|
conn *gwsConn |
|
|
|
|
payload *protocol.Payload |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
type GwsHandler struct { |
|
|
|
|
grpcFactory *generic.GrpcGenericClientFactory |
|
|
|
|
sessionStore session.Store // <string, *session.NetSubject> 当前连接用户,内存缓存
|
|
|
|
|
requests chan *soRequest |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func NewGwsHandler( |
|
|
|
|
@ -37,6 +42,93 @@ func NewGwsHandler(
|
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (c *GwsHandler) Init(ctx context.Context) { |
|
|
|
|
c.requests = make(chan *soRequest, 1024*16) |
|
|
|
|
|
|
|
|
|
// todo 针对不同服务的协程池
|
|
|
|
|
for i := 0; i < 32; i++ { |
|
|
|
|
go c.dispatchRequest(ctx) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// dispatchRequest 多个 goroutine 进行请求 dispatch
|
|
|
|
|
func (c *GwsHandler) dispatchRequest(ctx context.Context) { |
|
|
|
|
for { |
|
|
|
|
select { |
|
|
|
|
case <-ctx.Done(): |
|
|
|
|
return |
|
|
|
|
case request := <-c.requests: |
|
|
|
|
func() { |
|
|
|
|
conn, payload := request.conn, request.payload |
|
|
|
|
header := payload.Header |
|
|
|
|
service, method := header.Svc, header.Target |
|
|
|
|
|
|
|
|
|
// grpc generic call
|
|
|
|
|
ctx := context.Background() |
|
|
|
|
ctx2, cancel2 := context.WithTimeout(ctx, time.Second*3) |
|
|
|
|
defer cancel2() |
|
|
|
|
grpcClient, err := c.grpcFactory.GetClient(ctx2, service) |
|
|
|
|
if err != nil { |
|
|
|
|
logger.Error("get grpc generic client error: ", err) |
|
|
|
|
writeError(conn, header.SeqId, fmt.Sprintf("service %s not avaliable: %s", header.Svc, err.Error())) |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// put grpc request session
|
|
|
|
|
if conn.uid != "" { |
|
|
|
|
ctx = session.PutSubject(ctx, session.NewRpcSubject(conn.uid)) |
|
|
|
|
} |
|
|
|
|
ctx3, cancel3 := context.WithTimeout(ctx, time.Second*10) |
|
|
|
|
defer cancel3() |
|
|
|
|
resp, err := grpcClient.InvokeUnary(ctx3, header.Target, payload.Body) |
|
|
|
|
if err != nil { |
|
|
|
|
writeError(conn, header.SeqId, unwrapRpcError(err.Error())) |
|
|
|
|
logger.Error("grpc generic call error: ", err) |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// write response
|
|
|
|
|
if resp == nil { // proto.Empty
|
|
|
|
|
header.Target = "" |
|
|
|
|
} else { |
|
|
|
|
payload.Body, err = resp.Marshal() |
|
|
|
|
if err != nil { |
|
|
|
|
logger.Error("generic call response marshal error: ", err) |
|
|
|
|
writeError(conn, header.SeqId, "server error") |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
header.Target = resp.XXX_MessageName() |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// auth verify success
|
|
|
|
|
if service == "Auth" && method == "Verify" { |
|
|
|
|
var channel *session.Channel |
|
|
|
|
conn.uid, channel, err = c.extractAuthVerify(conn, payload.Body) |
|
|
|
|
if err != nil { |
|
|
|
|
logger.Error("extractAuthVerify error: ", err) |
|
|
|
|
writeError(conn, header.SeqId, "server error") |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
// channel todo dispatch on conn open
|
|
|
|
|
go c.dispatch(channel) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
header.Type = protocol.TypeResponse |
|
|
|
|
resMessage, err := protocol.EncodeSo(payload) |
|
|
|
|
if err != nil { |
|
|
|
|
logger.Error("grpc generic call error: ", err) |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
err = conn.Write(resMessage) |
|
|
|
|
if err != nil { |
|
|
|
|
logger.Error("gws write response error: ", err) |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
}() |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (c *GwsHandler) OnOpen(socket *gws.Conn) { |
|
|
|
|
_ = socket.SetDeadline(time.Time{}) // time.Now().Add(PingInterval + PingWait)
|
|
|
|
|
socket.Session().Store(SessionGwsConnKey, newGwsConn(socket)) |
|
|
|
|
@ -46,13 +138,15 @@ func (c *GwsHandler) OnClose(socket *gws.Conn, err error) {
|
|
|
|
|
if err != nil { |
|
|
|
|
logger.Error("gws conn close error: ", err) |
|
|
|
|
} |
|
|
|
|
val, ok := socket.Session().Load(SessionUidKey) |
|
|
|
|
val, ok := socket.Session().Load(SessionGwsConnKey) |
|
|
|
|
if !ok { |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
uid := val.(string) |
|
|
|
|
c.sessionStore.Delete(uid) |
|
|
|
|
logger.Info("subject offline: ", uid) |
|
|
|
|
conn := val.(*gwsConn) |
|
|
|
|
if conn.uid != "" { |
|
|
|
|
c.sessionStore.Delete(conn.uid) |
|
|
|
|
logger.Info("subject offline: ", conn.uid) |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
func (c *GwsHandler) OnPing(socket *gws.Conn, payload []byte) { |
|
|
|
|
@ -70,6 +164,7 @@ func (c *GwsHandler) OnMessage(socket *gws.Conn, message *gws.Message) {
|
|
|
|
|
} |
|
|
|
|
}() |
|
|
|
|
if message.Opcode != gws.OpcodeBinary { |
|
|
|
|
logger.Info("receive websocket message type: ", message.Opcode) |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
@ -79,12 +174,7 @@ func (c *GwsHandler) OnMessage(socket *gws.Conn, message *gws.Message) {
|
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
conn := val.(*gwsConn) |
|
|
|
|
|
|
|
|
|
val, authorized := socket.Session().Load(SessionUidKey) |
|
|
|
|
var sessionUid string |
|
|
|
|
if authorized { |
|
|
|
|
sessionUid = val.(string) |
|
|
|
|
} |
|
|
|
|
authorized := conn.uid != "" |
|
|
|
|
|
|
|
|
|
messageBytes := message.Bytes() |
|
|
|
|
payload, err := protocol.DecodeSo(messageBytes) |
|
|
|
|
@ -100,67 +190,12 @@ func (c *GwsHandler) OnMessage(socket *gws.Conn, message *gws.Message) {
|
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// grpc generic call
|
|
|
|
|
ctx := context.Background() |
|
|
|
|
ctx2, cancel2 := context.WithTimeout(ctx, time.Second*3) |
|
|
|
|
defer cancel2() |
|
|
|
|
grpcClient, err := c.grpcFactory.GetClient(ctx2, service) |
|
|
|
|
if err != nil { |
|
|
|
|
logger.Error("get grpc generic client error: ", err) |
|
|
|
|
writeError(conn, header.SeqId, fmt.Sprintf("service %s not avaliable: %s", header.Svc, err.Error())) |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
// put session
|
|
|
|
|
if authorized { |
|
|
|
|
ctx = session.PutSubject(ctx, session.NewRpcSubject(sessionUid)) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
ctx3, cancel3 := context.WithTimeout(ctx, time.Second*10) |
|
|
|
|
defer cancel3() |
|
|
|
|
resp, err := grpcClient.InvokeUnary(ctx3, header.Target, payload.Body) |
|
|
|
|
if err != nil { |
|
|
|
|
writeError(conn, header.SeqId, unwrapRpcError(err.Error())) |
|
|
|
|
logger.Error("grpc generic call error: ", err) |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// write response
|
|
|
|
|
if resp == nil { // proto.Empty
|
|
|
|
|
header.Target = "" |
|
|
|
|
} else { |
|
|
|
|
payload.Body, err = resp.Marshal() |
|
|
|
|
if err != nil { |
|
|
|
|
logger.Error("generic call response marshal error: ", err) |
|
|
|
|
writeError(conn, header.SeqId, "server error") |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
header.Target = resp.XXX_MessageName() |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
// auth verify success
|
|
|
|
|
if service == "Auth" && method == "Verify" { |
|
|
|
|
var channel *session.Channel |
|
|
|
|
sessionUid, channel, err = c.extractAuthVerify(conn, payload.Body) |
|
|
|
|
if err != nil { |
|
|
|
|
logger.Error("extractAuthVerify error: ", err) |
|
|
|
|
writeError(conn, header.SeqId, "server error") |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
socket.Session().Store(SessionUidKey, sessionUid) |
|
|
|
|
// channel
|
|
|
|
|
go c.dispatch(channel) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
header.Type = protocol.TypeResponse |
|
|
|
|
resMessage, err := protocol.EncodeSo(payload) |
|
|
|
|
if err != nil { |
|
|
|
|
logger.Error("grpc generic call error: ", err) |
|
|
|
|
return |
|
|
|
|
} |
|
|
|
|
err = conn.Write(resMessage) |
|
|
|
|
if err != nil { |
|
|
|
|
logger.Error("gws write response error: ", err) |
|
|
|
|
return |
|
|
|
|
// payload -> processing channel
|
|
|
|
|
request := &soRequest{conn: conn, payload: payload} |
|
|
|
|
select { |
|
|
|
|
case c.requests <- request: |
|
|
|
|
default: |
|
|
|
|
writeError(conn, header.SeqId, "server busy") |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
tangmingyou
3 years ago