cn.dreampie.common.plugin.db.DbConfig Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jfinal-dreampie Show documentation
jfinal shiro-freemarker plugins
There is a newer version: 1.2.2
package cn.dreampie.common.plugin.db;

import cn.dreampie.common.util.FileUtils;
import cn.dreampie.common.util.PropertiesUtils;
import com.google.common.collect.Maps;
import com.google.common.collect.Sets;
import com.jfinal.kit.PathKit;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.*;

/**
 * Created by wangrenhui on 14-5-5.
 */
public class DbConfig {
  private Logger logger = LoggerFactory.getLogger(getClass());

  private String config = "/application.properties";
  private Properties properties;

  public DbConfig() {
    properties = PropertiesUtils.me().loadPropertyFile(config);
  }

  public DbConfig(String config) {
    this.config = config;
    properties = PropertiesUtils.me().loadPropertyFile(config);
  }

  public List getAllDbNames() {
    Set dbNames = Sets.newHashSet();
    Enumeration enums = properties.keys();
    String key = null;
    String dbName = null;
    while (enums.hasMoreElements()) {
      key = enums.nextElement() + "";
      if (key.startsWith("db.")) {
        dbName = key.split("\\.")[1];
        dbNames.add(dbName);
      }
    }
    return new ArrayList(dbNames);
  }

  public Map getAllDbSources() {
    Map dbSourceMap = Maps.newHashMap();
    List dbNames = getAllDbNames();
    for (String dbName : dbNames) {
      dbSourceMap.put(dbName, new DbSource(properties.getProperty("db." + dbName + ".driver"), properties.getProperty("db." + dbName + ".url"), properties.getProperty("db." + dbName + ".user"), properties.getProperty("db." + dbName + ".password")));
    }
    return dbSourceMap;
  }


  public boolean initOnMigrate(String dbName) {
    return properties.getProperty("db." + dbName + ".migration.initOnMigrate", "false").equals("true");
  }

  public boolean migrateAuto(String dbName) {
    return properties.getProperty("db." + dbName + ".migration.auto", "false").equals("true");
  }

  public boolean migrationFileDirectoryExists(String path) {
    if (FileUtils.me().exist(PathKit.getRootClassPath() + path)) {
      logger.debug("Directory for migration files found." + path);
      return true;
    } else {
      logger.warn("Directory for migration files not found." + path);
      return false;
    }
  }

  public boolean isClean(String dbName) {
    return PropertiesUtils.me().loadPropertyFile(config).getProperty("db." + dbName + ".valid.clean", "false").equals("true");
  }

  public boolean isDev() {
    return PropertiesUtils.me().loadPropertyFile(config).getProperty("devMode", "false").equals("true");
  }
}