From 72df0b6343c874dc112fc1daad919392cc2c1e15 Mon Sep 17 00:00:00 2001
From: Levente Batuska <mr.levente97@gmail.com>
Date: Wed, 11 Dec 2024 12:19:09 +0100
Subject: [PATCH] feat(Logger)!: add StopLogger function (closes channel)

---
 logger/consolelogger.go | 4 ++++
 logger/filelogger.go    | 4 ++++
 logger/interfaces.go    | 2 ++
 logger/nulllogger.go    | 2 ++
 4 files changed, 12 insertions(+)

diff --git a/logger/consolelogger.go b/logger/consolelogger.go
index 30dfdf8..1431602 100644
--- a/logger/consolelogger.go
+++ b/logger/consolelogger.go
@@ -20,6 +20,10 @@ func (logger *ConsoleLoggerImpl) StartLogger() {
 	})
 }
 
+func (logger *ConsoleLoggerImpl) StopLogger() {
+	close(logger.messages)
+}
+
 func (logger *ConsoleLoggerImpl) Write(message string) {
 	logger.messages <- time.Now().Format(time.UnixDate) + " : " + message + "\n"
 }
diff --git a/logger/filelogger.go b/logger/filelogger.go
index 3c537b3..90aab48 100644
--- a/logger/filelogger.go
+++ b/logger/filelogger.go
@@ -68,6 +68,10 @@ func (logger *FileLoggerImpl) StartLogger() {
 	// logger.mutex.Unlock()
 }
 
+func (logger *FileLoggerImpl) StopLogger() {
+	close(logger.messages)
+}
+
 func (logger *FileLoggerImpl) Write(message string) {
 	logger.messages <- time.Now().Format(time.UnixDate) + " : " + message + "\n"
 }
diff --git a/logger/interfaces.go b/logger/interfaces.go
index 4cdba60..99c2825 100644
--- a/logger/interfaces.go
+++ b/logger/interfaces.go
@@ -20,6 +20,7 @@ type (
 		init()
 		// Start an infinite loop to write out messages from the channel
 		StartLogger()
+		StopLogger()
 		Write(message string)
 		WriteRequest(message string, uuid string)
 		// If an error that is not nill passed in it logs the error and returns 1, otherwise 0
@@ -32,6 +33,7 @@ type (
 		init()
 		// Start an infinite loop to write out messages from the channel
 		StartLogger()
+		StopLogger()
 		WriteDebug(message string)
 		WriteRequestDebug(message string, uuid string)
 		WriteErrDebug(err error) (errnum int)
diff --git a/logger/nulllogger.go b/logger/nulllogger.go
index 720af23..416df12 100644
--- a/logger/nulllogger.go
+++ b/logger/nulllogger.go
@@ -4,6 +4,8 @@ func (lgr *NullLoggerImpl) init() {}
 
 func (logger *NullLoggerImpl) StartLogger() {}
 
+func (logger *NullLoggerImpl) StopLogger() {}
+
 func (logger *NullLoggerImpl) Write(message string) {}
 
 func (logger *NullLoggerImpl) WriteRequest(message string, request string) {}