warnings) {
60 | this.warnings = warnings;
61 | }
62 |
63 | public ProgressCallback getProgressCallback() {
64 | return progressCallback;
65 | }
66 |
67 | public void setProgressCallback(ProgressCallback progressCallback) {
68 | this.progressCallback = progressCallback;
69 | }
70 | }
71 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/internal/db/ActualTableName.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2007 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 |
17 | package org.mybatis.generator.internal.db;
18 |
19 | import static org.mybatis.generator.internal.util.StringUtility.composeFullyQualifiedTableName;
20 |
21 | /**
22 | * This class holds the actual catalog, schema, and table name returned from the
23 | * database introspection.
24 | *
25 | * @author Jeff Butler
26 | *
27 | */
28 | public class ActualTableName {
29 |
30 | private String tableName;
31 | private String catalog;
32 | private String schema;
33 | private String fullName;
34 |
35 | public ActualTableName(String catalog, String schema, String tableName) {
36 | this.catalog = catalog;
37 | this.schema = schema;
38 | this.tableName = tableName;
39 | fullName = composeFullyQualifiedTableName(catalog,
40 | schema, tableName, '.');
41 | }
42 |
43 | public String getCatalog() {
44 | return catalog;
45 | }
46 |
47 | public String getSchema() {
48 | return schema;
49 | }
50 |
51 | public String getTableName() {
52 | return tableName;
53 | }
54 |
55 | @Override
56 | public boolean equals(Object obj) {
57 | if (obj == null || !(obj instanceof ActualTableName)) {
58 | return false;
59 | }
60 |
61 | return obj.toString().equals(this.toString());
62 | }
63 |
64 | @Override
65 | public int hashCode() {
66 | return fullName.hashCode();
67 | }
68 |
69 | @Override
70 | public String toString() {
71 | return fullName;
72 | }
73 | }
74 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/mybatis3/xmlmapper/elements/DeleteByExampleElementGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2009 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.codegen.mybatis3.xmlmapper.elements;
17 |
18 | import org.mybatis.generator.api.dom.xml.Attribute;
19 | import org.mybatis.generator.api.dom.xml.TextElement;
20 | import org.mybatis.generator.api.dom.xml.XmlElement;
21 |
22 | /**
23 | *
24 | * @author Jeff Butler
25 | *
26 | */
27 | public class DeleteByExampleElementGenerator extends
28 | AbstractXmlElementGenerator {
29 |
30 | public DeleteByExampleElementGenerator() {
31 | super();
32 | }
33 |
34 | @Override
35 | public void addElements(XmlElement parentElement) {
36 | XmlElement answer = new XmlElement("delete"); //$NON-NLS-1$
37 |
38 | String fqjt = introspectedTable.getExampleType();
39 |
40 | answer.addAttribute(new Attribute(
41 | "id", introspectedTable.getDeleteByExampleStatementId())); //$NON-NLS-1$
42 | answer.addAttribute(new Attribute("parameterType", fqjt)); //$NON-NLS-1$
43 |
44 | context.getCommentGenerator().addComment(answer);
45 |
46 | StringBuilder sb = new StringBuilder();
47 | sb.append("delete from "); //$NON-NLS-1$
48 | sb.append(introspectedTable
49 | .getAliasedFullyQualifiedTableNameAtRuntime());
50 | answer.addElement(new TextElement(sb.toString()));
51 | answer.addElement(getExampleIncludeElement());
52 |
53 | if (context.getPlugins().sqlMapDeleteByExampleElementGenerated(
54 | answer, introspectedTable)) {
55 | parentElement.addElement(answer);
56 | }
57 | }
58 | }
59 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/mybatis3/javamapper/elements/annotated/AnnotatedInsertSelectiveMethodGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2010 The MyBatis Team
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
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.InsertSelectiveMethodGenerator;
22 | import org.mybatis.generator.config.GeneratedKey;
23 |
24 | /**
25 | *
26 | * @author Jeff Butler
27 | */
28 | public class AnnotatedInsertSelectiveMethodGenerator extends
29 | InsertSelectiveMethodGenerator {
30 |
31 | public AnnotatedInsertSelectiveMethodGenerator() {
32 | super();
33 | }
34 |
35 | @Override
36 | public void addMapperAnnotations(Interface interfaze, Method method) {
37 | FullyQualifiedJavaType fqjt = new FullyQualifiedJavaType(introspectedTable.getMyBatis3SqlProviderType());
38 | interfaze.addImportedType(new FullyQualifiedJavaType("org.apache.ibatis.annotations.InsertProvider")); //$NON-NLS-1$
39 | StringBuilder sb = new StringBuilder();
40 | sb.append("@InsertProvider(type="); //$NON-NLS-1$
41 | sb.append(fqjt.getShortName());
42 | sb.append(".class, method=\""); //$NON-NLS-1$
43 | sb.append(introspectedTable.getInsertSelectiveStatementId());
44 | sb.append("\")"); //$NON-NLS-1$
45 |
46 | method.addAnnotation(sb.toString());
47 |
48 | GeneratedKey gk = introspectedTable.getGeneratedKey();
49 | if (gk != null) {
50 | addGeneratedKeyAnnotation(interfaze, method, gk);
51 | }
52 | }
53 | }
54 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/internal/util/EqualsUtil.java:
--------------------------------------------------------------------------------
1 | package org.mybatis.generator.internal.util;
2 |
3 | /**
4 | * This class is from javapractices.com:
5 | *
6 | * http://www.javapractices.com/Topic17.cjp
7 | *
8 | * Collected methods which allow easy implementation of equals.
9 | *
10 | * Example use case in a class called Car:
11 | *
12 | *
13 | * public boolean equals(Object that) {
14 | * if (this == that)
15 | * return true;
16 | * if (!(that instanceof Car))
17 | * return false;
18 | * Car thatCar = (Car) that;
19 | * return EqualsUtil.areEqual(this.fName, that.fName)
20 | * && EqualsUtil.areEqual(this.fNumDoors, that.fNumDoors)
21 | * && EqualsUtil.areEqual(this.fGasMileage, that.fGasMileage)
22 | * && EqualsUtil.areEqual(this.fColor, that.fColor)
23 | * && Arrays.equals(this.fMaintenanceChecks, that.fMaintenanceChecks); //array!
24 | * }
25 | *
26 | *
27 | * Arrays are not handled by this class. This is because the
28 | * Arrays.equals methods should be used for array fields.
29 | */
30 | public final class EqualsUtil {
31 |
32 | static public boolean areEqual(boolean aThis, boolean aThat) {
33 | return aThis == aThat;
34 | }
35 |
36 | static public boolean areEqual(char aThis, char aThat) {
37 | return aThis == aThat;
38 | }
39 |
40 | static public boolean areEqual(long aThis, long aThat) {
41 | /*
42 | * Implementation Note Note that byte, short, and int are handled by
43 | * this method, through implicit conversion.
44 | */
45 | return aThis == aThat;
46 | }
47 |
48 | static public boolean areEqual(float aThis, float aThat) {
49 | return Float.floatToIntBits(aThis) == Float.floatToIntBits(aThat);
50 | }
51 |
52 | static public boolean areEqual(double aThis, double aThat) {
53 | return Double.doubleToLongBits(aThis) == Double.doubleToLongBits(aThat);
54 | }
55 |
56 | /**
57 | * Possibly-null object field.
58 | *
59 | * Includes type-safe enumerations and collections, but does not include
60 | * arrays. See class comment.
61 | */
62 | static public boolean areEqual(Object aThis, Object aThat) {
63 | return aThis == null ? aThat == null : aThis.equals(aThat);
64 | }
65 | }
66 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/config/xml/ParserEntityResolver.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2005 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.config.xml;
17 |
18 | import java.io.IOException;
19 | import java.io.InputStream;
20 |
21 | import org.mybatis.generator.codegen.XmlConstants;
22 | import org.xml.sax.EntityResolver;
23 | import org.xml.sax.InputSource;
24 | import org.xml.sax.SAXException;
25 |
26 | /**
27 | * @author Jeff Butler
28 | */
29 | public class ParserEntityResolver implements EntityResolver {
30 |
31 | /**
32 | *
33 | */
34 | public ParserEntityResolver() {
35 | super();
36 | }
37 |
38 | /*
39 | * (non-Javadoc)
40 | *
41 | * @see org.xml.sax.EntityResolver#resolveEntity(java.lang.String,
42 | * java.lang.String)
43 | */
44 | public InputSource resolveEntity(String publicId, String systemId)
45 | throws SAXException, IOException {
46 | if (XmlConstants.IBATOR_CONFIG_PUBLIC_ID.equalsIgnoreCase(publicId)) {
47 | InputStream is = getClass().getClassLoader().getResourceAsStream(
48 | "org/mybatis/generator/config/xml/ibator-config_1_0.dtd"); //$NON-NLS-1$
49 | InputSource ins = new InputSource(is);
50 |
51 | return ins;
52 | } else if (XmlConstants.MYBATIS_GENERATOR_CONFIG_PUBLIC_ID
53 | .equalsIgnoreCase(publicId)) {
54 | InputStream is = getClass()
55 | .getClassLoader()
56 | .getResourceAsStream(
57 | "org/mybatis/generator/config/xml/mybatis-generator-config_1_0.dtd"); //$NON-NLS-1$
58 | InputSource ins = new InputSource(is);
59 |
60 | return ins;
61 | } else {
62 | return null;
63 | }
64 | }
65 | }
66 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/mybatis3/xmlmapper/elements/CountByExampleElementGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2009 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.codegen.mybatis3.xmlmapper.elements;
17 |
18 | import org.mybatis.generator.api.dom.xml.Attribute;
19 | import org.mybatis.generator.api.dom.xml.TextElement;
20 | import org.mybatis.generator.api.dom.xml.XmlElement;
21 |
22 | /**
23 | *
24 | * @author Jeff Butler
25 | *
26 | */
27 | public class CountByExampleElementGenerator extends AbstractXmlElementGenerator {
28 |
29 | public CountByExampleElementGenerator() {
30 | super();
31 | }
32 |
33 | @Override
34 | public void addElements(XmlElement parentElement) {
35 | XmlElement answer = new XmlElement("select"); //$NON-NLS-1$
36 |
37 | String fqjt = introspectedTable.getExampleType();
38 |
39 | answer.addAttribute(new Attribute(
40 | "id", introspectedTable.getCountByExampleStatementId())); //$NON-NLS-1$
41 | answer.addAttribute(new Attribute("parameterType", fqjt)); //$NON-NLS-1$
42 | answer.addAttribute(new Attribute("resultType", "java.lang.Integer")); //$NON-NLS-1$ //$NON-NLS-2$
43 |
44 | context.getCommentGenerator().addComment(answer);
45 |
46 | StringBuilder sb = new StringBuilder();
47 | sb.append("select count(*) from "); //$NON-NLS-1$
48 | sb.append(introspectedTable
49 | .getAliasedFullyQualifiedTableNameAtRuntime());
50 | answer.addElement(new TextElement(sb.toString()));
51 | answer.addElement(getExampleIncludeElement());
52 |
53 | if (context.getPlugins().sqlMapCountByExampleElementGenerated(
54 | answer, introspectedTable)) {
55 | parentElement.addElement(answer);
56 | }
57 | }
58 | }
59 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/internal/rules/HierarchicalModelRules.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2006 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 |
17 | package org.mybatis.generator.internal.rules;
18 |
19 | import org.mybatis.generator.api.IntrospectedTable;
20 |
21 | /**
22 | * This class encapsulates all the code generation rules for a table using the
23 | * hierarchical model.
24 | *
25 | * @author Jeff Butler
26 | *
27 | */
28 | public class HierarchicalModelRules extends BaseRules {
29 |
30 | /**
31 | *
32 | */
33 | public HierarchicalModelRules(IntrospectedTable introspectedTable) {
34 | super(introspectedTable);
35 | }
36 |
37 | /**
38 | * Implements the rule for determining whether to generate a primary key
39 | * class. If the physical table has a primary key, then we generate the
40 | * class.
41 | *
42 | * @return true if the primary key should be generated
43 | */
44 | public boolean generatePrimaryKeyClass() {
45 | return introspectedTable.hasPrimaryKeyColumns();
46 | }
47 |
48 | /**
49 | * Implements the rule for generating a base record. If the table has fields
50 | * that are not in the primary key, and non-BLOB fields, then generate the
51 | * class.
52 | *
53 | * @return true if the class should be generated
54 | */
55 | public boolean generateBaseRecordClass() {
56 | return introspectedTable.hasBaseColumns();
57 | }
58 |
59 | /**
60 | * Implements the rule for generating a record with BLOBs. A record with
61 | * BLOBs is generated if the table contains any BLOB fields.
62 | *
63 | * @return true if the record with BLOBs class should be generated
64 | */
65 | public boolean generateRecordWithBLOBsClass() {
66 | return introspectedTable.hasBLOBColumns();
67 | }
68 | }
69 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/api/GeneratedXmlFile.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2005 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.api;
17 |
18 | import org.mybatis.generator.api.dom.xml.Document;
19 |
20 | /**
21 | * @author Jeff Butler
22 | */
23 | public class GeneratedXmlFile extends GeneratedFile {
24 | private Document document;
25 |
26 | private String fileName;
27 |
28 | private String targetPackage;
29 |
30 | private boolean isMergeable;
31 |
32 | private XmlFormatter xmlFormatter;
33 |
34 | /**
35 | *
36 | * @param document
37 | * @param fileName
38 | * @param targetPackage
39 | * @param targetProject
40 | * @param isMergeable
41 | * true if the file can be merged by the built in XML file
42 | * merger.
43 | */
44 | public GeneratedXmlFile(Document document, String fileName,
45 | String targetPackage, String targetProject, boolean isMergeable,
46 | XmlFormatter xmlFormatter) {
47 | super(targetProject);
48 | this.document = document;
49 | this.fileName = fileName;
50 | this.targetPackage = targetPackage;
51 | this.isMergeable = isMergeable;
52 | this.xmlFormatter = xmlFormatter;
53 | }
54 |
55 | @Override
56 | public String getFormattedContent() {
57 | return xmlFormatter.getFormattedContent(document);
58 | }
59 |
60 | /**
61 | * @return Returns the fileName.
62 | */
63 | @Override
64 | public String getFileName() {
65 | return fileName;
66 | }
67 |
68 | /**
69 | * @return Returns the targetPackage.
70 | */
71 | @Override
72 | public String getTargetPackage() {
73 | return targetPackage;
74 | }
75 |
76 | @Override
77 | public boolean isMergeable() {
78 | return isMergeable;
79 | }
80 | }
81 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/internal/util/ClassloaderUtility.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2008 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 |
17 | package org.mybatis.generator.internal.util;
18 |
19 | import static org.mybatis.generator.internal.util.messages.Messages.getString;
20 |
21 | import java.io.File;
22 | import java.net.MalformedURLException;
23 | import java.net.URL;
24 | import java.net.URLClassLoader;
25 | import java.util.ArrayList;
26 | import java.util.Collection;
27 | import java.util.List;
28 |
29 | /**
30 | * This class holds methods useful for constructing custom classloaders.
31 | *
32 | * @author Jeff Butler
33 | *
34 | */
35 | public class ClassloaderUtility {
36 |
37 | /**
38 | * Utility Class - No Instances
39 | */
40 | private ClassloaderUtility() {
41 | }
42 |
43 | public static ClassLoader getCustomClassloader(Collection entries) {
44 | List urls = new ArrayList();
45 | File file;
46 |
47 | if (entries != null) {
48 | for (String classPathEntry : entries) {
49 | file = new File(classPathEntry);
50 | if (!file.exists()) {
51 | throw new RuntimeException(getString(
52 | "RuntimeError.9", classPathEntry)); //$NON-NLS-1$
53 | }
54 |
55 | try {
56 | urls.add(file.toURI().toURL());
57 | } catch (MalformedURLException e) {
58 | // this shouldn't happen, but just in case...
59 | throw new RuntimeException(getString(
60 | "RuntimeError.9", classPathEntry)); //$NON-NLS-1$
61 | }
62 | }
63 | }
64 |
65 | ClassLoader parent = Thread.currentThread().getContextClassLoader();
66 |
67 | URLClassLoader ucl = new URLClassLoader(urls.toArray(new URL[urls
68 | .size()]), parent);
69 |
70 | return ucl;
71 | }
72 | }
73 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/XmlConstants.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2005 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.codegen;
17 |
18 | /**
19 | * @author Jeff Butler
20 | */
21 | public class XmlConstants {
22 |
23 | /**
24 | * Utility Class, no instances
25 | */
26 | private XmlConstants() {
27 | super();
28 | }
29 |
30 | public static final String IBATIS2_SQL_MAP_SYSTEM_ID = "http://ibatis.apache.org/dtd/sql-map-2.dtd"; //$NON-NLS-1$
31 |
32 | public static final String IBATIS2_SQL_MAP_PUBLIC_ID = "-//ibatis.apache.org//DTD SQL Map 2.0//EN"; //$NON-NLS-1$
33 |
34 | public static final String IBATIS2_SQL_MAP_CONFIG_SYSTEM_ID = "http://ibatis.apache.org/dtd/sql-map-config-2.dtd"; //$NON-NLS-1$
35 |
36 | public static final String IBATIS2_SQL_MAP_CONFIG_PUBLIC_ID = "-//ibatis.apache.org//DTD SQL Map Config 2.0//EN"; //$NON-NLS-1$
37 |
38 | public static final String MYBATIS3_MAPPER_SYSTEM_ID = "http://mybatis.org/dtd/mybatis-3-mapper.dtd"; //$NON-NLS-1$
39 |
40 | public static final String MYBATIS3_MAPPER_PUBLIC_ID = "-//mybatis.org//DTD Mapper 3.0//EN"; //$NON-NLS-1$
41 |
42 | public static final String MYBATIS3_MAPPER_CONFIG_SYSTEM_ID = "http://mybatis.org/dtd/mybatis-3-config.dtd"; //$NON-NLS-1$
43 |
44 | public static final String MYBATIS3_MAPPER_CONFIG_PUBLIC_ID = "-//mybatis.org//DTD Config 3.0//EN"; //$NON-NLS-1$
45 |
46 | public static final String IBATOR_CONFIG_SYSTEM_ID = "http://ibatis.apache.org/dtd/ibator-config_1_0.dtd"; //$NON-NLS-1$
47 |
48 | public static final String IBATOR_CONFIG_PUBLIC_ID = "-//Apache Software Foundation//DTD Apache iBATIS Ibator Configuration 1.0//EN"; //$NON-NLS-1$
49 |
50 | public static final String MYBATIS_GENERATOR_CONFIG_SYSTEM_ID = "http://mybatis.org/dtd/mybatis-generator-config_1_0.dtd"; //$NON-NLS-1$
51 |
52 | public static final String MYBATIS_GENERATOR_CONFIG_PUBLIC_ID = "-//mybatis.org//DTD MyBatis Generator Configuration 1.0//EN"; //$NON-NLS-1$
53 | }
54 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/internal/util/messages/Messages.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2006 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.internal.util.messages;
17 |
18 | import java.text.MessageFormat;
19 | import java.util.MissingResourceException;
20 | import java.util.ResourceBundle;
21 |
22 | /**
23 | * @author Jeff Butler
24 | */
25 | public class Messages {
26 | private static final String BUNDLE_NAME = "org.mybatis.generator.internal.util.messages.messages"; //$NON-NLS-1$
27 |
28 | private static final ResourceBundle RESOURCE_BUNDLE = ResourceBundle
29 | .getBundle(BUNDLE_NAME);
30 |
31 | private Messages() {
32 | }
33 |
34 | public static String getString(String key) {
35 | try {
36 | return RESOURCE_BUNDLE.getString(key);
37 | } catch (MissingResourceException e) {
38 | return '!' + key + '!';
39 | }
40 | }
41 |
42 | public static String getString(String key, String parm1) {
43 | try {
44 | return MessageFormat.format(RESOURCE_BUNDLE.getString(key),
45 | new Object[] { parm1 });
46 | } catch (MissingResourceException e) {
47 | return '!' + key + '!';
48 | }
49 | }
50 |
51 | public static String getString(String key, String parm1, String parm2) {
52 | try {
53 | return MessageFormat.format(RESOURCE_BUNDLE.getString(key),
54 | new Object[] { parm1, parm2 });
55 | } catch (MissingResourceException e) {
56 | return '!' + key + '!';
57 | }
58 | }
59 |
60 | public static String getString(String key, String parm1, String parm2,
61 | String parm3) {
62 | try {
63 | return MessageFormat.format(RESOURCE_BUNDLE.getString(key),
64 | new Object[] { parm1, parm2, parm3 });
65 | } catch (MissingResourceException e) {
66 | return '!' + key + '!';
67 | }
68 | }
69 | }
70 |
--------------------------------------------------------------------------------
/src/test/java/demo/MBPMain.java:
--------------------------------------------------------------------------------
1 | package demo;
2 |
3 | import org.mybatis.generator.api.MyBatisGenerator;
4 | import org.mybatis.generator.api.ProgressCallback;
5 | import org.mybatis.generator.config.Configuration;
6 | import org.mybatis.generator.config.xml.ConfigurationParser;
7 | import org.mybatis.generator.exception.InvalidConfigurationException;
8 | import org.mybatis.generator.exception.XMLParserException;
9 | import org.mybatis.generator.internal.DefaultShellCallback;
10 |
11 | import java.io.File;
12 | import java.io.IOException;
13 | import java.sql.SQLException;
14 | import java.util.ArrayList;
15 | import java.util.List;
16 |
17 | public class MBPMain {
18 |
19 | public static final String CFG_FILE_PATH = MBPMain.class.getResource("/MybatisGeneratorCfg.xml").getFile();
20 |
21 | public static void main(String[] args) {
22 | List warnings = new ArrayList<>();
23 | boolean overwrite = true;
24 | File configFile = new File(CFG_FILE_PATH);
25 | ConfigurationParser cp = new ConfigurationParser(warnings);
26 | Configuration config = null;
27 | try {
28 | config = cp.parseConfiguration(configFile);
29 | }
30 | catch (IOException e) {
31 | e.printStackTrace();
32 | }
33 | catch (XMLParserException e) {
34 | e.printStackTrace();
35 | }
36 | DefaultShellCallback callback = new DefaultShellCallback(overwrite);
37 | MyBatisGenerator myBatisGenerator = null;
38 | try {
39 | myBatisGenerator = new MyBatisGenerator(config, callback, warnings);
40 | }
41 | catch (InvalidConfigurationException e) {
42 | e.printStackTrace();
43 | }
44 | try {
45 | myBatisGenerator.generate(new ProgressCallback() {
46 | @Override
47 | public void startTask(String taskName) {
48 | System.out.println("startTask(String "+taskName+")");
49 | }
50 | @Override
51 | public void saveStarted(int totalTasks) {
52 | System.out.println("saveStarted(int "+totalTasks+")");
53 | }
54 | @Override
55 | public void introspectionStarted(int totalTasks) {
56 | System.out.println("introspectionStarted(int "+totalTasks+")");
57 | }
58 | @Override
59 | public void generationStarted(int totalTasks) {
60 | System.out.println("generationStarted(int "+totalTasks+")");
61 | }
62 | @Override
63 | public void done() {
64 | System.out.println("done()");
65 | }
66 | @Override
67 | public void checkCancel() throws InterruptedException {
68 | System.out.println("checkCancel()");
69 | }
70 | });
71 | }
72 | catch (SQLException e) {
73 | e.printStackTrace();
74 | }
75 | catch (IOException e) {
76 | e.printStackTrace();
77 | }
78 | catch (InterruptedException e) {
79 | e.printStackTrace();
80 | }
81 | }
82 |
83 | }
84 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/api/GeneratedFile.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2005 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.api;
17 |
18 |
19 | /**
20 | * Abstract class that holds information common to all generated files.
21 | *
22 | * @author Jeff Butler
23 | */
24 | public abstract class GeneratedFile {
25 | protected String targetProject;
26 |
27 | /**
28 | *
29 | */
30 | public GeneratedFile(String targetProject) {
31 | super();
32 | this.targetProject = targetProject;
33 | }
34 |
35 | /**
36 | * This method returns the entire contents of the generated file. Clients
37 | * can simply save the value returned from this method as the file contents.
38 | * Subclasses such as @see org.mybatis.generator.api.GeneratedJavaFile offer
39 | * more fine grained access to file parts, but still implement this method
40 | * in the event that the entire contents are desired.
41 | *
42 | * @return Returns the content.
43 | */
44 | public abstract String getFormattedContent();
45 |
46 | /**
47 | * Get the file name (without any path). Clients should use this method to
48 | * determine how to save the results.
49 | *
50 | * @return Returns the file name.
51 | */
52 | public abstract String getFileName();
53 |
54 | /**
55 | * Gets the target project. Clients can call this method to determine how to
56 | * save the results.
57 | *
58 | * @return the target project
59 | */
60 | public String getTargetProject() {
61 | return targetProject;
62 | }
63 |
64 | /**
65 | * Get the target package for the file. Clients should use this method to
66 | * determine how to save the results.
67 | *
68 | * @return Returns the target project.
69 | */
70 | public abstract String getTargetPackage();
71 |
72 | @Override
73 | public String toString() {
74 | return getFormattedContent();
75 | }
76 |
77 | public abstract boolean isMergeable();
78 | }
79 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/ibatis2/sqlmap/elements/DeleteByExampleElementGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2008 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.codegen.ibatis2.sqlmap.elements;
17 |
18 | import org.mybatis.generator.api.dom.xml.Attribute;
19 | import org.mybatis.generator.api.dom.xml.TextElement;
20 | import org.mybatis.generator.api.dom.xml.XmlElement;
21 |
22 | /**
23 | *
24 | * @author Jeff Butler
25 | *
26 | */
27 | public class DeleteByExampleElementGenerator extends
28 | AbstractXmlElementGenerator {
29 |
30 | public DeleteByExampleElementGenerator() {
31 | super();
32 | }
33 |
34 | @Override
35 | public void addElements(XmlElement parentElement) {
36 | XmlElement answer = new XmlElement("delete"); //$NON-NLS-1$
37 |
38 | answer.addAttribute(new Attribute(
39 | "id", introspectedTable.getDeleteByExampleStatementId())); //$NON-NLS-1$
40 | answer.addAttribute(new Attribute(
41 | "parameterClass", introspectedTable.getExampleType())); //$NON-NLS-1$
42 |
43 | context.getCommentGenerator().addComment(answer);
44 |
45 | StringBuilder sb = new StringBuilder();
46 | sb.append("delete from "); //$NON-NLS-1$
47 | sb.append(introspectedTable
48 | .getAliasedFullyQualifiedTableNameAtRuntime());
49 | answer.addElement(new TextElement(sb.toString()));
50 |
51 | XmlElement includeElement = new XmlElement("include"); //$NON-NLS-1$
52 | sb.setLength(0);
53 | sb.append(introspectedTable.getIbatis2SqlMapNamespace());
54 | sb.append('.');
55 | sb.append(introspectedTable.getExampleWhereClauseId());
56 | includeElement.addAttribute(new Attribute("refid", //$NON-NLS-1$
57 | sb.toString()));
58 |
59 | answer.addElement(includeElement);
60 |
61 | if (context.getPlugins().sqlMapDeleteByExampleElementGenerated(
62 | answer, introspectedTable)) {
63 | parentElement.addElement(answer);
64 | }
65 | }
66 | }
67 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/config/xml/ParserErrorHandler.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2005 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.config.xml;
17 |
18 | import static org.mybatis.generator.internal.util.messages.Messages.getString;
19 |
20 | import java.util.List;
21 |
22 | import org.xml.sax.ErrorHandler;
23 | import org.xml.sax.SAXException;
24 | import org.xml.sax.SAXParseException;
25 |
26 | /**
27 | * @author Jeff Butler
28 | */
29 | public class ParserErrorHandler implements ErrorHandler {
30 | private List warnings;
31 |
32 | private List errors;
33 |
34 | /**
35 | *
36 | */
37 | public ParserErrorHandler(List warnings, List errors) {
38 | super();
39 | this.warnings = warnings;
40 | this.errors = errors;
41 | }
42 |
43 | /*
44 | * (non-Javadoc)
45 | *
46 | * @see org.xml.sax.ErrorHandler#warning(org.xml.sax.SAXParseException)
47 | */
48 | public void warning(SAXParseException exception) throws SAXException {
49 | warnings.add(getString("Warning.7", //$NON-NLS-1$
50 | Integer.toString(exception.getLineNumber()), exception
51 | .getMessage()));
52 | }
53 |
54 | /*
55 | * (non-Javadoc)
56 | *
57 | * @see org.xml.sax.ErrorHandler#error(org.xml.sax.SAXParseException)
58 | */
59 | public void error(SAXParseException exception) throws SAXException {
60 | errors.add(getString("RuntimeError.4", //$NON-NLS-1$
61 | Integer.toString(exception.getLineNumber()), exception
62 | .getMessage()));
63 | }
64 |
65 | /*
66 | * (non-Javadoc)
67 | *
68 | * @see org.xml.sax.ErrorHandler#fatalError(org.xml.sax.SAXParseException)
69 | */
70 | public void fatalError(SAXParseException exception) throws SAXException {
71 | errors.add(getString("RuntimeError.4", //$NON-NLS-1$
72 | Integer.toString(exception.getLineNumber()), exception
73 | .getMessage()));
74 | }
75 | }
76 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/mybatis3/javamapper/elements/CountByExampleMethodGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2009 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
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 | * @author Jeff Butler
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 importedTypes = new TreeSet();
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")); //$NON-NLS-1$
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 | return;
66 | }
67 | }
68 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/mybatis3/javamapper/elements/DeleteByExampleMethodGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2009 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
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 | * @author Jeff Butler
30 | *
31 | */
32 | public class DeleteByExampleMethodGenerator extends
33 | AbstractJavaMapperMethodGenerator {
34 |
35 | public DeleteByExampleMethodGenerator() {
36 | super();
37 | }
38 |
39 | @Override
40 | public void addInterfaceElements(Interface interfaze) {
41 | Set importedTypes = new TreeSet();
42 | FullyQualifiedJavaType type = new FullyQualifiedJavaType(
43 | introspectedTable.getExampleType());
44 | importedTypes.add(type);
45 |
46 | Method method = new Method();
47 | method.setVisibility(JavaVisibility.PUBLIC);
48 | method.setReturnType(FullyQualifiedJavaType.getIntInstance());
49 | method.setName(introspectedTable.getDeleteByExampleStatementId());
50 | method.addParameter(new Parameter(type, "example")); //$NON-NLS-1$
51 |
52 | context.getCommentGenerator().addGeneralMethodComment(method,
53 | introspectedTable);
54 |
55 | addMapperAnnotations(interfaze, method);
56 |
57 | if (context.getPlugins().clientDeleteByExampleMethodGenerated(
58 | method, interfaze, introspectedTable)) {
59 | interfaze.addImportedTypes(importedTypes);
60 | interfaze.addMethod(method);
61 | }
62 | }
63 |
64 | public void addMapperAnnotations(Interface interfaze, Method method) {
65 | return;
66 | }
67 | }
68 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/ibatis2/sqlmap/elements/BaseColumnListElementGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2009 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.codegen.ibatis2.sqlmap.elements;
17 |
18 | import java.util.Iterator;
19 |
20 | import org.mybatis.generator.api.IntrospectedColumn;
21 | import org.mybatis.generator.api.dom.xml.Attribute;
22 | import org.mybatis.generator.api.dom.xml.TextElement;
23 | import org.mybatis.generator.api.dom.xml.XmlElement;
24 | import org.mybatis.generator.codegen.ibatis2.Ibatis2FormattingUtilities;
25 |
26 | /**
27 | *
28 | * @author Jeff Butler
29 | *
30 | */
31 | public class BaseColumnListElementGenerator extends AbstractXmlElementGenerator {
32 |
33 | public BaseColumnListElementGenerator() {
34 | super();
35 | }
36 |
37 | @Override
38 | public void addElements(XmlElement parentElement) {
39 | XmlElement answer = new XmlElement("sql"); //$NON-NLS-1$
40 |
41 | answer.addAttribute(new Attribute("id", //$NON-NLS-1$
42 | introspectedTable.getBaseColumnListId()));
43 |
44 | context.getCommentGenerator().addComment(answer);
45 |
46 | StringBuilder sb = new StringBuilder();
47 | Iterator iter = introspectedTable
48 | .getNonBLOBColumns().iterator();
49 | while (iter.hasNext()) {
50 | sb.append(Ibatis2FormattingUtilities.getSelectListPhrase(iter
51 | .next()));
52 |
53 | if (iter.hasNext()) {
54 | sb.append(", "); //$NON-NLS-1$
55 | }
56 |
57 | if (sb.length() > 80) {
58 | answer.addElement(new TextElement(sb.toString()));
59 | sb.setLength(0);
60 | }
61 | }
62 |
63 | if (sb.length() > 0) {
64 | answer.addElement((new TextElement(sb.toString())));
65 | }
66 |
67 | if (context.getPlugins().sqlMapBaseColumnListElementGenerated(
68 | answer, introspectedTable)) {
69 | parentElement.addElement(answer);
70 | }
71 | }
72 | }
73 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/ibatis2/sqlmap/elements/BlobColumnListElementGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2009 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.codegen.ibatis2.sqlmap.elements;
17 |
18 | import java.util.Iterator;
19 |
20 | import org.mybatis.generator.api.IntrospectedColumn;
21 | import org.mybatis.generator.api.dom.xml.Attribute;
22 | import org.mybatis.generator.api.dom.xml.TextElement;
23 | import org.mybatis.generator.api.dom.xml.XmlElement;
24 | import org.mybatis.generator.codegen.ibatis2.Ibatis2FormattingUtilities;
25 |
26 | /**
27 | *
28 | * @author Jeff Butler
29 | *
30 | */
31 | public class BlobColumnListElementGenerator extends AbstractXmlElementGenerator {
32 |
33 | public BlobColumnListElementGenerator() {
34 | super();
35 | }
36 |
37 | @Override
38 | public void addElements(XmlElement parentElement) {
39 | XmlElement answer = new XmlElement("sql"); //$NON-NLS-1$
40 |
41 | answer.addAttribute(new Attribute("id", //$NON-NLS-1$
42 | introspectedTable.getBlobColumnListId()));
43 |
44 | context.getCommentGenerator().addComment(answer);
45 |
46 | StringBuilder sb = new StringBuilder();
47 |
48 | Iterator iter = introspectedTable.getBLOBColumns()
49 | .iterator();
50 | while (iter.hasNext()) {
51 | sb.append(Ibatis2FormattingUtilities.getSelectListPhrase(iter
52 | .next()));
53 |
54 | if (iter.hasNext()) {
55 | sb.append(", "); //$NON-NLS-1$
56 | }
57 |
58 | if (sb.length() > 80) {
59 | answer.addElement(new TextElement(sb.toString()));
60 | sb.setLength(0);
61 | }
62 | }
63 |
64 | if (sb.length() > 0) {
65 | answer.addElement((new TextElement(sb.toString())));
66 | }
67 |
68 | if (context.getPlugins().sqlMapBlobColumnListElementGenerated(
69 | answer, introspectedTable)) {
70 | parentElement.addElement(answer);
71 | }
72 | }
73 | }
74 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/mybatis3/xmlmapper/elements/BaseColumnListElementGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2009 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.codegen.mybatis3.xmlmapper.elements;
17 |
18 | import java.util.Iterator;
19 |
20 | import org.mybatis.generator.api.IntrospectedColumn;
21 | import org.mybatis.generator.api.dom.xml.Attribute;
22 | import org.mybatis.generator.api.dom.xml.TextElement;
23 | import org.mybatis.generator.api.dom.xml.XmlElement;
24 | import org.mybatis.generator.codegen.mybatis3.MyBatis3FormattingUtilities;
25 |
26 | /**
27 | *
28 | * @author Jeff Butler
29 | *
30 | */
31 | public class BaseColumnListElementGenerator extends AbstractXmlElementGenerator {
32 |
33 | public BaseColumnListElementGenerator() {
34 | super();
35 | }
36 |
37 | @Override
38 | public void addElements(XmlElement parentElement) {
39 | XmlElement answer = new XmlElement("sql"); //$NON-NLS-1$
40 |
41 | answer.addAttribute(new Attribute("id", //$NON-NLS-1$
42 | introspectedTable.getBaseColumnListId()));
43 |
44 | context.getCommentGenerator().addComment(answer);
45 |
46 | StringBuilder sb = new StringBuilder();
47 | Iterator iter = introspectedTable
48 | .getNonBLOBColumns().iterator();
49 | while (iter.hasNext()) {
50 | sb.append(MyBatis3FormattingUtilities.getSelectListPhrase(iter
51 | .next()));
52 |
53 | if (iter.hasNext()) {
54 | sb.append(", "); //$NON-NLS-1$
55 | }
56 |
57 | if (sb.length() > 80) {
58 | answer.addElement(new TextElement(sb.toString()));
59 | sb.setLength(0);
60 | }
61 | }
62 |
63 | if (sb.length() > 0) {
64 | answer.addElement((new TextElement(sb.toString())));
65 | }
66 |
67 | if (context.getPlugins().sqlMapBaseColumnListElementGenerated(
68 | answer, introspectedTable)) {
69 | parentElement.addElement(answer);
70 | }
71 | }
72 | }
73 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/mybatis3/javamapper/elements/InsertSelectiveMethodGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2009 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
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 | * @author Jeff Butler
30 | *
31 | */
32 | public class InsertSelectiveMethodGenerator extends
33 | AbstractJavaMapperMethodGenerator {
34 |
35 | public InsertSelectiveMethodGenerator() {
36 | super();
37 | }
38 |
39 | @Override
40 | public void addInterfaceElements(Interface interfaze) {
41 | Set importedTypes = new TreeSet();
42 | Method method = new Method();
43 |
44 | method.setReturnType(FullyQualifiedJavaType.getIntInstance());
45 | method.setVisibility(JavaVisibility.PUBLIC);
46 | method.setName(introspectedTable.getInsertSelectiveStatementId());
47 |
48 | FullyQualifiedJavaType parameterType = introspectedTable.getRules()
49 | .calculateAllFieldsClass();
50 |
51 | importedTypes.add(parameterType);
52 | method.addParameter(new Parameter(parameterType, "record")); //$NON-NLS-1$
53 |
54 | context.getCommentGenerator().addGeneralMethodComment(method,
55 | introspectedTable);
56 |
57 | addMapperAnnotations(interfaze, method);
58 |
59 | if (context.getPlugins().clientInsertSelectiveMethodGenerated(
60 | method, interfaze, introspectedTable)) {
61 | interfaze.addImportedTypes(importedTypes);
62 | interfaze.addMethod(method);
63 | }
64 | }
65 |
66 | public void addMapperAnnotations(Interface interfaze, Method method) {
67 | return;
68 | }
69 | }
70 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/mybatis3/xmlmapper/elements/BlobColumnListElementGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2009 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.codegen.mybatis3.xmlmapper.elements;
17 |
18 | import java.util.Iterator;
19 |
20 | import org.mybatis.generator.api.IntrospectedColumn;
21 | import org.mybatis.generator.api.dom.xml.Attribute;
22 | import org.mybatis.generator.api.dom.xml.TextElement;
23 | import org.mybatis.generator.api.dom.xml.XmlElement;
24 | import org.mybatis.generator.codegen.mybatis3.MyBatis3FormattingUtilities;
25 |
26 | /**
27 | *
28 | * @author Jeff Butler
29 | *
30 | */
31 | public class BlobColumnListElementGenerator extends AbstractXmlElementGenerator {
32 |
33 | public BlobColumnListElementGenerator() {
34 | super();
35 | }
36 |
37 | @Override
38 | public void addElements(XmlElement parentElement) {
39 | XmlElement answer = new XmlElement("sql"); //$NON-NLS-1$
40 |
41 | answer.addAttribute(new Attribute("id", //$NON-NLS-1$
42 | introspectedTable.getBlobColumnListId()));
43 |
44 | context.getCommentGenerator().addComment(answer);
45 |
46 | StringBuilder sb = new StringBuilder();
47 |
48 | Iterator iter = introspectedTable.getBLOBColumns()
49 | .iterator();
50 | while (iter.hasNext()) {
51 | sb.append(MyBatis3FormattingUtilities.getSelectListPhrase(iter
52 | .next()));
53 |
54 | if (iter.hasNext()) {
55 | sb.append(", "); //$NON-NLS-1$
56 | }
57 |
58 | if (sb.length() > 80) {
59 | answer.addElement(new TextElement(sb.toString()));
60 | sb.setLength(0);
61 | }
62 | }
63 |
64 | if (sb.length() > 0) {
65 | answer.addElement((new TextElement(sb.toString())));
66 | }
67 |
68 | if (context.getPlugins().sqlMapBlobColumnListElementGenerated(
69 | answer, introspectedTable)) {
70 | parentElement.addElement(answer);
71 | }
72 | }
73 | }
74 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/ibatis2/sqlmap/elements/CountByExampleElementGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2008 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.codegen.ibatis2.sqlmap.elements;
17 |
18 | import org.mybatis.generator.api.dom.xml.Attribute;
19 | import org.mybatis.generator.api.dom.xml.TextElement;
20 | import org.mybatis.generator.api.dom.xml.XmlElement;
21 |
22 | /**
23 | *
24 | * @author Jeff Butler
25 | *
26 | */
27 | public class CountByExampleElementGenerator extends AbstractXmlElementGenerator {
28 |
29 | public CountByExampleElementGenerator() {
30 | super();
31 | }
32 |
33 | @Override
34 | public void addElements(XmlElement parentElement) {
35 | XmlElement answer = new XmlElement("select"); //$NON-NLS-1$
36 |
37 | answer.addAttribute(new Attribute(
38 | "id", introspectedTable.getCountByExampleStatementId())); //$NON-NLS-1$
39 | answer.addAttribute(new Attribute(
40 | "parameterClass", introspectedTable.getExampleType())); //$NON-NLS-1$
41 | answer.addAttribute(new Attribute("resultClass", "java.lang.Integer")); //$NON-NLS-1$ //$NON-NLS-2$
42 |
43 | context.getCommentGenerator().addComment(answer);
44 |
45 | StringBuilder sb = new StringBuilder();
46 | sb.append("select count(*) from "); //$NON-NLS-1$
47 | sb.append(introspectedTable
48 | .getAliasedFullyQualifiedTableNameAtRuntime());
49 | answer.addElement(new TextElement(sb.toString()));
50 |
51 | XmlElement includeElement = new XmlElement("include"); //$NON-NLS-1$
52 | sb.setLength(0);
53 | sb.append(introspectedTable.getIbatis2SqlMapNamespace());
54 | sb.append('.');
55 | sb.append(introspectedTable.getExampleWhereClauseId());
56 | includeElement.addAttribute(new Attribute("refid", //$NON-NLS-1$
57 | sb.toString()));
58 |
59 | answer.addElement(includeElement);
60 |
61 | if (context.getPlugins().sqlMapCountByExampleElementGenerated(
62 | answer, introspectedTable)) {
63 | parentElement.addElement(answer);
64 | }
65 | }
66 | }
67 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/api/dom/xml/Document.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2006 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.api.dom.xml;
17 |
18 | import org.mybatis.generator.api.dom.OutputUtilities;
19 |
20 | /**
21 | * @author Jeff Butler
22 | */
23 | public class Document {
24 | private String publicId;
25 |
26 | private String systemId;
27 |
28 | private XmlElement rootElement;
29 |
30 | /**
31 | *
32 | */
33 | public Document(String publicId, String systemId) {
34 | super();
35 | this.publicId = publicId;
36 | this.systemId = systemId;
37 | }
38 |
39 | public Document() {
40 | super();
41 | }
42 |
43 | /**
44 | * @return Returns the rootElement.
45 | */
46 | public XmlElement getRootElement() {
47 | return rootElement;
48 | }
49 |
50 | /**
51 | * @param rootElement
52 | * The rootElement to set.
53 | */
54 | public void setRootElement(XmlElement rootElement) {
55 | this.rootElement = rootElement;
56 | }
57 |
58 | /**
59 | * @return Returns the publicId.
60 | */
61 | public String getPublicId() {
62 | return publicId;
63 | }
64 |
65 | /**
66 | * @return Returns the systemId.
67 | */
68 | public String getSystemId() {
69 | return systemId;
70 | }
71 |
72 | public String getFormattedContent() {
73 | StringBuilder sb = new StringBuilder();
74 |
75 | sb.append(""); //$NON-NLS-1$
76 |
77 | if (publicId != null && systemId != null) {
78 | OutputUtilities.newLine(sb);
79 | sb.append(""); //$NON-NLS-1$
86 | }
87 |
88 | OutputUtilities.newLine(sb);
89 | sb.append(rootElement.getFormattedContent(0));
90 |
91 | return sb.toString();
92 | }
93 | }
94 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/ibatis2/dao/templates/SpringDAOTemplate.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2006 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.codegen.ibatis2.dao.templates;
17 |
18 | import org.mybatis.generator.api.dom.java.FullyQualifiedJavaType;
19 | import org.mybatis.generator.api.dom.java.JavaVisibility;
20 | import org.mybatis.generator.api.dom.java.Method;
21 |
22 | /**
23 | * @author Jeff Butler
24 | */
25 | public class SpringDAOTemplate extends AbstractDAOTemplate {
26 |
27 | /**
28 | *
29 | */
30 | public SpringDAOTemplate() {
31 | super();
32 | }
33 |
34 | @Override
35 | protected void configureConstructorTemplate() {
36 | Method method = new Method();
37 | method.setConstructor(true);
38 | method.setVisibility(JavaVisibility.PUBLIC);
39 | method.addBodyLine("super();"); //$NON-NLS-1$
40 | setConstructorTemplate(method);
41 | }
42 |
43 | @Override
44 | protected void configureDeleteMethodTemplate() {
45 | setDeleteMethodTemplate("getSqlMapClientTemplate().delete(\"{0}.{1}\", {2});"); //$NON-NLS-1$
46 | }
47 |
48 | @Override
49 | protected void configureInsertMethodTemplate() {
50 | setInsertMethodTemplate("getSqlMapClientTemplate().insert(\"{0}.{1}\", {2});"); //$NON-NLS-1$
51 | }
52 |
53 | @Override
54 | protected void configureQueryForListMethodTemplate() {
55 | setQueryForListMethodTemplate("getSqlMapClientTemplate().queryForList(\"{0}.{1}\", {2});"); //$NON-NLS-1$
56 | }
57 |
58 | @Override
59 | protected void configureQueryForObjectMethodTemplate() {
60 | setQueryForObjectMethodTemplate("getSqlMapClientTemplate().queryForObject(\"{0}.{1}\", {2});"); //$NON-NLS-1$
61 | }
62 |
63 | @Override
64 | protected void configureSuperClass() {
65 | setSuperClass(new FullyQualifiedJavaType(
66 | "org.springframework.orm.ibatis.support.SqlMapClientDaoSupport")); //$NON-NLS-1$
67 | }
68 |
69 | @Override
70 | protected void configureUpdateMethodTemplate() {
71 | setUpdateMethodTemplate("getSqlMapClientTemplate().update(\"{0}.{1}\", {2});"); //$NON-NLS-1$
72 | }
73 | }
74 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/mybatis3/javamapper/elements/UpdateByPrimaryKeyWithoutBLOBsMethodGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2009 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
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 | * @author Jeff Butler
30 | *
31 | */
32 | public class UpdateByPrimaryKeyWithoutBLOBsMethodGenerator extends
33 | AbstractJavaMapperMethodGenerator {
34 |
35 | public UpdateByPrimaryKeyWithoutBLOBsMethodGenerator() {
36 | super();
37 | }
38 |
39 | @Override
40 | public void addInterfaceElements(Interface interfaze) {
41 | Set importedTypes = new TreeSet();
42 | FullyQualifiedJavaType parameterType = new FullyQualifiedJavaType(
43 | introspectedTable.getBaseRecordType());
44 | importedTypes.add(parameterType);
45 |
46 | Method method = new Method();
47 | method.setVisibility(JavaVisibility.PUBLIC);
48 | method.setReturnType(FullyQualifiedJavaType.getIntInstance());
49 | method.setName(introspectedTable.getUpdateByPrimaryKeyStatementId());
50 | method.addParameter(new Parameter(parameterType, "record")); //$NON-NLS-1$
51 |
52 | context.getCommentGenerator().addGeneralMethodComment(method,
53 | introspectedTable);
54 |
55 | addMapperAnnotations(interfaze, method);
56 |
57 | if (context.getPlugins()
58 | .clientUpdateByPrimaryKeyWithoutBLOBsMethodGenerated(method,
59 | interfaze, introspectedTable)) {
60 | interfaze.addImportedTypes(importedTypes);
61 | interfaze.addMethod(method);
62 | }
63 | }
64 |
65 | public void addMapperAnnotations(Interface interfaze, Method method) {
66 | return;
67 | }
68 | }
69 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/mybatis3/javamapper/elements/SelectAllMethodGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2012 The MyBatis Team
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
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 | * The selectAll method is only generated by the Mybatis3Simple runtime.
28 | *
29 | * @author Jeff Butler
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 importedTypes = new TreeSet();
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 | return;
71 | }
72 | }
73 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/config/SqlMapGeneratorConfiguration.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2005 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.config;
17 |
18 | import static org.mybatis.generator.internal.util.StringUtility.stringHasValue;
19 | import static org.mybatis.generator.internal.util.messages.Messages.getString;
20 |
21 | import java.util.List;
22 |
23 | import org.mybatis.generator.api.dom.xml.Attribute;
24 | import org.mybatis.generator.api.dom.xml.XmlElement;
25 |
26 | /**
27 | * @author Jeff Butler
28 | */
29 | public class SqlMapGeneratorConfiguration extends PropertyHolder {
30 | private String targetPackage;
31 |
32 | private String targetProject;
33 |
34 | /**
35 | *
36 | */
37 | public SqlMapGeneratorConfiguration() {
38 | super();
39 | }
40 |
41 | public String getTargetProject() {
42 | return targetProject;
43 | }
44 |
45 | public void setTargetProject(String targetProject) {
46 | this.targetProject = targetProject;
47 | }
48 |
49 | public String getTargetPackage() {
50 | return targetPackage;
51 | }
52 |
53 | public void setTargetPackage(String targetPackage) {
54 | this.targetPackage = targetPackage;
55 | }
56 |
57 | public XmlElement toXmlElement() {
58 | XmlElement answer = new XmlElement("sqlMapGenerator"); //$NON-NLS-1$
59 |
60 | if (targetPackage != null) {
61 | answer.addAttribute(new Attribute("targetPackage", targetPackage)); //$NON-NLS-1$
62 | }
63 |
64 | if (targetProject != null) {
65 | answer.addAttribute(new Attribute("targetProject", targetProject)); //$NON-NLS-1$
66 | }
67 |
68 | addPropertyXmlElements(answer);
69 |
70 | return answer;
71 | }
72 |
73 | public void validate(List errors, String contextId) {
74 | if (!stringHasValue(targetProject)) {
75 | errors.add(getString("ValidationError.1", contextId)); //$NON-NLS-1$
76 | }
77 |
78 | if (!stringHasValue(targetPackage)) {
79 | errors.add(getString("ValidationError.12", //$NON-NLS-1$
80 | "SQLMapGenerator", contextId)); //$NON-NLS-1$
81 | }
82 | }
83 | }
84 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/config/JavaModelGeneratorConfiguration.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2005 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.config;
17 |
18 | import static org.mybatis.generator.internal.util.StringUtility.stringHasValue;
19 | import static org.mybatis.generator.internal.util.messages.Messages.getString;
20 |
21 | import java.util.List;
22 |
23 | import org.mybatis.generator.api.dom.xml.Attribute;
24 | import org.mybatis.generator.api.dom.xml.XmlElement;
25 |
26 | /**
27 | * @author Jeff Butler
28 | */
29 | public class JavaModelGeneratorConfiguration extends PropertyHolder {
30 |
31 | private String targetPackage;
32 |
33 | private String targetProject;
34 |
35 | /**
36 | *
37 | */
38 | public JavaModelGeneratorConfiguration() {
39 | super();
40 | }
41 |
42 | public String getTargetProject() {
43 | return targetProject;
44 | }
45 |
46 | public void setTargetProject(String targetProject) {
47 | this.targetProject = targetProject;
48 | }
49 |
50 | public String getTargetPackage() {
51 | return targetPackage;
52 | }
53 |
54 | public void setTargetPackage(String targetPackage) {
55 | this.targetPackage = targetPackage;
56 | }
57 |
58 | public XmlElement toXmlElement() {
59 | XmlElement answer = new XmlElement("javaModelGenerator"); //$NON-NLS-1$
60 |
61 | if (targetPackage != null) {
62 | answer.addAttribute(new Attribute("targetPackage", targetPackage)); //$NON-NLS-1$
63 | }
64 |
65 | if (targetProject != null) {
66 | answer.addAttribute(new Attribute("targetProject", targetProject)); //$NON-NLS-1$
67 | }
68 |
69 | addPropertyXmlElements(answer);
70 |
71 | return answer;
72 | }
73 |
74 | public void validate(List errors, String contextId) {
75 | if (!stringHasValue(targetProject)) {
76 | errors.add(getString("ValidationError.0", contextId)); //$NON-NLS-1$
77 | }
78 |
79 | if (!stringHasValue(targetPackage)) {
80 | errors.add(getString("ValidationError.12", //$NON-NLS-1$
81 | "JavaModelGenerator", contextId)); //$NON-NLS-1$
82 | }
83 | }
84 | }
85 |
--------------------------------------------------------------------------------
/src/test/java/demo/domain/oob/OperateLog.java:
--------------------------------------------------------------------------------
1 | package demo.domain.oob;
2 |
3 | import java.io.Serializable;
4 |
5 | public class OperateLog implements Serializable {
6 | private Long id;
7 |
8 | private String action;
9 |
10 | private String data;
11 |
12 | private Long authorId;
13 |
14 | private String ip;
15 |
16 | private Long created;
17 |
18 | private static final long serialVersionUID = 1L;
19 |
20 | public Long getId() {
21 | return id;
22 | }
23 |
24 | public void setId(Long id) {
25 | this.id = id;
26 | }
27 |
28 | public String getAction() {
29 | return action;
30 | }
31 |
32 | public void setAction(String action) {
33 | this.action = action;
34 | }
35 |
36 | public String getData() {
37 | return data;
38 | }
39 |
40 | public void setData(String data) {
41 | this.data = data;
42 | }
43 |
44 | public Long getAuthorId() {
45 | return authorId;
46 | }
47 |
48 | public void setAuthorId(Long authorId) {
49 | this.authorId = authorId;
50 | }
51 |
52 | public String getIp() {
53 | return ip;
54 | }
55 |
56 | public void setIp(String ip) {
57 | this.ip = ip;
58 | }
59 |
60 | public Long getCreated() {
61 | return created;
62 | }
63 |
64 | public void setCreated(Long created) {
65 | this.created = created;
66 | }
67 |
68 | private OperateLog(Builder b) {
69 | id = b.id;
70 | action = b.action;
71 | data = b.data;
72 | authorId = b.authorId;
73 | ip = b.ip;
74 | created = b.created;
75 | }
76 |
77 | public OperateLog() {
78 | super();
79 | }
80 |
81 | public static class Builder {
82 | private Long id;
83 |
84 | private String action;
85 |
86 | private String data;
87 |
88 | private Long authorId;
89 |
90 | private String ip;
91 |
92 | private Long created;
93 |
94 | public Builder id(Long id) {
95 | this.id = id;
96 | return this;
97 | }
98 |
99 | public Builder action(String action) {
100 | this.action = action;
101 | return this;
102 | }
103 |
104 | public Builder data(String data) {
105 | this.data = data;
106 | return this;
107 | }
108 |
109 | public Builder authorId(Long authorId) {
110 | this.authorId = authorId;
111 | return this;
112 | }
113 |
114 | public Builder ip(String ip) {
115 | this.ip = ip;
116 | return this;
117 | }
118 |
119 | public Builder created(Long created) {
120 | this.created = created;
121 | return this;
122 | }
123 |
124 | public OperateLog build() {
125 | return new OperateLog(this);
126 | }
127 | }
128 | }
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/logging/LogFactory.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2009 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.logging;
17 |
18 | import static org.mybatis.generator.internal.util.messages.Messages.getString;
19 |
20 | import org.mybatis.generator.internal.ObjectFactory;
21 |
22 | /**
23 | * Factory for creating loggers. Uses runtime introspection to determine the
24 | * AbstractLogFactory implementation.
25 | *
26 | * @author Jeff Butler
27 | *
28 | */
29 | public class LogFactory {
30 | private static AbstractLogFactory logFactory;
31 |
32 | static {
33 | try {
34 | ObjectFactory.internalClassForName("org.apache.log4j.Logger"); //$NON-NLS-1$
35 | logFactory = new Log4jLoggingLogFactory();
36 | } catch (Exception e) {
37 | logFactory = new JdkLoggingLogFactory();
38 | }
39 | }
40 |
41 | public static Log getLog(Class clazz) {
42 | try {
43 | return logFactory.getLog(clazz);
44 | } catch (Throwable t) {
45 | throw new RuntimeException(getString("RuntimeError.21", //$NON-NLS-1$
46 | clazz.getName(), t.getMessage()), t);
47 | }
48 | }
49 |
50 | /**
51 | * This method will switch the logging implementation to Java native
52 | * logging. This is useful in situations where you want to use Java native
53 | * logging to log activity but Log4J is on the classpath. Note that
54 | * this method is only effective for log classes obtained after calling this
55 | * method. If you intend to use this method you should call it before
56 | * calling any other method.
57 | */
58 | public static synchronized void forceJavaLogging() {
59 | logFactory = new JdkLoggingLogFactory();
60 | }
61 |
62 | private static class JdkLoggingLogFactory implements AbstractLogFactory {
63 | public Log getLog(Class clazz) {
64 | return new JdkLoggingImpl(clazz);
65 | }
66 | }
67 |
68 | private static class Log4jLoggingLogFactory implements AbstractLogFactory {
69 | public Log getLog(Class clazz) {
70 | return new Log4jImpl(clazz);
71 | }
72 | }
73 |
74 | public static void setLogFactory(AbstractLogFactory logFactory) {
75 | LogFactory.logFactory = logFactory;
76 | }
77 | }
78 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/mybatis3/javamapper/elements/InsertMethodGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2009 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
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 | * @author Jeff Butler
30 | *
31 | */
32 | public class InsertMethodGenerator extends AbstractJavaMapperMethodGenerator {
33 |
34 | boolean isSimple;
35 |
36 | public InsertMethodGenerator(boolean isSimple) {
37 | super();
38 | this.isSimple = isSimple;
39 | }
40 |
41 | @Override
42 | public void addInterfaceElements(Interface interfaze) {
43 | Set importedTypes = new TreeSet();
44 | Method method = new Method();
45 |
46 | method.setReturnType(FullyQualifiedJavaType.getIntInstance());
47 | method.setVisibility(JavaVisibility.PUBLIC);
48 | method.setName(introspectedTable.getInsertStatementId());
49 |
50 | FullyQualifiedJavaType parameterType;
51 | if (isSimple) {
52 | parameterType = new FullyQualifiedJavaType(
53 | introspectedTable.getBaseRecordType());
54 | } else {
55 | parameterType = introspectedTable.getRules()
56 | .calculateAllFieldsClass();
57 | }
58 |
59 | importedTypes.add(parameterType);
60 | method.addParameter(new Parameter(parameterType, "record")); //$NON-NLS-1$
61 |
62 | context.getCommentGenerator().addGeneralMethodComment(method,
63 | introspectedTable);
64 |
65 | addMapperAnnotations(interfaze, method);
66 |
67 | if (context.getPlugins().clientInsertMethodGenerated(method, interfaze,
68 | introspectedTable)) {
69 | interfaze.addImportedTypes(importedTypes);
70 | interfaze.addMethod(method);
71 | }
72 | }
73 |
74 | public void addMapperAnnotations(Interface interfaze, Method method) {
75 | return;
76 | }
77 | }
78 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/api/GeneratedJavaFile.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2005 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.api;
17 |
18 | import org.mybatis.generator.api.dom.java.CompilationUnit;
19 |
20 | /**
21 | * @author Jeff Butler
22 | */
23 | public class GeneratedJavaFile extends GeneratedFile {
24 | private CompilationUnit compilationUnit;
25 | private String fileEncoding;
26 | private JavaFormatter javaFormatter;
27 |
28 | /**
29 | * Default constructor
30 | */
31 | public GeneratedJavaFile(CompilationUnit compilationUnit,
32 | String targetProject,
33 | String fileEncoding,
34 | JavaFormatter javaFormatter) {
35 | super(targetProject);
36 | this.compilationUnit = compilationUnit;
37 | this.fileEncoding = fileEncoding;
38 | this.javaFormatter = javaFormatter;
39 | }
40 |
41 | public GeneratedJavaFile(CompilationUnit compilationUnit,
42 | String targetProject,
43 | JavaFormatter javaFormatter) {
44 | this(compilationUnit, targetProject, null, javaFormatter);
45 | }
46 |
47 | @Override
48 | public String getFormattedContent() {
49 | return javaFormatter.getFormattedContent(compilationUnit);
50 | }
51 |
52 | @Override
53 | public String getFileName() {
54 | return compilationUnit.getType().getShortName() + ".java"; //$NON-NLS-1$
55 | }
56 |
57 | public String getTargetPackage() {
58 | return compilationUnit.getType().getPackageName();
59 | }
60 |
61 | /**
62 | * This method is required by the Eclipse Java merger. If you are not
63 | * running in Eclipse, or some other system that implements the Java merge
64 | * function, you may return null from this method.
65 | *
66 | * @return the CompilationUnit associated with this file, or null if the
67 | * file is not mergeable.
68 | */
69 | public CompilationUnit getCompilationUnit() {
70 | return compilationUnit;
71 | }
72 |
73 | /**
74 | * A Java file is mergeable if the getCompilationUnit() method returns a
75 | * valid compilation unit.
76 | *
77 | */
78 | @Override
79 | public boolean isMergeable() {
80 | return true;
81 | }
82 |
83 | public String getFileEncoding() {
84 | return fileEncoding;
85 | }
86 | }
87 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/internal/rules/ConditionalModelRules.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2006 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 |
17 | package org.mybatis.generator.internal.rules;
18 |
19 | import org.mybatis.generator.api.IntrospectedTable;
20 |
21 | /**
22 | * This class encapsulates all the code generation rules for a table using the
23 | * conditional model. In this model we do not generate primary key or record
24 | * with BLOBs classes if the class would only hold one field.
25 | *
26 | * @author Jeff Butler
27 | *
28 | */
29 | public class ConditionalModelRules extends BaseRules {
30 |
31 | /**
32 | *
33 | */
34 | public ConditionalModelRules(IntrospectedTable introspectedTable) {
35 | super(introspectedTable);
36 | }
37 |
38 | /**
39 | * We generate a primary key if there is more than one primary key field.
40 | *
41 | * @return true if the primary key should be generated
42 | */
43 | public boolean generatePrimaryKeyClass() {
44 | return introspectedTable.getPrimaryKeyColumns().size() > 1;
45 | }
46 |
47 | /**
48 | * Generate a base record if there are any base columns, or if there is only
49 | * one primary key coulmn (in which case we will not generate a primary key
50 | * class), or if there is only one BLOB column (in which case we will not
51 | * generate a record with BLOBs class).
52 | *
53 | * @return true if the class should be generated
54 | */
55 | public boolean generateBaseRecordClass() {
56 | return introspectedTable.getBaseColumns().size() > 0
57 | || introspectedTable.getPrimaryKeyColumns().size() == 1
58 | || (introspectedTable.getBLOBColumns().size() > 0 && !generateRecordWithBLOBsClass());
59 |
60 | }
61 |
62 | /**
63 | * We generate a record with BLOBs class if there is more than one BLOB
64 | * column. Do not generate a BLOBs class if any other super class would only
65 | * contain one field
66 | *
67 | * @return true if the record with BLOBs class should be generated
68 | */
69 | public boolean generateRecordWithBLOBsClass() {
70 | int otherColumnCount = introspectedTable.getPrimaryKeyColumns().size()
71 | + introspectedTable.getBaseColumns().size();
72 |
73 | return otherColumnCount > 1
74 | && introspectedTable.getBLOBColumns().size() > 1;
75 | }
76 | }
77 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/ibatis2/dao/templates/IbatisDAOTemplate.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2006 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.codegen.ibatis2.dao.templates;
17 |
18 | import org.mybatis.generator.api.dom.java.FullyQualifiedJavaType;
19 | import org.mybatis.generator.api.dom.java.JavaVisibility;
20 | import org.mybatis.generator.api.dom.java.Method;
21 | import org.mybatis.generator.api.dom.java.Parameter;
22 |
23 | /**
24 | * @author Jeff Butler
25 | */
26 | public class IbatisDAOTemplate extends AbstractDAOTemplate {
27 |
28 | private FullyQualifiedJavaType fqjt = new FullyQualifiedJavaType(
29 | "com.ibatis.dao.client.DaoManager"); //$NON-NLS-1$
30 |
31 | /**
32 | *
33 | */
34 | public IbatisDAOTemplate() {
35 | super();
36 | }
37 |
38 | @Override
39 | protected void configureConstructorTemplate() {
40 | Method method = new Method();
41 | method.setConstructor(true);
42 | method.setVisibility(JavaVisibility.PUBLIC);
43 | method.addParameter(new Parameter(fqjt, "daoManager")); //$NON-NLS-1$
44 | method.addBodyLine("super(daoManager);"); //$NON-NLS-1$
45 | setConstructorTemplate(method);
46 | }
47 |
48 | @Override
49 | protected void configureDeleteMethodTemplate() {
50 | setDeleteMethodTemplate("delete(\"{0}.{1}\", {2});"); //$NON-NLS-1$
51 | }
52 |
53 | @Override
54 | protected void configureImplementationImports() {
55 | addImplementationImport(fqjt);
56 | }
57 |
58 | @Override
59 | protected void configureInsertMethodTemplate() {
60 | setInsertMethodTemplate("insert(\"{0}.{1}\", {2});"); //$NON-NLS-1$
61 | }
62 |
63 | @Override
64 | protected void configureQueryForListMethodTemplate() {
65 | setQueryForListMethodTemplate("queryForList(\"{0}.{1}\", {2});"); //$NON-NLS-1$
66 | }
67 |
68 | @Override
69 | protected void configureQueryForObjectMethodTemplate() {
70 | setQueryForObjectMethodTemplate("queryForObject(\"{0}.{1}\", {2});"); //$NON-NLS-1$
71 | }
72 |
73 | @Override
74 | protected void configureSuperClass() {
75 | setSuperClass(new FullyQualifiedJavaType(
76 | "com.ibatis.dao.client.template.SqlMapDaoTemplate")); //$NON-NLS-1$
77 | }
78 |
79 | @Override
80 | protected void configureUpdateMethodTemplate() {
81 | setUpdateMethodTemplate("update(\"{0}.{1}\", {2});"); //$NON-NLS-1$
82 | }
83 | }
84 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/mybatis3/javamapper/elements/UpdateByPrimaryKeyWithBLOBsMethodGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2009 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
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 | * @author Jeff Butler
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 importedTypes = new TreeSet();
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")); //$NON-NLS-1$
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 | return;
77 | }
78 | }
79 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/mybatis3/javamapper/elements/UpdateByPrimaryKeySelectiveMethodGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2009 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
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 | * @author Jeff Butler
30 | *
31 | */
32 | public class UpdateByPrimaryKeySelectiveMethodGenerator extends
33 | AbstractJavaMapperMethodGenerator {
34 |
35 | public UpdateByPrimaryKeySelectiveMethodGenerator() {
36 | super();
37 | }
38 |
39 | @Override
40 | public void addInterfaceElements(Interface interfaze) {
41 | Set importedTypes = new TreeSet();
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 | method.setName(introspectedTable
58 | .getUpdateByPrimaryKeySelectiveStatementId());
59 | method.addParameter(new Parameter(parameterType, "record")); //$NON-NLS-1$
60 |
61 | context.getCommentGenerator().addGeneralMethodComment(method,
62 | introspectedTable);
63 |
64 | addMapperAnnotations(interfaze, method);
65 |
66 | if (context.getPlugins()
67 | .clientUpdateByPrimaryKeySelectiveMethodGenerated(method,
68 | interfaze, introspectedTable)) {
69 | interfaze.addImportedTypes(importedTypes);
70 | interfaze.addMethod(method);
71 | }
72 | }
73 |
74 | public void addMapperAnnotations(Interface interfaze, Method method) {
75 | return;
76 | }
77 | }
78 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/api/dom/java/Parameter.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2006 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.api.dom.java;
17 |
18 | import java.util.ArrayList;
19 | import java.util.List;
20 |
21 | /**
22 | * @author Jeff Butler
23 | */
24 | public class Parameter {
25 | private String name;
26 | private FullyQualifiedJavaType type;
27 | private boolean isVarargs;
28 |
29 | private List annotations;
30 |
31 | public Parameter(FullyQualifiedJavaType type, String name, boolean isVarargs) {
32 | super();
33 | this.name = name;
34 | this.type = type;
35 | this.isVarargs = isVarargs;
36 | annotations = new ArrayList();
37 | }
38 |
39 | public Parameter(FullyQualifiedJavaType type, String name) {
40 | this(type, name, false);
41 | }
42 |
43 | public Parameter(FullyQualifiedJavaType type, String name, String annotation) {
44 | this(type, name, false);
45 | addAnnotation(annotation);
46 | }
47 |
48 | public Parameter(FullyQualifiedJavaType type, String name, String annotation, boolean isVarargs) {
49 | this(type, name, isVarargs);
50 | addAnnotation(annotation);
51 | }
52 |
53 | /**
54 | * @return Returns the name.
55 | */
56 | public String getName() {
57 | return name;
58 | }
59 |
60 | /**
61 | * @return Returns the type.
62 | */
63 | public FullyQualifiedJavaType getType() {
64 | return type;
65 | }
66 |
67 | public List getAnnotations() {
68 | return annotations;
69 | }
70 |
71 | public void addAnnotation(String annotation) {
72 | annotations.add(annotation);
73 | }
74 |
75 | public String getFormattedContent() {
76 | StringBuilder sb = new StringBuilder();
77 |
78 | for (String annotation : annotations) {
79 | sb.append(annotation);
80 | sb.append(' ');
81 | }
82 |
83 | sb.append(type.getShortName());
84 | sb.append(' ');
85 | if (isVarargs) {
86 | sb.append("... "); //$NON-NLS-1$
87 | }
88 | sb.append(name);
89 |
90 | return sb.toString();
91 | }
92 |
93 | @Override
94 | public String toString() {
95 | return getFormattedContent();
96 | }
97 |
98 | public boolean isVarargs() {
99 | return isVarargs;
100 | }
101 | }
102 |
--------------------------------------------------------------------------------
/maven-plugin/src/main/java/org/ihansen/mbp/extend/maven/MavenShellCallback.java:
--------------------------------------------------------------------------------
1 | /**
2 | * Copyright 2006-2017 the original author or authors.
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.ihansen.mbp.extend.maven;
17 |
18 | import org.mybatis.generator.exception.ShellException;
19 | import org.mybatis.generator.internal.DefaultShellCallback;
20 | import org.mybatis.generator.internal.util.messages.Messages;
21 |
22 | import java.io.File;
23 | import java.util.StringTokenizer;
24 |
25 | /**
26 | * Shell callback that calculates the Maven output directory.
27 | *
28 | * @author Jeff Butler
29 | */
30 | public class MavenShellCallback extends DefaultShellCallback {
31 | private MyBatisGeneratorMojo mybatisGeneratorMojo;
32 |
33 | public MavenShellCallback(MyBatisGeneratorMojo mybatisGeneratorMojo, boolean overwrite) {
34 | super(overwrite);
35 | this.mybatisGeneratorMojo = mybatisGeneratorMojo;
36 | }
37 |
38 | @Override
39 | public File getDirectory(String targetProject, String targetPackage)
40 | throws ShellException {
41 | if (!"MAVEN".equals(targetProject)) {
42 | return super.getDirectory(targetProject, targetPackage);
43 | }
44 |
45 | // targetProject is the output directory from the MyBatis generator
46 | // Mojo. It will be created if necessary
47 | //
48 | // targetPackage is interpreted as a sub directory, but in package
49 | // format (with dots instead of slashes). The sub directory will be created
50 | // if it does not already exist
51 |
52 | File project = mybatisGeneratorMojo.getOutputDirectory();
53 | if (!project.exists()) {
54 | project.mkdirs();
55 | }
56 |
57 | if (!project.isDirectory()) {
58 | throw new ShellException(Messages.getString("Warning.9", //$NON-NLS-1$
59 | project.getAbsolutePath()));
60 | }
61 |
62 | StringBuilder sb = new StringBuilder();
63 | StringTokenizer st = new StringTokenizer(targetPackage, "."); //$NON-NLS-1$
64 | while (st.hasMoreTokens()) {
65 | sb.append(st.nextToken());
66 | sb.append(File.separatorChar);
67 | }
68 |
69 | File directory = new File(project, sb.toString());
70 | if (!directory.isDirectory()) {
71 | boolean rc = directory.mkdirs();
72 | if (!rc) {
73 | throw new ShellException(Messages.getString("Warning.10", //$NON-NLS-1$
74 | directory.getAbsolutePath()));
75 | }
76 | }
77 |
78 | return directory;
79 | }
80 | }
81 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/config/IgnoredColumn.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2006 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 |
17 | package org.mybatis.generator.config;
18 |
19 | import static org.mybatis.generator.internal.util.StringUtility.stringContainsSpace;
20 | import static org.mybatis.generator.internal.util.StringUtility.stringHasValue;
21 | import static org.mybatis.generator.internal.util.messages.Messages.getString;
22 |
23 | import java.util.List;
24 |
25 | import org.mybatis.generator.api.dom.xml.Attribute;
26 | import org.mybatis.generator.api.dom.xml.XmlElement;
27 |
28 | /**
29 | * @author Jeff Butler
30 | *
31 | */
32 | public class IgnoredColumn {
33 |
34 | private String columnName;
35 |
36 | private boolean isColumnNameDelimited;
37 |
38 | private String configuredDelimitedColumnName;
39 |
40 | /**
41 | *
42 | */
43 | public IgnoredColumn(String columnName) {
44 | super();
45 | this.columnName = columnName;
46 | isColumnNameDelimited = stringContainsSpace(columnName);
47 | }
48 |
49 | public String getColumnName() {
50 | return columnName;
51 | }
52 |
53 | public boolean isColumnNameDelimited() {
54 | return isColumnNameDelimited;
55 | }
56 |
57 | public void setColumnNameDelimited(boolean isColumnNameDelimited) {
58 | this.isColumnNameDelimited = isColumnNameDelimited;
59 | configuredDelimitedColumnName = isColumnNameDelimited ? "true" : "false"; //$NON-NLS-1$ //$NON-NLS-2$
60 | }
61 |
62 | public boolean equals(Object obj) {
63 | if (obj == null || !(obj instanceof IgnoredColumn)) {
64 | return false;
65 | }
66 |
67 | return columnName.equals(((IgnoredColumn) obj).getColumnName());
68 | }
69 |
70 | public int hashCode() {
71 | return columnName.hashCode();
72 | }
73 |
74 | public XmlElement toXmlElement() {
75 | XmlElement xmlElement = new XmlElement("ignoreColumn"); //$NON-NLS-1$
76 | xmlElement.addAttribute(new Attribute("column", columnName)); //$NON-NLS-1$
77 |
78 | if (stringHasValue(configuredDelimitedColumnName)) {
79 | xmlElement.addAttribute(new Attribute(
80 | "delimitedColumnName", configuredDelimitedColumnName)); //$NON-NLS-1$
81 | }
82 |
83 | return xmlElement;
84 | }
85 |
86 | public void validate(List errors, String tableName) {
87 | if (!stringHasValue(columnName)) {
88 | errors.add(getString("ValidationError.21", //$NON-NLS-1$
89 | tableName));
90 | }
91 | }
92 | }
93 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/codegen/mybatis3/javamapper/elements/UpdateByExampleSelectiveMethodGenerator.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2009 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
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 | * @author Jeff Butler
30 | *
31 | */
32 | public class UpdateByExampleSelectiveMethodGenerator extends
33 | AbstractJavaMapperMethodGenerator {
34 |
35 | @Override
36 | public void addInterfaceElements(Interface interfaze) {
37 | Set importedTypes = new TreeSet();
38 | Method method = new Method();
39 | method.setVisibility(JavaVisibility.PUBLIC);
40 | method.setReturnType(FullyQualifiedJavaType.getIntInstance());
41 | method.setName(introspectedTable
42 | .getUpdateByExampleSelectiveStatementId());
43 |
44 | FullyQualifiedJavaType parameterType =
45 | introspectedTable.getRules().calculateAllFieldsClass();
46 | method.addParameter(new Parameter(parameterType,
47 | "record", "@Param(\"record\")")); //$NON-NLS-1$ //$NON-NLS-2$
48 | importedTypes.add(parameterType);
49 |
50 | FullyQualifiedJavaType exampleType = new FullyQualifiedJavaType(
51 | introspectedTable.getExampleType());
52 | method.addParameter(new Parameter(exampleType,
53 | "example", "@Param(\"example\")")); //$NON-NLS-1$ //$NON-NLS-2$
54 | importedTypes.add(exampleType);
55 |
56 | importedTypes.add(new FullyQualifiedJavaType(
57 | "org.apache.ibatis.annotations.Param")); //$NON-NLS-1$
58 |
59 | context.getCommentGenerator().addGeneralMethodComment(method,
60 | introspectedTable);
61 |
62 | addMapperAnnotations(interfaze, method);
63 |
64 | if (context.getPlugins()
65 | .clientUpdateByExampleSelectiveMethodGenerated(method, interfaze,
66 | introspectedTable)) {
67 | interfaze.addImportedTypes(importedTypes);
68 | interfaze.addMethod(method);
69 | }
70 | }
71 |
72 | public void addMapperAnnotations(Interface interfaze, Method method) {
73 | return;
74 | }
75 | }
76 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/internal/db/ConnectionFactory.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2005 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.internal.db;
17 |
18 | import static org.mybatis.generator.internal.util.StringUtility.stringHasValue;
19 | import static org.mybatis.generator.internal.util.messages.Messages.getString;
20 |
21 | import java.sql.Connection;
22 | import java.sql.Driver;
23 | import java.sql.SQLException;
24 | import java.util.Properties;
25 |
26 | import org.mybatis.generator.config.JDBCConnectionConfiguration;
27 | import org.mybatis.generator.internal.ObjectFactory;
28 |
29 | /**
30 | * This class assumes that classes are cached elsewhere for performance reasons,
31 | * but also to make sure that any native libraries are only loaded one time
32 | * (avoids the dreaded UnsatisfiedLinkError library loaded in another
33 | * classloader)
34 | *
35 | * @author Jeff Butler
36 | */
37 | public class ConnectionFactory {
38 |
39 | private static ConnectionFactory instance = new ConnectionFactory();
40 |
41 | public static ConnectionFactory getInstance() {
42 | return instance;
43 | }
44 |
45 | /**
46 | *
47 | */
48 | private ConnectionFactory() {
49 | super();
50 | }
51 |
52 | public Connection getConnection(JDBCConnectionConfiguration config)
53 | throws SQLException {
54 | Driver driver = getDriver(config);
55 |
56 | Properties props = new Properties();
57 |
58 | if (stringHasValue(config.getUserId())) {
59 | props.setProperty("user", config.getUserId()); //$NON-NLS-1$
60 | }
61 |
62 | if (stringHasValue(config.getPassword())) {
63 | props.setProperty("password", config.getPassword()); //$NON-NLS-1$
64 | }
65 |
66 | props.putAll(config.getProperties());
67 |
68 | Connection conn = driver.connect(config.getConnectionURL(), props);
69 |
70 | if (conn == null) {
71 | throw new SQLException(getString("RuntimeError.7")); //$NON-NLS-1$
72 | }
73 |
74 | return conn;
75 | }
76 |
77 | private Driver getDriver(JDBCConnectionConfiguration connectionInformation) {
78 | String driverClass = connectionInformation.getDriverClass();
79 | Driver driver;
80 |
81 | try {
82 | Class clazz = ObjectFactory.externalClassForName(driverClass);
83 | driver = (Driver) clazz.newInstance();
84 | } catch (Exception e) {
85 | throw new RuntimeException(getString("RuntimeError.8"), e); //$NON-NLS-1$
86 | }
87 |
88 | return driver;
89 | }
90 | }
91 |
--------------------------------------------------------------------------------
/src/test/java/demo/HelloTest.java:
--------------------------------------------------------------------------------
1 | package demo;
2 |
3 |
4 | import demo.domain.oob.OperateLog;
5 | import demo.domain.oob.OperateLogExample;
6 | import demo.domain.ooc.UserVisitLog;
7 | import demo.domain.ooc.UserVisitLogExample;
8 | import demo.mapper.oob.OperateLogMapper;
9 | import demo.mapper.ooc.UserVisitLogMapper;
10 | import org.apache.ibatis.session.SqlSession;
11 | import org.apache.ibatis.session.SqlSessionFactory;
12 | import org.apache.ibatis.session.SqlSessionFactoryBuilder;
13 | import org.junit.Before;
14 | import org.junit.Test;
15 |
16 | import java.io.FileInputStream;
17 | import java.io.FileNotFoundException;
18 | import java.io.InputStream;
19 | import java.util.ArrayList;
20 | import java.util.List;
21 |
22 | public class HelloTest {
23 |
24 | SqlSession sqlSession;
25 |
26 | OperateLogMapper operateLogMapper;
27 |
28 | UserVisitLogMapper userVisitLogMapper;
29 |
30 | @Before
31 | public void before() throws FileNotFoundException {
32 | // Mapper的配置文件
33 | String resource = HelloTest.class.getResource("/mybatis-config.xml").getFile();
34 | // 使用类加载器加载mybatis的配置文件（它也加载关联的映射文件）
35 | InputStream is = new FileInputStream(resource);
36 | // 构建sqlSession的工厂
37 | SqlSessionFactory sessionFactory = new SqlSessionFactoryBuilder().build(is);
38 | sqlSession = sessionFactory.openSession();
39 | operateLogMapper = sqlSession.getMapper(OperateLogMapper.class);
40 | userVisitLogMapper = sqlSession.getMapper(UserVisitLogMapper.class);
41 | }
42 |
43 | /**
44 | * insert Test
45 | * @author 吴帅
46 | * @parameter @throws Exception
47 | * @createDate 2015年12月18日 下午4:11:18
48 | */
49 | @Test
50 | public void insertTest() throws Exception {
51 | OperateLog operateLog = new OperateLog.Builder()
52 | .action("insert_test")
53 | .build();
54 | operateLogMapper.insertSelective(operateLog);
55 | sqlSession.commit();
56 | }
57 |
58 | /**
59 | * insertBatch Test
60 | * @author 吴帅
61 | * @parameter @throws Exception
62 | * @createDate 2015年12月18日 下午4:11:33
63 | */
64 | @Test
65 | public void insertBatchTest() throws Exception {
66 | List operateLogList = new ArrayList<>();
67 | for (int i = 0; i < 5; i++) {
68 | OperateLog operateLog = new OperateLog.Builder()
69 | .action("insertBatch_test"+i)
70 | .build();
71 | operateLogList.add(operateLog);
72 | }
73 | operateLogMapper.insertBatch(operateLogList);
74 | sqlSession.commit();
75 | }
76 |
77 |
78 | /**
79 | * select page Test
80 | * @throws Exception
81 | */
82 | @Test
83 | public void selectPageTest() throws Exception {
84 | OperateLogExample relationshipsExample = new OperateLogExample();
85 | relationshipsExample.setPagination(0l,2l);
86 | List operateLogList = operateLogMapper.selectByExample(relationshipsExample);
87 | //TODO verify
88 | System.out.println(operateLogList);
89 | }
90 |
91 | @Test
92 | public void selectByBigOffsetTest(){
93 | UserVisitLogExample userVisitLogExample = new UserVisitLogExample();
94 | userVisitLogExample.createCriteria().andIdLessThan(100);
95 | userVisitLogExample.setPagination(0,10);
96 | List userVisitLogList = userVisitLogMapper.selectByBigOffset(userVisitLogExample);
97 | System.out.println(userVisitLogList);
98 | }
99 |
100 |
101 |
102 | }
103 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/internal/db/DatabaseDialects.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2005 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.internal.db;
17 |
18 | /**
19 | * Typesafe enum of known database dialects
20 | *
21 | * @author Jeff Butler
22 | */
23 | public enum DatabaseDialects {
24 |
25 | DB2("VALUES IDENTITY_VAL_LOCAL()"), //$NON-NLS-1$
26 | MYSQL("SELECT LAST_INSERT_ID()"), //$NON-NLS-1$
27 | SQLSERVER("SELECT SCOPE_IDENTITY()"), //$NON-NLS-1$
28 | CLOUDSCAPE("VALUES IDENTITY_VAL_LOCAL()"), //$NON-NLS-1$
29 | DERBY("VALUES IDENTITY_VAL_LOCAL()"), //$NON-NLS-1$
30 | HSQLDB("CALL IDENTITY()"), //$NON-NLS-1$
31 | SYBASE("SELECT @@IDENTITY"), //$NON-NLS-1$
32 | DB2_MF("SELECT IDENTITY_VAL_LOCAL() FROM SYSIBM.SYSDUMMY1"), //$NON-NLS-1$
33 | INFORMIX("select dbinfo('sqlca.sqlerrd1') from systables where tabid=1"); //$NON-NLS-1$
34 |
35 | private String identityRetrievalStatement;
36 |
37 | /**
38 | *
39 | */
40 | private DatabaseDialects(String identityRetrievalStatement) {
41 | this.identityRetrievalStatement = identityRetrievalStatement;
42 | }
43 |
44 | public String getIdentityRetrievalStatement() {
45 | return identityRetrievalStatement;
46 | }
47 |
48 | /**
49 | *
50 | * @param database
51 | * @return the database dialect for the selected database. May return null
52 | * if there is no known dialect for the selected db
53 | */
54 | public static DatabaseDialects getDatabaseDialect(String database) {
55 | DatabaseDialects returnValue = null;
56 |
57 | if ("DB2".equalsIgnoreCase(database)) { //$NON-NLS-1$
58 | returnValue = DB2;
59 | } else if ("MySQL".equalsIgnoreCase(database)) { //$NON-NLS-1$
60 | returnValue = MYSQL;
61 | } else if ("SqlServer".equalsIgnoreCase(database)) { //$NON-NLS-1$
62 | returnValue = SQLSERVER;
63 | } else if ("Cloudscape".equalsIgnoreCase(database)) { //$NON-NLS-1$
64 | returnValue = CLOUDSCAPE;
65 | } else if ("Derby".equalsIgnoreCase(database)) { //$NON-NLS-1$
66 | returnValue = DERBY;
67 | } else if ("HSQLDB".equalsIgnoreCase(database)) { //$NON-NLS-1$
68 | returnValue = HSQLDB;
69 | } else if ("SYBASE".equalsIgnoreCase(database)) { //$NON-NLS-1$
70 | returnValue = SYBASE;
71 | } else if ("DB2_MF".equalsIgnoreCase(database)) { //$NON-NLS-1$
72 | returnValue = DB2_MF;
73 | } else if ("Informix".equalsIgnoreCase(database)) { //$NON-NLS-1$
74 | returnValue = INFORMIX;
75 | }
76 |
77 | return returnValue;
78 | }
79 | }
80 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/internal/DefaultShellCallback.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2005 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.internal;
17 |
18 | import static org.mybatis.generator.internal.util.messages.Messages.getString;
19 |
20 | import java.io.File;
21 | import java.util.StringTokenizer;
22 |
23 | import org.mybatis.generator.api.ShellCallback;
24 | import org.mybatis.generator.exception.ShellException;
25 |
26 | /**
27 | * @author Jeff Butler
28 | */
29 | public class DefaultShellCallback implements ShellCallback {
30 | private boolean overwrite;
31 |
32 | /**
33 | *
34 | */
35 | public DefaultShellCallback(boolean overwrite) {
36 | super();
37 | this.overwrite = overwrite;
38 | }
39 |
40 | public File getDirectory(String targetProject, String targetPackage)
41 | throws ShellException {
42 | // targetProject is interpreted as a directory that must exist
43 | //
44 | // targetPackage is interpreted as a sub directory, but in package
45 | // format (with dots instead of slashes). The sub directory will be
46 | // created
47 | // if it does not already exist
48 |
49 | File project = new File(targetProject);
50 | if (!project.isDirectory()) {
51 | throw new ShellException(getString("Warning.9", //$NON-NLS-1$
52 | targetProject));
53 | }
54 |
55 | StringBuilder sb = new StringBuilder();
56 | StringTokenizer st = new StringTokenizer(targetPackage, "."); //$NON-NLS-1$
57 | while (st.hasMoreTokens()) {
58 | sb.append(st.nextToken());
59 | sb.append(File.separatorChar);
60 | }
61 |
62 | File directory = new File(project, sb.toString());
63 | if (!directory.isDirectory()) {
64 | boolean rc = directory.mkdirs();
65 | if (!rc) {
66 | throw new ShellException(getString("Warning.10", //$NON-NLS-1$
67 | directory.getAbsolutePath()));
68 | }
69 | }
70 |
71 | return directory;
72 | }
73 |
74 | public void refreshProject(String project) {
75 | // nothing to do in the default shell callback
76 | }
77 |
78 | public boolean isMergeSupported() {
79 | return false;
80 | }
81 |
82 | public boolean isOverwriteEnabled() {
83 | return overwrite;
84 | }
85 |
86 | public String mergeJavaFile(String newFileSource,
87 | String existingFileFullPath, String[] javadocTags, String fileEncoding)
88 | throws ShellException {
89 | throw new UnsupportedOperationException();
90 | }
91 | }
92 |
--------------------------------------------------------------------------------
/src/main/java/org/mybatis/generator/config/ColumnRenamingRule.java:
--------------------------------------------------------------------------------
1 | /*
2 | * Copyright 2007 The Apache Software Foundation
3 | *
4 | * Licensed under the Apache License, Version 2.0 (the "License");
5 | * you may not use this file except in compliance with the License.
6 | * You may obtain a copy of the License at
7 | *
8 | * http://www.apache.org/licenses/LICENSE-2.0
9 | *
10 | * Unless required by applicable law or agreed to in writing, software
11 | * distributed under the License is distributed on an "AS IS" BASIS,
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | * See the License for the specific language governing permissions and
14 | * limitations under the License.
15 | */
16 | package org.mybatis.generator.config;
17 |
18 | import static org.mybatis.generator.internal.util.StringUtility.stringHasValue;
19 | import static org.mybatis.generator.internal.util.messages.Messages.getString;
20 |
21 | import java.util.List;
22 |
23 | import org.mybatis.generator.api.dom.xml.Attribute;
24 | import org.mybatis.generator.api.dom.xml.XmlElement;
25 |
26 | /**
27 | * This class is used to specify a renaming rule for columns in a table. This
28 | * renaming rule will be run against all column names before calculating the
29 | * corresponding property name. The most common use case is when columns in a
30 | * table are all prefixed by a certain value.
31 | *
32 | * For example, if columns in a table are named:
33 | *
34 | *
35 | * - CUST_NAME
36 | * - CUST_ADDRESS
37 | * - CUST_CITY
38 | * - CUST_STATE
39 | *
40 | *
41 | * it might be annoying to have the generated properties all containing the CUST
42 | * prefix. This class can be used to remove the prefix by specifying
43 | *
44 | *
45 | * - searchString = "^CUST"
46 | * - replaceString=""
47 | *
48 | *
49 | * Note that internally, the generator uses the
50 | * java.util.regex.Matcher.replaceAll method for this function. See
51 | * the documentation of that method for example of the regular expression
52 | * language used in Java.
53 | *
54 | * @author Jeff Butler
55 | *
56 | */
57 | public class ColumnRenamingRule {
58 | private String searchString;
59 | private String replaceString;
60 |
61 | public String getReplaceString() {
62 | return replaceString;
63 | }
64 |
65 | public void setReplaceString(String replaceString) {
66 | this.replaceString = replaceString;
67 | }
68 |
69 | public String getSearchString() {
70 | return searchString;
71 | }
72 |
73 | public void setSearchString(String searchString) {
74 | this.searchString = searchString;
75 | }
76 |
77 | public void validate(List errors, String tableName) {
78 | if (!stringHasValue(searchString)) {
79 | errors.add(getString("ValidationError.14", tableName)); //$NON-NLS-1$
80 | }
81 | }
82 |
83 | public XmlElement toXmlElement() {
84 | XmlElement xmlElement = new XmlElement("columnRenamingRule"); //$NON-NLS-1$
85 | xmlElement.addAttribute(new Attribute("searchString", searchString)); //$NON-NLS-1$
86 |
87 | if (replaceString != null) {
88 | xmlElement.addAttribute(new Attribute(
89 | "replaceString", replaceString)); //$NON-NLS-1$
90 | }
91 |
92 | return xmlElement;
93 | }
94 | }
95 |
--------------------------------------------------------------------------------