1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package org.mybatis.generator.codegen.mybatis3.javamapper.elements;
17
18 import java.util.Set;
19 import java.util.TreeSet;
20
21 import org.mybatis.generator.api.dom.java.FullyQualifiedJavaType;
22 import org.mybatis.generator.api.dom.java.Interface;
23 import org.mybatis.generator.api.dom.java.JavaVisibility;
24 import org.mybatis.generator.api.dom.java.Method;
25
26 public class DeleteByPrimaryKeyMethodGenerator extends AbstractJavaMapperMethodGenerator {
27
28 private final boolean isSimple;
29
30 public DeleteByPrimaryKeyMethodGenerator(boolean isSimple) {
31 super();
32 this.isSimple = isSimple;
33 }
34
35 @Override
36 public void addInterfaceElements(Interface interfaze) {
37 Method method = new Method(introspectedTable.getDeleteByPrimaryKeyStatementId());
38 method.setVisibility(JavaVisibility.PUBLIC);
39 method.setAbstract(true);
40 method.setReturnType(FullyQualifiedJavaType.getIntInstance());
41
42 Set<FullyQualifiedJavaType> importedTypes = new TreeSet<>();
43 addPrimaryKeyMethodParameters(isSimple, method, importedTypes);
44
45 addMapperAnnotations(method);
46
47 context.getCommentGenerator().addGeneralMethodComment(method, introspectedTable);
48
49 if (context.getPlugins().clientDeleteByPrimaryKeyMethodGenerated(method, interfaze, introspectedTable)) {
50 addExtraImports(interfaze);
51 interfaze.addImportedTypes(importedTypes);
52 interfaze.addMethod(method);
53 }
54 }
55
56 public void addMapperAnnotations(Method method) {
57
58 }
59
60 public void addExtraImports(Interface interfaze) {
61
62 }
63 }