org.mybatis.generator.config.PropertyRegistry Java Examples

@Override
public List<GeneratedJavaFile> getGeneratedJavaFiles() {
    List<GeneratedJavaFile> answer = new ArrayList<>();

    for (AbstractJavaGenerator javaGenerator : javaGenerators) {
        List<CompilationUnit> compilationUnits = javaGenerator.getCompilationUnits();
        for (CompilationUnit compilationUnit : compilationUnits) {
            GeneratedJavaFile gjf = new GeneratedJavaFile(compilationUnit,
                            javaGenerator.getProject(),
                            context.getProperty(PropertyRegistry.CONTEXT_JAVA_FILE_ENCODING),
                            context.getJavaFormatter());
            answer.add(gjf);
        }
    }

    return answer;
}
 

public JavaVisibility getExampleMethodVisibility() {
    if (exampleMethodVisibility == null) {
        String type = context
                .getJavaClientGeneratorConfiguration()
                .getProperty(PropertyRegistry.DAO_EXAMPLE_METHOD_VISIBILITY);
        if (stringHasValue(type)) {
            if ("public".equalsIgnoreCase(type)) { //$NON-NLS-1$
                exampleMethodVisibility = JavaVisibility.PUBLIC;
            } else if ("private".equalsIgnoreCase(type)) { //$NON-NLS-1$
                exampleMethodVisibility = JavaVisibility.PRIVATE;
            } else if ("protected".equalsIgnoreCase(type)) { //$NON-NLS-1$
                exampleMethodVisibility = JavaVisibility.PROTECTED;
            } else if ("default".equalsIgnoreCase(type)) { //$NON-NLS-1$
                exampleMethodVisibility = JavaVisibility.DEFAULT;
            } else {
                exampleMethodVisibility = JavaVisibility.PUBLIC;
                warnings.add(getString("Warning.16", type)); //$NON-NLS-1$
            }
        } else {
            exampleMethodVisibility = JavaVisibility.PUBLIC;
        }
    }

    return exampleMethodVisibility;
}
 

public DAOMethodNameCalculator getDAOMethodNameCalculator() {
    if (dAOMethodNameCalculator == null) {
        String type = context.getJavaClientGeneratorConfiguration()
                .getProperty(PropertyRegistry.DAO_METHOD_NAME_CALCULATOR);
        if (stringHasValue(type)) {
            if ("extended".equalsIgnoreCase(type)) { //$NON-NLS-1$
                type = ExtendedDAOMethodNameCalculator.class.getName();
            } else if ("default".equalsIgnoreCase(type)) { //$NON-NLS-1$
                type = DefaultDAOMethodNameCalculator.class.getName();
            }
        } else {
            type = DefaultDAOMethodNameCalculator.class.getName();
        }

        try {
            dAOMethodNameCalculator = (DAOMethodNameCalculator) ObjectFactory
                    .createInternalObject(type);
        } catch (Exception e) {
            dAOMethodNameCalculator = new DefaultDAOMethodNameCalculator();
            warnings.add(getString(
                    "Warning.17", type, e.getMessage())); //$NON-NLS-1$
        }
    }

    return dAOMethodNameCalculator;
}
 

public JavaVisibility getExampleMethodVisibility() {
    if (exampleMethodVisibility == null) {
        String type = context
                .getJavaClientGeneratorConfiguration()
                .getProperty(PropertyRegistry.DAO_EXAMPLE_METHOD_VISIBILITY);
        if (stringHasValue(type)) {
            if ("public".equalsIgnoreCase(type)) { //$NON-NLS-1$
                exampleMethodVisibility = JavaVisibility.PUBLIC;
            } else if ("private".equalsIgnoreCase(type)) { //$NON-NLS-1$
                exampleMethodVisibility = JavaVisibility.PRIVATE;
            } else if ("protected".equalsIgnoreCase(type)) { //$NON-NLS-1$
                exampleMethodVisibility = JavaVisibility.PROTECTED;
            } else if ("default".equalsIgnoreCase(type)) { //$NON-NLS-1$
                exampleMethodVisibility = JavaVisibility.DEFAULT;
            } else {
                exampleMethodVisibility = JavaVisibility.PUBLIC;
                warnings.add(getString("Warning.16", type)); //$NON-NLS-1$
            }
        } else {
            exampleMethodVisibility = JavaVisibility.PUBLIC;
        }
    }

    return exampleMethodVisibility;
}
 

public DAOMethodNameCalculator getDAOMethodNameCalculator() {
    if (dAOMethodNameCalculator == null) {
        String type = context.getJavaClientGeneratorConfiguration()
                .getProperty(PropertyRegistry.DAO_METHOD_NAME_CALCULATOR);
        if (stringHasValue(type)) {
            if ("extended".equalsIgnoreCase(type)) { //$NON-NLS-1$
                type = ExtendedDAOMethodNameCalculator.class.getName();
            } else if ("default".equalsIgnoreCase(type)) { //$NON-NLS-1$
                type = DefaultDAOMethodNameCalculator.class.getName();
            }
        } else {
            type = DefaultDAOMethodNameCalculator.class.getName();
        }

        try {
            dAOMethodNameCalculator = (DAOMethodNameCalculator) ObjectFactory
                    .createInternalObject(type);
        } catch (Exception e) {
            dAOMethodNameCalculator = new DefaultDAOMethodNameCalculator();
            warnings.add(getString(
                    "Warning.17", type, e.getMessage())); //$NON-NLS-1$
        }
    }

    return dAOMethodNameCalculator;
}
 

@Override
public List<CompilationUnit> getExtraCompilationUnits() {
    boolean useLegacyBuilder = false;

    String prop = context.getJavaClientGeneratorConfiguration()
            .getProperty(PropertyRegistry.CLIENT_USE_LEGACY_BUILDER);
    if (StringUtility.stringHasValue(prop)) {
        useLegacyBuilder = Boolean.valueOf(prop);
    }
    SqlProviderGenerator sqlProviderGenerator = new SqlProviderGenerator(getProject(), useLegacyBuilder);
    sqlProviderGenerator.setContext(context);
    sqlProviderGenerator.setIntrospectedTable(introspectedTable);
    sqlProviderGenerator.setProgressCallback(progressCallback);
    sqlProviderGenerator.setWarnings(warnings);
    return sqlProviderGenerator.getCompilationUnits();
}
 

/**
 * Checks if is constructor based.
 *
 * @return true, if is constructor based
 */
public boolean isConstructorBased() {
    if (isImmutable()) {
        return true;
    }

    Properties properties;

    if (tableConfiguration.getProperties().containsKey(PropertyRegistry.ANY_CONSTRUCTOR_BASED)) {
        properties = tableConfiguration.getProperties();
    } else {
        properties = context.getJavaModelGeneratorConfiguration().getProperties();
    }

    return isTrue(properties.getProperty(PropertyRegistry.ANY_CONSTRUCTOR_BASED));
}
 

private Interface createBasicInterface() {
    FullyQualifiedJavaType type = new FullyQualifiedJavaType(
            introspectedTable.getMyBatis3JavaMapperType());
    Interface interfaze = new Interface(type);
    interfaze.setVisibility(JavaVisibility.PUBLIC);
    context.getCommentGenerator().addJavaFileComment(interfaze);
    interfaze.addImportedType(new FullyQualifiedJavaType("org.apache.ibatis.annotations.Mapper"));
    interfaze.addAnnotation("@Mapper");

    String rootInterface = introspectedTable.getTableConfigurationProperty(PropertyRegistry.ANY_ROOT_INTERFACE);
    if (!stringHasValue(rootInterface)) {
        rootInterface = context.getJavaClientGeneratorConfiguration().getProperty(PropertyRegistry.ANY_ROOT_INTERFACE);
    }

    if (stringHasValue(rootInterface)) {
        FullyQualifiedJavaType fqjt = new FullyQualifiedJavaType(rootInterface);
        interfaze.addSuperInterface(fqjt);
        interfaze.addImportedType(fqjt);
    }
    return interfaze;
}
 

@Override
public void addConfigurationProperties(Properties properties) {
    this.properties.putAll(properties);

    suppressDate = isTrue(properties
            .getProperty(PropertyRegistry.COMMENT_GENERATOR_SUPPRESS_DATE));
    
    suppressAllComments = isTrue(properties
            .getProperty(PropertyRegistry.COMMENT_GENERATOR_SUPPRESS_ALL_COMMENTS));

    addRemarkComments = isTrue(properties
            .getProperty(PropertyRegistry.COMMENT_GENERATOR_ADD_REMARK_COMMENTS));
    
    String dateFormatString = properties.getProperty(PropertyRegistry.COMMENT_GENERATOR_DATE_FORMAT);
    if (StringUtility.stringHasValue(dateFormatString)) {
        dateFormat = new SimpleDateFormat(dateFormatString);
    }
}
 

/**
 * Checks if is trim strings enabled.
 *
 * @param context
 *            the context
 * @return true, if is trim strings enabled
 */
private static boolean isTrimStringsEnabled(Context context) {
    Properties properties = context
            .getJavaModelGeneratorConfiguration().getProperties();
    boolean rc = isTrue(properties
            .getProperty(PropertyRegistry.MODEL_GENERATOR_TRIM_STRINGS));
    return rc;
}
 

/**
 * Creates a new Object object.
 *
 * @param context
 *            the context
 * @return the xml formatter
 */
public static XmlFormatter createXmlFormatter(Context context) {
    String type = context.getProperty(PropertyRegistry.CONTEXT_XML_FORMATTER);
    if (!StringUtility.stringHasValue(type)) {
        type = DefaultXmlFormatter.class.getName();
    }

    XmlFormatter answer = (XmlFormatter) createInternalObject(type);

    answer.setContext(context);

    return answer;
}
 

/**
 * Creates a new Object object.
 *
 * @param context
 *            the context
 * @return the java formatter
 */
public static JavaFormatter createJavaFormatter(Context context) {
    String type = context.getProperty(PropertyRegistry.CONTEXT_JAVA_FORMATTER);
    if (!StringUtility.stringHasValue(type)) {
        type = DefaultJavaFormatter.class.getName();
    }

    JavaFormatter answer = (JavaFormatter) createInternalObject(type);

    answer.setContext(context);

    return answer;
}
 

public void addConfigurationProperties(Properties properties) {
    this.properties.putAll(properties);

    suppressDate = isTrue(properties
            .getProperty(PropertyRegistry.COMMENT_GENERATOR_SUPPRESS_DATE));

    suppressAllComments = isTrue(properties
            .getProperty(PropertyRegistry.COMMENT_GENERATOR_SUPPRESS_ALL_COMMENTS));

    addRemarkComments = isTrue(properties
            .getProperty(PropertyRegistry.COMMENT_GENERATOR_ADD_REMARK_COMMENTS));
}
 

public String getRootClass() {
    String rootClass = introspectedTable
            .getTableConfigurationProperty(PropertyRegistry.ANY_ROOT_CLASS);
    if (rootClass == null) {
        Properties properties = context
                .getJavaModelGeneratorConfiguration().getProperties();
        rootClass = properties.getProperty(PropertyRegistry.ANY_ROOT_CLASS);
    }

    return rootClass;
}
 

/**
 * Instantiates a new base rules.
 *
 * @param introspectedTable
 *            the introspected table
 */
public BaseRules(IntrospectedTable introspectedTable) {
    super();
    this.introspectedTable = introspectedTable;
    this.tableConfiguration = introspectedTable.getTableConfiguration();
    String modelOnly = tableConfiguration.getProperty(PropertyRegistry.TABLE_MODEL_ONLY);
    isModelOnly = StringUtility.isTrue(modelOnly);
}
 

private static boolean isTrimStringsEnabled(IntrospectedTable table) {
    TableConfiguration tableConfiguration = table.getTableConfiguration();
    String trimSpaces = tableConfiguration.getProperties().getProperty(
            PropertyRegistry.MODEL_GENERATOR_TRIM_STRINGS);
    if (trimSpaces != null) {
        return isTrue(trimSpaces);
    }
    return isTrimStringsEnabled(table.getContext());
}
 

private static boolean isTrimStringsEnabled(IntrospectedColumn column) {
    String trimSpaces = column.getProperties().getProperty(PropertyRegistry.MODEL_GENERATOR_TRIM_STRINGS);
    if (trimSpaces != null) {
        return isTrue(trimSpaces);
    }
    return isTrimStringsEnabled(column.getIntrospectedTable());
}
 

/**
 * Checks if is immutable.
 *
 * @return true, if is immutable
 */
public boolean isImmutable() {
    Properties properties;

    if (tableConfiguration.getProperties().containsKey(PropertyRegistry.ANY_IMMUTABLE)) {
        properties = tableConfiguration.getProperties();
    } else {
        properties = context.getJavaModelGeneratorConfiguration().getProperties();
    }

    return isTrue(properties.getProperty(PropertyRegistry.ANY_IMMUTABLE));
}
 

public void addConfigurationProperties(Properties properties) {
      this.properties.putAll(properties);
      forceBigDecimals = StringUtility
              .isTrue(properties
                      .getProperty(PropertyRegistry.TYPE_RESOLVER_FORCE_BIG_DECIMALS));
      this.jdbcDateToJavaType = StringUtility.escapeStringForJava(properties.getProperty("DATE"));
this.jdbcTimestampToJavaType =  StringUtility.escapeStringForJava(properties.getProperty("TIME"));
      this.jdbcTimeToJavaType =  StringUtility.escapeStringForJava(properties.getProperty("TIMESTAMP"));
  }
 

public String getRootClass() {
    String rootClass = introspectedTable
            .getTableConfigurationProperty(PropertyRegistry.ANY_ROOT_CLASS);
    if (rootClass == null) {
        Properties properties = context
                .getJavaModelGeneratorConfiguration().getProperties();
        rootClass = properties.getProperty(PropertyRegistry.ANY_ROOT_CLASS);
    }

    return rootClass;
}
 

@Override
public List<CompilationUnit> getExtraCompilationUnits() {
	boolean useLegacyBuilder = false;
	
	String prop = context.getJavaClientGeneratorConfiguration().getProperty(PropertyRegistry.CLIENT_USE_LEGACY_BUILDER);
	if (StringUtility.stringHasValue(prop)) {
		useLegacyBuilder = Boolean.valueOf(prop);
	}
    SqlProviderGenerator sqlProviderGenerator = new SqlProviderGenerator(useLegacyBuilder);
    sqlProviderGenerator.setContext(context);
    sqlProviderGenerator.setIntrospectedTable(introspectedTable);
    sqlProviderGenerator.setProgressCallback(progressCallback);
    sqlProviderGenerator.setWarnings(warnings);
    return sqlProviderGenerator.getCompilationUnits();
}
 

public boolean isImmutable() {
    Properties properties;
    
    if (tableConfiguration.getProperties().containsKey(PropertyRegistry.ANY_IMMUTABLE)) {
        properties = tableConfiguration.getProperties();
    } else {
        properties = context.getJavaModelGeneratorConfiguration().getProperties();
    }
    
    return isTrue(properties.getProperty(PropertyRegistry.ANY_IMMUTABLE));
}
 

public boolean isConstructorBased() {
    if (isImmutable()) {
        return true;
    }
    
    Properties properties;
    
    if (tableConfiguration.getProperties().containsKey(PropertyRegistry.ANY_CONSTRUCTOR_BASED)) {
        properties = tableConfiguration.getProperties();
    } else {
        properties = context.getJavaModelGeneratorConfiguration().getProperties();
    }
    
    return isTrue(properties.getProperty(PropertyRegistry.ANY_CONSTRUCTOR_BASED));
}
 

/**
 * 
 */
public BaseRules(IntrospectedTable introspectedTable) {
    super();
    this.introspectedTable = introspectedTable;
    this.tableConfiguration = introspectedTable.getTableConfiguration();
    String modelOnly = tableConfiguration.getProperty(PropertyRegistry.TABLE_MODEL_ONLY);
    isModelOnly = StringUtility.isTrue(modelOnly);
}
 

public void addConfigurationProperties(Properties properties) {
    this.properties.putAll(properties);

    suppressDate = isTrue(properties
            .getProperty(PropertyRegistry.COMMENT_GENERATOR_SUPPRESS_DATE));
    
    suppressAllComments = isTrue(properties
            .getProperty(PropertyRegistry.COMMENT_GENERATOR_SUPPRESS_ALL_COMMENTS));
}
 

public static JavaFormatter createJavaFormatter(Context context) {
    String type = context.getProperty(PropertyRegistry.CONTEXT_JAVA_FORMATTER);
    if (!StringUtility.stringHasValue(type)) {
        type = DefaultJavaFormatter.class.getName();
    }

    JavaFormatter answer = (JavaFormatter) createInternalObject(type);

    answer.setContext(context);

    return answer;
}
 

public static XmlFormatter createXmlFormatter(Context context) {
    String type = context.getProperty(PropertyRegistry.CONTEXT_XML_FORMATTER);
    if (!StringUtility.stringHasValue(type)) {
        type = DefaultXmlFormatter.class.getName();
    }

    XmlFormatter answer = (XmlFormatter) createInternalObject(type);

    answer.setContext(context);

    return answer;
}
 

public boolean isTrimStringsEnabled() {
    Properties properties = context
            .getJavaModelGeneratorConfiguration().getProperties();
    boolean rc = isTrue(properties
            .getProperty(PropertyRegistry.MODEL_GENERATOR_TRIM_STRINGS));
    return rc;
}
 

public String getRootClass() {
    String rootClass = introspectedTable
            .getTableConfigurationProperty(PropertyRegistry.ANY_ROOT_CLASS);
    if (rootClass == null) {
        Properties properties = context
                .getJavaModelGeneratorConfiguration().getProperties();
        rootClass = properties.getProperty(PropertyRegistry.ANY_ROOT_CLASS);
    }

    return rootClass;
}
 

@Override
public void addConfigurationProperties(Properties properties) {
    super.addConfigurationProperties(properties);

    // stolen from the parent
    suppressDate = isTrue(properties
            .getProperty(PropertyRegistry.COMMENT_GENERATOR_SUPPRESS_DATE));

    suppressAllComments = isTrue(properties
            .getProperty(PropertyRegistry.COMMENT_GENERATOR_SUPPRESS_ALL_COMMENTS));

    if(suppressAllComments) return;

    // 获取国际化资源的路径
    i18nPath = properties.getProperty(XMBG_CG_I18N_PATH_KEY, XMBG_CG_I18N_DEFAULT_PATH);
    LOG.info("use the i18n resources under {}",i18nPath);

    // 获取项目开始时间，用在版权声明中
    String startYearStr = properties.getProperty(XMBG_CG_PROJECT_START_YEAR);
    if(StringUtils.isNotEmpty(startYearStr)){
        startYear = startYearStr;
    }else{
        startYear = XMBG_CG_PROJECT_START_DEFAULT_YEAR;
    }

    // 初始化资源
    String localeStr = properties.getProperty(XMBG_CG_I18N_LOCALE_KEY);
    Locale locale = Locale.getDefault();
    if(localeStr != null && StringUtils.isNoneEmpty(localeStr)) {
        String[] localeAras = localeStr.trim().split("_");
        locale = new Locale(localeAras[0], localeAras[1]);
    }
    LOG.info("use the locale {}",locale);
    try {
        initResources(locale);
    } catch (MalformedURLException e) {
        e.printStackTrace();
    }
}