From 520f730557c908ee256f96846e61b18820f76494 Mon Sep 17 00:00:00 2001 From: Sven Njegac Date: Thu, 5 Oct 2023 16:10:56 +0200 Subject: [PATCH] fix: return error if flush fails --- errors.go | 7 +++++++ writer.go | 4 +++- 2 files changed, 10 insertions(+), 1 deletion(-) create mode 100644 errors.go diff --git a/errors.go b/errors.go new file mode 100644 index 0000000..c42838f --- /dev/null +++ b/errors.go @@ -0,0 +1,7 @@ +package zlogsentry + +import ( + "errors" +) + +var ErrFlushTimeout = errors.New("zlogsentry flush timeout") diff --git a/writer.go b/writer.go index c5fc3b4..5f1a87e 100644 --- a/writer.go +++ b/writer.go @@ -83,7 +83,9 @@ func (w *Writer) WriteLevel(level zerolog.Level, p []byte) (n int, err error) { // Close forces client to flush all pending events. // Can be useful before application exits. func (w *Writer) Close() error { - w.hub.Flush(w.flushTimeout) + if ok := w.hub.Flush(w.flushTimeout); !ok { + return ErrFlushTimeout + } return nil }