/** 添加方法 */ protected Method selectByPrimaryKey(IntrospectedTable introspectedTable, String tableName) { Method method = new Method(); method.setName("selectByPrimaryKey"); method.setReturnType(pojoType); if (introspectedTable.getRules().generatePrimaryKeyClass()) { FullyQualifiedJavaType type = new FullyQualifiedJavaType(introspectedTable.getPrimaryKeyType()); method.addParameter(new Parameter(type, "key")); } else { for (IntrospectedColumn introspectedColumn : introspectedTable.getPrimaryKeyColumns()) { FullyQualifiedJavaType type = introspectedColumn.getFullyQualifiedJavaType(); method.addParameter(new Parameter(type, introspectedColumn.getJavaProperty())); } } method.setVisibility(JavaVisibility.PUBLIC); StringBuilder sb = new StringBuilder(); // method.addBodyLine("try {"); sb.append("return this."); sb.append(getDaoShort()); sb.append("selectByPrimaryKey"); sb.append("("); for (IntrospectedColumn introspectedColumn : introspectedTable.getPrimaryKeyColumns()) { sb.append(introspectedColumn.getJavaProperty()); sb.append(","); } sb.setLength(sb.length() - 1); sb.append(");"); method.addBodyLine(sb.toString()); // method.addBodyLine("} catch (Exception e) {"); // method.addBodyLine("logger.error(\"Exception: \", e);"); // method.addBodyLine("return null;"); // method.addBodyLine("}"); return method; }
/** * @param introspectedColumn * @param inMethod if true generates an "in" method, else generates a "not in" method * @return a generated method for the in or not in method */ private Method getSetInOrNotInMethod(IntrospectedColumn introspectedColumn, boolean inMethod) { Method method = new Method(); method.setVisibility(JavaVisibility.PUBLIC); FullyQualifiedJavaType type = FullyQualifiedJavaType.getNewListInstance(); if (introspectedColumn.getFullyQualifiedJavaType().isPrimitive()) { type.addTypeArgument( introspectedColumn.getFullyQualifiedJavaType().getPrimitiveTypeWrapper()); } else { type.addTypeArgument(introspectedColumn.getFullyQualifiedJavaType()); } method.addParameter(new Parameter(type, "values")); // $NON-NLS-1$ StringBuilder sb = new StringBuilder(); sb.append(introspectedColumn.getJavaProperty()); sb.setCharAt(0, Character.toUpperCase(sb.charAt(0))); sb.insert(0, "and"); // $NON-NLS-1$ if (inMethod) { sb.append("In"); // $NON-NLS-1$ } else { sb.append("NotIn"); // $NON-NLS-1$ } method.setName(sb.toString()); method.setReturnType(FullyQualifiedJavaType.getCriteriaInstance()); sb.setLength(0); if (introspectedColumn.isJDBCDateColumn()) { sb.append("addCriterionForJDBCDate(\""); // $NON-NLS-1$ } else if (introspectedColumn.isJDBCTimeColumn()) { sb.append("addCriterionForJDBCTime(\""); // $NON-NLS-1$ } else if (stringHasValue(introspectedColumn.getTypeHandler())) { sb.append("add"); // $NON-NLS-1$ sb.append(introspectedColumn.getJavaProperty()); sb.setCharAt(3, Character.toUpperCase(sb.charAt(3))); sb.append("Criterion(\""); // $NON-NLS-1$ } else { sb.append("addCriterion(\""); // $NON-NLS-1$ } sb.append(MyBatis3FormattingUtilities.getAliasedActualColumnName(introspectedColumn)); if (inMethod) { sb.append(" in"); // $NON-NLS-1$ } else { sb.append(" not in"); // $NON-NLS-1$ } sb.append("\", values, \""); // $NON-NLS-1$ sb.append(introspectedColumn.getJavaProperty()); sb.append("\");"); // $NON-NLS-1$ method.addBodyLine(sb.toString()); method.addBodyLine("return (Criteria) this;"); // $NON-NLS-1$ return method; }
private Method getSingleValueMethod( IntrospectedColumn introspectedColumn, String nameFragment, String operator) { Method method = new Method(); method.setVisibility(JavaVisibility.PUBLIC); method.addParameter( new Parameter(introspectedColumn.getFullyQualifiedJavaType(), "value")); // $NON-NLS-1$ StringBuilder sb = new StringBuilder(); sb.append(introspectedColumn.getJavaProperty()); sb.setCharAt(0, Character.toUpperCase(sb.charAt(0))); sb.insert(0, "and"); // $NON-NLS-1$ sb.append(nameFragment); method.setName(sb.toString()); method.setReturnType(FullyQualifiedJavaType.getCriteriaInstance()); sb.setLength(0); if (introspectedColumn.isJDBCDateColumn()) { sb.append("addCriterionForJDBCDate(\""); // $NON-NLS-1$ } else if (introspectedColumn.isJDBCTimeColumn()) { sb.append("addCriterionForJDBCTime(\""); // $NON-NLS-1$ } else if (stringHasValue(introspectedColumn.getTypeHandler())) { sb.append("add"); // $NON-NLS-1$ sb.append(introspectedColumn.getJavaProperty()); sb.setCharAt(3, Character.toUpperCase(sb.charAt(3))); sb.append("Criterion(\""); // $NON-NLS-1$ } else { sb.append("addCriterion(\""); // $NON-NLS-1$ } sb.append(MyBatis3FormattingUtilities.getAliasedActualColumnName(introspectedColumn)); sb.append(' '); sb.append(operator); sb.append("\", "); // $NON-NLS-1$ if (introspectedColumn.getFullyQualifiedJavaType().isPrimitive()) { sb.append("new "); // $NON-NLS-1$ sb.append( introspectedColumn.getFullyQualifiedJavaType().getPrimitiveTypeWrapper().getShortName()); sb.append("(value)"); // $NON-NLS-1$ } else { sb.append("value"); // $NON-NLS-1$ } sb.append(", \""); // $NON-NLS-1$ sb.append(introspectedColumn.getJavaProperty()); sb.append("\");"); // $NON-NLS-1$ method.addBodyLine(sb.toString()); method.addBodyLine("return (Criteria) this;"); // $NON-NLS-1$ return method; }
@Override public void initialized(IntrospectedTable introspectedTable) { List<IntrospectedColumn> columns = introspectedTable.getAllColumns(); for (IntrospectedColumn introspectedColumn : columns) { String javaProperty = this.getJavaName( introspectedColumn.getActualColumnName(), introspectedColumn.getJavaProperty()); introspectedColumn.setJavaProperty(javaProperty); } super.initialized(introspectedTable); }
private void addParameterizedConstructor(TopLevelClass topLevelClass) { Method method = new Method(); method.setVisibility(JavaVisibility.PUBLIC); method.setConstructor(true); method.setName(topLevelClass.getType().getShortName()); context.getCommentGenerator().addGeneralMethodComment(method, introspectedTable); for (IntrospectedColumn introspectedColumn : introspectedTable.getAllColumns()) { method.addParameter( new Parameter( introspectedColumn.getFullyQualifiedJavaType(), introspectedColumn.getJavaProperty())); } boolean comma = false; StringBuilder sb = new StringBuilder(); sb.append("super("); // $NON-NLS-1$ for (IntrospectedColumn introspectedColumn : introspectedTable.getNonBLOBColumns()) { if (comma) { sb.append(", "); // $NON-NLS-1$ } else { comma = true; } sb.append(introspectedColumn.getJavaProperty()); } sb.append(");"); // $NON-NLS-1$ method.addBodyLine(sb.toString()); for (IntrospectedColumn introspectedColumn : introspectedTable.getBLOBColumns()) { sb.setLength(0); sb.append("this."); // $NON-NLS-1$ sb.append(introspectedColumn.getJavaProperty()); sb.append(" = "); // $NON-NLS-1$ sb.append(introspectedColumn.getJavaProperty()); sb.append(';'); method.addBodyLine(sb.toString()); } topLevelClass.addMethod(method); }
/** type 的意义 pojo 1 key 2 example 3 pojo+example 4 */ protected String addParams(IntrospectedTable introspectedTable, Method method, int type1) { switch (type1) { case 1: method.addParameter(new Parameter(pojoType, "record")); return "record"; case 2: if (introspectedTable.getRules().generatePrimaryKeyClass()) { FullyQualifiedJavaType type = new FullyQualifiedJavaType(introspectedTable.getPrimaryKeyType()); method.addParameter(new Parameter(type, "key")); } else { for (IntrospectedColumn introspectedColumn : introspectedTable.getPrimaryKeyColumns()) { FullyQualifiedJavaType type = introspectedColumn.getFullyQualifiedJavaType(); method.addParameter(new Parameter(type, introspectedColumn.getJavaProperty())); } } StringBuffer sb = new StringBuffer(); for (IntrospectedColumn introspectedColumn : introspectedTable.getPrimaryKeyColumns()) { sb.append(introspectedColumn.getJavaProperty()); sb.append(","); } sb.setLength(sb.length() - 1); return sb.toString(); case 3: method.addParameter(new Parameter(pojoCriteriaType, "example")); return "example"; case 4: method.addParameter(0, new Parameter(pojoType, "record")); method.addParameter(1, new Parameter(pojoCriteriaType, "example")); if (method.getName().equals("updateByExampleSelective") || method.getName().equals("updateByExample")) { return "record, example.getCondition()"; } return "record, example"; default: break; } return null; }
/** * 给字段添加数据库备注 * * @param field * @param introspectedTable * @param introspectedColumn */ public void addFieldComment( Field field, IntrospectedTable introspectedTable, IntrospectedColumn introspectedColumn) { if (StringUtility.stringHasValue(introspectedColumn.getRemarks())) { field.addJavaDocLine("/**"); StringBuilder sb = new StringBuilder(); sb.append(" * "); sb.append(introspectedColumn.getRemarks()); field.addJavaDocLine(sb.toString()); field.addJavaDocLine(" */"); } // 添加注解 if (field.isTransient()) { // @Column field.addAnnotation("@Transient"); } for (IntrospectedColumn column : introspectedTable.getPrimaryKeyColumns()) { if (introspectedColumn == column) { field.addAnnotation("@Id"); break; } } String column = introspectedColumn.getActualColumnName(); if (StringUtility.stringContainsSpace(column) || introspectedTable.getTableConfiguration().isAllColumnDelimitingEnabled()) { column = introspectedColumn.getContext().getBeginningDelimiter() + column + introspectedColumn.getContext().getEndingDelimiter(); } if (!column.equals(introspectedColumn.getJavaProperty())) { // @Column field.addAnnotation("@Column(name = \"" + column + "\")"); } if (introspectedColumn.isIdentity()) { if (introspectedTable .getTableConfiguration() .getGeneratedKey() .getRuntimeSqlStatement() .equals("JDBC")) { field.addAnnotation("@GeneratedValue(generator = \"JDBC\")"); } else { field.addAnnotation("@GeneratedValue(strategy = GenerationType.IDENTITY)"); } } else if (introspectedColumn.isSequenceColumn()) { field.addAnnotation( "@SequenceGenerator(name=\"\",sequenceName=\"" + introspectedTable.getTableConfiguration().getGeneratedKey().getRuntimeSqlStatement() + "\")"); } }
private Method getNoValueMethod( IntrospectedColumn introspectedColumn, String nameFragment, String operator) { Method method = new Method(); method.setVisibility(JavaVisibility.PUBLIC); StringBuilder sb = new StringBuilder(); sb.append(introspectedColumn.getJavaProperty()); sb.setCharAt(0, Character.toUpperCase(sb.charAt(0))); sb.insert(0, "and"); // $NON-NLS-1$ sb.append(nameFragment); method.setName(sb.toString()); method.setReturnType(FullyQualifiedJavaType.getCriteriaInstance()); sb.setLength(0); sb.append("addCriterion(\""); // $NON-NLS-1$ sb.append(MyBatis3FormattingUtilities.getAliasedActualColumnName(introspectedColumn)); sb.append(' '); sb.append(operator); sb.append("\");"); // $NON-NLS-1$ method.addBodyLine(sb.toString()); method.addBodyLine("return (Criteria) this;"); // $NON-NLS-1$ return method; }
@Override public void addElements(XmlElement parentElement) { XmlElement answer = new XmlElement("insert"); // $NON-NLS-1$ answer.addAttribute( new Attribute("id", introspectedTable.getInsertSelectiveStatementId())); // $NON-NLS-1$ FullyQualifiedJavaType parameterType = introspectedTable.getRules().calculateAllFieldsClass(); answer.addAttribute( new Attribute( "parameterClass", //$NON-NLS-1$ parameterType.getFullyQualifiedName())); context.getCommentGenerator().addComment(answer); GeneratedKey gk = introspectedTable.getGeneratedKey(); if (gk != null && gk.isPlacedBeforeInsertInIbatis2()) { IntrospectedColumn introspectedColumn = introspectedTable.getColumn(gk.getColumn()); // if the column is null, then it's a configuration error. The // warning has already been reported if (introspectedColumn != null) { // pre-generated key answer.addElement(getSelectKey(introspectedColumn, gk)); } } StringBuilder sb = new StringBuilder(); sb.append("insert into "); // $NON-NLS-1$ sb.append(introspectedTable.getFullyQualifiedTableNameAtRuntime()); answer.addElement(new TextElement(sb.toString())); XmlElement insertElement = new XmlElement("dynamic"); // $NON-NLS-1$ insertElement.addAttribute(new Attribute("prepend", "(")); // $NON-NLS-1$ //$NON-NLS-2$ answer.addElement(insertElement); answer.addElement(new TextElement("values")); // $NON-NLS-1$ XmlElement valuesElement = new XmlElement("dynamic"); // $NON-NLS-1$ valuesElement.addAttribute(new Attribute("prepend", "(")); // $NON-NLS-1$ //$NON-NLS-2$ answer.addElement(valuesElement); for (IntrospectedColumn introspectedColumn : introspectedTable.getAllColumns()) { if (introspectedColumn.isIdentity()) { // cannot set values on identity fields continue; } XmlElement insertNotNullElement = new XmlElement("isNotNull"); // $NON-NLS-1$ insertNotNullElement.addAttribute(new Attribute("prepend", ",")); // $NON-NLS-1$ //$NON-NLS-2$ insertNotNullElement.addAttribute( new Attribute("property", introspectedColumn.getJavaProperty())); // $NON-NLS-1$ insertNotNullElement.addElement( new TextElement(Ibatis2FormattingUtilities.getEscapedColumnName(introspectedColumn))); insertElement.addElement(insertNotNullElement); XmlElement valuesNotNullElement = new XmlElement("isNotNull"); // $NON-NLS-1$ valuesNotNullElement.addAttribute(new Attribute("prepend", ",")); // $NON-NLS-1$ //$NON-NLS-2$ valuesNotNullElement.addAttribute( new Attribute("property", introspectedColumn.getJavaProperty())); // $NON-NLS-1$ valuesNotNullElement.addElement( new TextElement(Ibatis2FormattingUtilities.getParameterClause(introspectedColumn))); valuesElement.addElement(valuesNotNullElement); } insertElement.addElement(new TextElement(")")); // $NON-NLS-1$ valuesElement.addElement(new TextElement(")")); // $NON-NLS-1$ if (gk != null && !gk.isPlacedBeforeInsertInIbatis2()) { IntrospectedColumn introspectedColumn = introspectedTable.getColumn(gk.getColumn()); // if the column is null, then it's a configuration error. The // warning has already been reported if (introspectedColumn != null) { // pre-generated key answer.addElement(getSelectKey(introspectedColumn, gk)); } } if (context.getPlugins().sqlMapInsertSelectiveElementGenerated(answer, introspectedTable)) { parentElement.addElement(answer); } }
/** * This method adds all the extra methods and fields required to support a user defined type * handler on some column. * * @param introspectedColumn * @param constructor * @param innerClass * @return the name of the List added to the class by this method */ private String addtypeHandledObjectsAndMethods( IntrospectedColumn introspectedColumn, Method constructor, InnerClass innerClass) { String answer; StringBuilder sb = new StringBuilder(); // add new private field and public accessor in the class sb.setLength(0); sb.append(introspectedColumn.getJavaProperty()); sb.append("Criteria"); // $NON-NLS-1$ answer = sb.toString(); Field field = new Field(); field.setVisibility(JavaVisibility.PROTECTED); field.setType(new FullyQualifiedJavaType("java.util.List<Criterion>")); // $NON-NLS-1$ field.setName(answer); innerClass.addField(field); Method method = new Method(); method.setVisibility(JavaVisibility.PUBLIC); method.setReturnType(field.getType()); method.setName(getGetterMethodName(field.getName(), field.getType())); sb.insert(0, "return "); // $NON-NLS-1$ sb.append(';'); method.addBodyLine(sb.toString()); innerClass.addMethod(method); // add constructor initialization sb.setLength(0); sb.append(field.getName()); sb.append(" = new ArrayList<Criterion>();"); // $NON-NLS-1$; constructor.addBodyLine(sb.toString()); // now add the methods for simplifying the individual field set methods method = new Method(); method.setVisibility(JavaVisibility.PROTECTED); sb.setLength(0); sb.append("add"); // $NON-NLS-1$ sb.append(introspectedColumn.getJavaProperty()); sb.setCharAt(3, Character.toUpperCase(sb.charAt(3))); sb.append("Criterion"); // $NON-NLS-1$ method.setName(sb.toString()); method.addParameter( new Parameter(FullyQualifiedJavaType.getStringInstance(), "condition")); // $NON-NLS-1$ method.addParameter( new Parameter(FullyQualifiedJavaType.getObjectInstance(), "value")); // $NON-NLS-1$ method.addParameter( new Parameter(FullyQualifiedJavaType.getStringInstance(), "property")); // $NON-NLS-1$ method.addBodyLine("if (value == null) {"); // $NON-NLS-1$ method.addBodyLine( "throw new RuntimeException(\"Value for \" + property + \" cannot be null\");"); //$NON-NLS-1$ method.addBodyLine("}"); // $NON-NLS-1$ method.addBodyLine( String.format( "%s.add(new Criterion(condition, value, \"%s\"));", //$NON-NLS-1$ field.getName(), introspectedColumn.getTypeHandler())); method.addBodyLine("allCriteria = null;"); // $NON-NLS-1$ innerClass.addMethod(method); sb.setLength(0); sb.append("add"); // $NON-NLS-1$ sb.append(introspectedColumn.getJavaProperty()); sb.setCharAt(3, Character.toUpperCase(sb.charAt(3))); sb.append("Criterion"); // $NON-NLS-1$ method = new Method(); method.setVisibility(JavaVisibility.PROTECTED); method.setName(sb.toString()); method.addParameter( new Parameter(FullyQualifiedJavaType.getStringInstance(), "condition")); // $NON-NLS-1$ method.addParameter( new Parameter(introspectedColumn.getFullyQualifiedJavaType(), "value1")); // $NON-NLS-1$ method.addParameter( new Parameter(introspectedColumn.getFullyQualifiedJavaType(), "value2")); // $NON-NLS-1$ method.addParameter( new Parameter(FullyQualifiedJavaType.getStringInstance(), "property")); // $NON-NLS-1$ method.addBodyLine("if (value1 == null || value2 == null) {"); // $NON-NLS-1$ method.addBodyLine( "throw new RuntimeException(\"Between values for \" + property + \" cannot be null\");"); //$NON-NLS-1$ method.addBodyLine("}"); // $NON-NLS-1$ method.addBodyLine( String.format( "%s.add(new Criterion(condition, value1, value2, \"%s\"));", //$NON-NLS-1$ field.getName(), introspectedColumn.getTypeHandler())); method.addBodyLine("allCriteria = null;"); // $NON-NLS-1$ innerClass.addMethod(method); return answer; }
private XmlElement getMiddleForEachElement(IntrospectedColumn introspectedColumn) { StringBuilder sb = new StringBuilder(); String criteriaAttribute; boolean typeHandled; String typeHandlerString; if (introspectedColumn == null) { criteriaAttribute = "criteria.criteria"; // $NON-NLS-1$ typeHandled = false; typeHandlerString = null; } else { sb.setLength(0); sb.append("criteria."); // $NON-NLS-1$ sb.append(introspectedColumn.getJavaProperty()); sb.append("Criteria"); // $NON-NLS-1$ criteriaAttribute = sb.toString(); typeHandled = true; sb.setLength(0); sb.append(",typeHandler="); // $NON-NLS-1$ sb.append(introspectedColumn.getTypeHandler()); typeHandlerString = sb.toString(); } XmlElement middleForEachElement = new XmlElement("foreach"); // $NON-NLS-1$ middleForEachElement.addAttribute( new Attribute("collection", criteriaAttribute)); // $NON-NLS-1$ middleForEachElement.addAttribute( new Attribute("item", "criterion")); // $NON-NLS-1$ //$NON-NLS-2$ XmlElement chooseElement = new XmlElement("choose"); // $NON-NLS-1$ middleForEachElement.addElement(chooseElement); XmlElement when = new XmlElement("when"); // $NON-NLS-1$ when.addAttribute(new Attribute("test", "criterion.noValue")); // $NON-NLS-1$ //$NON-NLS-2$ when.addElement(new TextElement("and ${criterion.condition}")); // $NON-NLS-1$ chooseElement.addElement(when); when = new XmlElement("when"); // $NON-NLS-1$ when.addAttribute(new Attribute("test", "criterion.singleValue")); // $NON-NLS-1$ //$NON-NLS-2$ sb.setLength(0); sb.append("and ${criterion.condition} #{criterion.value"); // $NON-NLS-1$ if (typeHandled) { sb.append(typeHandlerString); } sb.append('}'); when.addElement(new TextElement(sb.toString())); chooseElement.addElement(when); when = new XmlElement("when"); // $NON-NLS-1$ when.addAttribute(new Attribute("test", "criterion.betweenValue")); // $NON-NLS-1$ //$NON-NLS-2$ sb.setLength(0); sb.append("and ${criterion.condition} #{criterion.value"); // $NON-NLS-1$ if (typeHandled) { sb.append(typeHandlerString); } sb.append("} and #{criterion.secondValue"); // $NON-NLS-1$ if (typeHandled) { sb.append(typeHandlerString); } sb.append('}'); when.addElement(new TextElement(sb.toString())); chooseElement.addElement(when); when = new XmlElement("when"); // $NON-NLS-1$ when.addAttribute(new Attribute("test", "criterion.listValue")); // $NON-NLS-1$ //$NON-NLS-2$ when.addElement(new TextElement("and ${criterion.condition}")); // $NON-NLS-1$ XmlElement innerForEach = new XmlElement("foreach"); // $NON-NLS-1$ innerForEach.addAttribute( new Attribute("collection", "criterion.value")); // $NON-NLS-1$ //$NON-NLS-2$ innerForEach.addAttribute(new Attribute("item", "listItem")); // $NON-NLS-1$ //$NON-NLS-2$ innerForEach.addAttribute(new Attribute("open", "(")); // $NON-NLS-1$ //$NON-NLS-2$ innerForEach.addAttribute(new Attribute("close", ")")); // $NON-NLS-1$ //$NON-NLS-2$ innerForEach.addAttribute(new Attribute("separator", ",")); // $NON-NLS-1$ //$NON-NLS-2$ sb.setLength(0); sb.append("#{listItem"); // $NON-NLS-1$ if (typeHandled) { sb.append(typeHandlerString); } sb.append('}'); innerForEach.addElement(new TextElement(sb.toString())); when.addElement(innerForEach); chooseElement.addElement(when); return middleForEachElement; }