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.CountByExampleMethodGenerator;
22
23
24
25
26
27 public class AnnotatedCountByExampleMethodGenerator extends
28 CountByExampleMethodGenerator {
29
30 public AnnotatedCountByExampleMethodGenerator() {
31 super();
32 }
33
34 @Override
35 public void addMapperAnnotations(Interface interfaze, Method method) {
36 FullyQualifiedJavaType fqjt = new FullyQualifiedJavaType(introspectedTable.getMyBatis3SqlProviderType());
37 interfaze.addImportedType(new FullyQualifiedJavaType("org.apache.ibatis.annotations.SelectProvider"));
38 StringBuilder sb = new StringBuilder();
39 sb.append("@SelectProvider(type=");
40 sb.append(fqjt.getShortName());
41 sb.append(".class, method=\"");
42 sb.append(introspectedTable.getCountByExampleStatementId());
43 sb.append("\")");
44
45 method.addAnnotation(sb.toString());
46 }
47 }