From d55a0c7100323b04887afc2f8ca70c649144885d Mon Sep 17 00:00:00 2001
From: Shengliang Guan <slguan@taosdata.com>
Date: Thu, 8 Aug 2024 16:18:18 +0800
Subject: [PATCH] enh: enable log in udf

---
 include/libs/function/taosudf.h | 24 ++++++++++++++++++++++++
 include/util/tlog.h             |  4 ++--
 source/util/src/tlog.c          |  4 ++--
 3 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/include/libs/function/taosudf.h b/include/libs/function/taosudf.h
index 04b92a897a..0b59d7c2f5 100644
--- a/include/libs/function/taosudf.h
+++ b/include/libs/function/taosudf.h
@@ -320,6 +320,30 @@ typedef int32_t (*TScriptUdfDestoryFunc)(void *udfCtx);
 typedef int32_t (*TScriptOpenFunc)(SScriptUdfEnvItem *items, int numItems);
 typedef int32_t (*TScriptCloseFunc)();
 
+// clang-format off
+#ifdef WINDOWS
+  #define fnFatal(...) {}
+  #define fnError(...) {}
+  #define fnWarn(...)  {}
+  #define fnInfo(...)  {}
+  #define fnDebug(...) {}
+  #define fnTrace(...) {}
+#else
+  DLL_EXPORT void taosPrintLog(const char *flags, int32_t level, int32_t dflag, const char *format, ...)
+#ifdef __GNUC__
+    __attribute__((format(printf, 4, 5)))
+#endif
+    ;
+  extern int32_t udfDebugFlag;
+  #define udfFatal(...) { if (udfDebugFlag & 1) { taosPrintLog("UDF FATAL ", 1, 255, __VA_ARGS__); }}
+  #define udfError(...) { if (udfDebugFlag & 1) { taosPrintLog("UDF ERROR ", 1, 255, __VA_ARGS__); }}
+  #define udfWarn(...)  { if (udfDebugFlag & 2) { taosPrintLog("UDF WARN ",  2, 255, __VA_ARGS__); }}
+  #define udfInfo(...)  { if (udfDebugFlag & 2) { taosPrintLog("UDF ",       2, 255, __VA_ARGS__); }}
+  #define udfDebug(...) { if (udfDebugFlag & 4) { taosPrintLog("UDF ",       4, udfDebugFlag, __VA_ARGS__); }}
+  #define udfTrace(...) { if (udfDebugFlag & 8) { taosPrintLog("UDF ",       8, udfDebugFlag, __VA_ARGS__); }}
+#endif
+// clang-format on
+
 #ifdef __cplusplus
 }
 #endif
diff --git a/include/util/tlog.h b/include/util/tlog.h
index 67aafbfe44..b4dbbef532 100644
--- a/include/util/tlog.h
+++ b/include/util/tlog.h
@@ -74,13 +74,13 @@ void    taosCloseLog();
 void    taosResetLog();
 void    taosDumpData(uint8_t *msg, int32_t len);
 
-void taosPrintLog(const char *flags, ELogLevel level, int32_t dflag, const char *format, ...)
+void taosPrintLog(const char *flags, int32_t level, int32_t dflag, const char *format, ...)
 #ifdef __GNUC__
     __attribute__((format(printf, 4, 5)))
 #endif
     ;
 
-void taosPrintLongString(const char *flags, ELogLevel level, int32_t dflag, const char *format, ...)
+void taosPrintLongString(const char *flags, int32_t level, int32_t dflag, const char *format, ...)
 #ifdef __GNUC__
     __attribute__((format(printf, 4, 5)))
 #endif
diff --git a/source/util/src/tlog.c b/source/util/src/tlog.c
index 1946a0a274..86dc767adc 100644
--- a/source/util/src/tlog.c
+++ b/source/util/src/tlog.c
@@ -567,7 +567,7 @@ static inline void taosPrintLogImp(ELogLevel level, int32_t dflag, const char *b
   }
 }
 
-void taosPrintLog(const char *flags, ELogLevel level, int32_t dflag, const char *format, ...) {
+void taosPrintLog(const char *flags, int32_t level, int32_t dflag, const char *format, ...) {
   if (!(dflag & DEBUG_FILE) && !(dflag & DEBUG_SCREEN)) return;
 
   char    buffer[LOG_MAX_LINE_BUFFER_SIZE];
@@ -590,7 +590,7 @@ void taosPrintLog(const char *flags, ELogLevel level, int32_t dflag, const char
   }
 }
 
-void taosPrintLongString(const char *flags, ELogLevel level, int32_t dflag, const char *format, ...) {
+void taosPrintLongString(const char *flags, int32_t level, int32_t dflag, const char *format, ...) {
   if (!osLogSpaceAvailable()) return;
   if (!(dflag & DEBUG_FILE) && !(dflag & DEBUG_SCREEN)) return;