Added name support

joeinman · Oct 8, 2024 · 4a4771f · 4a4771f
1 parent 4ff24e5
commit 4a4771f
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 5 deletions.
diff --git a/include/EmbedLog/EmbedLog.hpp b/include/EmbedLog/EmbedLog.hpp
@@ -1,3 +1,5 @@
+#pragma once
+
 #include <functional>
 #include <string>
 
@@ -13,7 +15,7 @@ namespace EmbedLog
     class EmbedLog
     {
     public:
-        EmbedLog(OpenFunction openFunc, CloseFunction closeFunc, PrintFunction printFunc, MicrosecondFunction microsecondFunc, LogLevel logLevel = LogLevel::INFO);
+        EmbedLog(OpenFunction openFunc, CloseFunction closeFunc, PrintFunction printFunc, MicrosecondFunction microsecondFunc, std::string name = "", LogLevel logLevel = LogLevel::INFO);
         ~EmbedLog();
 
         bool open();
@@ -27,8 +29,10 @@ namespace EmbedLog
         CloseFunction closeFunc;
         PrintFunction printFunc;
         MicrosecondFunction microsecondFunc;
-        LogLevel logLevel;
+
         bool isOpen = false;
+        LogLevel logLevel;
+        std::string name = " ";
 
         std::string getTimestamp();
     };

diff --git a/src/EmbedLog.cpp b/src/EmbedLog.cpp
@@ -2,9 +2,20 @@
 
 namespace EmbedLog
 {
-    EmbedLog::EmbedLog(OpenFunction openFunc, CloseFunction closeFunc, PrintFunction printFunc, MicrosecondFunction microsecondFunc, LogLevel logLevel)
-        : openFunc(openFunc), closeFunc(closeFunc), printFunc(printFunc), microsecondFunc(microsecondFunc), logLevel(logLevel)
+    EmbedLog::EmbedLog(OpenFunction openFunc,
+                       CloseFunction closeFunc,
+                       PrintFunction printFunc,
+                       MicrosecondFunction microsecondFunc,
+                       std::string name,
+                       LogLevel logLevel)
+        : openFunc(openFunc),
+          closeFunc(closeFunc),
+          printFunc(printFunc),
+          microsecondFunc(microsecondFunc),
+          logLevel(logLevel)
     {
+        if (!name.empty())
+            this->name = " " + name + " ";
     }
 
     EmbedLog::~EmbedLog()
@@ -45,7 +56,7 @@ namespace EmbedLog
                 break;
             }
 
-            printFunc(getTimestamp() + " [" + logLevelString + "] " + message);
+            printFunc("[" + getTimestamp() + name + logLevelString + "] " + message);
         }
     }