peer-conn-msg-writer.go | 4 ++-- diff --git a/peer-conn-msg-writer.go b/peer-conn-msg-writer.go index 293c838896ba08d6de01f75c45e8c6e8ecdd7a06..16892546e77618de610199d884324000a9c442e1 100644 --- a/peer-conn-msg-writer.go +++ b/peer-conn-msg-writer.go @@ -63,6 +63,8 @@ var ( lastWrite time.Time = time.Now() keepAliveTimer *time.Timer ) + cn.mu.Lock() + defer cn.mu.Unlock() keepAliveTimer = time.AfterFunc(keepAliveTimeout, func() { cn.mu.Lock() defer cn.mu.Unlock() @@ -71,8 +73,6 @@ cn.writeCond.Broadcast() } keepAliveTimer.Reset(keepAliveTimeout) }) - cn.mu.Lock() - defer cn.mu.Unlock() defer keepAliveTimer.Stop() frontBuf := new(bytes.Buffer) for {