Spring Data 工具类快速生成代码

以前使用MyBatis可以进行数据库反向生成，后来转用了Spring Data逆向生成数据库，感觉再去生成control、service 不是很方便，与实际开发不是很相符。现在公司的后台开发框架已经定性，代码的书写都是按着标准，能做的就是按着模板添加新功能。于是我手写了一个工具类，指定了存放entity的实体路径，扫描所有文件，依靠文件生成最简单的controller、service等。还有很多功能不完善，比如常用的方法以前生成了。

 

代码如下：

package com.shuju.display.util;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;


public class GenerateUtil {

    /**
     * 实体类路径
     */
    private static String entityDir = "E:/xx/src/main/java/com/xx/xx/entity";

    /**
     * 新建模块路径
     */
    private static String baseDir = "xx/display";

    /**
     * controller模板
     */
    private static String controllerContent = "package com.shuju.display.controller;\n" +
            "\n" +
            "import com.shuju.sys.controller.AbstractController;\n" +
            "import io.swagger.annotations.Api;\n" +
            "import org.springframework.web.bind.annotation.RequestMapping;\n" +
            "import org.springframework.web.bind.annotation.RestController;\n" +
            "\n" +
            "@RestController\n" +
            "@RequestMapping(\"/custom/product\")\n" +
            "@Api(\"定制产品相关\")\n" +
            "public class CustomProductController extends AbstractController {\n" +
            "}";

    private static String repositoryContent = "package com.shuju.display.repository;\n" +
            "\n" +
            "import com.shuju.core.repository.BaseRepository;\n" +
            "import com.shuju.farming.entity.CustomProduct;\n" +
            "\n" +
            "\n" +
            "public interface CustomProductRepository extends BaseRepository<CustomProduct, String> {\n" +
            "}\n";

    private static String serviceContent = "package com.shuju.display.service;\n" +
            "\n" +
            "import com.shuju.core.service.BaseService;\n" +
            "import com.shuju.farming.entity.CustomProduct;\n" +
            "\n" +
            "public interface CustomProductService extends BaseService<CustomProduct, String> {\n" +
            "}";

    private static String serviceImplContent = "package com.shuju.display.service.impl;\n" +
            "\n" +
            "import com.shuju.core.service.impl.BaseServiceImpl;\n" +
            "import com.shuju.display.repository.CustomProductRepository;\n" +
            "import com.shuju.display.service.CustomProductService;\n" +
            "import com.shuju.farming.entity.CustomProduct;\n" +
            "import org.springframework.stereotype.Service;\n" +
            "\n" +
            "\n" +
            "@Service(\"customProductService\")\n" +
            "public class CustomProductServiceImpl extends BaseServiceImpl<CustomProduct, String, CustomProductRepository> implements CustomProductService {\n" +
            "}\n";


    public static void main(String[] args) {
        File entityFile = new File(entityDir);
        File controllerDir = new File(baseDir + "/controller");
        File repositoryDir = new File(baseDir + "/repository");
        File serviceDir = new File(baseDir + "/service");
        File serviceImplDir = new File(baseDir + "/service/impl");
        String[] list = entityFile.list();
//        createFile(controllerDir, list, "Controller.", controllerContent);
//        createFile(repositoryDir, list, "Repository.", repositoryContent);
        createFile(serviceDir, list, "Service.", serviceContent);
        createFile(serviceImplDir, list, "ServiceImpl.", serviceImplContent);
    }

    private static void createFile(File dir, String[] list, String replace, String content) {
        for (String s : list) {
            String name = s.replace(".java", "");
            String newName = s.replace(".", replace);
            File controllerFile = new File(dir + "/" + newName);
            try (FileWriter writer = new FileWriter(controllerFile, true)) {
                if (!controllerFile.exists()) {
                    controllerFile.createNewFile();
                }
                writer.append(content.replace("CustomProduct", name));
                writer.flush();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

}