cn.dreampie.common.plugin.tablebind.AutoMultiSourceTableBindPlugin 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.tablebind;

import com.google.common.collect.Lists;
import com.jfinal.ext.plugin.tablebind.INameStyle;
import com.jfinal.ext.plugin.tablebind.SimpleNameStyles;
import com.jfinal.ext.plugin.tablebind.TableBind;
import com.jfinal.kit.StrKit;
import com.jfinal.log.Logger;
import com.jfinal.plugin.activerecord.ActiveRecordPlugin;
import com.jfinal.plugin.activerecord.IDataSourceProvider;
import com.jfinal.plugin.activerecord.Model;

import javax.sql.DataSource;
import java.util.List;

/**
 * Created by wangrenhui on 14-4-11.
 */
public class AutoMultiSourceTableBindPlugin extends ActiveRecordPlugin {

  protected final Logger log = Logger.getLogger(getClass());

  @SuppressWarnings("rawtypes")
  private List> excludeClasses = Lists.newArrayList();
  private List> includeClasses = Lists.newArrayList();
  private List includeClassPaths = Lists.newArrayList();
  private List includeJars = Lists.newArrayList();
  private boolean autoScan = false;
  private boolean includeAllJarsInLib;
  private INameStyle nameStyle;

  public AutoMultiSourceTableBindPlugin(DataSource dataSource) {
    this(dataSource, SimpleNameStyles.DEFAULT);
  }

  public AutoMultiSourceTableBindPlugin(DataSource dataSource, INameStyle nameStyle) {
    super(dataSource);
    this.nameStyle = nameStyle;
  }

  public AutoMultiSourceTableBindPlugin(IDataSourceProvider dataSourceProvider) {
    this(dataSourceProvider, SimpleNameStyles.DEFAULT);
  }

  public AutoMultiSourceTableBindPlugin(IDataSourceProvider dataSourceProvider, INameStyle nameStyle) {
    super(dataSourceProvider);
    this.nameStyle = nameStyle;
  }

  public AutoMultiSourceTableBindPlugin(String dataSourceName, IDataSourceProvider dataSourceProvider, INameStyle nameStyle) {
    super(dataSourceName, dataSourceProvider);
    this.nameStyle = nameStyle;
  }


  @SuppressWarnings({"rawtypes", "unchecked"})
  public AutoMultiSourceTableBindPlugin addExcludeClasses(Class... clazzes) {
    for (Class clazz : clazzes) {
      excludeClasses.add(clazz);
    }
    return this;
  }

  @SuppressWarnings("rawtypes")
  public AutoMultiSourceTableBindPlugin addExcludeClasses(List> clazzes) {
    if (clazzes != null) {
      excludeClasses.addAll(clazzes);
    }
    return this;
  }

  @SuppressWarnings({"rawtypes", "unchecked"})
  public AutoMultiSourceTableBindPlugin addIncludeClasses(Class... clazzes) {
    for (Class clazz : clazzes) {
      includeClasses.add(clazz);
    }
    return this;
  }

  @SuppressWarnings("rawtypes")
  public AutoMultiSourceTableBindPlugin addIncludeClasses(List> clazzes) {
    if (clazzes != null) {
      includeClasses.addAll(clazzes);
    }
    return this;
  }

  public AutoMultiSourceTableBindPlugin addIncludePaths(String... paths) {
    for (String path : paths) {
      includeClassPaths.add(path);
    }
    return this;
  }

  public AutoMultiSourceTableBindPlugin addExcludePaths(String... paths) {
    List> clazzes = ClassSearcherExt.of(Model.class).classpaths(paths).search();
    for (Class clazz : clazzes) {
      excludeClasses.add(clazz);
    }
    return this;
  }

  public AutoMultiSourceTableBindPlugin addJars(List jars) {
    if (jars != null) {
      includeJars.addAll(jars);
    }
    return this;
  }

  public AutoMultiSourceTableBindPlugin addJars(String... jars) {
    if (jars != null) {
      for (String jar : jars) {
        includeJars.add(jar);
      }
    }
    return this;
  }

  @SuppressWarnings({"unchecked", "rawtypes"})
  @Override
  public boolean start() {
    if (includeClasses.size() <= 0) {
      includeClasses = ClassSearcherExt.of(Model.class).classpaths(includeClassPaths).injars(includeJars).includeAllJarsInLib(includeAllJarsInLib).search();
    }
    TableBind tb = null;
    for (Class modelClass : includeClasses) {
      if (excludeClasses.contains(modelClass)) {
        continue;
      }
      tb = (TableBind) modelClass.getAnnotation(TableBind.class);
      String tableName;
      if (tb == null) {
        if (!autoScan) {
          continue;
        }
        tableName = nameStyle.name(modelClass.getSimpleName());
        this.addMapping(tableName, modelClass);
        log.debug("addMapping(" + tableName + ", " + modelClass.getName() + ")");
      } else {
        tableName = tb.tableName();
        if (StrKit.notBlank(tb.pkName())) {
          this.addMapping(tableName, tb.pkName(), modelClass);
          log.debug("addMapping(" + tableName + ", " + tb.pkName() + "," + modelClass.getName() + ")");
        } else {
          this.addMapping(tableName, modelClass);
          log.debug("addMapping(" + tableName + ", " + modelClass.getName() + ")");
        }
      }
    }
    return super.start();
  }

  @Override
  public boolean stop() {
    return super.stop();
  }

  public AutoMultiSourceTableBindPlugin autoScan(boolean autoScan) {
    this.autoScan = autoScan;
    return this;
  }

  public AutoMultiSourceTableBindPlugin includeAllJarsInLib(boolean includeAllJarsInLib) {
    this.includeAllJarsInLib = includeAllJarsInLib;
    return this;
  }
}