🐛 fix(writer.go): add check for mainWriter buffer length in Close() m…

…ethod for DiscordWriter and TelegramWriter interfaces
kamushadenes · Mar 29, 2023 · 22c9afc · 22c9afc
1 parent 22a4228
commit 22c9afc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/interfaces/discord/writer.go b/interfaces/discord/writer.go
@@ -52,7 +52,8 @@ func (w *DiscordWriter) Flush() {
 }
 
 func (w *DiscordWriter) Close() error {
-	if len(w.bufs) > 0 && w.bufs[0].Len() > 0 {
+	if (len(w.bufs) > 0 && w.bufs[0].Len() > 0) ||
+		(w.mainWriter != nil && len(w.mainWriter.bufs) > 0 && w.mainWriter.bufs[0].Len() > 0) {
 		switch w.Type {
 		case "text":
 			return w.closeText()

diff --git a/interfaces/telegram/writer.go b/interfaces/telegram/writer.go
@@ -52,7 +52,8 @@ func (w *TelegramWriter) Flush() {
 }
 
 func (w *TelegramWriter) Close() error {
-	if len(w.bufs) > 0 && w.bufs[0].Len() > 0 {
+	if (len(w.bufs) > 0 && w.bufs[0].Len() > 0) ||
+		(w.mainWriter != nil && len(w.mainWriter.bufs) > 0 && w.mainWriter.bufs[0].Len() > 0) {
 		switch w.Type {
 		case "text":
 			return w.closeText()