生成的时候写了一些个性化的东西，需要提前集成好mybatis plus lombok，这些东西。
（安利一下mybatis plus ,太好用了）
好，进入正题
首选在idea 内连接数据库
image.png
填好了以后连上以后右边看到一堆表
image.png
大概是这个样子
然后随便找张表右键
image.png
然后进入脚本
image.png
最开始是有个pojo的，可以直接生成，后面是我自己写的，反正就是做一些个性化的东西没改啥，这里把代码贴出来，也可以根据我的来改
POJO
import com.intellij.database.model.DasTable
import com.intellij.database.util.Case
import com.intellij.database.util.DasUtil
/*
* Available context bindings:
* SELECTION Iterable
* PROJECT project
* FILES files helper
*/
packageName = "com.cjkj.sbdcp.domain;"
typeMapping = [
(~/(?i)int/) : "long",
(~/(?i)float|double|decimal|real/): "double",
(~/(?i)datetime|timestamp/) : "java.sql.Timestamp",
(~/(?i)date/) : "java.sql.Date",
(~/(?i)time/) : "java.sql.Time",
(~/(?i)/) : "String"
]
FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->
SELECTION.filter { it instanceof DasTable }.each { generate(it, dir) }
}
def generate(table, dir) {
def className = javaName(table.getName(), true)
def fields = calcFields(table)
new File(dir, className + ".java").withPrintWriter { out -> generate(out, className, table.getName(), fields) }
}
def generate(out, className, tableName, fields) {
out.println "package $packageName"
out.println ""
out.println "import com.baomidou.mybatisplus.annotation.TableName;"
out.println "import lombok.Builder;"
out.println "import lombok.Data;"
out.println ""
out.println "@Data"
out.println "@Builder"
out.println "@TableName(\"$tableName\")"
out.println "public class $className {"
out.println ""
fields.each() {
if (it.annos != "") out.println " ${it.annos}"
out.println " private ${it.type} ${it.name};"
}
out.println ""
out.println " public $className() {"
out.println " }"
out.println ""
out.print " public $className("
StringBuilder stringBuilder = new StringBuilder()
fields.each() {
stringBuilder.append("${it.type} ${it.name}, ")
}
stringBuilder.delete(stringBuilder.length() -2 , stringBuilder.length())
out.print stringBuilder
out.println ") {"
fields.each() {
out.println " this.${it.name} = ${it.name};"
}
out.println " }"
out.println "}"
}
def calcFields(table) {
DasUtil.getColumns(table).reduce([]) { fields, col ->
def spec = Case.LOWER.apply(col.getDataType().getSpecification())
def typeStr = typeMapping.find { p, t -> p.matcher(spec).find() }.value
fields += [[
name : javaName(col.getName(), false),
type : typeStr,
annos: ""]]
}
}
def javaName(str, capitalize) {
def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
.collect { Case.LOWER.apply(it).capitalize() }
.join("")
.replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
}
Mapper
import com.intellij.database.model.DasTable
import com.intellij.database.util.Case
import com.intellij.database.util.DasUtil
/*
* Available context bindings:
* SELECTION Iterable
* PROJECT project
* FILES files helper
*/
packageName = "com.cjkj.sbdcp.mapper;"
typeMapping = [
(~/(?i)int/) : "long",
(~/(?i)float|double|decimal|real/): "double",
(~/(?i)datetime|timestamp/) : "java.sql.Timestamp",
(~/(?i)date/) : "java.sql.Date",
(~/(?i)time/) : "java.sql.Time",
(~/(?i)/) : "String"
]
FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->
SELECTION.filter { it instanceof DasTable }.each { generate(it, dir) }
}
def generate(table, dir) {
def className = javaName(table.getName(), true)
def fields = calcFields(table)
new File(dir, className + "Mapper.java").withPrintWriter { out -> generate(out, className, table.getName(), fields) }
}
def generate(out, className, tableName, fields) {
out.println "package $packageName"
out.println ""
out.println "import com.baomidou.mybatisplus.core.mapper.BaseMapper;"
out.println "import com.cjkj.sbdcp.domain.$className;"
out.println "import org.apache.ibatis.annotations.Mapper;"
out.println "import org.springframework.stereotype.Repository;"
out.println ""
out.println "@Repository"
out.println "@Mapper"
out.println "public interface " + className + "Mapper extends BaseMapper<$className> {"
out.println "}"
}
def calcFields(table) {
DasUtil.getColumns(table).reduce([]) { fields, col ->
def spec = Case.LOWER.apply(col.getDataType().getSpecification())
def typeStr = typeMapping.find { p, t -> p.matcher(spec).find() }.value
fields += [[
name : javaName(col.getName(), false),
type : typeStr,
annos: ""]]
}
}
def javaName(str, capitalize) {
def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
.collect { Case.LOWER.apply(it).capitalize() }
.join("")
.replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
}
Service
import com.intellij.database.model.DasTable
import com.intellij.database.util.Case
import com.intellij.database.util.DasUtil
/*
* Available context bindings:
* SELECTION Iterable
* PROJECT project
* FILES files helper
*/
packageName = "com.cjkj.sbdcp.service;"
typeMapping = [
(~/(?i)int/) : "long",
(~/(?i)float|double|decimal|real/): "double",
(~/(?i)datetime|timestamp/) : "java.sql.Timestamp",
(~/(?i)date/) : "java.sql.Date",
(~/(?i)time/) : "java.sql.Time",
(~/(?i)/) : "String"
]
FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->
SELECTION.filter { it instanceof DasTable }.each { generate(it, dir) }
}
def generate(table, dir) {
def className = javaName(table.getName(), true)
def fields = calcFields(table)
new File(dir, className + "Service.java").withPrintWriter { out -> generate(out, className, table.getName(), fields) }
}
def generate(out, className, tableName, fields) {
out.println "package $packageName"
out.println ""
out.println "import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;"
out.println "import com.cjkj.sbdcp.domain.$className;"
out.println "import com.cjkj.sbdcp.mapper.dcp." + className + "Mapper;"
out.println "import org.springframework.stereotype.Service;"
out.println ""
out.println "@Service"
out.println "public class " + className + "Service extends ServiceImpl<" + className + "Mapper,$className> {"
out.println "}"
}
def calcFields(table) {
DasUtil.getColumns(table).reduce([]) { fields, col ->
def spec = Case.LOWER.apply(col.getDataType().getSpecification())
def typeStr = typeMapping.find { p, t -> p.matcher(spec).find() }.value
fields += [[
name : javaName(col.getName(), false),
type : typeStr,
annos: ""]]
}
}
def javaName(str, capitalize) {
def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
.collect { Case.LOWER.apply(it).capitalize() }
.join("")
.replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
}
Controller
import com.intellij.database.model.DasTable
import com.intellij.database.util.Case
import com.intellij.database.util.DasUtil
/*
* Available context bindings:
* SELECTION Iterable
* PROJECT project
* FILES files helper
*/
packageName = "com.cjkj.sbdcp.controller;"
typeMapping = [
(~/(?i)int/) : "long",
(~/(?i)float|double|decimal|real/): "double",
(~/(?i)datetime|timestamp/) : "java.sql.Timestamp",
(~/(?i)date/) : "java.sql.Date",
(~/(?i)time/) : "java.sql.Time",
(~/(?i)/) : "String"
]
FILES.chooseDirectoryAndSave("Choose directory", "Choose where to store generated files") { dir ->
SELECTION.filter { it instanceof DasTable }.each { generate(it, dir) }
}
def generate(table, dir) {
def className = javaName(table.getName(), true)
def fields = calcFields(table)
new File(dir, className + "Controller.java").withPrintWriter { out -> generate(out, className, table.getName(), fields) }
}
def generate(out, className, tableName, fields) {
out.println "package $packageName"
out.println ""
out.println "import com.cjkj.sbdcp.common.annotation.CrossOriginRestController;"
out.println "import org.apache.shiro.authz.annotation.RequiresPermissions;"
out.println "import org.springframework.web.bind.annotation.RequestBody;"
out.println "import org.springframework.web.bind.annotation.RequestMapping;"
out.println "import org.springframework.web.bind.annotation.RequestMethod;"
out.println "import com.cjkj.sbdcp.common.BaseResult;"
out.println "import com.cjkj.sbdcp.common.BaseController;"
out.println "import com.cjkj.sbdcp.domain.$className;"
def model = ""
if (className.indexOf("Dss") > -1) {
model = "dss"
out.println "import com.cjkj.sbdcp.service.dss." + className + "Service;"
} else if (className.indexOf("Sgf") > -1) {
model = "sgf"
out.println "import com.cjkj.sbdcp.service.sgf." + className + "Service;"
} else if (className.indexOf("Ws") > -1) {
model = "sgf"
out.println "import com.cjkj.sbdcp.service.ws." + className + "Service;"
} else if (className.indexOf("sys") > -1) {
model = "sys"
out.println "import com.cjkj.sbdcp.service.system." + className + "Service;"
} else {
out.println "import com.cjkj.sbdcp.service." + className + "Service;"
}
def javaClassName = className.substring(0, 1).toLowerCase() + className.substring(1, className.length())
out.println ""
out.println "@CrossOriginRestController(\"" + model + "/" + javaClassName + "\")"
out.println "public class " + className + "Controller extends BaseController<" + className + "Service,$className> {"
out.println ""
out.println " @RequiresPermissions(\"" + javaClassName + ":list\")"
out.println " @RequestMapping(value = \"list\", method = RequestMethod.GET)"
out.println " public BaseResult list() {"
out.println " return super.list();"
out.println " }"
out.println ""
out.println " @RequiresPermissions(\"" + javaClassName + ":list\")"
out.println " @RequestMapping(value = \"list\", method = RequestMethod.POST)"
out.println " public BaseResult list(@RequestBody " + className + " " + javaClassName + ") {"
out.println " return super.list(" + javaClassName + ");"
out.println " }"
out.println ""
out.println " @RequiresPermissions(\"" + javaClassName + ":add\")"
out.println " @RequestMapping(value = \"save\", method = RequestMethod.POST)"
out.println " public BaseResult save(@RequestBody " + className + " " + javaClassName + ") {"
out.println " return super.save(" + javaClassName + ");"
out.println " }"
out.println ""
out.println " @RequiresPermissions(\"" + javaClassName + ":update\")"
out.println " @RequestMapping(value = \"save\", method = RequestMethod.PUT)"
out.println " public BaseResult update(@RequestBody " + className + " " + javaClassName + ") {"
out.println " return super.updateById(" + javaClassName + ");"
out.println " }"
out.println ""
out.println " @RequiresPermissions(\"" + javaClassName + ":delete\")"
out.println " @RequestMapping(value = \"del\", method = RequestMethod.DELETE)"
out.println " public BaseResult delete(Long id) {"
out.println " return super.deleteById(id);"
out.println " }"
out.println ""
out.println ""
out.println "}"
}
def calcFields(table) {
DasUtil.getColumns(table).reduce([]) { fields, col ->
def spec = Case.LOWER.apply(col.getDataType().getSpecification())
def typeStr = typeMapping.find { p, t -> p.matcher(spec).find() }.value
fields += [[
name : javaName(col.getName(), false),
type : typeStr,
annos: ""]]
}
}
def javaName(str, capitalize) {
def s = com.intellij.psi.codeStyle.NameUtil.splitNameIntoWords(str)
.collect { Case.LOWER.apply(it).capitalize() }
.join("")
.replaceAll(/[^\p{javaJavaIdentifierPart}[_]]/, "_")
capitalize || s.length() == 1 ? s : Case.LOWER.apply(s[0]) + s[1..-1]
}
BaseController
public class BaseController, T> {
@Autowired
protected S s;
public BaseResult list() {
return BaseResult.createSuccResult().addData("list", s.list(null));
}
public BaseResult list(T t) {
return BaseResult.createSuccResult().addData("list", s.list(new QueryWrapper().setEntity(t)));
}
public BaseResult page(int current, int size) {
return BaseResult.createSuccResult().addData("page", s.page(new Page<>(current, size), null));
}
public BaseResult page(T t, int current, int size) {
return BaseResult.createSuccResult().addData("page", s.page(new Page<>(current, size), new QueryWrapper().setEntity(t)));
}
public BaseResult save(T t) {
CommonUtils.insertFilter(t);
s.save(t);
return BaseResult.createSuccResult();
}
public BaseResult saveBatch(Collection ts){
ts.forEach(t -> {
CommonUtils.insertFilter(t);
s.save(t);
});
return BaseResult.createSuccResult();
}
public BaseResult updateById(T t) {
CommonUtils.updateFilter(t);
s.updateById(t);
return BaseResult.createSuccResult();
}
public BaseResult deleteById(long id) {
s.removeById(id);
return BaseResult.createSuccResult();
}
}
现在是每生成一个层的类要点一下，下个版本整合一下所有的生成和路径选择