Merge branch 'dev-1.10.0-webank' into dev-1.10.0-bug-fix

WeDataSphere · Dec 25, 2024 · 1783615 · 1783615
2 parents f550f49 + 24af105
commit 1783615
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/...nhancements/linkis-udf-service/src/main/java/org/apache/linkis/udf/api/UDFRestfulApi.java b/...nhancements/linkis-udf-service/src/main/java/org/apache/linkis/udf/api/UDFRestfulApi.java
@@ -1450,9 +1450,12 @@ public Message getRegisterFunctions(HttpServletRequest req, @RequestParam("path"
         || StringUtils.endsWithIgnoreCase(path, Constants.FILE_EXTENSION_SCALA)) {
       if (StringUtils.startsWithIgnoreCase(path, StorageUtils$.MODULE$.FILE_SCHEMA())) {
         try {
+          // 获取登录用户
+          String userName = ModuleUserUtils.getOperationUser(req, "get-register-functions");
+
           FsPath fsPath = new FsPath(path);
           // 获取文件系统实例
-          FileSystem fileSystem = (FileSystem) FSFactory.getFs(fsPath);
+          FileSystem fileSystem = (FileSystem) FSFactory.getFsByProxyUser(fsPath, userName);
           fileSystem.init(null);
           if (fileSystem.canRead(fsPath)) {
             return Message.ok()

diff --git a/...c-enhancements/linkis-udf-service/src/main/java/org/apache/linkis/udf/utils/UdfUtils.java b/...c-enhancements/linkis-udf-service/src/main/java/org/apache/linkis/udf/utils/UdfUtils.java
@@ -324,12 +324,14 @@ public static List<String> extractPythonMethodNames(String udfPath) throws Excep
     String exec =
         Utils.exec(
             (new String[] {
+              "sudo",
               Constants.PYTHON_COMMAND.getValue(),
               Configuration.getLinkisHome() + "/admin/" + "linkis_udf_get_python_methods.py",
               localPath
             }));
     logger.info(
-        "execute python script to get python method name...{} {} {}",
+        "execute python script to get python method name...{} {} {} {}",
+        "sudo",
         Constants.PYTHON_COMMAND.getValue(),
         Configuration.getLinkisHome() + "/admin/" + "linkis_udf_get_python_methods.py",
         localPath);