Skip to content

Commit cb7c788

Browse files
committed
Improve field names in conn
1 parent 26dd4ac commit cb7c788

File tree

1 file changed

+21
-20
lines changed

1 file changed

+21
-20
lines changed

websocket.go

+21-20
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@ type Conn struct {
4646
closeErr error
4747
closed chan struct{}
4848

49+
// messageWriter state.
4950
// writeMsgLock is acquired to write a data message.
5051
writeMsgLock chan struct{}
5152
// writeFrameLock is acquired to write a single frame.
@@ -72,11 +73,11 @@ type Conn struct {
7273
readHeaderBuf []byte
7374
controlPayloadBuf []byte
7475

75-
// messageReader state
76-
readMsgCtx context.Context
77-
readMsgHeader header
78-
readFrameEOF bool
79-
readMaskPos int
76+
// messageReader state.
77+
readerMsgCtx context.Context
78+
readerMsgHeader header
79+
readerFrameEOF bool
80+
readerMaskPos int
8081

8182
setReadTimeout chan context.Context
8283
setWriteTimeout chan context.Context
@@ -360,7 +361,7 @@ func (c *Conn) Reader(ctx context.Context) (MessageType, io.Reader, error) {
360361
}
361362

362363
func (c *Conn) reader(ctx context.Context) (MessageType, io.Reader, error) {
363-
if c.activeReader != nil && !c.readFrameEOF {
364+
if c.activeReader != nil && !c.readerFrameEOF {
364365
// The only way we know for sure the previous reader is not yet complete is
365366
// if there is an active frame not yet fully read.
366367
// Otherwise, a user may have read the last byte but not the EOF if the EOF
@@ -396,10 +397,10 @@ func (c *Conn) reader(ctx context.Context) (MessageType, io.Reader, error) {
396397
return 0, nil, err
397398
}
398399

399-
c.readMsgCtx = ctx
400-
c.readMsgHeader = h
401-
c.readFrameEOF = false
402-
c.readMaskPos = 0
400+
c.readerMsgCtx = ctx
401+
c.readerMsgHeader = h
402+
c.readerFrameEOF = false
403+
c.readerMaskPos = 0
403404
c.readMsgLeft = c.msgReadLimit
404405

405406
r := &messageReader{
@@ -468,8 +469,8 @@ func (r *messageReader) read(p []byte) (int, error) {
468469
p = p[:r.c.readMsgLeft]
469470
}
470471

471-
if r.c.readFrameEOF {
472-
h, err := r.c.readTillMsg(r.c.readMsgCtx)
472+
if r.c.readerFrameEOF {
473+
h, err := r.c.readTillMsg(r.c.readerMsgCtx)
473474
if err != nil {
474475
return 0, err
475476
}
@@ -480,31 +481,31 @@ func (r *messageReader) read(p []byte) (int, error) {
480481
return 0, err
481482
}
482483

483-
r.c.readMsgHeader = h
484-
r.c.readFrameEOF = false
485-
r.c.readMaskPos = 0
484+
r.c.readerMsgHeader = h
485+
r.c.readerFrameEOF = false
486+
r.c.readerMaskPos = 0
486487
}
487488

488-
h := r.c.readMsgHeader
489+
h := r.c.readerMsgHeader
489490
if int64(len(p)) > h.payloadLength {
490491
p = p[:h.payloadLength]
491492
}
492493

493-
n, err := r.c.readFramePayload(r.c.readMsgCtx, p)
494+
n, err := r.c.readFramePayload(r.c.readerMsgCtx, p)
494495

495496
h.payloadLength -= int64(n)
496497
r.c.readMsgLeft -= int64(n)
497498
if h.masked {
498-
r.c.readMaskPos = fastXOR(h.maskKey, r.c.readMaskPos, p)
499+
r.c.readerMaskPos = fastXOR(h.maskKey, r.c.readerMaskPos, p)
499500
}
500-
r.c.readMsgHeader = h
501+
r.c.readerMsgHeader = h
501502

502503
if err != nil {
503504
return n, err
504505
}
505506

506507
if h.payloadLength == 0 {
507-
r.c.readFrameEOF = true
508+
r.c.readerFrameEOF = true
508509

509510
if h.fin {
510511
r.c.activeReader = nil

0 commit comments

Comments
 (0)