package com.nf147.policy_publishing_platform.util.auto;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.sql.*;
import java.util.ArrayList;
import java.util.List;
/**
* @author: WBG
* @description: 自动创建service、impl、controller
* @date: 2019/02/18 08:15
*/
public class AutoCreate {
//驱动
static String DBDRIVER = "org.mariadb.jdbc.Driver";
//服务器地址
static String DBURL = "jdbc:mariadb://localhost:3306/policydb";
//登录用户名
static String DBUID = "root";
//密码
static String DBPWD = "123456";
//项目包名
static String Package = "com.nf147.policy_publishing_platform";
private static String tablename;
private String[] colnames; // 列名数组
private String[] colTypes; // 列名类型数组
private int[] colSizes; // 列名大小数组
public static void main(String[] args) throws Exception {
AutoCreate auto = new AutoCreate();
//获取所有数据表
List<String> list = auto.TBlist();
for (String table : list) {
createStart(tables(table));
}
}
/**
* 获取指定数据库中包含的表 TBlist
*
* @return 返回所有表名(将表名放到一个集合中)
* @throws Exception
* @time 2017年7月14日下午5:54:52
* @packageName com.util
*/
public List<String> TBlist() throws Exception {
// 访问数据库 采用 JDBC方式
Class.forName(DBDRIVER);
Connection con = DriverManager.getConnection(DBURL, DBUID, DBPWD);
DatabaseMetaData md = con.getMetaData();
List<String> list = null;
ResultSet rs = md.getTables(null, null, null, null);
if (rs != null) {
list = new ArrayList<String>();
}
while (rs.next()) {
String tableName = rs.getString("TABLE_NAME");
list.add(tableName);
}
rs = null;
md = null;
con = null;
return list;
}
/**
* 把输入字符串的首字母改成大写
*
* @param str
* @return
*/
private static String initcap(String str) {
char[] ch = str.toCharArray();
if (ch[0] >= 'a' && ch[0] <= 'z') {
ch[0] = (char) (ch[0] - 32);
}
return new String(ch);
}
/**
* 把输入字符串的首字母改成小写
*
* @param str
* @return
*/
private static String initlow(String str) {
char[] ch = str.toCharArray();
if (ch[0] >= 'A' && ch[0] <= 'Z') {
ch[0] = (char) (ch[0] + 32);
}
return new String(ch);
}
//首字母转换和下划线转换
private static String tables(String table) {
String[] tables = table.split("_");
table = "";
for (String s : tables) {
table += initcap(s);
}
return table;
}
/**
* 创建Dao
*/
private static String createDao(String tableName) {
String service = "package "+Package+".dao;
" +
"
" +
"import "+Package+".entity."+tableName+";
" +
"import org.springframework.stereotype.Repository;
" +
"
" +
"import java.util.List;
" +
"
" +
"@Repository
" +
"public interface "+tableName+"Mapper {
" +
" /**
" +
" * 根据id删除操作
" +
" *
" +
" * @param id
" +
" * @return
" +
" */
" +
" int deleteByPrimaryKey(Integer id);
" +
"
" +
" /**
" +
" * 添加操作
" +
" *
" +
" * @param "+initlow(tableName)+"
" +
" * @return
" +
" */
" +
" int insert("+tableName+" "+initlow(tableName)+");
" +
"
" +
" /**
" +
" * 根据id查询操作
" +
" *
" +
" * @param id
" +
" * @return
" +
" */
" +
" "+tableName+" selectByPrimaryKey(Integer id);
" +
"
" +
" /**
" +
" * 全部查询操作
" +
" *
" +
" * @return
" +
" */
" +
" List<"+tableName+"> selectAll();
" +
"
" +
" /**
" +
" * 根据id全部修改操作
" +
" *
" +
" * @param "+initlow(tableName)+"
" +
" * @return
" +
" */
" +
" int updateByPrimaryKey("+tableName+" "+initlow(tableName)+");
" +
"}";
return service;
}
/**
* 创建Service
*
* @param tableName 数据库表
*/
private static String createService(String tableName) {
String service = "package "+Package+".service;
" +
"
" +
"import "+Package+".entity." + tableName + ";
" +
"
" +
"import java.util.List;
" +
"
" +
"public interface " + tableName + "Service {
" +
"
" +
" /**
" +
" * 删除操作 根据id
" +
" *
" +
" * @param id
" +
" * @return
" +
" */" +
"
" +
" int deleteByPrimaryKey(Integer id);
" +
"
" +
" /**
" +
" * 添加操作
" +
" *
" +
" * @param " + initlow(tableName) + "
" +
" * @return
" +
" */" +
"
" +
" int insert(" + tableName + " " + initlow(tableName) + ");
" +
"
" +
" /**
" +
" * 根据id查询操作
" +
" *
" +
" * @param id
" +
" * @return
" +
" */" +
"
" +
" " + tableName + " selectByPrimaryKey(Integer id);
" +
"
" +
" /**
" +
" * 全部查询操作
" +
" *
" +
" * @return
" +
" */" +
"
" +
" List<" + tableName + "> selectAll();
" +
"
" +
" /**
" +
" * 修改操作
" +
" *
" +
" * @param " + initlow(tableName) + "
" +
" * @return
" +
" */" +
"
" +
" int updateByPrimaryKey(" + tableName + " " + initlow(tableName) + ");
" +
"}";
return service;
}
/**
* 创建ServiceImpl
*
* @param tableName 数据库表
*/
private static String createServiceImpl(String tableName) {
String serviceImpl = "package "+Package+".impl;
" +
"
" +
"import "+Package+".dao." + tableName + "Mapper;
" +
"import "+Package+".entity." + tableName + ";
" +
"import "+Package+".service." + tableName + "Service;
" +
"import org.springframework.beans.factory.annotation.Autowired;
" +
"import org.springframework.stereotype.Service;
" +
"
" +
"import java.util.List;
" +
"
" +
"@Service
" +
"public class " + tableName + "ServiceImpl implements " + tableName + "Service {
" +
"
" +
" @Autowired
" +
" private " + tableName + "Mapper " + initlow(tableName) + "Mapper;
" +
"
" +
" /**
" +
" * 删除操作 根据id删除
" +
" *
" +
" * @param id
" +
" * @return
" +
" */
" +
" @Override
" +
" public int deleteByPrimaryKey(Integer id) {
" +
" return " + initlow(tableName) + "Mapper.deleteByPrimaryKey(id);
" +
" }
" +
"
" +
" /**
" +
" * 添加操作
" +
" *
" +
" * @param " + initlow(tableName) + "
" +
" * @return
" +
" */
" +
" @Override
" +
" public int insert(" + tableName + " " + initlow(tableName) + ") {
" +
" return " + initlow(tableName) + "Mapper.insert(" + initlow(tableName) + ");
" +
" }
" +
"
" +
" /**
" +
" * 根据id查询操作
" +
" *
" +
" * @param id
" +
" * @return
" +
" */
" +
" @Override
" +
" public " + tableName + " selectByPrimaryKey(Integer id) {
" +
" return " + initlow(tableName) + "Mapper.selectByPrimaryKey(id);
" +
" }
" +
"
" +
" /**
" +
" * 全部查询操作
" +
" *
" +
" * @return
" +
" */
" +
" @Override
" +
" public List<" + tableName + "> selectAll() {
" +
" return " + initlow(tableName) + "Mapper.selectAll();
" +
" }
" +
"
" +
" /**
" +
" * 修改操作
" +
" *
" +
" * @param " + initlow(tableName) + "
" +
" * @return
" +
" */
" +
" @Override
" +
" public int updateByPrimaryKey(" + tableName + " " + initlow(tableName) + ") {
" +
" return " + initlow(tableName) + "Mapper.updateByPrimaryKey(" + initlow(tableName) + ");
" +
" }
" +
"}
";
return serviceImpl;
}
/**
* 创建Controller
*
* @param tableName 数据库表
*/
private static String createController(String tableName) {
String controller = "package "+Package+".controller;
" +
"import "+Package+".entity." + tableName + ";
" +
"import "+Package+".service." + tableName + "Service;
" +
"import "+Package+".util.Result;
" +
"import org.springframework.beans.factory.annotation.Autowired;
" +
"import org.springframework.web.bind.annotation.*;
" +
"import java.util.List;
" +
"
" +
"@RestController
" +
"@RequestMapping("/" + initlow(tableName) + "")
" +
"public class " + tableName + "Controller {
" +
" @Autowired
" +
" private " + tableName + "Service " + initlow(tableName) + "Service;
" +
"
" +
" /**
" +
" * 根据aId删除
" +
" * 要求转入 aId
" +
" *
" +
" * @param " + initlow(tableName) + "
" +
" * @return
" +
" */
" +
" @GetMapping("/deleteByPrimaryKey")
" +
" public Result deleteByPrimaryKey(" + tableName + " " + initlow(tableName) + ") {
" +
" try {
" +
"
" +
" return " + initlow(tableName) + "Service.deleteByPrimaryKey(" + initlow(tableName) + ".getId()) > 0 ? new Result().successMessage("删除成功") : new Result("删除失败");
" +
" } catch (Exception ex) {
" +
" return new Result().error("出错,请重试！");
" +
" }
" +
" }
" +
"
" +
" /**
" +
" * 添加对象" + initlow(tableName) + "
" +
" *
" +
" * @param " + initlow(tableName) + "
" +
" * @return
" +
" */
" +
" @PostMapping("/insert")
" +
" public Result insert(@RequestBody " + tableName + " " + initlow(tableName) + ") {
" +
" try {
" +
" return " + initlow(tableName) + "Service.insert(" + initlow(tableName) + ") > 0 ? new Result().successMessage("添加成功！") : new Result("添加失败！");
" +
" } catch (Exception ex) {
" +
" return new Result().error("出错,请重试！");
" +
" }
" +
"
" +
" }
" +
"
" +
" /**
" +
" * 根据aid查找对象 最多只能返回一个对象
" +
" *
" +
" * @param " + initlow(tableName) + "
" +
" * @return
" +
" */
" +
" @GetMapping("/selectByPrimaryKey")
" +
" public Result selectByPrimaryKey(" + tableName + " " + initlow(tableName) + ") {
" +
" try {
" +
" " + tableName + " " + initlow(tableName) + "1 = " + initlow(tableName) + "Service.selectByPrimaryKey(" + initlow(tableName) + ".getId());
" +
" if (" + initlow(tableName) + "1 == null) {
" +
" return new Result().successMessage("无数据");
" +
" } else {
" +
" return new Result().success(" + initlow(tableName) + "1);
" +
" }
" +
" } catch (Exception ex) {
" +
" return new Result().error("出错,请重试！");
" +
" }
" +
" }
" +
"
" +
" /**
" +
" * 查询所有数据
" +
" *
" +
" * @return
" +
" */
" +
" @GetMapping("/selectAll")
" +
" public Result selectAll() {
" +
" //public Result selectAll(@RequestParam(defaultValue = "1") int pageNum, @RequestParam(defaultValue = "10") int pageSize) {
" +
" try {
" +
" //分页
" +
" //PageHelper.startPage(pageNum, pageSize);
" +
" List<" + tableName + "> list = " + initlow(tableName) + "Service.selectAll();
" +
" if (list == null) {
" +
" return new Result().successMessage("无数据");
" +
" } else {
" +
" // return new Result().success(list, " + initlow(tableName) + "Service.count(""));
" +
" return new Result().success(list);
" +
" }
" +
" } catch (Exception ex) {
" +
" return new Result().error("出错,请重试！");
" +
" }
" +
" }
" +
"
" +
" /**
" +
" * 根据id修改全部字段
" +
" *
" +
" * @param " + initlow(tableName) + "
" +
" * @return
" +
" */
" +
" @PostMapping(value = "/updateByPrimaryKey")
" +
" public Result updateByPrimaryKey(@RequestBody " + tableName + " " + initlow(tableName) + ") {
" +
" try {
" +
" return " + initlow(tableName) + "Service.updateByPrimaryKey(" + initlow(tableName) + ") > 0 ? new Result().successMessage("修改成功") : new Result("修改失败");
" +
" } catch (Exception ex) {
" +
" return new Result().error("出错,请重试！");
" +
" }
" +
"
" +
"
" +
" }
" +
"}
";
return controller;
}
/**
* 开始创建
*
* @param tableName 数据库表
*/
static void createStart(String tableName) {
//获取当前项目的路径
String url = System.getProperty("user.dir");
url += "\src\main\java\com\nf147\policy_publishing_platform\";
//创建Dao
createFile(new File(url + "dao\" + tableName + "Mapper.java"), createDao(tableName));
//创建Service
createFile(new File(url + "service\" + tableName + "Service.java"), createService(tableName));
//创建ServiceImpl
createFile(new File(url + "impl\" + tableName + "ServiceImpl.java"), createServiceImpl(tableName));
//创建Controller
createFile(new File(url + "controller\" + tableName + "Controller.java"), createController(tableName));
}
/**
* @param file 创建的文件
* @param context 文件里面的内容
*/
static boolean createFile(File file, String context) {
//获取文件
File parent = file.getParentFile();
//如果是目录
if (parent != null) {
//创建目录
parent.mkdirs();
}
try {
//创建文件
file.createNewFile();
FileWriter fileWriter = null;
try {
fileWriter = new FileWriter(file);
fileWriter.write(context);
fileWriter.flush();
fileWriter.close();
} catch (IOException e) {
return false;
}
} catch (IOException e) {
System.out.println("创建文件失败:" + e.getMessage());
}
return true;
}
}