1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package org.mybatis.generator.codegen.mybatis3.javamapper.elements.sqlprovider;
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.TopLevelClass;
23 import org.mybatis.generator.codegen.AbstractGenerator;
24
25 public abstract class AbstractJavaProviderMethodGenerator extends AbstractGenerator {
26
27 protected static final FullyQualifiedJavaType BUILDER_IMPORT =
28 new FullyQualifiedJavaType("org.apache.ibatis.jdbc.SQL");
29
30 protected AbstractJavaProviderMethodGenerator() {
31 super();
32 }
33
34 protected Set<FullyQualifiedJavaType> initializeImportedTypes() {
35 Set<FullyQualifiedJavaType> importedTypes = new TreeSet<>();
36
37 importedTypes.add(BUILDER_IMPORT);
38
39 return importedTypes;
40 }
41
42 protected Set<FullyQualifiedJavaType> initializeImportedTypes(String extraType) {
43 return initializeImportedTypes(new FullyQualifiedJavaType(extraType));
44 }
45
46 protected Set<FullyQualifiedJavaType> initializeImportedTypes(FullyQualifiedJavaType extraType) {
47 Set<FullyQualifiedJavaType> importedTypes = initializeImportedTypes();
48
49 importedTypes.add(extraType);
50
51 return importedTypes;
52 }
53
54 public abstract void addClassElements(TopLevelClass topLevelClass);
55 }