| 369 | } |
| 370 | |
| 371 | func (s *cappedWriteSyncer) Write(p []byte) (int, error) { |
| 372 | s.mu.Lock() |
| 373 | defer s.mu.Unlock() |
| 374 | written := s.written.Load() |
| 375 | if written >= s.cap { |
| 376 | s.capped.Store(true) |
| 377 | return len(p), nil |
| 378 | } |
| 379 | remaining := s.cap - written |
| 380 | if int64(len(p)) > remaining { |
| 381 | n, err := s.inner.Write(p[:remaining]) |
| 382 | s.written.Add(int64(n)) |
| 383 | s.capped.Store(true) |
| 384 | // Report we accepted the full slice so zap doesn't retry; the |
| 385 | // overflow is intentionally dropped. |
| 386 | return len(p), err |
| 387 | } |
| 388 | n, err := s.inner.Write(p) |
| 389 | s.written.Add(int64(n)) |
| 390 | return n, err |
| 391 | } |
| 392 | |
| 393 | func (s *cappedWriteSyncer) Sync() error { |
| 394 | s.mu.Lock() |