zapcore: add stopped for Stop method called exactly once

zapcore/buffered_write_syncer.go

@@ -74,6 +74,7 @@ type BufferedWriteSyncer struct {
 	writer      *bufio.Writer
 	ticker      *time.Ticker
 	stop        chan struct{} // closed when flushLoop should stop
+	stopOnce    sync.Once
 	done        chan struct{} // closed when flushLoop has stopped
 }
 
@@ -155,22 +156,24 @@ func (s *BufferedWriteSyncer) flushLoop() {
 
 // Stop closes the buffer, cleans up background goroutines, and flushes
 // remaining unwritten data.
-//
-// This must be called exactly once.
-func (s *BufferedWriteSyncer) Stop() error {
-	// Critical section.
-	func() {
-		s.mu.Lock()
-		defer s.mu.Unlock()
-
-		if !s.initialized {
-			return
-		}
-
-		s.ticker.Stop()
-		close(s.stop) // tell flushLoop to stop
-		<-s.done      // and wait until it has
-	}()
-
-	return s.Sync()
+func (s *BufferedWriteSyncer) Stop() (err error) {
+	s.stopOnce.Do(func() {
+		// Critical section.
+		func() {
+			s.mu.Lock()
+			defer s.mu.Unlock()
+
+			if !s.initialized {
+				return
+			}
+
+			s.ticker.Stop()
+			close(s.stop) // tell flushLoop to stop
+			<-s.done      // and wait until it has
+		}()
+
+		err = s.Sync()
+	})
+
+	return err
 }

zapcore/buffered_write_syncer_test.go

@@ -53,6 +53,14 @@ func TestBufferWriter(t *testing.T) {
 		assert.Equal(t, "foo", buf.String(), "Unexpected log string")
 	})
 
+	t.Run("stop twice", func(t *testing.T) {
+		ws := &BufferedWriteSyncer{WS: &ztest.FailWriter{}}
+		_, err := ws.Write([]byte("foo"))
+		require.NoError(t, err, "Unexpected error writing to WriteSyncer.")
+		assert.Error(t, ws.Stop(), "Expected stop to fail.")
+		assert.NoError(t, ws.Stop(), "Expected stop to not fail.")
+	})
+
 	t.Run("wrap twice", func(t *testing.T) {
 		buf := &bytes.Buffer{}
 		bufsync := &BufferedWriteSyncer{WS: AddSync(buf)}