Improve tls writer

This commit is contained in:
世界
2022-09-30 11:27:18 +08:00
parent 968430c338
commit 1173fdea64
16 changed files with 358 additions and 39 deletions

View File

@@ -26,7 +26,7 @@ func NewServerConn(wsConn *websocket.Conn, remoteAddr net.Addr) *WebsocketConn {
return &WebsocketConn{
Conn: wsConn,
remoteAddr: remoteAddr,
Writer: &Writer{wsConn, true},
Writer: NewWriter(wsConn, true),
}
}
@@ -117,7 +117,7 @@ func (c *EarlyWebsocketConn) Write(b []byte) (n int, err error) {
if err != nil {
return 0, wrapDialError(response, err)
}
c.conn = &WebsocketConn{Conn: conn, Writer: &Writer{conn, false}}
c.conn = &WebsocketConn{Conn: conn, Writer: NewWriter(conn, false)}
close(c.create)
if len(lateData) > 0 {
_, err = c.conn.Write(lateData)
@@ -160,7 +160,7 @@ func (c *EarlyWebsocketConn) WriteBuffer(buffer *buf.Buffer) error {
if err != nil {
return wrapDialError(response, err)
}
c.conn = &WebsocketConn{Conn: conn, Writer: &Writer{conn, false}}
c.conn = &WebsocketConn{Conn: conn, Writer: NewWriter(conn, false)}
close(c.create)
if len(lateData) > 0 {
_, err = c.conn.Write(lateData)