├── .gitignore
├── .idea
├── compiler.xml
├── emacs.xml
├── kotlinc.xml
├── libraries
│ ├── KotlinJavaRuntime.xml
│ └── commons_lang3_3_6.xml
├── misc.xml
├── modules.xml
├── vcs.xml
└── workspace.xml
├── MybatisGenerator.iml
├── MybatisGenerator.zip
├── README.md
├── _config.yml
├── lib
├── commons-lang3-3.6.jar
├── kotlin-reflect.jar
├── kotlin-stdlib-jre7-sources.jar
├── kotlin-stdlib-jre7.jar
├── kotlin-stdlib-jre8-sources.jar
├── kotlin-stdlib-jre8.jar
├── kotlin-stdlib-sources.jar
├── kotlin-stdlib.jar
├── kotlin-test.jar
└── mysql-connector-java-5.1.41.jar
└── src
└── main
├── java
└── com
│ └── yangyang
│ └── mybatis
│ └── generator
│ ├── MyConstant.kt
│ ├── actions
│ ├── ActionConfigPath.kt
│ ├── ActionConfigureDB.kt
│ ├── ActionGenerate.kt
│ └── ActionGroupMybatis.kt
│ └── utils
│ ├── NotificationUtil.kt
│ ├── PlatformUtil.kt
│ ├── UIUtil.kt
│ └── mybatis
│ ├── MyBatisGenConst.kt
│ └── MybatisCore.kt
└── resources
├── META-INF
└── plugin.xml
├── icons
├── ic_mybatis.png
└── ic_mybatis@2x.png
└── template
├── base
├── baseCriteria.txt
├── baseQuery.txt
├── criteria.txt
└── pageResult.txt
├── do.txt
├── manager.txt
├── managerImpl.txt
├── mapper-ext.txt
├── mapper.txt
├── query.txt
├── sqlmap-ext.txt
└── sqlmap.txt
/.gitignore:
--------------------------------------------------------------------------------
1 | .classpath
2 | .project
3 | .settings/
4 | .git/
5 |
6 | # Mac
7 | .DS_Store
8 | # Maven
9 | log/
10 | target/
11 |
12 | #Intellij
13 | out/
14 | .idea/workspace.xml
--------------------------------------------------------------------------------
/.idea/compiler.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 |
6 |
7 |
8 |
--------------------------------------------------------------------------------
/.idea/emacs.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 |
6 |
--------------------------------------------------------------------------------
/.idea/kotlinc.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 |
6 |
--------------------------------------------------------------------------------
/.idea/libraries/KotlinJavaRuntime.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 |
6 |
7 |
8 |
9 |
10 |
11 |
12 |
13 |
14 |
15 |
16 |
17 |
--------------------------------------------------------------------------------
/.idea/libraries/commons_lang3_3_6.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 |
6 |
7 |
8 |
9 |
10 |
--------------------------------------------------------------------------------
/.idea/misc.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 |
6 |
--------------------------------------------------------------------------------
/.idea/modules.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 |
6 |
7 |
8 |
--------------------------------------------------------------------------------
/.idea/vcs.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 |
6 |
--------------------------------------------------------------------------------
/.idea/workspace.xml:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 |
6 |
7 |
8 |
9 |
10 |
11 |
12 |
13 |
14 |
15 |
16 |
17 |
18 |
19 |
20 |
21 |
22 |
23 |
24 |
25 |
26 |
27 |
28 |
29 |
30 |
31 |
32 |
33 |
34 |
35 |
36 |
37 |
38 |
39 |
40 |
41 |
42 |
43 |
44 |
45 |
46 |
47 |
48 |
49 |
50 |
51 |
52 |
53 |
54 |
55 |
56 |
57 |
58 |
59 |
60 |
61 |
62 |
63 |
64 |
65 |
66 |
67 |
68 |
69 |
70 |
71 |
72 |
73 |
74 |
75 |
76 | sIsMHCStaff
77 | 卖好车
78 | MHC_STAFF
79 | baseQuery.txt
80 | sqlmapOutFile
81 | doOutFilePath
82 | doResult
83 | doOutFile
84 | className
85 | sqlmapResult
86 | .do
87 | /do
88 | 不替换旧文件
89 | cri
90 | BaseCriteria
91 | Criterion
92 | b2b
93 | pref
94 | getClassName
95 | managerPackagePath
96 | getclass
97 | do..
98 | do
99 | getColInfoList
100 | pks
101 | getSqlmapParamList
102 | doTemplate
103 | doresult
104 | merge
105 | VP_LIST
106 |
107 |
108 | 152
109 | param
110 | )
111 | sDaoPackageDir
112 | baseCriteria
113 | baseQuery
114 |
115 |
116 |
117 |
118 |
119 |
120 |
123 |
124 |
125 |
160 |
161 |
162 |
163 |
164 | true
165 |
166 | false
167 | true
168 | true
169 |
170 |
171 | true
172 | DEFINITION_ORDER
173 |
174 |
175 |
176 |
177 |
178 |
179 |
180 |
181 |
182 |
183 |
184 |
185 |
186 |
187 |
188 |
189 |
190 |
191 |
192 |
193 |
194 | ActionScript specificJavaScript
195 |
196 |
197 | Android > Lint > Lint
198 |
199 |
200 | Batch Applications Issues
201 |
202 |
203 | Flow type checkerJavaScript
204 |
205 |
206 | GeneralJavaScript
207 |
208 |
209 | Groovy
210 |
211 |
212 | HTML
213 |
214 |
215 | J2ME issuesJava
216 |
217 |
218 | Java
219 |
220 |
221 | JavaFX
222 |
223 |
224 | JavaScript
225 |
226 |
227 | OtherGroovy
228 |
229 |
230 | Portability issuesJava
231 |
232 |
233 | Spring
234 |
235 |
236 | Spring CoreSpring
237 |
238 |
239 | Struts
240 |
241 |
242 | Struts 1Struts
243 |
244 |
245 | TypeScript
246 |
247 |
248 | Web Services
249 |
250 |
251 | XML
252 |
253 |
254 | XMLSpring CoreSpring
255 |
256 |
257 |
258 |
259 |
260 |
261 |
262 |
263 |
264 |
265 |
266 |
267 |
268 |
269 |
270 |
271 |
272 |
273 |
274 |
275 |
276 |
277 |
278 |
279 |
280 |
281 |
282 |
283 |
284 |
285 |
286 |
12 | 
13 |
14 |
15 | * 入口：安装并重启后，在菜单栏Tools下可以看到插件的功能入口
16 |
17 | 
18 |
19 |
20 | * 配置：需要配置用于连接到目标数据库的链接，用户名和密码以及生成的各个java类和xml文件的路径。卖好车内部使用，请在"是否用于卖好车内部代码"出选择"是。
21 |
22 | * 生成mybatis代码：输入表名，点击确定，生成该表的mybatis代码。
23 |
24 | ### 实现
25 | ##### 配置流程
26 |
27 | 1. 配置数据库连接链接，用户名，密码，配置生成代码的目录路径；
28 |
29 | 2. 插件连接到数据库，获取表结构信息(不获取表内容信息)；
30 |
31 | 3. 通过Velocity ，将预先定义好的模板于数据库信息结合，生成类文件，写入目标路径中。
32 |
33 | ##### API
34 | * 实体
35 | DO，Query，Mapper，Manager, Mapper.xml。前缀为：表名（驼峰），比如表table_user，生成的实体分别为：TableUserDO，TableUserQuery,TableUserMapper,TableUserManager 和 TableUserMapper.xml。
36 | > * DO 表对象，每个字段代表一个列
37 | > * Query 查询对象，用于添加各类查询条件
38 | > * Mapper mybatis 需要的mapper的接口
39 | > * Manager 对外输出的API，包含了基本的增删改查
40 | > * Mapper.xml mybatis的mapper文件，sql的具体实现
41 |
42 | * 拓展
43 | 每个对象都可以创建一个子对象，包含在同级目录的ext子目录下。具体的实现可以自定义。
44 |
45 | * CodeExample
46 | ```
47 | public class example {
48 |
49 | @Autowired
50 | private UserManager userManager;
51 |
52 | public void findUser() {
53 | //select DO by id
54 | long userId = 1;
55 | UserDO userDOById = userManager.selectByPrimaryKey(userId);
56 |
57 | //select DO by query
58 | UserQuery userQuery = new UserQuery();
59 | userQuery.createCriteria()
60 | .andUserNameLike("JonSnow%");
61 | List userListResult = userManager.selectByQuery(userQuery);
62 |
63 | //select DO by page
64 | UserQuery pageQuery = new UserQuery();
65 | pageQuery.setPageNo(1);
66 | pageQuery.setPageSize(20);
67 | pageQuery.createCriteria()
68 | .andUserNameLike("JonSnow%");
69 | PageResult userPageResult = userManager.selectByQueryWithPage(pageQuery);
70 | }
71 | }
72 | ```
73 |
74 |
75 |
76 |
77 |
78 |
--------------------------------------------------------------------------------
/_config.yml:
--------------------------------------------------------------------------------
1 | theme: jekyll-theme-slate
--------------------------------------------------------------------------------
/lib/commons-lang3-3.6.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/maihaoche/mybatis-generator-plugin/f9255ade46443574766a6a212f7692f100cb8f48/lib/commons-lang3-3.6.jar
--------------------------------------------------------------------------------
/lib/kotlin-reflect.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/maihaoche/mybatis-generator-plugin/f9255ade46443574766a6a212f7692f100cb8f48/lib/kotlin-reflect.jar
--------------------------------------------------------------------------------
/lib/kotlin-stdlib-jre7-sources.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/maihaoche/mybatis-generator-plugin/f9255ade46443574766a6a212f7692f100cb8f48/lib/kotlin-stdlib-jre7-sources.jar
--------------------------------------------------------------------------------
/lib/kotlin-stdlib-jre7.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/maihaoche/mybatis-generator-plugin/f9255ade46443574766a6a212f7692f100cb8f48/lib/kotlin-stdlib-jre7.jar
--------------------------------------------------------------------------------
/lib/kotlin-stdlib-jre8-sources.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/maihaoche/mybatis-generator-plugin/f9255ade46443574766a6a212f7692f100cb8f48/lib/kotlin-stdlib-jre8-sources.jar
--------------------------------------------------------------------------------
/lib/kotlin-stdlib-jre8.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/maihaoche/mybatis-generator-plugin/f9255ade46443574766a6a212f7692f100cb8f48/lib/kotlin-stdlib-jre8.jar
--------------------------------------------------------------------------------
/lib/kotlin-stdlib-sources.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/maihaoche/mybatis-generator-plugin/f9255ade46443574766a6a212f7692f100cb8f48/lib/kotlin-stdlib-sources.jar
--------------------------------------------------------------------------------
/lib/kotlin-stdlib.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/maihaoche/mybatis-generator-plugin/f9255ade46443574766a6a212f7692f100cb8f48/lib/kotlin-stdlib.jar
--------------------------------------------------------------------------------
/lib/kotlin-test.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/maihaoche/mybatis-generator-plugin/f9255ade46443574766a6a212f7692f100cb8f48/lib/kotlin-test.jar
--------------------------------------------------------------------------------
/lib/mysql-connector-java-5.1.41.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/maihaoche/mybatis-generator-plugin/f9255ade46443574766a6a212f7692f100cb8f48/lib/mysql-connector-java-5.1.41.jar
--------------------------------------------------------------------------------
/src/main/java/com/yangyang/mybatis/generator/MyConstant.kt:
--------------------------------------------------------------------------------
1 | package main.java.com.yangyang.mybatis.generator
2 |
3 | /**
4 | * 类简介：项目的常量。存储一些配置数据的信息
5 | * 作者： yang
6 | * 时间： 17/7/24
7 | * 邮箱： yangyang1@maihaoche.com
8 | */
9 | class MyConstant {
10 |
11 | companion object {
12 | // -------------------------配置数据库连接的key-------------------------
13 | val DB_CONF_INITED = "db_conf_inited"//mysql是否初始化
14 | val DB_URL = "db_url"//数据库的url
15 | val DB_USER = "db_user"//数据库的用户名
16 | val DB_PW = "db_pw"//数据库的密码
17 | // -------------------------生成数据的一些路径的配置key-------------------------
18 | val DAO_MODULE = "dao_module"//dao的目录
19 | val DAO_PACKAGE = "dao_package"//dao的package名
20 | val MANAGER_MODULE = "namager_module"//manager所在的目录
21 | val MANAGER_PACKAGE = "manager_package"//manager生成目录
22 | val TABLE_PREFIX = "table_prefix"//表名的前缀
23 | val MHC_STAFF = "mhc_staff"//是否生成卖好车内部的代码
24 |
25 | //-------------------------内部字段-------------------------
26 | val TABLE_HISTORY = "table_history"//保存历史选中的表名。
27 |
28 | //-------------------------action的一些路径-------------------
29 |
30 | val ACTION_TOOLS_MENU = "ToolsMenu"
31 | val ACTION_MYBATIS = "Mybatis"
32 | val ACTION_MYBATIS_PATH_CONFIGURE = "配置路径"
33 | val ACTION_MYBATIS_DB_CONFIGURE = "配置数据库"
34 |
35 |
36 | }
37 | }
--------------------------------------------------------------------------------
/src/main/java/com/yangyang/mybatis/generator/actions/ActionConfigPath.kt:
--------------------------------------------------------------------------------
1 | package main.java.com.yangyang.mybatis.generator.actions
2 |
3 | import com.intellij.openapi.actionSystem.AnAction
4 | import com.intellij.openapi.actionSystem.AnActionEvent
5 | import com.intellij.openapi.externalSystem.util.ExternalSystemUiUtil
6 | import com.intellij.ui.CheckBoxList
7 | import com.intellij.ui.IdeBorderFactory
8 | import com.intellij.ui.components.CheckBox
9 | import main.java.com.yangyang.mybatis.generator.MyConstant
10 | import main.java.com.yangyang.mybatis.generator.utils.NotificationUtil
11 | import main.java.com.yangyang.mybatis.generator.utils.PlatformUtil
12 | import main.java.com.yangyang.mybatis.generator.utils.UIUtil
13 | import main.java.com.yangyang.mybatis.generator.utils.mybatis.MyBatisGenConst
14 | import java.awt.GridBagLayout
15 | import java.util.*
16 | import javax.swing.*
17 |
18 |
19 | /**
20 | * 类简介：配置的动作
21 | * 作者： yang
22 | * 时间： 17/7/24
23 | * 邮箱： yangyang1@maihaoche.com
24 | */
25 | class ActionConfigPath : AnAction() {
26 | /**
27 | * 执行相关的动作
28 | */
29 | override fun actionPerformed(event: AnActionEvent?) {
30 | if (event == null) {
31 | return
32 | }
33 | val project = event.getProject()
34 |
35 | val content = JPanel(GridBagLayout())
36 |
37 | val daoModule = UIUtil.addTextInputToPanel("DO,Query,Mapper所在的module名: ", MyConstant.DAO_MODULE, content)
38 | val daoPackagePath = UIUtil.addTextInputToPanel("DO,Query,Mapper所在的package名: ", MyConstant.DAO_PACKAGE, content)
39 | val managerModule = UIUtil.addTextInputToPanel("Manager所在的module名: ", MyConstant.MANAGER_MODULE, content)
40 | val managerPackagePath = UIUtil.addTextInputToPanel("Manager所在的package: ", MyConstant.MANAGER_PACKAGE, content)
41 | val tablePrefix = UIUtil.addTextInputToPanel("表名的前缀: ", MyConstant.TABLE_PREFIX, content)
42 | content.add(JLabel("是否用于卖好车内部代码: "), ExternalSystemUiUtil.getFillLineConstraints(0))
43 | // 卖好车内部圆通的选择
44 | val isMHCStaffCheck = CheckBoxList()
45 | isMHCStaffCheck.setItems(Arrays.asList("是"), null)
46 | isMHCStaffCheck.border = IdeBorderFactory.createEmptyBorder(8)
47 | content.add(isMHCStaffCheck, ExternalSystemUiUtil.getFillLineConstraints(0))
48 | if (PlatformUtil.getData(MyConstant.MHC_STAFF, false)) {
49 | isMHCStaffCheck.setItemSelected("是", true)
50 | }
51 | //清空缓存
52 | val clearCacheBtn = JButton("清除缓存")
53 | clearCacheBtn.addActionListener {
54 | PlatformUtil.setData(MyConstant.DAO_MODULE, "")
55 | PlatformUtil.setData(MyConstant.DAO_PACKAGE, "")
56 | PlatformUtil.setData(MyConstant.MANAGER_MODULE, "")
57 | PlatformUtil.setData(MyConstant.MANAGER_PACKAGE, "")
58 | isMHCStaffCheck.clearSelection()
59 | MyBatisGenConst.clearConfig()
60 | NotificationUtil.info("清除缓存成功", project)
61 | daoModule.text = ""
62 | daoPackagePath.text = ""
63 | managerModule.text = ""
64 | managerPackagePath.text = ""
65 | tablePrefix.text = ""
66 | }
67 | clearCacheBtn.border = IdeBorderFactory.createEmptyBorder(14, 4, 4, 4)
68 | content.add(clearCacheBtn, ExternalSystemUiUtil.getLabelConstraints(0))
69 | //显示dialog
70 | content.border = IdeBorderFactory.createEmptyBorder(0, 0, 8, 0)
71 |
72 | UIUtil.showDialog("路径配置参数", project, content, {
73 | if (it) {
74 | if (daoPackagePath.text.trim().isNullOrBlank()
75 | || managerModule.text.trim().isNullOrBlank()
76 | || managerPackagePath.text.trim().isNullOrBlank()
77 | ) {
78 | NotificationUtil.popError("路径配置不能为空", event)
79 | return@showDialog
80 | }
81 | PlatformUtil.setData(MyConstant.DAO_MODULE, daoModule.text)
82 | PlatformUtil.setData(MyConstant.DAO_PACKAGE, daoPackagePath.text)
83 | PlatformUtil.setData(MyConstant.MANAGER_MODULE, managerModule.text)
84 | PlatformUtil.setData(MyConstant.MANAGER_PACKAGE, managerPackagePath.text)
85 | PlatformUtil.setData(MyConstant.TABLE_PREFIX, tablePrefix.text)
86 | val isMHCStaff = isMHCStaffCheck.isItemSelected("是")
87 | PlatformUtil.setData(MyConstant.MHC_STAFF, isMHCStaff)
88 |
89 | MyBatisGenConst.setConfig(
90 | daoModule.text,
91 | daoPackagePath.text,
92 | managerModule.text,
93 | managerPackagePath.text,
94 | tablePrefix.text,
95 | isMHCStaff,
96 | project)
97 | NotificationUtil.popInfo("路径配置修改成功", event)
98 | }
99 | })
100 | }
101 |
102 | companion object {
103 | /**
104 | * 显示配置窗口
105 | */
106 | fun showConfigureDialog(msg: String, event: AnActionEvent) {
107 | //总的内容
108 | val content = JPanel(GridBagLayout())
109 | //设置切换为全module任务名
110 | content.add(JLabel(msg), ExternalSystemUiUtil.getFillLineConstraints(0))
111 |
112 | UIUtil.showDialog("错误提示", event.project, content, {
113 | if (it) {
114 | PlatformUtil.performActionWithPath(event, MyConstant.ACTION_TOOLS_MENU, MyConstant.ACTION_MYBATIS, MyConstant.ACTION_MYBATIS_PATH_CONFIGURE)
115 | }
116 | })
117 | }
118 | }
119 | }
--------------------------------------------------------------------------------
/src/main/java/com/yangyang/mybatis/generator/actions/ActionConfigureDB.kt:
--------------------------------------------------------------------------------
1 | package com.yangyang.mybatis.generator.actions
2 |
3 | import com.intellij.openapi.actionSystem.AnAction
4 | import com.intellij.openapi.actionSystem.AnActionEvent
5 | import com.intellij.openapi.externalSystem.util.ExternalSystemUiUtil
6 | import com.intellij.ui.IdeBorderFactory
7 | import main.java.com.yangyang.mybatis.generator.MyConstant
8 | import main.java.com.yangyang.mybatis.generator.utils.NotificationUtil
9 | import main.java.com.yangyang.mybatis.generator.utils.PlatformUtil
10 | import main.java.com.yangyang.mybatis.generator.utils.UIUtil
11 | import java.awt.GridBagLayout
12 | import javax.swing.JButton
13 | import javax.swing.JLabel
14 | import javax.swing.JPanel
15 |
16 | /**
17 | * 类简介：配置mysql的action
18 | * 作者： yang
19 | * 时间： 17/7/26
20 | * 邮箱： yangyang1@maihaoche.com
21 | */
22 | class ActionConfigureDB : AnAction() {
23 | override fun actionPerformed(event: AnActionEvent?) {
24 | if (event == null) {
25 | return
26 | }
27 | val project = event.getProject()
28 | //总的内容
29 | val content = JPanel(GridBagLayout())
30 | content.add(JLabel("该版本仅支持mysql的驱动(com.mysql.jdbc.Driver): "), ExternalSystemUiUtil.getFillLineConstraints(0))
31 |
32 | val urlInput = UIUtil.addTextInputToPanel("请输入数据库链接的URL: ", MyConstant.DB_URL, content)
33 | val userNameInput = UIUtil.addTextInputToPanel("请输入用户名: ", MyConstant.DB_USER, content)
34 | val passwordInput = UIUtil.addTextInputToPanel("请输入密码: ", MyConstant.DB_PW, content)
35 |
36 | //清空缓存
37 | val clearCacheBtn = JButton("清除缓存")
38 | clearCacheBtn.addActionListener {
39 | PlatformUtil.setData(MyConstant.DB_URL, "")
40 | PlatformUtil.setData(MyConstant.DB_USER, "")
41 | PlatformUtil.setData(MyConstant.DB_PW, "")
42 | PlatformUtil.setData(MyConstant.DB_CONF_INITED, false)
43 | NotificationUtil.info("清除缓存成功", project)
44 | urlInput.text = ""
45 | userNameInput.text = ""
46 | passwordInput.text = ""
47 | }
48 | clearCacheBtn.border = IdeBorderFactory.createEmptyBorder(14, 4, 4, 4)
49 | content.add(clearCacheBtn, ExternalSystemUiUtil.getLabelConstraints(0))
50 | //显示dialog
51 | content.border = IdeBorderFactory.createEmptyBorder(0, 0, 8, 0)
52 |
53 | UIUtil.showDialog("数据库配置参数", project, content, {
54 | if (it) {
55 | if (urlInput.text.trim().isNullOrBlank()
56 | || userNameInput.text.trim().isNullOrBlank()
57 | || passwordInput.text.trim().isNullOrBlank()
58 | ) {
59 | NotificationUtil.popInfo("配置不能为空", event)
60 | return@showDialog
61 | }
62 | PlatformUtil.setData(MyConstant.DB_URL, urlInput.text)
63 | PlatformUtil.setData(MyConstant.DB_USER, userNameInput.text)
64 | PlatformUtil.setData(MyConstant.DB_PW, passwordInput.text)
65 | PlatformUtil.setData(MyConstant.DB_CONF_INITED, true)
66 | NotificationUtil.popInfo("数据库配置修改成功", event)
67 | }
68 | })
69 | }
70 | companion object {
71 | /**
72 | * 显示配置窗口
73 | */
74 | fun showConfigureDialog(msg: String, event: AnActionEvent) {
75 | //总的内容
76 | val content = JPanel(GridBagLayout())
77 | //设置切换为全module任务名
78 | content.add(JLabel(msg), ExternalSystemUiUtil.getFillLineConstraints(0))
79 |
80 | UIUtil.showDialog("错误提示", event.project, content, {
81 | if (it) {
82 | PlatformUtil.performActionWithPath(event, MyConstant.ACTION_TOOLS_MENU, MyConstant.ACTION_MYBATIS, MyConstant.ACTION_MYBATIS_DB_CONFIGURE)
83 | }
84 | })
85 | }
86 | }
87 | }
--------------------------------------------------------------------------------
/src/main/java/com/yangyang/mybatis/generator/actions/ActionGenerate.kt:
--------------------------------------------------------------------------------
1 | package main.java.com.yangyang.mybatis.generator.actions
2 |
3 | import com.intellij.openapi.actionSystem.AnAction
4 | import com.intellij.openapi.actionSystem.AnActionEvent
5 | import com.intellij.openapi.externalSystem.util.ExternalSystemUiUtil
6 | import com.intellij.openapi.progress.ProgressIndicator
7 | import com.intellij.openapi.progress.Task
8 | import com.intellij.openapi.progress.util.ProgressIndicatorUtils
9 | import com.intellij.ui.CheckBoxList
10 | import com.intellij.ui.IdeBorderFactory
11 | import com.intellij.util.containers.isNullOrEmpty
12 | import com.yangyang.mybatis.generator.actions.ActionConfigureDB
13 | import main.java.com.yangyang.mybatis.generator.MyConstant
14 | import main.java.com.yangyang.mybatis.generator.utils.NotificationUtil
15 | import main.java.com.yangyang.mybatis.generator.utils.PlatformUtil
16 | import main.java.com.yangyang.mybatis.generator.utils.UIUtil
17 | import main.java.com.yangyang.mybatis.generator.utils.mybatis.MyBatisGenConst
18 | import main.java.com.yangyang.mybatis.generator.utils.mybatis.MybatisCore
19 | import java.awt.GridBagLayout
20 | import javax.swing.JButton
21 | import javax.swing.JLabel
22 | import javax.swing.JPanel
23 | import javax.swing.ListSelectionModel
24 |
25 | /**
26 | * 类简介：产生模板代码的动作
27 | * 作者： yang
28 | * 时间： 17/7/24
29 | * 邮箱： yangyang1@maihaoche.com
30 | */
31 | class ActionGenerate : AnAction() {
32 |
33 |
34 | override fun actionPerformed(event: AnActionEvent?) {
35 | if (event == null) {
36 | return
37 | }
38 | val project = event.project
39 | if (project == null) {
40 | return
41 | }
42 |
43 | if (!MyBatisGenConst.initConfig(project)) {
44 | ActionConfigPath.showConfigureDialog("还没有设置生成代码路径，是否现在去设置?", event)
45 | return
46 | }
47 | if (!PlatformUtil.getData(MyConstant.DB_CONF_INITED, false)) {
48 | ActionConfigureDB.showConfigureDialog("还没有配置数据库连接，是否现在去设置?", event)
49 | return
50 | }
51 |
52 | val content = JPanel(GridBagLayout())
53 | content.add(JLabel("请输入需要产生代码的表名，以\",\"隔开: "), ExternalSystemUiUtil.getFillLineConstraints(0))
54 |
55 | val talbeNameInput = UIUtil.getTextField("")
56 | content.add(talbeNameInput, ExternalSystemUiUtil.getFillLineConstraints(0))
57 |
58 | var tableHistory = PlatformUtil.getDatas(MyConstant.TABLE_HISTORY)?.toList()
59 | val orphanModulesList = CheckBoxList()
60 | if (tableHistory != null && tableHistory.size > 0) {
61 | orphanModulesList.selectionMode = ListSelectionModel.MULTIPLE_INTERVAL_SELECTION
62 | orphanModulesList.setItems(tableHistory) { moduleName -> moduleName }
63 | orphanModulesList.border = IdeBorderFactory.createEmptyBorder(8)
64 | content.add(orphanModulesList, ExternalSystemUiUtil.getFillLineConstraints(0))
65 |
66 | //清空缓存
67 | val clearCacheBtn = JButton("清除缓存")
68 | clearCacheBtn.addActionListener {
69 | content.remove(orphanModulesList)
70 | content.updateUI()
71 | }
72 | clearCacheBtn.border = IdeBorderFactory.createEmptyBorder(14, 4, 4, 4)
73 | content.add(clearCacheBtn, ExternalSystemUiUtil.getLabelConstraints(0))
74 | }
75 | content.border = IdeBorderFactory.createEmptyBorder(0, 0, 8, 0)
76 |
77 | UIUtil.showDialog("请选择表名", project, content,
78 | {
79 | if (it) {
80 | var tableNames = mutableSetOf()
81 | talbeNameInput.text?.split(",")?.let { tableNames.addAll(it.filter { !it.isNullOrBlank() }) }
82 | if (tableHistory != null) {
83 | for (table in tableHistory) {
84 | if (table.isNullOrBlank()) {
85 | continue
86 | }
87 | if (orphanModulesList.isItemSelected(table)) {
88 | tableNames.add(table)
89 | }
90 | }
91 | }
92 | if (tableNames.size == 0) {
93 | return@showDialog
94 | }
95 | PlatformUtil.setDatas(MyConstant.TABLE_HISTORY, tableNames.toTypedArray())
96 | PlatformUtil.executeBackgroundTask(project
97 | , "正在生成代码"
98 | , Runnable {
99 | var exception = MybatisCore.batchGen(tableNames.toList(), project)
100 | if (exception == null) {
101 | NotificationUtil.popInfo("生成代码成功,详情见EventLog控制台输出", event)
102 | } else {
103 | NotificationUtil.popError("生成代码出错,详情见EventLog控制台输出", event)
104 | NotificationUtil.error("生成代码出错exception:" + exception )
105 | }
106 | PlatformUtil.notifyFileChanged(project)
107 | })
108 |
109 | }
110 | })
111 | }
112 | }
--------------------------------------------------------------------------------
/src/main/java/com/yangyang/mybatis/generator/actions/ActionGroupMybatis.kt:
--------------------------------------------------------------------------------
1 | package main.java.com.yangyang.mybatis.generator.actions
2 |
3 | import com.intellij.openapi.actionSystem.DefaultActionGroup
4 |
5 | /**
6 | * 类简介：
7 | * 作者： yang
8 | * 时间： 17/7/24
9 | * 邮箱： yangyang1@maihaoche.com
10 | */
11 | class ActionGroupMybatis : DefaultActionGroup()
--------------------------------------------------------------------------------
/src/main/java/com/yangyang/mybatis/generator/utils/NotificationUtil.kt:
--------------------------------------------------------------------------------
1 | package main.java.com.yangyang.mybatis.generator.utils
2 |
3 | import com.intellij.notification.NotificationDisplayType
4 | import com.intellij.notification.NotificationGroup
5 | import com.intellij.notification.NotificationType
6 | import com.intellij.notification.Notifications
7 | import com.intellij.openapi.actionSystem.AnActionEvent
8 | import com.intellij.openapi.actionSystem.DataKeys
9 | import com.intellij.openapi.diagnostic.Logger
10 | import com.intellij.openapi.project.Project
11 | import com.intellij.openapi.ui.MessageType
12 | import com.intellij.openapi.ui.popup.Balloon
13 | import com.intellij.openapi.ui.popup.JBPopupFactory
14 | import com.intellij.openapi.wm.WindowManager
15 | import com.intellij.ui.awt.RelativePoint
16 |
17 | /**
18 | * 类简介：
19 | * 作者： yang
20 | * 时间： 17/7/24
21 | * 邮箱： yangyang1@maihaoche.com
22 | */
23 | class NotificationUtil private constructor(){
24 |
25 | companion object {
26 | private val sPLUGIN_LOGGER = Logger.getInstance(NotificationUtil::class.java)
27 | private val GROUP_DISPLAY_ID_INFO_BALLOON = NotificationGroup("main.java.com.yangyang.mybatis.generator",
28 | NotificationDisplayType.BALLOON, true)
29 |
30 | private val LOGGING_NOTIFICATION = NotificationGroup("Gradle sync", NotificationDisplayType.NONE, true)
31 |
32 | /**
33 | * IntelliJ自带的API输出的info，android studio 会显示在底部statusbar下面。
34 |
35 | * @param info
36 | */
37 | fun infoToStatusBar(info: String) {
38 | sPLUGIN_LOGGER.info(info)
39 | }
40 |
41 | /**
42 | * 显示在gradle 的consol中的info
43 |
44 | * @param infoMsg
45 | * *
46 | * @param project
47 | */
48 | fun info(infoMsg: String?, project: Project?) {
49 | if (infoMsg == null || infoMsg.trim { it <= ' ' } == "" || project == null) {
50 | return
51 | }
52 | LOGGING_NOTIFICATION.createNotification(infoMsg, MessageType.INFO).notify(project)
53 | }
54 | /**
55 | * 控制台，显示错误信息
56 | */
57 | fun error(errorMsg: String?) {
58 | if (errorMsg == null || errorMsg.trim { it <= ' ' } == "") {
59 | return
60 | }
61 | val notificationX = GROUP_DISPLAY_ID_INFO_BALLOON.createNotification(errorMsg, NotificationType.ERROR)
62 | Notifications.Bus.notify(notificationX)
63 | }
64 |
65 | /**
66 | * 严重的错误，需要弹窗提醒。
67 | */
68 | fun popError(errorMsg: String?, event: AnActionEvent) {
69 | pop(errorMsg,event,MessageType.ERROR)
70 | }
71 |
72 | /**
73 | * 弹窗提醒。
74 | */
75 | fun popInfo(info: String?, event: AnActionEvent) {
76 | pop(info,event,MessageType.INFO)
77 | }
78 |
79 | /**
80 | * 弹出信息。
81 | */
82 | private fun pop(msg:String?,event:AnActionEvent,msgType:MessageType){
83 | if (msg == null || msg.trim { it <= ' ' } == "") {
84 | return
85 | }
86 | val statusBar = WindowManager.getInstance()
87 | .getStatusBar(DataKeys.PROJECT.getData(event.dataContext))
88 | JBPopupFactory.getInstance()
89 | .createHtmlTextBalloonBuilder(msg, msgType, null)
90 | .setFadeoutTime(3500)
91 | .createBalloon()
92 | .show(RelativePoint.getCenterOf(statusBar.component),
93 | Balloon.Position.atRight)
94 | }
95 |
96 |
97 | }
98 |
99 | }
--------------------------------------------------------------------------------
/src/main/java/com/yangyang/mybatis/generator/utils/PlatformUtil.kt:
--------------------------------------------------------------------------------
1 | package main.java.com.yangyang.mybatis.generator.utils
2 |
3 | import com.intellij.ide.util.PropertiesComponent
4 | import com.intellij.openapi.actionSystem.ActionManager
5 | import com.intellij.openapi.actionSystem.AnAction
6 | import com.intellij.openapi.actionSystem.AnActionEvent
7 | import com.intellij.openapi.actionSystem.DefaultActionGroup
8 | import com.intellij.openapi.application.ApplicationManager
9 | import com.intellij.openapi.application.TransactionGuard
10 | import com.intellij.openapi.progress.ProgressIndicator
11 | import com.intellij.openapi.progress.ProgressManager
12 | import com.intellij.openapi.progress.Task
13 | import com.intellij.openapi.progress.util.ProgressIndicatorUtils
14 | import com.intellij.openapi.progress.util.ReadTask
15 | import com.intellij.openapi.project.Project
16 | import com.intellij.openapi.roots.ex.ProjectRootManagerEx
17 | import com.intellij.openapi.vfs.LocalFileSystem
18 | import com.intellij.openapi.vfs.VirtualFile
19 | import com.intellij.openapi.vfs.VirtualFileManager
20 | import com.intellij.openapi.vfs.newvfs.RefreshQueue
21 | import com.intellij.util.ui.UIUtil
22 | import javax.swing.SwingUtilities
23 |
24 | /**
25 | * 类简介：
26 | * 作者： yang
27 | * 时间： 17/7/24
28 | * 邮箱： yangyang1@maihaoche.com
29 | */
30 | class PlatformUtil {
31 | companion object {
32 |
33 | /**
34 | * 在异步线程池中执行任务，真正的异步任务。不会阻塞主线程
35 | */
36 | fun executeBackgroundTask(project: Project,title: String, runnable: Runnable) {
37 | ApplicationManager.getApplication().invokeLater({
38 | ApplicationManager.getApplication().runReadAction({
39 | object : Task.Backgroundable(project,title) {
40 | override fun run(progressIndicator: ProgressIndicator) {
41 | ProgressIndicatorUtils.runWithWriteActionPriority({
42 | runnable.run()
43 | if (!progressIndicator.isCanceled) {
44 | progressIndicator.stop()
45 | }
46 | }, progressIndicator)
47 | }
48 | override fun shouldStartInBackground(): Boolean {
49 | return true
50 | }
51 | }.queue()
52 | })
53 | })
54 | }
55 |
56 | /**
57 | * 通知文件发生变化
58 | */
59 | fun notifyFileChanged(project: Project) {
60 | project?.getBaseDir()?.refresh(true, true);
61 | }
62 |
63 | /**
64 | * 涉及到Project修改的操作。在主线程执行某个耗时任务，需要调用invokeAndWaitIfNeeded
65 | * 注意，该方法执行的任务会阻塞主线程！
66 | */
67 | fun executeProjectChanges(project: Project, changes: Runnable) {
68 | UIUtil.invokeAndWaitIfNeeded(Runnable {
69 | //在UI主线程执行任务。执行耗时的任务
70 | //官方文档：http://www.jetbrains.org/intellij/sdk/docs/basics/architectural_overview/general_threading_rules.html
71 | //总结：文件修改，涉及project module的修改，等操作都需要用该API来run.
72 | ApplicationManager.getApplication().runWriteAction() {
73 | if (!project.isDisposed) {
74 | ProjectRootManagerEx.getInstanceEx(project).mergeRootsChangesDuring(changes)
75 | }
76 | }
77 | })
78 | }
79 |
80 |
81 | /**
82 | * 获取本地存储的key value ，string类型
83 | */
84 | fun getData(key: String, defaultValue: String): String {
85 | return PropertiesComponent.getInstance().getValue(key, defaultValue)
86 | }
87 |
88 | /**
89 | * 获取本地存储的key value ，string类型
90 | */
91 | fun getData(key: String, defaultValue: Boolean): Boolean {
92 | return PropertiesComponent.getInstance().getBoolean(key, defaultValue)
93 | }
94 |
95 |
96 | /**
97 | * 数组数据
98 | */
99 | fun getDatas(key: String): Array? {
100 | return PropertiesComponent.getInstance().getValues(key)
101 | }
102 |
103 | /**
104 | * 存储key value数据，string[]类型
105 | */
106 | fun setDatas(key: String, setValue: Array?) {
107 | PropertiesComponent.getInstance().setValues(key, setValue)
108 | }
109 |
110 | /**
111 | * 存储key value数据，string类型
112 | */
113 | fun setData(key: String, stringValue: String) {
114 | PropertiesComponent.getInstance().setValue(key, stringValue)
115 | }
116 |
117 | /**
118 | * 存储key value数据，boolean类型
119 | */
120 | fun setData(key: String, booleanValue: Boolean) {
121 | PropertiesComponent.getInstance().setValue(key, booleanValue)
122 | }
123 |
124 | /**
125 | * 根据路径，执行某个命令。最多三个层级。既：ToolsMenu-->Mybatis-->xxx
126 | */
127 | fun performActionWithPath(event: AnActionEvent, menuName: String?, vararg paths: String) {
128 | if (menuName == null || menuName.isEmpty()) {
129 | return
130 | }
131 | var firstLevelPath: String?
132 | var secondLevelPath = ""
133 | if (paths.size > 0) {
134 | firstLevelPath = paths[0]
135 | //至少有1级深度，切不为空
136 | if (firstLevelPath.isEmpty()) {
137 | NotificationUtil.error("在id为" + menuName + "的菜单栏,paths的第一级路径为空。")
138 | return
139 | }
140 | if (paths.size > 1) {
141 | secondLevelPath = paths[1]
142 | }
143 | } else {
144 | NotificationUtil.error("在id为" + menuName + "的菜单栏,paths为空。")
145 | return
146 | }
147 | val actionManager = ActionManager.getInstance()
148 | val actionGroup = actionManager.getAction(menuName) as DefaultActionGroup
149 | var found = false
150 | val anActions = actionGroup.childActionsOrStubs
151 | if (anActions.size > 0) {
152 | for (i in anActions.indices) {
153 | if (firstLevelPath == anActions[i].templatePresentation.text) {
154 | if (paths.size == 1) {
155 | anActions[i].actionPerformed(event)
156 | return
157 | } else if (secondLevelPath.isEmpty()) {
158 | return
159 | }
160 | var subActions: Array? = null
161 | if (anActions[i] != null && anActions[i] is DefaultActionGroup && anActions[i].templatePresentation.isEnabledAndVisible) {
162 | subActions = (anActions[i] as DefaultActionGroup).childActionsOrStubs
163 | }
164 | if (subActions != null && subActions.size > 0) {
165 | for (j in subActions.indices) {
166 | if (secondLevelPath == subActions[j].templatePresentation.text) {
167 | found = true
168 | subActions[j].actionPerformed(event)
169 | }
170 | }
171 | }
172 | }
173 | }
174 | }
175 | if (!found) {
176 | NotificationUtil.error("在id为" + menuName + "的菜单栏下没有找到Action：" + firstLevelPath + if (secondLevelPath.isEmpty()) "" else "->" + secondLevelPath)
177 | }
178 | }
179 |
180 | }
181 |
182 | }
--------------------------------------------------------------------------------
/src/main/java/com/yangyang/mybatis/generator/utils/UIUtil.kt:
--------------------------------------------------------------------------------
1 | package main.java.com.yangyang.mybatis.generator.utils
2 |
3 | import com.intellij.openapi.externalSystem.util.ExternalSystemUiUtil
4 | import com.intellij.openapi.project.Project
5 | import com.intellij.openapi.ui.DialogWrapper
6 | import com.intellij.ui.CheckBoxList
7 | import com.intellij.ui.IdeBorderFactory
8 | import com.intellij.ui.components.JBScrollPane
9 | import java.awt.GridBagLayout
10 | import java.awt.event.ItemEvent
11 | import java.util.ArrayList
12 | import javax.swing.*
13 |
14 | /**
15 | * 类简介：
16 | * 作者： yang
17 | * 时间： 17/7/24
18 | * 邮箱： yangyang1@maihaoche.com
19 | */
20 | class UIUtil {
21 | companion object {
22 | /**
23 | * 显示一个dialog的对话框
24 | */
25 | fun showDialog(titleText:String,project:Project?,content:JComponent, onOkClick:(ok:Boolean)->Unit){
26 | val dialog = object : DialogWrapper(project) {
27 | init {
28 | title = titleText
29 | init()
30 | }
31 |
32 | override fun createCenterPanel(): JComponent? {
33 | return JBScrollPane(content)
34 | }
35 |
36 | override fun createActions(): Array {
37 | return arrayOf(okAction)
38 | }
39 | }
40 | onOkClick.let {
41 | onOkClick(dialog.showAndGet())
42 | }
43 | }
44 | /**在对话框面中
45 | * 添加一个数据框的内容
46 | */
47 | fun addTextInputToPanel(label: String, dataKey: String, panel: JPanel): JTextField {
48 | panel.add(JLabel(label), ExternalSystemUiUtil.getFillLineConstraints(0))
49 | val textField = getTextField(PlatformUtil.getData(dataKey, ""))
50 | panel.add(textField, ExternalSystemUiUtil.getFillLineConstraints(0))
51 | return textField
52 | }
53 |
54 | /**
55 | * 创建一个文本输入框
56 | */
57 | fun getTextField(defaultStr: String): JTextField {
58 | val jTextField = JTextField(defaultStr)
59 | jTextField.border = IdeBorderFactory.createEmptyBorder(4, 4, 4, 4)
60 | return jTextField
61 | }
62 |
63 | /**
64 | * 弹窗选择,单选，则是radiogroup组件。复选，则是checkbox组件
65 |
66 | * @param project
67 | * @param title
68 | * @param datas
69 | * @param singleChoose
70 | * @return
71 | */
72 | private fun chooseFromData(project: Project, title: String, datas: ArrayList, singleChoose: Boolean): ArrayList? {
73 | val mChoosedModuleNames = ArrayList()
74 | //布局
75 | val content = JPanel(GridBagLayout())
76 | content.add(JLabel(title), ExternalSystemUiUtil.getFillLineConstraints(0))
77 | val radioButtons = ArrayList()
78 | val orphanModulesList = CheckBoxList()
79 | if (singleChoose) {
80 | val box = Box.createVerticalBox()
81 | for (i in datas.indices) {
82 | val radioButton = JRadioButton(datas[i])
83 | radioButtons.add(radioButton)
84 | radioButton.addItemListener { e ->
85 | //选中这个，取消其他的选中
86 | if (e.stateChange == ItemEvent.SELECTED) {
87 | for (j in radioButtons.indices) {
88 | if (radioButtons[j] !== radioButton) {
89 | radioButtons[j].isSelected = false
90 | }
91 | }
92 | }
93 | }
94 | box.add(radioButton)
95 | }
96 | box.border = IdeBorderFactory.createEmptyBorder(8)
97 | content.add(box, ExternalSystemUiUtil.getFillLineConstraints(0))
98 | } else {
99 | orphanModulesList.selectionMode = ListSelectionModel.MULTIPLE_INTERVAL_SELECTION
100 | orphanModulesList.setItems(datas) { moduleName -> moduleName }
101 | orphanModulesList.border = IdeBorderFactory.createEmptyBorder(8)
102 | content.add(orphanModulesList, ExternalSystemUiUtil.getFillLineConstraints(0))
103 | }
104 | content.border = IdeBorderFactory.createEmptyBorder(0, 0, 8, 0)
105 |
106 | showDialog("请选择",project,content,{
107 | if (!it) return@showDialog
108 | if (singleChoose) {
109 | for (i in radioButtons.indices) {
110 | if (radioButtons[i].isSelected) {
111 | mChoosedModuleNames.add(radioButtons[i].text)
112 | }
113 | }
114 | } else {
115 | for (i in datas.indices) {
116 | val module = datas[i]
117 | if (orphanModulesList.isItemSelected(i)) {
118 | mChoosedModuleNames.add(module)
119 | }
120 | }
121 | }
122 | })
123 | return mChoosedModuleNames
124 | }
125 | }
126 | }
--------------------------------------------------------------------------------
/src/main/java/com/yangyang/mybatis/generator/utils/mybatis/MyBatisGenConst.kt:
--------------------------------------------------------------------------------
1 | package main.java.com.yangyang.mybatis.generator.utils.mybatis
2 |
3 | import com.intellij.openapi.project.Project
4 | import main.java.com.yangyang.mybatis.generator.MyConstant
5 | import main.java.com.yangyang.mybatis.generator.utils.PlatformUtil
6 | import java.io.File
7 |
8 |
9 | /**
10 | * 类简介：
11 | * 作者： yang
12 | * 时间： 17/7/24
13 | * 邮箱： yangyang1@maihaoche.com
14 | */
15 | class MyBatisGenConst {
16 |
17 | /**
18 | * 伴生对象。companion中的内容，可以通过类直接访问。
19 | */
20 | companion object {
21 | var sConfigured = false
22 | // 包名
23 | var sDaoModule = ""
24 | var sDaoPackage = ""//do,query,mapper,mapper_ext所在的package名
25 | var sManagerModule = ""
26 | var sManagerPackage = ""
27 | var sManagerImplPackage = ""
28 |
29 | var sTablePrefix = ""//表名的前缀
30 |
31 | // 输出目录
32 | var sDaoPackageDir = ""
33 | var sBaseDir = ""//BaseCriteria,BaseQuery,PageResult所在的文件目录
34 | var sMapperXmlDir = ""
35 | var sMapperExtXmlDir = ""
36 | var sDoJavaDir = ""
37 | var sQueryJavaDir = ""
38 | var sMapperJavaDir = ""
39 | var sMapperExtJavaDir = ""
40 | var sManagerJavaDir = ""
41 | var sManagerImplJavaDir = ""
42 |
43 | //是否是mhc内部使用的代码
44 | var sIsMHCStaff = false
45 |
46 | //模板路径
47 | val DO_TEMPLATE = "/template/do.txt"
48 | val QUERY_TEMPLATE = "/template/query.txt"
49 | val SQLMAP_TEMPLATE = "/template/sqlmap.txt"
50 | val MAPPER_TEMPLATE = "/template/mapper.txt"
51 | val MANAGER_TEMPLATE = "/template/manager.txt"
52 | val MANAGER_IMPL_TEMPLATE = "/template/managerImpl.txt"
53 | val MAPPER_EXT_TEMPLATE = "/template/mapper-ext.txt"
54 | val SQLMAP_EXT_TEMPLATE = "/template/sqlmap-ext.txt"
55 |
56 | val PAGERESULT_TEMPLATE = "/template/base/pageResult.txt"
57 | val BASE_CRITERIA_TEMPLATE = "/template/base/baseCriteria.txt"
58 | val CRITERIA_TEMPLATE = "/template/base/criteria.txt"
59 | val BASE_QUERY_TEMPLATE = "/template/base/baseQuery.txt"
60 |
61 |
62 | val QUERY_PREFIX = "Query"
63 | val MAPPER_SUFFIX = "Mapper"
64 | val MANAGER_SUFFIX = "Manager"
65 | val MANAGER_IMPL_SUFFIX = "ManagerImpl"
66 | val MAPPER_EXT_SUFFIX = "ExtMapper"
67 | val DO_SUFFIX = "DO"
68 |
69 | val COMMON_COLUMN_STR = "gmt_create,gmt_modified,"
70 |
71 | // jdbc result set metadata collumn name
72 | val RSMD_COLUMN_NAME = "rsmdColumnName"
73 | val RSMD_COLUMN_CLASS_NAME = "columnClassName"
74 | val RSMD_COLUMN_TYPE_NAME = "columnTypeName"
75 | val RSMD_COLUMN_PRECISION = "Precision"
76 | val RSMD_COLUMN_SCALE = "Scale"
77 | val RSMD_COLUMN_PRIMARY_KEY = "PrimaryKey"
78 | val RSMD_COLUMN_REMARKS = "Remarks"
79 |
80 | val COLUMN_NAME = "COLUMN_NAME"
81 |
82 | // velocity param
83 | val VP_LIST = "list"
84 | val VP_QUERY_PREFIX = "queryPrefix"
85 | val VP_DO_SUFFIX = "doSuffix"
86 | val VP_MAPPER_SUFFIX = "mapperSuffix"
87 | val VP_MANAGER_SUFFIX = "managerSuffix"
88 |
89 | val VP_MANAGER_IMPL_SUFFIX = "managerImplSuffix"
90 | val VP_MAPPER_EXT_SUFFIX = "extMapperSuffix"
91 | val VP_CLASS_NAME = "className"
92 | val VP_MAPPER_PROPERTY_NAME = "mapperPropertyName"
93 | val VP_DAO_PACKAGE = "daoPackage"
94 | val VP_DO_PACKAGE = "doPackage"
95 | val VP_QUERY_PACKAGE = "queryPackage"
96 | val VP_MAPPER_PACKAGE = "mapperPackage"
97 | val VP_MANAGER_PACKAGE = "managerPackage"
98 | val VP_MANAGER_IMPL_PACKAGE = "managerImplPackage"
99 | val VP_MAPPER_EXT_PACKAGE = "mapperExtPackage"
100 | val VP_JAVA_TYPE = "javaType"
101 | val VP_PROP_NAME = "propName"
102 | val VP_GET_METHOD = "getMethod"
103 |
104 | val VP_SET_METHOD = "setMethod"
105 | val VP_COLUMN_NAME = "columnName"
106 | val VP_COLUMN_REMARKS = "columnRemarks"
107 | val VP_TABLE_NAME = "tableName"
108 | val VP_JDBC_TYPE = "jdbcType"
109 | val VP_COLS = "cols"
110 | val VP_COLS_WITHOUT_COMMON_COLUMNS = "colsWithoutCommColumns"
111 | val VP_SERIAL_VERSION_UID = "serialVersionUID"
112 |
113 | val VP_SERIAL_VERSION_UID2 = "serialVersionUID2"
114 | val VP_PRIMARY_KEY = "primaryKey"
115 | val VP_PROP_PRIMARY_KEY = "propPrimaryKey"
116 |
117 | //分库分表 表后缀regex
118 | val SHARDING_SUFFIX_REG = "_[\\d]{4}"
119 |
120 |
121 | fun initConfig(project: Project?): Boolean {
122 | val onNullReturn = {
123 | clearConfig()
124 | sConfigured = false
125 | }
126 |
127 | PlatformUtil.getData(MyConstant.DAO_MODULE, "").let {
128 | if (it.isNullOrBlank()) {
129 | sDaoModule = ""
130 | onNullReturn()
131 | } else {
132 | sDaoModule = it
133 | }
134 | }
135 | PlatformUtil.getData(MyConstant.DAO_PACKAGE, "").let {
136 | if (it.isNullOrBlank()) {
137 | sDaoPackage = ""
138 | onNullReturn()
139 | } else {
140 | sDaoPackage = it
141 | }
142 | }
143 | PlatformUtil.getData(MyConstant.MANAGER_MODULE, "").let {
144 | if (it.isNullOrBlank()) {
145 | sManagerModule = ""
146 | onNullReturn()
147 | } else {
148 | sManagerModule = it
149 | }
150 | }
151 | PlatformUtil.getData(MyConstant.MANAGER_PACKAGE, "").let {
152 | if (it.isNullOrBlank()) {
153 | sManagerPackage = ""
154 | onNullReturn()
155 | } else {
156 | sManagerPackage = it
157 | }
158 | }
159 | PlatformUtil.getData(MyConstant.TABLE_PREFIX, "").let {
160 | if (it.isNullOrBlank()) {
161 | sTablePrefix = ""
162 | onNullReturn()
163 | } else {
164 | sTablePrefix = it
165 | }
166 | }
167 |
168 |
169 | PlatformUtil.getData(MyConstant.MHC_STAFF, false).let {
170 | sIsMHCStaff = it
171 | }
172 | setConfig(sDaoModule,
173 | sDaoPackage,
174 | sManagerModule,
175 | sManagerPackage,
176 | sTablePrefix,
177 | sIsMHCStaff,
178 | project)
179 | return sConfigured
180 | }
181 |
182 | fun setConfig(
183 | daoModule: String,
184 | daoPackage: String,
185 | managerModule: String,
186 | managerPackage: String,
187 | tablePrefix:String,
188 | isMHCStaff: Boolean,
189 | project: Project?
190 | ) {
191 | if (null == project) {
192 | return
193 | }
194 | sIsMHCStaff = isMHCStaff
195 | sDaoModule = daoModule
196 | sDaoPackage = daoPackage
197 | sManagerModule = managerModule
198 | sManagerPackage = managerPackage
199 | sTablePrefix = tablePrefix
200 | sManagerImplPackage = sManagerPackage + ".impl"
201 | sDaoPackageDir = project.baseDir.path + File.separator + sDaoModule
202 | var managerPath = project.baseDir.path + File.separator + sManagerModule
203 | //资源文件路径
204 | sMapperXmlDir = sDaoPackageDir + "/src/main/resources/mapper/"
205 | sMapperExtXmlDir = sDaoPackageDir + "/src/main/resources/mapper/ext/"
206 | //代码文件路径
207 | sBaseDir = sDaoPackageDir + "/src/main/java/" + getFilePathFromPackage(sDaoPackage)
208 | sDoJavaDir = sDaoPackageDir + "/src/main/java/" + getFilePathFromPackage(sDaoPackage) + "/model"
209 | sQueryJavaDir = sDaoPackageDir + "/src/main/java/" + getFilePathFromPackage(sDaoPackage) + "/query"
210 | sMapperJavaDir = sDaoPackageDir + "/src/main/java/" + getFilePathFromPackage(sDaoPackage) + "/mapper"
211 | sMapperExtJavaDir = sDaoPackageDir + "/src/main/java/" + getFilePathFromPackage(sDaoPackage) + "/mapper/ext"
212 | sManagerJavaDir = managerPath + "/src/main/java/" + getFilePathFromPackage(sManagerPackage) + "/manager"
213 | sManagerImplJavaDir = sManagerJavaDir + "/impl"
214 | //
215 |
216 | sConfigured = true
217 | }
218 |
219 | fun getFilePathFromPackage(packageName: String): String {
220 | return packageName.replace(".", "/")
221 | }
222 |
223 | fun clearConfig() {
224 | sIsMHCStaff = false
225 | sDaoModule = ""
226 | sDaoPackage = ""
227 | sManagerModule = ""
228 | sManagerPackage = ""
229 | sManagerImplPackage = ""
230 | sTablePrefix = ""
231 | sMapperXmlDir = ""
232 | sMapperExtXmlDir = ""
233 | sDoJavaDir = ""
234 | sQueryJavaDir = ""
235 | sMapperJavaDir = ""
236 | sManagerJavaDir = ""
237 | sManagerImplJavaDir = ""
238 | sMapperExtJavaDir = ""
239 | sConfigured = false
240 | }
241 |
242 | }
243 | }
--------------------------------------------------------------------------------
/src/main/java/com/yangyang/mybatis/generator/utils/mybatis/MybatisCore.kt:
--------------------------------------------------------------------------------
1 | package main.java.com.yangyang.mybatis.generator.utils.mybatis
2 |
3 | import com.intellij.openapi.project.Project
4 | import main.java.com.yangyang.mybatis.generator.MyConstant
5 | import main.java.com.yangyang.mybatis.generator.utils.NotificationUtil
6 | import main.java.com.yangyang.mybatis.generator.utils.PlatformUtil
7 | import org.apache.commons.lang3.StringUtils
8 | import org.apache.commons.lang3.math.NumberUtils
9 | import org.apache.velocity.VelocityContext
10 | import org.apache.velocity.app.Velocity
11 | import java.io.*
12 | import java.nio.file.Files
13 | import java.nio.file.Paths
14 | import java.sql.*
15 | import java.util.*
16 | import java.util.regex.Pattern
17 | import kotlin.collections.ArrayList
18 |
19 |
20 | /**
21 | * 类简介：mybatis 生成文件的核心工具类
22 | * 作者： yang
23 | * 时间： 17/7/24
24 | * 邮箱： yangyang1@maihaoche.com
25 | */
26 | class MybatisCore {
27 |
28 |
29 | companion object {
30 |
31 | /**
32 | * 批量生成java数据对象类文件和sqlmap文件
33 |
34 | * @param tables 表 多个表用逗号分隔 table1,table2,table3
35 | * *
36 | * @throws Exception 抛出异常
37 | */
38 | fun batchGen(tables: List, project: Project): Exception? {
39 | if (!MyBatisGenConst.sConfigured) {
40 | return Exception("还没有配置参数，请先配置")
41 | }
42 | try {
43 | for (table in tables) {
44 | gen(table.trim { it <= ' ' }, project)
45 | }
46 | } catch (e: Exception) {
47 | return e
48 | }
49 | return null
50 | }
51 |
52 |
53 | /**
54 | * 根据表名获取字段信息
55 |
56 | * @param cn 数据库连接
57 | * *
58 | * @param table 数据库表名
59 | * *
60 | * @return 列infoList
61 | * *
62 | * @throws Exception 抛出错误
63 | */
64 | @Throws(Exception::class)
65 | private fun getColInfoList(cn: Connection, table: String): ArrayList> {
66 | val sql = "select * from $table where 1>2"
67 | var stmt: Statement? = null
68 | var rs: ResultSet? = null
69 | try {
70 | //获取主键串
71 | val dbmd = cn.metaData
72 | val primaryKeys = dbmd.getPrimaryKeys(null, null, table)
73 | val pks = getstrPimaryKeys(primaryKeys)
74 |
75 | stmt = cn.createStatement()
76 | rs = stmt!!.executeQuery(sql)
77 | // 获取结果集元数据信息
78 | val rsmd = rs!!.metaData
79 |
80 | val num = rsmd.columnCount
81 | var map: MutableMap
82 | val list = ArrayList>()
83 | for (i in 1..num) {
84 | map = HashMap()
85 | map.put(MyBatisGenConst.RSMD_COLUMN_NAME, rsmd.getColumnName(i))
86 | map.put(MyBatisGenConst.RSMD_COLUMN_CLASS_NAME, rsmd.getColumnClassName(i))
87 | map.put(MyBatisGenConst.RSMD_COLUMN_TYPE_NAME, rsmd.getColumnTypeName(i))
88 | map.put(MyBatisGenConst.RSMD_COLUMN_PRECISION, rsmd.getPrecision(i).toString() + "")
89 | map.put(MyBatisGenConst.RSMD_COLUMN_SCALE, rsmd.getScale(i).toString() + "")
90 | val remarks = getColRemarks(cn, table, rsmd.getColumnName(i))
91 | map.put(MyBatisGenConst.RSMD_COLUMN_REMARKS, remarks)
92 | list.add(map)
93 | }
94 | //主键串放入list
95 | map = HashMap()
96 | map.put(MyBatisGenConst.RSMD_COLUMN_PRIMARY_KEY, pks)
97 | list.add(map)
98 | return list
99 | } catch (e: Exception) {
100 | throw Exception(e.toString() + ",table=" + table, e)
101 | } finally {
102 | try {
103 | if (stmt != null) {
104 | stmt.close()
105 | }
106 | } catch (e2: Exception) {
107 | e2.printStackTrace()
108 | }
109 |
110 | try {
111 | if (rs != null) {
112 | rs.close()
113 | }
114 | } catch (e2: Exception) {
115 | e2.printStackTrace()
116 | }
117 |
118 | }
119 | }
120 |
121 | /**
122 | * 根据表列名获取字段注释
123 |
124 | * @param cn 数据库连接
125 | * *
126 | * @param table 数据库表名
127 | * *
128 | * @return 字段注释
129 | * *
130 | * @throws Exception 抛出异常
131 | */
132 | @Throws(Exception::class)
133 | private fun getColRemarks(cn: Connection, table: String, columnName: String): String {
134 | //获取主键串
135 | val dbmd = cn.metaData
136 | val resultSet = dbmd.getTables(null, "%", "%", arrayOf("TABLE"))
137 | while (resultSet.next()) {
138 | val tableName = resultSet.getString("TABLE_NAME")
139 | if (table == tableName) {
140 | val rs = dbmd.getColumns(null, "%", tableName, "%")
141 | while (rs.next()) {
142 | if (rs.getString("COLUMN_NAME") == columnName) {
143 | return rs.getString("REMARKS")
144 | }
145 | }
146 | }
147 | }
148 | return ""
149 | }
150 |
151 | /**
152 | * 获取列信息
153 |
154 | * @param table 表名
155 | * *
156 | * @return 列信息
157 | * *
158 | * @throws Exception 抛出错误
159 | */
160 | @Throws(Exception::class)
161 | private fun getColInfoList(table: String): ArrayList> {
162 | val cn = getConnection()
163 | try {
164 | return getColInfoList(cn, table)
165 | } finally {
166 | try {
167 | cn.close()
168 | } catch (e2: Exception) {
169 | e2.printStackTrace()
170 | }
171 |
172 | }
173 | }
174 |
175 | /**
176 | * 获取参数列表
177 |
178 | * @param colInfoList 列信息列表
179 | * *
180 | * @return 参数列表
181 | * *
182 | * @throws Exception 抛出异常
183 | */
184 | @Throws(Exception::class)
185 | private fun makeParamList(colInfoList: List>): ArrayList> {
186 | val list = ArrayList>()
187 | var map: Map
188 | var mapNew: MutableMap
189 | for (aColInfoList in colInfoList) {
190 | map = aColInfoList
191 | mapNew = HashMap()
192 | val columnName = map[MyBatisGenConst.RSMD_COLUMN_NAME]
193 | val columnClassName = map[MyBatisGenConst.RSMD_COLUMN_CLASS_NAME]
194 | val columnTypeName = map[MyBatisGenConst.RSMD_COLUMN_TYPE_NAME]
195 | val scaleStr = map[MyBatisGenConst.RSMD_COLUMN_SCALE]
196 | val scale = NumberUtils.toInt(scaleStr)
197 | val precisionStr = map[MyBatisGenConst.RSMD_COLUMN_PRECISION]
198 | val remarks = map[MyBatisGenConst.RSMD_COLUMN_REMARKS]
199 | val precision = NumberUtils.toInt(precisionStr)
200 | val javaType = getJavaType(columnClassName, columnTypeName, scale, precision)
201 | val jdbcType = getJdbcType(columnClassName, columnTypeName)
202 | val propName = getPropName(columnName)
203 | val setMethod = getSetMethod(propName)
204 | val getMethod = getGetMethod(propName)
205 | mapNew.put(MyBatisGenConst.VP_COLUMN_NAME, columnName?.toLowerCase())
206 | mapNew.put(MyBatisGenConst.VP_COLUMN_REMARKS, remarks)
207 | mapNew.put(MyBatisGenConst.VP_PROP_NAME, propName)
208 | mapNew.put(MyBatisGenConst.VP_JAVA_TYPE, javaType)
209 | mapNew.put(MyBatisGenConst.VP_JDBC_TYPE, jdbcType)
210 | mapNew.put(MyBatisGenConst.VP_GET_METHOD, getMethod)
211 | mapNew.put(MyBatisGenConst.VP_SET_METHOD, setMethod)
212 | list.add(mapNew)
213 | }
214 | return list
215 | }
216 |
217 | /**
218 | * 获取字段的java类型
219 |
220 | * @param columnClassName 字段类名
221 | * *
222 | * @param columnTypeName 字段类型名称
223 | * *
224 | * @param scale 精度 小数位数
225 | * *
226 | * @param precision 精度
227 | * *
228 | * @return java类型
229 | */
230 | private fun getJavaType(columnClassName: String?, columnTypeName: String?, scale: Int, precision: Int): String {
231 | if (columnClassName == null || columnTypeName == null) return ""
232 |
233 | if (columnClassName == "java.sql.Timestamp") {
234 | return "LocalDateTime"
235 | }
236 | if (columnClassName == "java.lang.String") {
237 | return "String"
238 | }
239 | if (columnTypeName == "DECIMAL" && scale < 1) {
240 | return "Long"
241 | }
242 | if (columnTypeName == "DECIMAL" && scale > 0) {
243 | return "java.math.BigDecimal"
244 | }
245 | if (columnTypeName.startsWith("BIGINT")) {
246 | return "Long"
247 | }
248 | if (columnTypeName.startsWith("INT")) {
249 | return "Integer"
250 | }
251 | if (columnTypeName.startsWith("TINYINT") && precision == 1) {
252 | return "Boolean"
253 | }
254 | if (columnTypeName.startsWith("TINYINT") && precision != 1) {
255 | return "Integer"
256 | }
257 | if (columnTypeName.startsWith("BIT") && precision > 1) {
258 | return "Integer"
259 | }
260 | if (columnTypeName.startsWith("BIT") && precision == 1) {
261 | return "Boolean"
262 | }
263 | if (columnTypeName.startsWith("SMALLINT")) {
264 | return "Integer"
265 | }
266 | return columnClassName
267 | }
268 |
269 | /**
270 | * 获取jdbc类型
271 |
272 | * @param columnClassName 字段类名
273 | * *
274 | * @param columnTypeName 字段类型名称
275 | * *
276 | * @return jdbc类型
277 | */
278 | private fun getJdbcType(columnClassName: String?, columnTypeName: String?): String? {
279 | if (columnClassName == null || columnTypeName == null) return ""
280 | if (columnClassName == "java.lang.String") {
281 | return "VARCHAR"
282 | }
283 | if (columnClassName.startsWith("java.sql.")) {
284 | return "TIMESTAMP"
285 | }
286 | if (columnTypeName.startsWith("NUMBER")) {
287 | return "DECIMAL"
288 | }
289 | if (columnTypeName.startsWith("INT")) {
290 | return "INTEGER"
291 | }
292 | return columnTypeName
293 | }
294 |
295 | /**
296 | * 根据表名获取java类型
297 |
298 | * @param tableName 表名
299 | * *
300 | * @return java类型
301 | */
302 | private fun getClassName(tableName: String): String {
303 | var t = tableName.toLowerCase()
304 | t = t.replace(MyBatisGenConst.sTablePrefix, "")
305 | val arr = t.split("_".toRegex()).dropLastWhile { it.isEmpty() }.toTypedArray()
306 | val s = StringBuilder()
307 | for (anArr in arr) {
308 | s.append(StringUtils.capitalize(anArr))
309 | }
310 | return s.toString()
311 | }
312 |
313 | /**
314 | * 根据字段名获取java数据对象属性名
315 |
316 | * @param columnName 字段名
317 | * *
318 | * @return 属性名
319 | */
320 | private fun getPropName(columnName: String?): String {
321 | if (columnName == null) return ""
322 | val t = columnName.toLowerCase()
323 | val arr = t.split("_".toRegex()).dropLastWhile { it.isEmpty() }.toTypedArray()
324 | val num = arr.size
325 | val s = StringBuilder()
326 | for (i in 0..num - 1) {
327 | if (i > 0) {
328 | s.append(StringUtils.capitalize(arr[i]))
329 | } else {
330 | s.append(arr[i])
331 | }
332 | }
333 | return s.toString()
334 | }
335 |
336 | private fun getSetMethod(propName: String): String {
337 | return "set" + StringUtils.capitalize(propName)
338 | }
339 |
340 | private fun getGetMethod(propName: String): String {
341 | return "get" + StringUtils.capitalize(propName)
342 | }
343 |
344 | private fun getColsStr(list: List>): String {
345 | val num = list.size
346 | var map: Map
347 | var colName: String?
348 | val stringBuilder = StringBuilder()
349 | for (i in 0..num - 1) {
350 | map = list[i]
351 | colName = map[MyBatisGenConst.VP_COLUMN_NAME]
352 | if (i > 0) {
353 | stringBuilder.append(",")
354 | }
355 | stringBuilder.append(colName)
356 | }
357 | return stringBuilder.toString()
358 | }
359 |
360 | @Throws(SQLException::class)
361 | private fun getstrPimaryKeys(primaryKeys: ResultSet): String {
362 | val sb = StringBuilder()
363 | val pks: String
364 | while (primaryKeys.next()) {
365 | sb.append(primaryKeys.getString(MyBatisGenConst.COLUMN_NAME)).append(",")
366 | }
367 | pks = sb.toString()
368 | return if (StringUtils.isBlank(pks)) "" else pks.substring(0, pks.length - 1)
369 | }
370 |
371 | private var sVelocityInited = false;
372 | /**
373 | * Velocity 模板合并
374 |
375 | * @param template 模板字符串 如 hello,${name}
376 | * *
377 | * @param paramMap 参数
378 | * *
379 | * @return 合并结果
380 | * *
381 | * @throws Exception 抛出错误
382 | */
383 | @Throws(Exception::class)
384 | private fun merge(template: String, paramMap: Map): String {
385 | val vc = VelocityContext(paramMap)
386 | val writer = StringWriter()
387 | Velocity.evaluate(vc, writer, "mybatis_code_gen", template)
388 | return writer.buffer.toString()
389 | }
390 |
391 | /**
392 | * 获取sqlmap 参数列表 去掉 主键 GMT_CREATE GMT_MODIFIED 字段
393 |
394 | * @param paramList 参数列表 Map
395 | * *
396 | * @param pks 主键
397 | * *
398 | * @return sql 参数列表
399 | * *
400 | * @throws Exception 抛出错误
401 | */
402 | @Throws(Exception::class)
403 | private fun getSqlmapParamList(paramList: ArrayList>, pks: String): List> {
404 | val list = ArrayList>()
405 | var tmp: Map
406 | var map: MutableMap
407 | for (aParamList in paramList) {
408 | tmp = aParamList
409 | val columnName = tmp[MyBatisGenConst.VP_COLUMN_NAME]
410 | if (columnName.equals(pks, ignoreCase = true)) {
411 | continue
412 | }
413 | if (columnName.equals("GMT_CREATE", ignoreCase = true)) {
414 | continue
415 | }
416 | if (columnName.equals("GMT_MODIFIED", ignoreCase = true)) {
417 | continue
418 | }
419 | map = HashMap()
420 | map.putAll(tmp)
421 | list.add(map)
422 | }
423 | return list
424 | }
425 |
426 | /**
427 | * 根据表名生成java数据对象类文件和sqlmap文件
428 |
429 | * @param table 表名
430 | * *
431 | * @throws Exception 抛出错误
432 | */
433 | @Throws(Exception::class)
434 | private fun gen(table: String, project: Project) {
435 | var tableLocal = table
436 | val colInfoList = getColInfoList(tableLocal)
437 | var pks: String? = ""
438 | if (colInfoList.size > 0) {
439 | pks = colInfoList.removeAt(colInfoList.size - 1)[MyBatisGenConst.RSMD_COLUMN_PRIMARY_KEY]
440 | }
441 | val paramList = makeParamList(colInfoList)
442 | val isSharding = Pattern.compile(MyBatisGenConst.SHARDING_SUFFIX_REG).matcher(tableLocal).find()
443 |
444 | if (isSharding) {
445 | // 去掉分库分表后面的表后缀，如_0001
446 | tableLocal = tableLocal.replace(MyBatisGenConst.SHARDING_SUFFIX_REG.toRegex(), "")
447 | }
448 | val className = getClassName(tableLocal)
449 |
450 |
451 | val param = HashMap()
452 |
453 | var fileGenerate = { packagePath: String, fileName: String, templatePath: String ->
454 | File(packagePath).mkdirs()
455 | if (!File(packagePath).exists()) {
456 | NotificationUtil.info(String.format("目录 %s 创建失败", packagePath), project)
457 | } else {
458 | val filePath = packagePath + File.separator + fileName
459 | val resultFile = File(filePath)
460 | if (!resultFile.exists()) {
461 | if (resultFile.createNewFile()) {
462 | val resultStr = merge(readFileToString(templatePath), param)
463 | writeStringToFile(resultFile, resultStr, project)
464 | } else {
465 | NotificationUtil.error(String.format("文件 %s 创建失败", filePath))
466 | }
467 | } else {
468 | NotificationUtil.error(String.format("文件 %s 已经存在,不替换旧文件", filePath))
469 | }
470 | }
471 | }
472 | if (MyBatisGenConst.sIsMHCStaff) {
473 | param.put(MyBatisGenConst.VP_DAO_PACKAGE, "com.subaru.common.query")
474 | } else {
475 | param.put(MyBatisGenConst.VP_DAO_PACKAGE, MyBatisGenConst.sDaoPackage)
476 | }
477 | param.put(MyBatisGenConst.VP_DO_PACKAGE, MyBatisGenConst.sDaoPackage + ".model")
478 | param.put(MyBatisGenConst.VP_QUERY_PACKAGE, MyBatisGenConst.sDaoPackage + ".query")
479 | param.put(MyBatisGenConst.VP_MAPPER_PACKAGE, MyBatisGenConst.sDaoPackage + ".mapper")
480 | param.put(MyBatisGenConst.VP_MAPPER_EXT_PACKAGE, MyBatisGenConst.sDaoPackage + ".mapper.ext")
481 | param.put(MyBatisGenConst.VP_MANAGER_PACKAGE, MyBatisGenConst.sManagerPackage + ".manager")
482 | param.put(MyBatisGenConst.VP_MANAGER_IMPL_PACKAGE, MyBatisGenConst.sManagerImplPackage)
483 | param.put(MyBatisGenConst.VP_CLASS_NAME, className)
484 | param.put(MyBatisGenConst.VP_MAPPER_PROPERTY_NAME, className.substring(0, 1).toLowerCase() + className.substring(1) + MyBatisGenConst.MAPPER_EXT_SUFFIX)
485 | param.put(MyBatisGenConst.VP_LIST, paramList)
486 | param.put(MyBatisGenConst.VP_QUERY_PREFIX, MyBatisGenConst.QUERY_PREFIX)
487 | param.put(MyBatisGenConst.VP_DO_SUFFIX, MyBatisGenConst.DO_SUFFIX)
488 | param.put(MyBatisGenConst.VP_MAPPER_SUFFIX, MyBatisGenConst.MAPPER_SUFFIX)
489 | param.put(MyBatisGenConst.VP_MANAGER_SUFFIX, MyBatisGenConst.MANAGER_SUFFIX)
490 | param.put(MyBatisGenConst.VP_MANAGER_IMPL_SUFFIX, MyBatisGenConst.MANAGER_IMPL_SUFFIX)
491 | param.put(MyBatisGenConst.VP_MAPPER_EXT_SUFFIX, MyBatisGenConst.MAPPER_EXT_SUFFIX)
492 |
493 | var vpTableName = tableLocal.toLowerCase()
494 | if (isSharding) {
495 | vpTableName += "_\$tabNum$"
496 | }
497 |
498 | param.put(MyBatisGenConst.VP_TABLE_NAME, vpTableName)
499 | param.put(MyBatisGenConst.VP_SERIAL_VERSION_UID, "" + (Math.random() * 1000000000000000000L).toLong())
500 |
501 | param.put(MyBatisGenConst.VP_SERIAL_VERSION_UID2, "" + (Math.random() * 1000000000000000000L).toLong())
502 |
503 | //先生成do的结果，再去掉主键
504 | //DO
505 | fileGenerate(MyBatisGenConst.sDoJavaDir, className + MyBatisGenConst.DO_SUFFIX + ".java", MyBatisGenConst.DO_TEMPLATE)
506 |
507 | val sqlmapParamList = getSqlmapParamList(paramList, pks.toString())
508 | param.put(MyBatisGenConst.VP_LIST, sqlmapParamList)
509 |
510 | val colsWithoutCommColumns = getColsStr(sqlmapParamList)
511 | param.put(MyBatisGenConst.VP_COLS_WITHOUT_COMMON_COLUMNS, colsWithoutCommColumns)
512 | val cols = pks + "," + MyBatisGenConst.COMMON_COLUMN_STR + colsWithoutCommColumns
513 | param.put(MyBatisGenConst.VP_COLS, cols)
514 | param.put(MyBatisGenConst.VP_PRIMARY_KEY, pks.toString())
515 | param.put(MyBatisGenConst.VP_PROP_PRIMARY_KEY, getPropName(pks))
516 |
517 |
518 | //如果不是mhc内部使用的代码，还需要额外生成几个基础类的文件
519 | if (!MyBatisGenConst.sIsMHCStaff) {
520 | //PageResult
521 | fileGenerate(MyBatisGenConst.sBaseDir, "PageResult.java", MyBatisGenConst.PAGERESULT_TEMPLATE)
522 | //Criteria
523 | fileGenerate(MyBatisGenConst.sBaseDir, "Criterion.java", MyBatisGenConst.CRITERIA_TEMPLATE)
524 | //BaseCriteria
525 | fileGenerate(MyBatisGenConst.sBaseDir, "BaseCriteria.java", MyBatisGenConst.BASE_CRITERIA_TEMPLATE)
526 | //BaseQuery
527 | fileGenerate(MyBatisGenConst.sBaseDir, "BaseQuery.java", MyBatisGenConst.BASE_QUERY_TEMPLATE)
528 | }
529 |
530 | //Query
531 | fileGenerate(MyBatisGenConst.sQueryJavaDir, className + MyBatisGenConst.QUERY_PREFIX + ".java", MyBatisGenConst.QUERY_TEMPLATE)
532 | //Mapper
533 | fileGenerate(MyBatisGenConst.sMapperJavaDir, className + MyBatisGenConst.MAPPER_SUFFIX + ".java", MyBatisGenConst.MAPPER_TEMPLATE)
534 | //MapperExt
535 | fileGenerate(MyBatisGenConst.sMapperExtJavaDir, className + MyBatisGenConst.MAPPER_EXT_SUFFIX + ".java", MyBatisGenConst.MAPPER_EXT_TEMPLATE)
536 | //Manager
537 | fileGenerate(MyBatisGenConst.sManagerJavaDir, className + MyBatisGenConst.MANAGER_SUFFIX + ".java", MyBatisGenConst.MANAGER_TEMPLATE)
538 | //ManagerImple
539 | fileGenerate(MyBatisGenConst.sManagerImplJavaDir, className + MyBatisGenConst.MANAGER_IMPL_SUFFIX + ".java", MyBatisGenConst.MANAGER_IMPL_TEMPLATE)
540 | //------------------------xml----------------------
541 | //map
542 | fileGenerate(MyBatisGenConst.sMapperXmlDir, className + MyBatisGenConst.MAPPER_SUFFIX + ".xml", MyBatisGenConst.SQLMAP_TEMPLATE)
543 | //map_ext
544 | fileGenerate(MyBatisGenConst.sMapperExtXmlDir, className + MyBatisGenConst.MAPPER_EXT_SUFFIX + ".xml", MyBatisGenConst.SQLMAP_EXT_TEMPLATE)
545 |
546 | }
547 |
548 | @Throws(Exception::class)
549 | private fun createAndWriteFile(managerResult: String, managerOutFilePath: String, managerOutFile: File, project: Project) {
550 | val success: Boolean
551 | success = managerOutFile.createNewFile()
552 | if (!success) {
553 | NotificationUtil.info(String.format("创建文件 %s 失败", managerOutFilePath), project)
554 | } else {
555 | writeStringToFile(managerOutFile, managerResult, project)
556 | }
557 | }
558 |
559 | @Throws(Exception::class)
560 | private fun writeStringToFile(file: File, data: String, project: Project) {
561 | val path = Paths.get(file.absolutePath)
562 | var writer: BufferedWriter? = null
563 | try {
564 | writer = Files.newBufferedWriter(path)
565 | writer.write(data)
566 | NotificationUtil.info("文件:" + path + "生成成功", project)
567 | } catch (e: Exception) {
568 | throw e
569 | } finally {
570 | writer?.close()
571 | }
572 | }
573 |
574 | @Throws(Exception::class)
575 | private fun readFileToString(fileRelativePath: String): String {
576 | var bufferReader = BufferedReader(InputStreamReader(javaClass.getResourceAsStream(fileRelativePath)))
577 | var fileStr = ""
578 | var lineStr: String?
579 | while (true) {
580 | lineStr = bufferReader.readLine()
581 | if (lineStr == null) {
582 | break
583 | }
584 | fileStr += lineStr + "\n"
585 | }
586 | return fileStr
587 | }
588 |
589 | /**
590 | * 获取数据库连接
591 |
592 | * @return 数据库连接
593 | * *
594 | * @throws ClassNotFoundException 类找不到异常
595 | * *
596 | * @throws SQLException SQL 异常
597 | * *
598 | * @throws IOException IO 异常
599 | */
600 | @Throws(ClassNotFoundException::class, SQLException::class, IOException::class)
601 | private fun getConnection(): Connection {
602 | Class.forName("com.mysql.jdbc.Driver")
603 | val url = PlatformUtil.getData(MyConstant.DB_URL, "")
604 | val user = PlatformUtil.getData(MyConstant.DB_USER, "")
605 | val psw = PlatformUtil.getData(MyConstant.DB_PW, "")
606 | var connection = DriverManager.getConnection(url, user, psw)
607 | if (connection.isValid(0)) {
608 | return connection
609 | } else {
610 | throw SQLException("数据库链接不上")
611 | }
612 | }
613 |
614 |
615 | }
616 | }
--------------------------------------------------------------------------------
/src/main/resources/META-INF/plugin.xml:
--------------------------------------------------------------------------------
1 |
2 | com.maihaoche.mclaren.mybatisgenerator
3 | MybatisGenerator
4 | 2.1.1
5 | MaiHaoChe.
6 |
7 |
12 |
13 |
15 | The version fix some serious bugs,i.e, no linefeed and blocking the ide.
16 | In addition, this version add a function to choose which module to put the manger codes,
17 | which is necessary for many projects.
18 | By now ,this version support only mysql and the target mybatis version is 1.2.0. That is
19 |
20 | org.mybatis.spring.boot
21 | mybatis-spring-boot-starter
22 | 1.2.0
23 | .
24 | Any advice:
25 | Git:https://github.com/maihaoche/MybatisGeneratorPlugin.
26 | ]]>
27 |
28 |
29 |
30 |
31 |
32 |
34 |
35 | com.intellij.modules.lang
36 |
37 |
38 |
39 |
40 |
41 |
42 |
43 |
46 |
47 |
49 |
52 |
54 |
55 |
56 |
57 |
--------------------------------------------------------------------------------
/src/main/resources/icons/ic_mybatis.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/maihaoche/mybatis-generator-plugin/f9255ade46443574766a6a212f7692f100cb8f48/src/main/resources/icons/ic_mybatis.png
--------------------------------------------------------------------------------
/src/main/resources/icons/ic_mybatis@2x.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/maihaoche/mybatis-generator-plugin/f9255ade46443574766a6a212f7692f100cb8f48/src/main/resources/icons/ic_mybatis@2x.png
--------------------------------------------------------------------------------
/src/main/resources/template/base/baseCriteria.txt:
--------------------------------------------------------------------------------
1 | package ${daoPackage};
2 |
3 | import java.io.Serializable;
4 | import java.util.ArrayList;
5 | import java.util.List;
6 |
7 | public class BaseCriteria implements Serializable {
8 | private static final long serialVersionUID = 1L;
9 | protected List criteria = new ArrayList();
10 |
11 | protected BaseCriteria() {
12 | }
13 |
14 | public boolean isValid() {
15 | return this.criteria.size() > 0;
16 | }
17 |
18 | public List getAllCriteria() {
19 | return this.criteria;
20 | }
21 |
22 | public List getCriteria() {
23 | return this.criteria;
24 | }
25 |
26 | protected void addCriterion(String condition) {
27 | if (condition == null) {
28 | throw new RuntimeException("Value for condition cannot be null");
29 | } else {
30 | this.criteria.add(new Criterion(condition));
31 | }
32 | }
33 |
34 | protected void addCriterion(String condition, Object value, String property) {
35 | if (value == null) {
36 | throw new RuntimeException("Value for " + property + " cannot be null");
37 | } else {
38 | this.criteria.add(new Criterion(condition, value));
39 | }
40 | }
41 |
42 | protected void addCriterion(String condition, Object value1, Object value2, String property) {
43 | if (value1 != null && value2 != null) {
44 | this.criteria.add(new Criterion(condition, value1, value2));
45 | } else {
46 | throw new RuntimeException("Between values for " + property + " cannot be null");
47 | }
48 | }
49 | }
--------------------------------------------------------------------------------
/src/main/resources/template/base/baseQuery.txt:
--------------------------------------------------------------------------------
1 | package ${daoPackage};
2 |
3 | import java.io.Serializable;
4 | import java.util.ArrayList;
5 | import java.util.List;
6 |
7 | public class BaseQuery implements Serializable {
8 | private static final long serialVersionUID = 1L;
9 | public static int DEFAULT_PAGE_SIZE = 20;
10 | public static int MAX_PAGE_SIZE = 100;
11 | public static int DEFAULT_PAGE = 1;
12 | protected String orderByClause;
13 | protected boolean distinct;
14 | protected List oredCriteria = new ArrayList();
15 | protected Integer pageOffset;
16 | protected Integer pageNo;
17 | protected Integer pageSize;
18 |
19 | public BaseQuery() {
20 | }
21 |
22 | public void setOrderByClause(String orderByClause) {
23 | this.orderByClause = orderByClause;
24 | }
25 |
26 | public String getOrderByClause() {
27 | return this.orderByClause;
28 | }
29 |
30 | public void setDistinct(boolean distinct) {
31 | this.distinct = distinct;
32 | }
33 |
34 | public boolean isDistinct() {
35 | return this.distinct;
36 | }
37 |
38 | public List getOredCriteria() {
39 | return this.oredCriteria;
40 | }
41 |
42 | public void or(BaseCriteria criteria) {
43 | this.oredCriteria.add(criteria);
44 | }
45 |
46 | public void clear() {
47 | this.oredCriteria.clear();
48 | this.orderByClause = null;
49 | this.distinct = false;
50 | }
51 |
52 | public void clearPage() {
53 | this.pageSize = null;
54 | this.pageNo = null;
55 | this.pageOffset = null;
56 | }
57 |
58 | protected void calPageOffset() {
59 | if (null != this.pageSize && null != this.pageNo) {
60 | this.pageOffset = (this.pageNo.intValue() - 1) * this.pageSize.intValue();
61 | } else {
62 | this.pageOffset = null;
63 | }
64 |
65 | }
66 |
67 | public boolean isValid() {
68 | for(int i = 0; i < this.oredCriteria.size(); ++i) {
69 | if (((BaseCriteria)this.oredCriteria.get(i)).isValid()) {
70 | return true;
71 | }
72 | }
73 |
74 | return false;
75 | }
76 |
77 | public void setPageOffset(Integer pageOffset) {
78 | this.pageOffset = pageOffset;
79 | }
80 |
81 | public Integer getPageOffset() {
82 | return this.pageOffset;
83 | }
84 |
85 | public void setPageSize(Integer pageSize) {
86 | if (pageSize != null && pageSize.intValue() > 0) {
87 | if (pageSize.intValue() > MAX_PAGE_SIZE) {
88 | this.pageSize = MAX_PAGE_SIZE;
89 | } else {
90 | this.pageSize = pageSize;
91 | }
92 | } else {
93 | this.clearPage();
94 | }
95 |
96 | this.calPageOffset();
97 | }
98 |
99 | public Integer getPageSize() {
100 | return this.pageSize;
101 | }
102 |
103 | public Integer getPageNo() {
104 | return this.pageNo;
105 | }
106 |
107 | public void setPageNo(Integer pageNo) {
108 | if (pageNo != null && pageNo.intValue() >= DEFAULT_PAGE) {
109 | this.pageNo = pageNo;
110 | } else {
111 | this.clearPage();
112 | }
113 |
114 | this.calPageOffset();
115 | }
116 | }
--------------------------------------------------------------------------------
/src/main/resources/template/base/criteria.txt:
--------------------------------------------------------------------------------
1 | package ${daoPackage};
2 |
3 | import java.io.Serializable;
4 | import java.util.List;
5 |
6 | public class Criterion implements Serializable {
7 | private static final long serialVersionUID = 1L;
8 | protected String condition;
9 | protected Object value;
10 | protected Object secondValue;
11 | protected boolean noValue;
12 | protected boolean singleValue;
13 | protected boolean betweenValue;
14 | protected boolean listValue;
15 | protected String typeHandler;
16 |
17 | public String getCondition() {
18 | return this.condition;
19 | }
20 |
21 | public Object getValue() {
22 | return this.value;
23 | }
24 |
25 | public Object getSecondValue() {
26 | return this.secondValue;
27 | }
28 |
29 | public boolean isNoValue() {
30 | return this.noValue;
31 | }
32 |
33 | public boolean isSingleValue() {
34 | return this.singleValue;
35 | }
36 |
37 | public boolean isBetweenValue() {
38 | return this.betweenValue;
39 | }
40 |
41 | public boolean isListValue() {
42 | return this.listValue;
43 | }
44 |
45 | public String getTypeHandler() {
46 | return this.typeHandler;
47 | }
48 |
49 | public Criterion(String condition) {
50 | this.condition = condition;
51 | this.typeHandler = null;
52 | this.noValue = true;
53 | }
54 |
55 | public Criterion(String condition, Object value, String typeHandler) {
56 | this.condition = condition;
57 | this.value = value;
58 | this.typeHandler = typeHandler;
59 | if (value instanceof List) {
60 | this.listValue = true;
61 | } else {
62 | this.singleValue = true;
63 | }
64 |
65 | }
66 |
67 | public Criterion(String condition, Object value) {
68 | this(condition, value, (String)null);
69 | }
70 |
71 | public Criterion(String condition, Object value, Object secondValue, String typeHandler) {
72 | this.condition = condition;
73 | this.value = value;
74 | this.secondValue = secondValue;
75 | this.typeHandler = typeHandler;
76 | this.betweenValue = true;
77 | }
78 |
79 | public Criterion(String condition, Object value, Object secondValue) {
80 | this(condition, value, secondValue, (String)null);
81 | }
82 |
83 | public String toString() {
84 | if (this.isNoValue()) {
85 | return this.condition;
86 | } else if (this.isSingleValue()) {
87 | return this.condition + " " + this.getValue();
88 | } else if (this.isBetweenValue()) {
89 | return this.condition + " " + this.value + " and " + this.secondValue;
90 | } else {
91 | return this.isListValue() ? this.condition + " in " + this.value : "invalid status";
92 | }
93 | }
94 | }
95 |
--------------------------------------------------------------------------------
/src/main/resources/template/base/pageResult.txt:
--------------------------------------------------------------------------------
1 | package ${daoPackage};
2 |
3 | import java.io.Serializable;
4 | import java.util.List;
5 |
6 | public class PageResult implements Serializable {
7 | private static final long serialVersionUID = 1L;
8 | private List result;
9 | private int pageNo;
10 | private int totalCount;
11 | private int pageSize;
12 |
13 | public PageResult() {
14 | this.pageNo = BaseQuery.DEFAULT_PAGE;
15 | this.pageSize = BaseQuery.DEFAULT_PAGE_SIZE;
16 | }
17 |
18 | public List getResult() {
19 | return this.result;
20 | }
21 |
22 | public void setResult(List result) {
23 | this.result = result;
24 | }
25 |
26 | public int getPageNo() {
27 | return this.pageNo;
28 | }
29 |
30 | public void setPageNo(int pageNo) {
31 | this.pageNo = pageNo;
32 | }
33 |
34 | public int getTotalPage() {
35 | if (this.pageNo < 1) {
36 | this.pageNo = BaseQuery.DEFAULT_PAGE;
37 | }
38 |
39 | return this.totalCount / this.pageSize + (this.totalCount % this.pageSize == 0 ? 0 : 1);
40 | }
41 |
42 | public int getTotalCount() {
43 | return this.totalCount;
44 | }
45 |
46 | public void setTotalCount(int totalCount) {
47 | this.totalCount = totalCount;
48 | }
49 |
50 | public int getPageSize() {
51 | return this.pageSize;
52 | }
53 |
54 | public void setPageSize(int pageSize) {
55 | this.pageSize = pageSize;
56 | }
57 | }
--------------------------------------------------------------------------------
/src/main/resources/template/do.txt:
--------------------------------------------------------------------------------
1 | package ${doPackage};
2 |
3 | import java.io.Serializable;
4 | import java.time.LocalDateTime;
5 | import lombok.Data;
6 |
7 | @Data
8 | public class ${className}${doSuffix} implements Serializable {
9 |
10 | private static final long serialVersionUID = 1L;
11 |
12 | #foreach($item in $list)
13 | /**
14 | * ${item.columnRemarks}
15 | * ${tableName}.${item.columnName}
16 | */
17 | private ${item.javaType} ${item.propName};
18 |
19 | #end
20 | }
--------------------------------------------------------------------------------
/src/main/resources/template/manager.txt:
--------------------------------------------------------------------------------
1 | package ${managerPackage};
2 |
3 | import ${daoPackage}.PageResult;
4 | import ${doPackage}.${className}${doSuffix};
5 | import ${queryPackage}.${className}${queryPrefix};
6 |
7 | import java.util.List;
8 |
9 |
10 | /**
11 | * Manager for ${className}.
12 | */
13 |
14 | public interface ${className}${managerSuffix} {
15 | /**
16 | * query count by query condition.
17 | */
18 | int countBy${queryPrefix}(${className}${queryPrefix} ${queryPrefix.toLowerCase()});
19 |
20 | /**
21 | * delete by query condition.
22 | */
23 | int deleteBy${queryPrefix}(${className}${queryPrefix} ${queryPrefix.toLowerCase()});
24 |
25 | /**
26 | * delete by primary key.
27 | */
28 | int deleteByPrimaryKey(${className}${doSuffix} record);
29 |
30 | /**
31 | * insert selective.
32 | */
33 | long insertSelective(${className}${doSuffix} record);
34 |
35 | /**
36 | * select by query condition.
37 | */
38 | List<${className}${doSuffix}> selectBy${queryPrefix}(${className}${queryPrefix} ${queryPrefix.toLowerCase()});
39 |
40 |
41 | /**
42 | * select by query condition with page.
43 | */
44 | PageResult<${className}${doSuffix}> selectBy${queryPrefix}WithPage(${className}${queryPrefix} ${queryPrefix.toLowerCase()});
45 |
46 | /**
47 | * select by primary key.
48 | */
49 | ${className}${doSuffix} selectByPrimaryKey(Long id);
50 |
51 | /**
52 | * update by query condition selective.
53 | */
54 | int updateBy${queryPrefix}Selective( ${className}${doSuffix} record, ${className}${queryPrefix} ${queryPrefix.toLowerCase()});
55 |
56 | /**
57 | * update by query condition.
58 | */
59 | int updateBy${queryPrefix}(${className}${doSuffix} record, ${className}${queryPrefix} ${queryPrefix.toLowerCase()});
60 |
61 | /**
62 | * update by primary key selective.
63 | */
64 | int updateByPrimaryKeySelective(${className}${doSuffix} record);
65 | }
--------------------------------------------------------------------------------
/src/main/resources/template/managerImpl.txt:
--------------------------------------------------------------------------------
1 | package ${managerImplPackage};
2 |
3 | import ${daoPackage}.PageResult;
4 | import ${doPackage}.${className}${doSuffix};
5 | import ${queryPackage}.${className}${queryPrefix};
6 | import ${mapperExtPackage}.${className}${extMapperSuffix};
7 | import ${managerPackage}.${className}${managerSuffix};
8 |
9 | import org.slf4j.Logger;
10 | import org.slf4j.LoggerFactory;
11 |
12 | import org.springframework.beans.factory.annotation.Autowired;
13 | import org.springframework.stereotype.Component;
14 |
15 | import java.util.List;
16 |
17 |
18 | /**
19 | * Manager for ${className}.
20 | */
21 |
22 | @Component
23 |
24 | public class ${className}${managerImplSuffix} implements ${className}${managerSuffix}{
25 |
26 | #set($managerMapperName= ${mapperPropertyName})
27 |
28 | @Autowired
29 | protected ${className}${extMapperSuffix} ${managerMapperName};
30 | /**
31 | * query count by query condition.
32 | */
33 | @Override
34 | public int countBy${queryPrefix}(${className}${queryPrefix} ${queryPrefix.toLowerCase()}){
35 | return ${managerMapperName}.countBy${queryPrefix}(${queryPrefix.toLowerCase()});
36 | }
37 |
38 | /**
39 | * delete by query condition.
40 | */
41 | @Override
42 | public int deleteBy${queryPrefix}(${className}${queryPrefix} ${queryPrefix.toLowerCase()}){
43 | return ${managerMapperName}.deleteBy${queryPrefix}(${queryPrefix.toLowerCase()});
44 | }
45 |
46 | /**
47 | * delete by primary key.
48 | */
49 | @Override
50 | public int deleteByPrimaryKey(${className}${doSuffix} record){
51 | return ${managerMapperName}.deleteByPrimaryKey(record);
52 | }
53 |
54 | /**
55 | * insert selective.
56 | */
57 | @Override
58 | public long insertSelective(${className}${doSuffix} record){
59 | return ${managerMapperName}.insertSelective(record);
60 | }
61 |
62 | /**
63 | * select by query condition.
64 | */
65 | @Override
66 | public List<${className}${doSuffix}> selectBy${queryPrefix}(${className}${queryPrefix} ${queryPrefix.toLowerCase()}){
67 | return ${managerMapperName}.selectBy${queryPrefix}(${queryPrefix.toLowerCase()});
68 | }
69 |
70 | /**
71 | * select by query condition with page.
72 | */
73 | @Override
74 | public PageResult<${className}${doSuffix}> selectBy${queryPrefix}WithPage(${className}${queryPrefix} ${queryPrefix.toLowerCase()}) {
75 | PageResult<${className}${doSuffix}> result = new PageResult<${className}${doSuffix}>();
76 | result.setPageSize(query.getPageSize());
77 | result.setPageNo(query.getPageNo());
78 | result.setTotalCount(this.countBy${queryPrefix}(${queryPrefix.toLowerCase()}));
79 | result.setResult(this.selectBy${queryPrefix}(${queryPrefix.toLowerCase()}));
80 | return result;
81 | }
82 |
83 | /**
84 | * select by primary key.
85 | */
86 | @Override
87 | public ${className}${doSuffix} selectByPrimaryKey(Long id){
88 | return ${managerMapperName}.selectByPrimaryKey(id);
89 | }
90 |
91 | /**
92 | * update by query condition selective.
93 | */
94 | @Override
95 | public int updateBy${queryPrefix}Selective( ${className}${doSuffix} record, ${className}${queryPrefix} ${queryPrefix.toLowerCase()}){
96 | return ${managerMapperName}.updateBy${queryPrefix}Selective(record, ${queryPrefix.toLowerCase()});
97 | }
98 |
99 | /**
100 | * update by query condition.
101 | */
102 | @Override
103 | public int updateBy${queryPrefix}( ${className}${doSuffix} record, ${className}${queryPrefix} ${queryPrefix.toLowerCase()}){
104 |
105 | return ${managerMapperName}.updateBy${queryPrefix}(record, ${queryPrefix.toLowerCase()});
106 | }
107 |
108 | /**
109 | * update by primary key selective.
110 | */
111 | @Override
112 | public int updateByPrimaryKeySelective(${className}${doSuffix} record){
113 | return ${managerMapperName}.updateByPrimaryKeySelective(record);
114 | }
115 | }
--------------------------------------------------------------------------------
/src/main/resources/template/mapper-ext.txt:
--------------------------------------------------------------------------------
1 | package ${mapperExtPackage};
2 |
3 | import ${mapperPackage}.${className}${mapperSuffix};
4 | import org.apache.ibatis.annotations.Mapper;
5 | import org.springframework.stereotype.Repository;
6 |
7 | /**
8 | * MyBatis Ext Mapper for ${className}.
9 | */
10 |
11 | @Mapper
12 | @Repository
13 |
14 | public interface ${className}${extMapperSuffix} extends ${className}${mapperSuffix} {
15 |
16 | }
--------------------------------------------------------------------------------
/src/main/resources/template/mapper.txt:
--------------------------------------------------------------------------------
1 | package ${mapperPackage};
2 |
3 | import ${doPackage}.${className}${doSuffix};
4 | import ${queryPackage}.${className}${queryPrefix};
5 |
6 | import java.util.List;
7 |
8 | import org.apache.ibatis.annotations.Mapper;
9 | import org.apache.ibatis.annotations.Param;
10 | import org.springframework.stereotype.Repository;
11 | /**
12 | * MyBatis Mapper for ${className}.
13 | */
14 | @Mapper
15 | @Repository
16 |
17 | public interface ${className}${mapperSuffix} {
18 | /**
19 | * query count by query condition.
20 | */
21 | int countBy${queryPrefix}(${className}${queryPrefix} ${queryPrefix.toLowerCase()});
22 |
23 | /**
24 | * delete by query condition.
25 | */
26 | int deleteBy${queryPrefix}(${className}${queryPrefix} ${queryPrefix.toLowerCase()});
27 |
28 | /**
29 | * delete by primary key.
30 | */
31 | int deleteByPrimaryKey(${className}${doSuffix} record);
32 |
33 | /**
34 | * insert selective.
35 | */
36 | int insertSelective(${className}${doSuffix} record);
37 |
38 | /**
39 | * select by query condition.
40 | */
41 | List<${className}${doSuffix}> selectBy${queryPrefix}(${className}${queryPrefix} ${queryPrefix.toLowerCase()});
42 |
43 | /**
44 | * select by primary key.
45 | */
46 | ${className}${doSuffix} selectByPrimaryKey(Long id);
47 |
48 | /**
49 | * update by query condition selective.
50 | */
51 | int updateBy${queryPrefix}Selective(@Param("record") ${className}${doSuffix} record, @Param("${queryPrefix.toLowerCase()}") ${className}${queryPrefix} ${queryPrefix.toLowerCase()});
52 |
53 | /**
54 | * update by query condition.
55 | */
56 | int updateBy${queryPrefix}(@Param("record") ${className}${doSuffix} record, @Param("${queryPrefix.toLowerCase()}") ${className}${queryPrefix} ${queryPrefix.toLowerCase()});
57 |
58 | /**
59 | * update by primary key selective.
60 | */
61 | int updateByPrimaryKeySelective(${className}${doSuffix} record);
62 | }
--------------------------------------------------------------------------------
/src/main/resources/template/query.txt:
--------------------------------------------------------------------------------
1 | package ${queryPackage};
2 |
3 |
4 | import ${daoPackage}.BaseCriteria;
5 | import ${daoPackage}.BaseQuery;
6 | import org.apache.commons.lang.builder.ReflectionToStringBuilder;
7 | import org.apache.commons.lang.builder.ToStringStyle;
8 |
9 | import java.io.Serializable;
10 |
11 | import java.time.LocalDateTime;
12 | import java.util.List;
13 |
14 | public class ${className}${queryPrefix} extends BaseQuery implements Serializable {
15 | private static final long serialVersionUID = 1L;
16 |
17 | public ${className}${queryPrefix}() {
18 | super();
19 | }
20 |
21 | public Criteria or() {
22 | Criteria criteria = createCriteriaInternal();
23 | super.oredCriteria.add(criteria);
24 | return criteria;
25 | }
26 |
27 | public Criteria createCriteria() {
28 | Criteria criteria = createCriteriaInternal();
29 | if (oredCriteria.size() == 0) {
30 | oredCriteria.add(criteria);
31 | }
32 | return criteria;
33 | }
34 |
35 | protected Criteria createCriteriaInternal() {
36 | Criteria criteria = new Criteria();
37 | return criteria;
38 | }
39 |
40 | /**
41 | * This class corresponds to the database table ${tableName}
42 | */
43 | #set($propPrimaryKey=$propPrimaryKey.substring(0, 1).toUpperCase() + $propPrimaryKey.substring(1))
44 | protected abstract static class GeneratedCriteria extends BaseCriteria {
45 |
46 | protected GeneratedCriteria() {
47 | super();
48 | }
49 |
50 | public Criteria and${propPrimaryKey}IsNull() {
51 | addCriterion("${primaryKey} is null");
52 | return (Criteria) this;
53 | }
54 |
55 | public Criteria and${propPrimaryKey}IsNotNull() {
56 | addCriterion("${primaryKey} is not null");
57 | return (Criteria) this;
58 | }
59 |
60 | public Criteria and${propPrimaryKey}EqualTo(Long value) {
61 | addCriterion("${primaryKey} =", value, "id");
62 | return (Criteria) this;
63 | }
64 |
65 | public Criteria and${propPrimaryKey}NotEqualTo(Long value) {
66 | addCriterion("${primaryKey} <>", value, "id");
67 | return (Criteria) this;
68 | }
69 |
70 | public Criteria an${propPrimaryKey}GreaterThan(Long value) {
71 | addCriterion("${primaryKey} >", value, "id");
72 | return (Criteria) this;
73 | }
74 |
75 | public Criteria and${propPrimaryKey}GreaterThanOrEqualTo(Long value) {
76 | addCriterion("${primaryKey} >=", value, "id");
77 | return (Criteria) this;
78 | }
79 |
80 | public Criteria and${propPrimaryKey}LessThan(Long value) {
81 | addCriterion("${primaryKey} <", value, "id");
82 | return (Criteria) this;
83 | }
84 |
85 | public Criteria and${propPrimaryKey}LessThanOrEqualTo(Long value) {
86 | addCriterion("${primaryKey} <=", value, "id");
87 | return (Criteria) this;
88 | }
89 |
90 | public Criteria and${propPrimaryKey}In(List values) {
91 | addCriterion("${primaryKey} in", values, "id");
92 | return (Criteria) this;
93 | }
94 |
95 | public Criteria and${propPrimaryKey}NotIn(List values) {
96 | addCriterion("${primaryKey} not in", values, "id");
97 | return (Criteria) this;
98 | }
99 |
100 | public Criteria and${propPrimaryKey}Between(Long value1, Long value2) {
101 | addCriterion("${primaryKey} between", value1, value2, "id");
102 | return (Criteria) this;
103 | }
104 |
105 | public Criteria and${propPrimaryKey}NotBetween(Long value1, Long value2) {
106 | addCriterion("${primaryKey} not between", value1, value2, "id");
107 | return (Criteria) this;
108 | }
109 |
110 | public Criteria andGmtCreateIsNull() {
111 | addCriterion("gmt_create is null");
112 | return (Criteria) this;
113 | }
114 |
115 | public Criteria andGmtCreateIsNotNull() {
116 | addCriterion("gmt_create is not null");
117 | return (Criteria) this;
118 | }
119 |
120 | public Criteria andGmtCreateEqualTo(LocalDateTime value) {
121 | addCriterion("gmt_create =", value, "gmtCreate");
122 | return (Criteria) this;
123 | }
124 |
125 | public Criteria andGmtCreateNotEqualTo(LocalDateTime value) {
126 | addCriterion("gmt_create <>", value, "gmtCreate");
127 | return (Criteria) this;
128 | }
129 |
130 | public Criteria andGmtCreateGreaterThan(LocalDateTime value) {
131 | addCriterion("gmt_create >", value, "gmtCreate");
132 | return (Criteria) this;
133 | }
134 |
135 | public Criteria andGmtCreateGreaterThanOrEqualTo(LocalDateTime value) {
136 | addCriterion("gmt_create >=", value, "gmtCreate");
137 | return (Criteria) this;
138 | }
139 |
140 | public Criteria andGmtCreateLessThan(LocalDateTime value) {
141 | addCriterion("gmt_create <", value, "gmtCreate");
142 | return (Criteria) this;
143 | }
144 |
145 | public Criteria andGmtCreateLessThanOrEqualTo(LocalDateTime value) {
146 | addCriterion("gmt_create <=", value, "gmtCreate");
147 | return (Criteria) this;
148 | }
149 |
150 | public Criteria andGmtCreateIn(List values) {
151 | addCriterion("gmt_create in", values, "gmtCreate");
152 | return (Criteria) this;
153 | }
154 |
155 | public Criteria andGmtCreateNotIn(List values) {
156 | addCriterion("gmt_create not in", values, "gmtCreate");
157 | return (Criteria) this;
158 | }
159 |
160 | public Criteria andGmtCreateBetween(LocalDateTime value1, LocalDateTime value2) {
161 | addCriterion("gmt_create between", value1, value2, "gmtCreate");
162 | return (Criteria) this;
163 | }
164 |
165 | public Criteria andGmtCreateNotBetween(LocalDateTime value1, LocalDateTime value2) {
166 | addCriterion("gmt_create not between", value1, value2, "gmtCreate");
167 | return (Criteria) this;
168 | }
169 |
170 | public Criteria andGmtModifiedIsNull() {
171 | addCriterion("gmt_modified is null");
172 | return (Criteria) this;
173 | }
174 |
175 | public Criteria andGmtModifiedIsNotNull() {
176 | addCriterion("gmt_modified is not null");
177 | return (Criteria) this;
178 | }
179 |
180 | public Criteria andGmtModifiedEqualTo(LocalDateTime value) {
181 | addCriterion("gmt_modified =", value, "gmtModified");
182 | return (Criteria) this;
183 | }
184 |
185 | public Criteria andGmtModifiedNotEqualTo(LocalDateTime value) {
186 | addCriterion("gmt_modified <>", value, "gmtModified");
187 | return (Criteria) this;
188 | }
189 |
190 | public Criteria andGmtModifiedGreaterThan(LocalDateTime value) {
191 | addCriterion("gmt_modified >", value, "gmtModified");
192 | return (Criteria) this;
193 | }
194 |
195 | public Criteria andGmtModifiedGreaterThanOrEqualTo(LocalDateTime value) {
196 | addCriterion("gmt_modified >=", value, "gmtModified");
197 | return (Criteria) this;
198 | }
199 |
200 | public Criteria andGmtModifiedLessThan(LocalDateTime value) {
201 | addCriterion("gmt_modified <", value, "gmtModified");
202 | return (Criteria) this;
203 | }
204 |
205 | public Criteria andGmtModifiedLessThanOrEqualTo(LocalDateTime value) {
206 | addCriterion("gmt_modified <=", value, "gmtModified");
207 | return (Criteria) this;
208 | }
209 |
210 | public Criteria andGmtModifiedIn(List values) {
211 | addCriterion("gmt_modified in", values, "gmtModified");
212 | return (Criteria) this;
213 | }
214 |
215 | public Criteria andGmtModifiedNotIn(List values) {
216 | addCriterion("gmt_modified not in", values, "gmtModified");
217 | return (Criteria) this;
218 | }
219 |
220 | public Criteria andGmtModifiedBetween(LocalDateTime value1, LocalDateTime value2) {
221 | addCriterion("gmt_modified between", value1, value2, "gmtModified");
222 | return (Criteria) this;
223 | }
224 |
225 | public Criteria andGmtModifiedNotBetween(LocalDateTime value1, LocalDateTime value2) {
226 | addCriterion("gmt_modified not between", value1, value2, "gmtModified");
227 | return (Criteria) this;
228 | }
229 |
230 | #foreach($item in $list)
231 | #set($colName=$item.propName.substring(0, 1).toUpperCase() + $item.propName.substring(1))
232 | public Criteria and${colName}IsNull() {
233 | addCriterion("${item.columnName} is null");
234 | return (Criteria) this;
235 | }
236 |
237 | public Criteria and${colName}IsNotNull() {
238 | addCriterion("${item.columnName} is not null");
239 | return (Criteria) this;
240 | }
241 |
242 | public Criteria and${colName}EqualTo(${item.javaType} value) {
243 | addCriterion("${item.columnName} =", value, "${item.propName}");
244 | return (Criteria) this;
245 | }
246 |
247 | public Criteria and${colName}NotEqualTo(${item.javaType} value) {
248 | addCriterion("${item.columnName} <>", value, "${item.propName}");
249 | return (Criteria) this;
250 | }
251 |
252 | public Criteria and${colName}GreaterThan(${item.javaType} value) {
253 | addCriterion("${item.columnName} >", value, "${item.propName}");
254 | return (Criteria) this;
255 | }
256 |
257 | public Criteria and${colName}GreaterThanOrEqualTo(${item.javaType} value) {
258 | addCriterion("${item.columnName} >=", value, "${item.propName}");
259 | return (Criteria) this;
260 | }
261 |
262 | public Criteria and${colName}LessThan(${item.javaType} value) {
263 | addCriterion("${item.columnName} <", value, "${item.propName}");
264 | return (Criteria) this;
265 | }
266 |
267 | public Criteria and${colName}LessThanOrEqualTo(${item.javaType} value) {
268 | addCriterion("${item.columnName} <=", value, "${item.propName}");
269 | return (Criteria) this;
270 | }
271 |
272 | public Criteria and${colName}Like(${item.javaType} value) {
273 | addCriterion("${item.columnName} like", value, "${item.propName}");
274 | return (Criteria) this;
275 | }
276 |
277 | public Criteria and${colName}NotLike(${item.javaType} value) {
278 | addCriterion("${item.columnName} not like", value, "${item.propName}");
279 | return (Criteria) this;
280 | }
281 |
282 | public Criteria and${colName}In(List<${item.javaType}> values) {
283 | addCriterion("${item.columnName} in", values, "${item.propName}");
284 | return (Criteria) this;
285 | }
286 |
287 | public Criteria and${colName}NotIn(List<${item.javaType}> values) {
288 | addCriterion("${item.columnName} not in", values, "${item.propName}");
289 | return (Criteria) this;
290 | }
291 |
292 | public Criteria and${colName}Between(${item.javaType} value1, ${item.javaType} value2) {
293 | addCriterion("${item.columnName} between", value1, value2, "${item.propName}");
294 | return (Criteria) this;
295 | }
296 |
297 | public Criteria and${colName}NotBetween(${item.javaType} value1, ${item.javaType} value2) {
298 | addCriterion("${item.columnName} not between", value1, value2, "${item.propName}");
299 | return (Criteria) this;
300 | }
301 |
302 | #end
303 | }
304 |
305 | /**
306 | * This class corresponds to the database table ${tableName}
307 | */
308 | public static class Criteria extends GeneratedCriteria{
309 | protected Criteria() {
310 | super();
311 | }
312 | }
313 |
314 |
315 |
316 | @Override
317 | public String toString(){
318 | return ReflectionToStringBuilder.toString(this, ToStringStyle.DEFAULT_STYLE);
319 | }
320 | }
--------------------------------------------------------------------------------
/src/main/resources/template/sqlmap-ext.txt:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
--------------------------------------------------------------------------------
/src/main/resources/template/sqlmap.txt:
--------------------------------------------------------------------------------
1 |
2 |
3 |
4 |
5 |
6 |
7 |
8 | #foreach($item in $list)
9 |
10 | #end
11 |
12 |
13 |
14 |
15 |
16 |
17 |
18 |
19 |
20 | and ${criterion.condition}
21 |
22 |
23 | and ${criterion.condition} #{criterion.value}
24 |
25 |
26 | and ${criterion.condition} #{criterion.value} and #{criterion.secondValue}
27 |
28 |
29 | and ${criterion.condition}
30 |
31 | #{listItem}
32 |
33 |
34 |
35 |
36 |
37 |
38 |
39 |
40 |
41 |
42 |
43 |
44 |
45 |
46 |
47 |
48 |
49 |
50 | and ${criterion.condition}
51 |
52 |
53 | and ${criterion.condition} #{criterion.value}
54 |
55 |
56 | and ${criterion.condition} #{criterion.value} and #{criterion.secondValue}
57 |
58 |
59 | and ${criterion.condition}
60 |
61 | #{listItem}
62 |
63 |
64 |
65 |
66 |
67 |
68 |
69 |
70 |
71 |
72 |
73 | ${cols}
74 |
75 |
76 |
91 |
92 |
95 |
96 |
97 | delete from ${tableName} where ${primaryKey} = #{${propPrimaryKey},jdbcType=BIGINT}
98 |
99 |
100 |
101 | delete from ${tableName}
102 |
103 |
104 |
105 |
106 |
107 |
108 |
109 | SELECT LAST_INSERT_ID()
110 |
111 | insert into ${tableName}
112 |
113 | gmt_create,gmt_modified,
114 | #foreach($item in $list)
115 |
116 | ${item.columnName},
117 |
118 | #end
119 |
120 |
121 | now(),now(),
122 | #foreach($item in $list)
123 |
124 | #{${item.propName},jdbcType=${item.jdbcType}},
125 |
126 | #end
127 |
128 |
129 |
130 |
136 |
137 |
138 | update ${tableName}
139 |
140 | gmt_modified=now(),
141 | #foreach($item in $list)
142 |
143 | ${item.columnName} = #{record.${item.propName},jdbcType=${item.jdbcType}},
144 |
145 | #end
146 |
147 |
148 |
149 |
150 |
151 |
152 |
153 | update ${tableName} set
154 | #foreach($item in $list)
155 | ${item.columnName} = #{record.${item.propName},jdbcType=${item.jdbcType}},
156 | #end
157 | gmt_modified = now()
158 |
159 |
160 |
161 |
162 |
163 |
164 | update ${tableName}
165 |
166 | gmt_modified=now(),
167 | #foreach($item in $list)
168 |
169 | ${item.columnName} = #{${item.propName},jdbcType=${item.jdbcType}},
170 |
171 | #end
172 |
173 | where ${primaryKey} = #{${propPrimaryKey},jdbcType=BIGINT}
174 |
175 |
176 |
177 |
178 |
179 |
180 |
181 |
182 |
183 |
--------------------------------------------------------------------------------