org.beangle.commons.entity.orm.AbstractPersistModule Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of beangle-commons-model Show documentation
The newest version!
/*
 * Beangle, Agile Development Scaffold and Toolkits.
 *
 * Copyright © 2005, The Beangle Software.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see .
 */
package org.beangle.commons.entity.orm;

import java.util.List;

import org.beangle.commons.collection.CollectUtils;
import org.beangle.commons.entity.Entity;

public abstract class AbstractPersistModule {

  private EntityPersistConfig config = null;

  abstract protected void doConfig();

  protected final EntityHolder add(Class>... classes) {
    for (Class> cls : classes)
      config.addEntity(new EntityPersistConfig.EntityDefinition(cls));
    return new EntityHolder(classes, config);
  }

  protected final CacheHolder cache(String region) {
    return new CacheHolder(config).cache(region).usage(config.getCache().getUsage());
  }

  protected final CacheHolder cache() {
    return new CacheHolder(config).cache(config.getCache().region).usage(config.getCache().getUsage());
  }

  protected final List collection(Class clazz,
      String... properties) {
    List definitions = CollectUtils.newArrayList(properties.length);
    for (String property : properties) {
      definitions.add(new EntityPersistConfig.CollectionDefinition(clazz, property));
    }
    return definitions;
  }

  protected final void defaultCache(String region, String usage) {
    config.cache.setRegion(region);
    config.cache.setUsage(usage);
  }

  public final EntityPersistConfig getConfig() {
    config = new EntityPersistConfig();
    doConfig();
    return config;
  }

  public static final class CacheHolder {
    final EntityPersistConfig config;
    String cacheUsage;
    String cacheRegion;

    public CacheHolder(EntityPersistConfig config) {
      super();
      this.config = config;
    }

    public CacheHolder add(List... definitionLists) {
      for (List definitions : definitionLists) {
        for (EntityPersistConfig.CollectionDefinition definition : definitions) {
          config.addCollection(definition.cache(cacheRegion, cacheUsage));
        }
      }
      return this;
    }

    public CacheHolder add(Class>... classes) {
      for (Class> clazz : classes)
        config.getEntity(clazz).cache(cacheRegion, cacheUsage);
      return this;
    }

    public CacheHolder usage(String cacheUsage) {
      this.cacheUsage = cacheUsage;
      return this;
    }

    public CacheHolder cache(String cacheRegion) {
      this.cacheRegion = cacheRegion;
      return this;
    }

  }

  public static final class EntityHolder {

    final Class[] classes;

    final EntityPersistConfig config;

    public EntityHolder(Class[] classes, EntityPersistConfig config) {
      super();
      this.classes = classes;
      this.config = config;
    }

    public EntityHolder cacheable() {
      for (Class clazz : classes) {
        config.getEntity(clazz).cache(config.getCache().getRegion(), config.getCache().getUsage());
      }
      return this;
    }

    public EntityHolder cache(String region) {
      for (Class clazz : classes) {
        config.getEntity(clazz).setCacheRegion(region);
      }
      return this;
    }

    public EntityHolder usage(String usage) {
      for (Class clazz : classes) {
        config.getEntity(clazz).setCacheUsage(usage);
      }
      return this;
    }

  }
}