From 24183d3f86abff528100fb4fdb8330e289ce0dba Mon Sep 17 00:00:00 2001
From: xsren <285808407@qq.com>
Date: Thu, 18 Jul 2024 09:58:23 +0800
Subject: [PATCH] fix: start udfd with env var

---
 source/libs/function/src/tudf.c | 2 ++
 source/libs/function/src/udfd.c | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/source/libs/function/src/tudf.c b/source/libs/function/src/tudf.c
index 5f7764f342..a38dc4e42a 100644
--- a/source/libs/function/src/tudf.c
+++ b/source/libs/function/src/tudf.c
@@ -71,6 +71,7 @@ void udfUdfdExit(uv_process_t *process, int64_t exitStatus, int termSignal) {
   }
 }
 
+extern char **environ;
 static int32_t udfSpawnUdfd(SUdfdData *pData) {
   fnInfo("start to init udfd");
   uv_process_options_t options = {0};
@@ -117,6 +118,7 @@ static int32_t udfSpawnUdfd(SUdfdData *pData) {
   child_stdio[2].data.fd = 2;
   options.stdio_count = 3;
   options.stdio = child_stdio;
+  options.env = environ;
 
   options.flags = UV_PROCESS_DETACHED;
 
diff --git a/source/libs/function/src/udfd.c b/source/libs/function/src/udfd.c
index df97e873aa..75bed73bb3 100644
--- a/source/libs/function/src/udfd.c
+++ b/source/libs/function/src/udfd.c
@@ -1421,22 +1421,26 @@ int main(int argc, char *argv[]) {
 
   if (taosInitCfg(configDir, NULL, NULL, NULL, NULL, 0) != 0) {
     fnError("failed to start since read config error");
+    taosCloseLog();
     return -2;
   }
 
   initEpSetFromCfg(tsFirst, tsSecond, &global.mgmtEp);
   if (udfdOpenClientRpc() != 0) {
     fnError("open rpc connection to mnode failed");
+    taosCloseLog();
     return -3;
   }
 
   if (udfdCreateUdfSourceDir() != 0) {
     fnError("create udf source directory failed");
+    taosCloseLog();
     return -4;
   }
 
   if (udfdUvInit() != 0) {
     fnError("uv init failure");
+    taosCloseLog();
     return -5;
   }
 
@@ -1452,6 +1456,7 @@ int main(int argc, char *argv[]) {
 
   udfdDeinitScriptPlugins();
 
+  taosCloseLog();
   udfdCleanup();
   return 0;
 }