improve

lxzan · Jan 6, 2023 · 625c577 · 625c577
1 parent 5009697
commit 625c577
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/compress.go b/compress.go
@@ -24,7 +24,7 @@ type compressor struct {
 
 // Compress 压缩
 func (c *compressor) Compress(content []byte) ([]byte, error) {
-	if c.writeBuffer.Cap() > internal.Lv3 {
+	if c.writeBuffer.Cap() > internal.Lv4 {
 		c.writeBuffer = internal.NewBuffer(nil)
 	}
 

diff --git a/protocol.go b/protocol.go
@@ -15,6 +15,8 @@ func (c Opcode) IsDataFrame() bool {
 	return c <= OpcodeBinary
 }
 
+// WebSocket Event
+// OnError and OnClose will not both be called
 type Event interface {
 	OnOpen(socket *Conn)
 	OnError(socket *Conn, err error)

diff --git a/updrader.go b/updrader.go
@@ -69,6 +69,7 @@ func handshake(conn net.Conn, headers http.Header, websocketKey string) error {
 }
 
 // Accept http protocol upgrade to websocket
+// ctx done means server stopping
 func Accept(ctx context.Context, w http.ResponseWriter, r *http.Request, eventHandler Event, config Config) (*Conn, error) {
 	config.initialize()