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
27
28
29
30
31
32 public class SelectAllMethodGenerator extends AbstractJavaMapperMethodGenerator {
33
34 public SelectAllMethodGenerator() {
35 super();
36 }
37
38 @Override
39 public void addInterfaceElements(Interface interfaze) {
40 Set<FullyQualifiedJavaType> importedTypes = new TreeSet<FullyQualifiedJavaType>();
41 importedTypes.add(FullyQualifiedJavaType.getNewListInstance());
42
43 Method method = new Method();
44 method.setVisibility(JavaVisibility.PUBLIC);
45
46 FullyQualifiedJavaType returnType = FullyQualifiedJavaType
47 .getNewListInstance();
48 FullyQualifiedJavaType listType;
49 listType = new FullyQualifiedJavaType(
50 introspectedTable.getBaseRecordType());
51
52 importedTypes.add(listType);
53 returnType.addTypeArgument(listType);
54 method.setReturnType(returnType);
55 method.setName(introspectedTable.getSelectAllStatementId());
56
57 context.getCommentGenerator().addGeneralMethodComment(method,
58 introspectedTable);
59
60 addMapperAnnotations(interfaze, method);
61
62 if (context.getPlugins().clientSelectAllMethodGenerated(method,
63 interfaze, introspectedTable)) {
64 interfaze.addImportedTypes(importedTypes);
65 interfaze.addMethod(method);
66 }
67 }
68
69 public void addMapperAnnotations(Interface interfaze, Method method) {
70 }
71 }