1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package org.mybatis.generator.codegen.mybatis3.javamapper.elements.annotated;
17
18 import static org.mybatis.generator.api.dom.OutputUtilities.javaIndent;
19 import static org.mybatis.generator.internal.util.StringUtility.escapeStringForJava;
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.Method;
24 import org.mybatis.generator.codegen.mybatis3.javamapper.elements.DeleteByPrimaryKeyMethodGenerator;
25
26 public class AnnotatedDeleteByPrimaryKeyMethodGenerator extends DeleteByPrimaryKeyMethodGenerator {
27
28 public AnnotatedDeleteByPrimaryKeyMethodGenerator(boolean isSimple) {
29 super(isSimple);
30 }
31
32 @Override
33 public void addMapperAnnotations(Method method) {
34
35 method.addAnnotation("@Delete({");
36
37 StringBuilder sb = new StringBuilder();
38 javaIndent(sb, 1);
39 sb.append("\"delete from ");
40 sb.append(escapeStringForJava(introspectedTable.getFullyQualifiedTableNameAtRuntime()));
41 sb.append("\",");
42 method.addAnnotation(sb.toString());
43
44 buildByPrimaryKeyWhereClause().forEach(method::addAnnotation);
45
46 method.addAnnotation("})");
47 }
48
49 @Override
50 public void addExtraImports(Interface interfaze) {
51 interfaze.addImportedType(new FullyQualifiedJavaType("org.apache.ibatis.annotations.Delete"));
52 }
53 }