1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package org.mybatis.generator.codegen.mybatis3.javamapper;
17
18 import org.mybatis.generator.api.dom.java.Interface;
19 import org.mybatis.generator.codegen.AbstractXmlGenerator;
20 import org.mybatis.generator.codegen.mybatis3.javamapper.elements.AbstractJavaMapperMethodGenerator;
21 import org.mybatis.generator.codegen.mybatis3.javamapper.elements.annotated.AnnotatedDeleteByPrimaryKeyMethodGenerator;
22 import org.mybatis.generator.codegen.mybatis3.javamapper.elements.annotated.AnnotatedInsertMethodGenerator;
23 import org.mybatis.generator.codegen.mybatis3.javamapper.elements.annotated.AnnotatedSelectByPrimaryKeyMethodGenerator;
24 import org.mybatis.generator.codegen.mybatis3.javamapper.elements.annotated.AnnotatedUpdateByPrimaryKeyWithBLOBsMethodGenerator;
25 import org.mybatis.generator.codegen.mybatis3.javamapper.elements.annotated.AnnotatedUpdateByPrimaryKeyWithoutBLOBsMethodGenerator;
26 import org.mybatis.generator.codegen.mybatis3.xmlmapper.MixedMapperGenerator;
27
28
29
30
31
32
33
34 public class MixedClientGenerator extends JavaMapperGenerator {
35
36 public MixedClientGenerator(String project) {
37 super(project, true);
38 }
39
40 @Override
41 protected void addDeleteByPrimaryKeyMethod(Interface interfaze) {
42 if (introspectedTable.getRules().generateDeleteByPrimaryKey()) {
43 AbstractJavaMapperMethodGenerator methodGenerator = new AnnotatedDeleteByPrimaryKeyMethodGenerator(false);
44 initializeAndExecuteGenerator(methodGenerator, interfaze);
45 }
46 }
47
48 @Override
49 protected void addInsertMethod(Interface interfaze) {
50 if (introspectedTable.getRules().generateInsert()) {
51 AbstractJavaMapperMethodGenerator methodGenerator = new AnnotatedInsertMethodGenerator(false);
52 initializeAndExecuteGenerator(methodGenerator, interfaze);
53 }
54 }
55
56 @Override
57 protected void addSelectByPrimaryKeyMethod(Interface interfaze) {
58 if (introspectedTable.getRules().generateSelectByPrimaryKey()) {
59 AbstractJavaMapperMethodGenerator methodGenerator =
60 new AnnotatedSelectByPrimaryKeyMethodGenerator(true, false);
61 initializeAndExecuteGenerator(methodGenerator, interfaze);
62 }
63 }
64
65 @Override
66 protected void addUpdateByPrimaryKeyWithBLOBsMethod(Interface interfaze) {
67 if (introspectedTable.getRules().generateUpdateByPrimaryKeyWithBLOBs()) {
68 AbstractJavaMapperMethodGenerator methodGenerator =
69 new AnnotatedUpdateByPrimaryKeyWithBLOBsMethodGenerator();
70 initializeAndExecuteGenerator(methodGenerator, interfaze);
71 }
72 }
73
74 @Override
75 protected void addUpdateByPrimaryKeyWithoutBLOBsMethod(Interface interfaze) {
76 if (introspectedTable.getRules().generateUpdateByPrimaryKeyWithoutBLOBs()) {
77 AbstractJavaMapperMethodGenerator methodGenerator =
78 new AnnotatedUpdateByPrimaryKeyWithoutBLOBsMethodGenerator(false);
79 initializeAndExecuteGenerator(methodGenerator, interfaze);
80 }
81 }
82
83 @Override
84 public AbstractXmlGenerator getMatchedXMLGenerator() {
85 return new MixedMapperGenerator();
86 }
87 }