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 org.mybatis.generator.api.dom.java.FullyQualifiedJavaType;
19 import org.mybatis.generator.api.dom.java.Interface;
20 import org.mybatis.generator.api.dom.java.Method;
21 import org.mybatis.generator.codegen.mybatis3.javamapper.elements.UpdateByPrimaryKeyWithoutBLOBsMethodGenerator;
22
23 public class AnnotatedUpdateByPrimaryKeyWithoutBLOBsMethodGenerator
24 extends UpdateByPrimaryKeyWithoutBLOBsMethodGenerator {
25
26 private final boolean isSimple;
27
28 public AnnotatedUpdateByPrimaryKeyWithoutBLOBsMethodGenerator(boolean isSimple) {
29 super();
30 this.isSimple = isSimple;
31 }
32
33 @Override
34 public void addMapperAnnotations(Method method) {
35 if (isSimple) {
36 buildUpdateByPrimaryKeyAnnotations(introspectedTable.getNonPrimaryKeyColumns())
37 .forEach(method::addAnnotation);
38 } else {
39 buildUpdateByPrimaryKeyAnnotations(introspectedTable.getBaseColumns()).forEach(method::addAnnotation);
40 }
41 }
42
43 @Override
44 public void addExtraImports(Interface interfaze) {
45 interfaze.addImportedType(new FullyQualifiedJavaType("org.apache.ibatis.annotations.Update"));
46 }
47 }