fix go race bug

8 years ago · 0d5eca9683
1 changed files with 2 additions and 2 deletions
--- a/command/agent/gated_writer.go
+++ b/command/agent/gated_writer.go
@ -29,8 +29,8 @@ func (w *GatedWriter) Flush() {
 }

 func (w *GatedWriter) Write(p []byte) (n int, err error) {
-	w.lock.RLock()
-	defer w.lock.RUnlock()
+	w.lock.Lock()
+	defer w.lock.Unlock()

 	if w.flush {
 		return w.Writer.Write(p)