good morning!!!!

Skip to content
Snippets Groups Projects
Unverified Commit 2389eb18 authored by Anmol Sethi's avatar Anmol Sethi
Browse files

Fix CI

parent ce145504
No related branches found
No related tags found
No related merge requests found
...@@ -99,6 +99,6 @@ func benchConn(b *testing.B, stream bool) { ...@@ -99,6 +99,6 @@ func benchConn(b *testing.B, stream bool) {
} }
func BenchmarkConn(b *testing.B) { func BenchmarkConn(b *testing.B) {
benchConn(b, true)
benchConn(b, false) benchConn(b, false)
benchConn(b, true)
} }
...@@ -460,7 +460,7 @@ type messageWriter struct { ...@@ -460,7 +460,7 @@ type messageWriter struct {
// Write writes the given bytes to the WebSocket connection. // Write writes the given bytes to the WebSocket connection.
func (w messageWriter) Write(p []byte) (int, error) { func (w messageWriter) Write(p []byte) (int, error) {
n, err := w.Write(p) n, err := w.write(p)
if err != nil { if err != nil {
return n, xerrors.Errorf("failed to write: %w", err) return n, xerrors.Errorf("failed to write: %w", err)
} }
...@@ -522,14 +522,14 @@ func (w messageWriter) close() error { ...@@ -522,14 +522,14 @@ func (w messageWriter) close() error {
// You can only read a single message at a time so do not call this method // You can only read a single message at a time so do not call this method
// concurrently. // concurrently.
func (c *Conn) Reader(ctx context.Context) (MessageType, io.Reader, error) { func (c *Conn) Reader(ctx context.Context) (MessageType, io.Reader, error) {
typ, r, err := c.Reader(ctx) typ, r, err := c.reader(ctx)
if err != nil { if err != nil {
return 0, nil, xerrors.Errorf("failed to get reader: %w", err) return 0, nil, xerrors.Errorf("failed to get reader: %w", err)
} }
return typ, r, nil return typ, r, nil
} }
func (c *Conn) reader(ctx context.Context) (MessageType, io.Reader, error) { func (c *Conn) reader(ctx context.Context) (MessageType, io.Reader, error) {
for !atomic.CompareAndSwapInt64(&c.activeReader, 0, 1) { for !atomic.CompareAndSwapInt64(&c.activeReader, 0, 1) {
select { select {
case <-c.closed: case <-c.closed:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment