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 import org.mybatis.generator.api.dom.java.Parameter;
26
27
28
29
30
31
32 public class CountByExampleMethodGenerator extends
33 AbstractJavaMapperMethodGenerator {
34
35 public CountByExampleMethodGenerator() {
36 super();
37 }
38
39 @Override
40 public void addInterfaceElements(Interface interfaze) {
41 FullyQualifiedJavaType fqjt = new FullyQualifiedJavaType(
42 introspectedTable.getExampleType());
43
44 Set<FullyQualifiedJavaType> importedTypes = new TreeSet<FullyQualifiedJavaType>();
45 importedTypes.add(fqjt);
46
47 Method method = new Method();
48 method.setVisibility(JavaVisibility.PUBLIC);
49 method.setReturnType(FullyQualifiedJavaType.getIntInstance());
50 method.setName(introspectedTable.getCountByExampleStatementId());
51 method.addParameter(new Parameter(fqjt, "example"));
52 context.getCommentGenerator().addGeneralMethodComment(method,
53 introspectedTable);
54
55 addMapperAnnotations(interfaze, method);
56
57 if (context.getPlugins().clientCountByExampleMethodGenerated(method,
58 interfaze, introspectedTable)) {
59 interfaze.addImportedTypes(importedTypes);
60 interfaze.addMethod(method);
61 }
62 }
63
64 public void addMapperAnnotations(Interface interfaze, Method method) {
65 }
66 }