ing
This commit is contained in:
parent
6a1e7aa3e2
commit
af159726d6
@ -17,4 +17,3 @@ import:
|
||||
- package: git.loafle.net/commons_go/util
|
||||
subpackages:
|
||||
- channel
|
||||
- package: git.loafle.net/commons_go/logging
|
||||
|
@ -4,10 +4,8 @@ import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"io"
|
||||
"log"
|
||||
|
||||
"go.uber.org/zap"
|
||||
|
||||
"git.loafle.net/commons_go/logging"
|
||||
gws "git.loafle.net/overflow/overflow_gateway_websocket"
|
||||
)
|
||||
|
||||
@ -16,16 +14,14 @@ type MessageHandler interface {
|
||||
}
|
||||
|
||||
type messageHandler struct {
|
||||
ctx context.Context
|
||||
logger *zap.Logger
|
||||
o *Options
|
||||
ctx context.Context
|
||||
o *Options
|
||||
}
|
||||
|
||||
func NewHandler(ctx context.Context, o *Options) MessageHandler {
|
||||
h := &messageHandler{
|
||||
ctx: ctx,
|
||||
logger: logging.WithContext(ctx),
|
||||
o: o,
|
||||
ctx: ctx,
|
||||
o: o,
|
||||
}
|
||||
|
||||
return h
|
||||
@ -62,7 +58,7 @@ func (h *messageHandler) onRequest(soc gws.Socket, req *ServerRequest) []byte {
|
||||
|
||||
j, err := json.Marshal(res)
|
||||
if nil != err {
|
||||
h.logger.Error("JSON RPC error", zap.Any("err", err))
|
||||
log.Printf("JSON RPC error: %v", err)
|
||||
}
|
||||
|
||||
return j
|
||||
@ -71,6 +67,6 @@ func (h *messageHandler) onRequest(soc gws.Socket, req *ServerRequest) []byte {
|
||||
func (h *messageHandler) onNotify(soc gws.Socket, req *ServerRequest) {
|
||||
err := h.o.OnNotify(soc, req.Method, req.Params)
|
||||
if nil != err {
|
||||
h.logger.Error("JSON RPC error", zap.Any("err", err))
|
||||
log.Printf("JSON RPC error: %v", err)
|
||||
}
|
||||
}
|
||||
|
@ -4,9 +4,6 @@ import (
|
||||
"context"
|
||||
"net/http"
|
||||
|
||||
"go.uber.org/zap"
|
||||
|
||||
"git.loafle.net/commons_go/logging"
|
||||
channelUtil "git.loafle.net/commons_go/util/channel"
|
||||
"git.loafle.net/overflow/overflow_gateway_websocket/websocket"
|
||||
"github.com/valyala/fasthttp"
|
||||
@ -26,7 +23,6 @@ type Server interface {
|
||||
|
||||
type server struct {
|
||||
_ctx context.Context
|
||||
_logger *zap.Logger
|
||||
_option *ServerOptions
|
||||
_upgrader *websocket.Upgrader
|
||||
_handlers map[string]*SocketOptions
|
||||
@ -37,7 +33,6 @@ type server struct {
|
||||
func NewServer(ctx context.Context, o *ServerOptions) Server {
|
||||
s := &server{
|
||||
_ctx: ctx,
|
||||
_logger: logging.WithContext(ctx),
|
||||
_option: o.Validate(),
|
||||
_handlers: make(map[string]*SocketOptions, 1),
|
||||
_sockets: make(map[string]Socket, 100),
|
||||
@ -110,7 +105,6 @@ func (s *server) listenHandler() {
|
||||
for {
|
||||
select {
|
||||
case <-s._ctx.Done():
|
||||
s._logger.Info("websocket server: Context Done")
|
||||
return
|
||||
case ca := <-s._socketsCh:
|
||||
switch ca.Type {
|
||||
|
10
socket.go
10
socket.go
@ -3,11 +3,9 @@ package overflow_gateway_websocket
|
||||
import (
|
||||
"context"
|
||||
"io"
|
||||
"log"
|
||||
"time"
|
||||
|
||||
"go.uber.org/zap"
|
||||
|
||||
"git.loafle.net/commons_go/logging"
|
||||
"git.loafle.net/overflow/overflow_gateway_websocket/websocket"
|
||||
)
|
||||
|
||||
@ -21,7 +19,6 @@ type Socket interface {
|
||||
|
||||
type socket struct {
|
||||
ctx context.Context
|
||||
logger *zap.Logger
|
||||
id string
|
||||
o *SocketOptions
|
||||
conn *websocket.Conn
|
||||
@ -34,7 +31,6 @@ type socket struct {
|
||||
func NewSocket(ctx context.Context, id string, path string, o *SocketOptions, conn *websocket.Conn) Socket {
|
||||
c := &socket{
|
||||
ctx: ctx,
|
||||
logger: logging.WithContext(ctx),
|
||||
id: id,
|
||||
o: o,
|
||||
conn: conn,
|
||||
@ -112,12 +108,12 @@ func (soc *socket) listenWrite() {
|
||||
case w := <-soc.writeCh:
|
||||
if writeTimeout := soc.o.WriteTimeout; writeTimeout > 0 {
|
||||
err := soc.conn.SetWriteDeadline(time.Now().Add(writeTimeout))
|
||||
soc.logger.Error("Socket timeout", zap.Any("err", err))
|
||||
log.Printf("Socket timeout: %v", err)
|
||||
}
|
||||
|
||||
err := soc.conn.WriteMessage(soc.messageType, w)
|
||||
if nil != err {
|
||||
soc.logger.Error("Socket Write error", zap.Any("err", err))
|
||||
log.Printf("Socket Write error: %v", err)
|
||||
}
|
||||
|
||||
// receive done request
|
||||
|
Loading…
x
Reference in New Issue
Block a user