项目根路径获取方式改进，支持独立部署和页面指定路径

Author: davidfantasy

src/frontend/src/pages/index/table/index.vue

@@ -3,10 +3,10 @@
     <div class="content-header"></div>
     <div class="content">
       <div class="container">
-        <div style="margin-bottom:10px;">
+        <div style="margin-bottom: 10px">
           <el-input
             clearable
-            style="width:300px;margin-right:20px;"
+            style="width: 300px; margin-right: 20px"
             placeholder="请输入表名进行查询"
             prefix-icon="el-icon-search"
             v-model="searchKey"
@@ -21,24 +21,33 @@
           @selection-change="handleTableSelection"
         >
           <el-table-column type="selection" width="55"></el-table-column>
-          <el-table-column type="index" width="50" label="序号"></el-table-column>
+          <el-table-column
+            type="index"
+            width="50"
+            label="序号"
+          ></el-table-column>
           <el-table-column sortable prop="name" label="table名称">
             <template slot-scope="scope">{{ scope.row.name }}</template>
           </el-table-column>
           <el-table-column prop="comment" label="table注释"></el-table-column>
         </el-table>
       </div>
     </div>
-    <el-dialog :visible.sync="showGenSettingDialog" title="输出配置" width="70%" top="5vh">
+    <el-dialog
+      :visible.sync="showGenSettingDialog"
+      title="输出配置"
+      width="70%"
+      top="5vh"
+    >
       <el-form label-width="220px">
         <el-form-item label="代码作者">
-          <el-input v-model="genSetting.author" style="width:260px"></el-input>
+          <el-input v-model="genSetting.author" style="width: 260px"></el-input>
         </el-form-item>
         <el-form-item label="功能模块名">
           <el-input
             v-model="genSetting.moduleName"
             placeholder="模块名将加入到输出包名之后，用于划分不同的模块"
-            style="width:260px"
+            style="width: 400px"
           ></el-input>
         </el-form-item>
         <el-form-item label="本次需生成的文件">
@@ -47,11 +56,15 @@
               v-for="file in userConfig.outputFiles"
               :label="file.fileType"
               :key="file.fileType"
-            >{{ file.fileType }}</el-checkbox>
+              >{{ file.fileType }}</el-checkbox
+            >
           </el-checkbox-group>
         </el-form-item>
         <el-form-item label="需生成的Controller方法" v-if="isControllerChecked">
-          <el-alert title="注意：如果更换了Controller的模板，以下选项可能不会生效，需自行在模板中实现" type="warning"></el-alert>
+          <el-alert
+            title="注意：如果更换了Controller的模板，以下选项可能不会生效，需自行在模板中实现"
+            type="warning"
+          ></el-alert>
           <el-checkbox-group v-model="genSetting.choosedControllerMethods">
             <el-checkbox label="list" key="list">列表查询</el-checkbox>
             <el-checkbox label="getById" key="getById">按ID查询</el-checkbox>
@@ -63,6 +76,15 @@
         <el-form-item label="存在同名文件时是否覆盖">
           <el-switch v-model="genSetting.override"></el-switch>
         </el-form-item>
+        <el-form-item label="目标项目根目录">
+          <el-input
+            v-model="genSetting.rootPath"
+            style="width: 400px"
+          ></el-input>
+          <help-tip
+            content="最终生成的代码位置等于：目标项目根目录 + 具体某个类别的源码设置的包目录"
+          ></help-tip>
+        </el-form-item>
         <el-form-item>
           <el-button type="primary" @click="genCode()">开始生成</el-button>
         </el-form-item>
@@ -92,6 +114,7 @@ export default {
         override: false,
         moduleName: "",
         choosedControllerMethods: [],
+        rootPath: "",
       },
       userConfig: {},
       outputFileInfos: [],
@@ -154,6 +177,11 @@ export default {
           this.genSetting.moduleName = "";
           this.genSetting.override = false;
         }
+        if (!this.genSetting.rootPath) {
+          axios.get("/api/output-file-info/project-root-path").then((res) => {
+            this.genSetting.rootPath = res;
+          });
+        }
       } catch (e) {
         console.error(e);
       }

src/main/java/com/github/davidfantasy/mybatisplus/generatorui/ProjectPathResolver.java

@@ -30,26 +30,28 @@ public class ProjectPathResolver {
     public ProjectPathResolver(String basePackage) {
         this.basePackage = basePackage;
         ClassLoader contextLoader = Thread.currentThread().getContextClassLoader();
-        String projectDir = System.getProperty("user.dir");
+        String rootPath = "";
         if (contextLoader.getResource(".") != null) {
-            projectDir = contextLoader.getResource(".").getPath();
+            String projectDir = contextLoader.getResource(".").getPath();
             projectDir = getUTF8String(projectDir);
             String[] paths = projectDir.split("/");
             StringBuilder temp = new StringBuilder();
+            //去掉目录的最后两个子目录，通常是/target/classes
             for (int i = 0; i < paths.length; i++) {
                 String path = paths[i];
+                if (Strings.isNullOrEmpty(path)) {
+                    continue;
+                }
                 if (i < paths.length - 2) {
                     temp.append(path);
-                    temp.append("/");
+                    temp.append(File.separator);
                 }
             }
-            baseProjectPath = temp.toString();
+            rootPath = temp.toString();
         } else {
-            projectDir = getUTF8String(projectDir).replace("\\", "/") + "/";
-            baseProjectPath = projectDir;
+            rootPath = getUTF8String(System.getProperty("user.dir")) + File.separator;
         }
-        sourcePath = new File(baseProjectPath + "src/main/java").toString();
-        resourcePath = new File(baseProjectPath + "src/main/resources").toString();
+        refreshBaseProjectPath(rootPath);
     }
 
     /**
@@ -58,7 +60,7 @@ public ProjectPathResolver(String basePackage) {
      * @param basePath
      * @return
      */
-    private String getUTF8String(String basePath) {
+    public String getUTF8String(String basePath) {
         try {
             basePath = URLDecoder.decode(basePath, "UTF-8");
         } catch (UnsupportedEncodingException e) {
@@ -135,4 +137,12 @@ public String resolveMapperXmlPackage() {
         return PACKAGE_RESOURCES_PREFIX + "mapper";
     }
 
+    public synchronized void refreshBaseProjectPath(String rootPath) {
+        if (baseProjectPath == null || !baseProjectPath.equals(rootPath)) {
+            this.baseProjectPath = rootPath;
+            sourcePath = new File(baseProjectPath + "src/main/java".replace("/", File.separator)).toString();
+            resourcePath = new File(baseProjectPath + "src/main/resources".replace("/", File.separator)).toString();
+        }
+    }
+
 }

src/main/java/com/github/davidfantasy/mybatisplus/generatorui/controller/OutputFileInfoController.java

@@ -1,5 +1,6 @@
 package com.github.davidfantasy.mybatisplus.generatorui.controller;
 
+import com.github.davidfantasy.mybatisplus.generatorui.ProjectPathResolver;
 import com.github.davidfantasy.mybatisplus.generatorui.common.Result;
 import com.github.davidfantasy.mybatisplus.generatorui.common.ResultGenerator;
 import com.github.davidfantasy.mybatisplus.generatorui.dto.OutputFileInfo;
@@ -21,14 +22,21 @@ public class OutputFileInfoController {
     @Autowired
     private UserConfigStore userConfigStore;
 
+    @Autowired
+    private ProjectPathResolver projectPathResolver;
+
     @GetMapping("/user-config")
     public Result getUserConfig() {
         return ResultGenerator.genSuccessResult(userConfigStore.getDefaultUserConfig());
     }
 
+    @GetMapping("/project-root-path")
+    public Result getRootPath() {
+        return ResultGenerator.genSuccessResult(projectPathResolver.getBaseProjectPath());
+    }
+
     @PostMapping("/delete")
     public Result deleteOutputInfos(@RequestBody OutputFileInfo outputFileInfo) throws IOException {
-
         outputFileInfoService.deleteOutputFileInfo(outputFileInfo);
         return ResultGenerator.genSuccessResult();
     }

src/main/java/com/github/davidfantasy/mybatisplus/generatorui/dto/GenSetting.java

@@ -36,4 +36,9 @@ public class GenSetting {
      */
     private List<String> choosedControllerMethods;
 
+    /**
+     * 目标项目根目录
+     */
+    private String rootPath;
+
 }

src/main/java/com/github/davidfantasy/mybatisplus/generatorui/service/MbpGeneratorService.java

@@ -1,11 +1,13 @@
 package com.github.davidfantasy.mybatisplus.generatorui.service;
 
+import cn.hutool.core.io.FileUtil;
 import com.baomidou.mybatisplus.generator.AutoGenerator;
 import com.baomidou.mybatisplus.generator.config.*;
 import com.baomidou.mybatisplus.generator.config.po.TableField;
 import com.baomidou.mybatisplus.generator.config.po.TableInfo;
 import com.github.davidfantasy.mybatisplus.generatorui.GeneratorConfig;
 import com.github.davidfantasy.mybatisplus.generatorui.ProjectPathResolver;
+import com.github.davidfantasy.mybatisplus.generatorui.common.ServiceException;
 import com.github.davidfantasy.mybatisplus.generatorui.dto.GenSetting;
 import com.github.davidfantasy.mybatisplus.generatorui.dto.OutputFileInfo;
 import com.github.davidfantasy.mybatisplus.generatorui.dto.UserConfig;
@@ -21,6 +23,7 @@
 import org.springframework.stereotype.Service;
 
 import javax.annotation.PostConstruct;
+import java.io.File;
 import java.util.ArrayList;
 import java.util.List;
 
@@ -66,6 +69,8 @@ public void initGenerator() {
     }
 
     public void genCodeBatch(GenSetting genSetting, List<String> tables) {
+        checkGenSetting(genSetting);
+        projectPathResolver.refreshBaseProjectPath(genSetting.getRootPath());
         //自定义参数配置
         mpg.setCfg(new TableInjectionConfig(generatorConfig, genSetting));
         //生成策略配置
@@ -152,4 +157,18 @@ private StrategyConfig getCurrentStrategy(UserConfig userConfig) {
     }
 
 
+    private void checkGenSetting(GenSetting genSetting) {
+        if (Strings.isNullOrEmpty(genSetting.getRootPath())) {
+            throw new ServiceException("目标项目根目录不能为空");
+        }
+        genSetting.setRootPath(projectPathResolver.getUTF8String(genSetting.getRootPath()));
+        if (!FileUtil.isDirectory(genSetting.getRootPath())) {
+            throw new ServiceException("目标项目根目录错误，请确认目录有效且存在：" + genSetting.getRootPath());
+        }
+        if (!genSetting.getRootPath().endsWith(File.separator)) {
+            genSetting.setRootPath(genSetting.getRootPath() + File.separator);
+        }
+    }
+
+
 }