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 UpdateByPrimaryKeyWithBLOBsMethodGenerator extends
33 AbstractJavaMapperMethodGenerator {
34
35 public UpdateByPrimaryKeyWithBLOBsMethodGenerator() {
36 super();
37 }
38
39 @Override
40 public void addInterfaceElements(Interface interfaze) {
41 Set<FullyQualifiedJavaType> importedTypes = new TreeSet<FullyQualifiedJavaType>();
42 FullyQualifiedJavaType parameterType;
43
44 if (introspectedTable.getRules().generateRecordWithBLOBsClass()) {
45 parameterType = new FullyQualifiedJavaType(introspectedTable
46 .getRecordWithBLOBsType());
47 } else {
48 parameterType = new FullyQualifiedJavaType(introspectedTable
49 .getBaseRecordType());
50 }
51
52 importedTypes.add(parameterType);
53
54 Method method = new Method();
55 method.setVisibility(JavaVisibility.PUBLIC);
56 method.setReturnType(FullyQualifiedJavaType.getIntInstance());
57
58 method.setName(introspectedTable
59 .getUpdateByPrimaryKeyWithBLOBsStatementId());
60 method.addParameter(new Parameter(parameterType, "record"));
61
62 context.getCommentGenerator().addGeneralMethodComment(method,
63 introspectedTable);
64
65 addMapperAnnotations(interfaze, method);
66
67 if (context.getPlugins()
68 .clientUpdateByPrimaryKeyWithBLOBsMethodGenerated(method,
69 interfaze, introspectedTable)) {
70 interfaze.addImportedTypes(importedTypes);
71 interfaze.addMethod(method);
72 }
73 }
74
75 public void addMapperAnnotations(Interface interfaze, Method method) {
76 }
77 }