mirror of
https://github.com/yusing/godoxy.git
synced 2026-04-23 08:48:32 +02:00
feat: add websocket writer and error handling utilities
This commit is contained in:
29
internal/net/gphttp/gpwebsocket/writer.go
Normal file
29
internal/net/gphttp/gpwebsocket/writer.go
Normal file
@@ -0,0 +1,29 @@
|
||||
package gpwebsocket
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/coder/websocket"
|
||||
)
|
||||
|
||||
type Writer struct {
|
||||
conn *websocket.Conn
|
||||
msgType websocket.MessageType
|
||||
ctx context.Context
|
||||
}
|
||||
|
||||
func NewWriter(ctx context.Context, conn *websocket.Conn, msgType websocket.MessageType) *Writer {
|
||||
return &Writer{
|
||||
ctx: ctx,
|
||||
conn: conn,
|
||||
msgType: msgType,
|
||||
}
|
||||
}
|
||||
|
||||
func (w *Writer) Write(p []byte) (int, error) {
|
||||
return len(p), w.conn.Write(w.ctx, w.msgType, p)
|
||||
}
|
||||
|
||||
func (w *Writer) Close() error {
|
||||
return w.conn.CloseNow()
|
||||
}
|
||||
18
internal/net/gphttp/server/error.go
Normal file
18
internal/net/gphttp/server/error.go
Normal file
@@ -0,0 +1,18 @@
|
||||
package server
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"net/http"
|
||||
|
||||
"github.com/rs/zerolog"
|
||||
)
|
||||
|
||||
func HandleError(logger *zerolog.Logger, err error, msg string) {
|
||||
switch {
|
||||
case err == nil, errors.Is(err, http.ErrServerClosed), errors.Is(err, context.Canceled):
|
||||
return
|
||||
default:
|
||||
logger.Fatal().Err(err).Msg(msg)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user