Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.makeandbuild.vessl.fixture.FixtureImpl Maven / Gradle / Ivy
package com.makeandbuild.vessl.fixture;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.util.List;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import com.makeandbuild.vessl.persistence.ObjectNotFoundException;
@SuppressWarnings("rawtypes")
public class FixtureImpl implements Fixture {
protected List entityLoaders;
protected List entityManagers;
protected DumperFactory dumperFactory = new DumperFactoryImpl();
protected Log logger = LogFactory.getLog(this.getClass());
public List getEntityLoaders() {
return entityLoaders;
}
public void setEntityLoaders(List entityLoaders) {
this.entityLoaders = entityLoaders;
}
public List getEntityManagers() {
return entityManagers;
}
public void setEntityManagers(List entityManagers) {
this.entityManagers = entityManagers;
}
private EntityManager getManager(Class entityClass, String subtype){
for (EntityManager entityManager : entityManagers){
if (entityManager.getEntityClass().equals(entityClass) && compareTypes(subtype, entityManager.getSubtype())){
return entityManager;
}
}
throw new RuntimeException("no entity maanager for class "+entityClass);
}
private EntityLoader getEntityLoader(Class entityClass, String subtype){
for (EntityLoader loader : entityLoaders){
if (loader.getEntityClass().equals(entityClass) && compareTypes(subtype, loader.getSubtype())){
return loader;
}
}
throw new RuntimeException("no loader for class "+entityClass);
}
private boolean compareTypes(String subtype1, String subtype2){
if (subtype1==null && subtype2==null){
return true;
} else {
return subtype1!=null && subtype1.equals(subtype2);
}
}
@Override
public void load() throws IOException {
for (EntityLoader loader : entityLoaders){
EntityManager manager = getManager(loader.getEntityClass(), loader.getSubtype());
doLoad(loader, manager);
}
}
private void doLoad(EntityLoader loader, EntityManager manager) throws IOException{
if (loader instanceof IteratedLoader){
IteratedLoader iteratedLoader = (IteratedLoader)loader;
Object value = iteratedLoader.read();
while(value != null){
manager.save(value);
value = iteratedLoader.read();
}
} else {
List entities = loader.load();
for (Object entity : entities) {
manager.save(entity);
}
}
}
@Override
public void load(String resourceName) throws ClassNotFoundException, IOException {
EntityLoader loader = new ResourceEntityLoaderImpl(resourceName);
EntityManager manager = getManager(loader.getEntityClass(), loader.getSubtype());
doLoad(loader, manager);
}
@Override
public void purge() throws IOException {
for (EntityManager entityManager : entityManagers){
Class clazz = entityManager.getEntityClass();
purge(clazz, entityManager.getSubtype());
}
}
@Override
public void purge(Class entityClass, String subtype) throws IOException {
EntityManager manager = getManager(entityClass, subtype);
EntityLoader loader = getEntityLoader(entityClass, subtype);
doPurge(manager, loader);
}
private void doPurge(EntityManager manager, EntityLoader loader) throws IOException {
if (loader instanceof IteratedLoader){
IteratedLoader iteratedLoader = (IteratedLoader)loader;
Object value = iteratedLoader.read();
while(value != null){
manager.delete(value);
value = iteratedLoader.read();
}
} else {
List entities = loader.loadReverse();
for (Object entity : entities) {
try {
manager.delete(entity);
}catch (ObjectNotFoundException squash){
logger.info("not found for "+entity);
}
}
}
}
@Override
public void load(Class entityClass, String subtype) throws IOException {
EntityLoader loader = getEntityLoader(entityClass, subtype);
this.load(loader);
}
@Override
public void load(EntityLoader loader) throws IOException {
EntityManager manager = getManager(loader.getEntityClass(), loader.getSubtype());
doLoad(loader, manager);
}
@Override
public void dump(Class entityClass, String subtype, File directory) throws IOException {
EntityManager manager = getManager(entityClass, subtype);
if (!directory.exists())
directory.mkdir();
dumperFactory.setDirectory(directory);
Object minimumKey = getMiniumKey(entityClass, subtype);
Dumper dumper = dumperFactory.create(entityClass, subtype, manager, minimumKey);
if (dumper != null)
dumper.dump();
}
@Override
public void dump(File directory) throws IOException {
for (EntityManager entityManager : entityManagers){
Class entityClass = entityManager.getEntityClass();
String subtype = entityManager.getSubtype();
dump(entityClass, subtype, directory);
}
}
public DumperFactory getDumperFactory() {
return dumperFactory;
}
public void setDumperFactory(DumperFactory dumperFactory) {
this.dumperFactory = dumperFactory;
}
@Override
public Object getMiniumKey(Class entityClass, String subtype) throws IOException {
EntityLoader loader = getEntityLoader(entityClass, subtype);
EntityManager manager = getManager(entityClass, subtype);
List entities = loader.load();
Object currentKey = null;
for (Object entity : entities) {
Object entityKey = manager.getId(entity);
if (entityKey.getClass().equals(Long.class) || entityKey.getClass().equals(Long.TYPE)){
if (currentKey == null){
currentKey = entityKey;
} else if ((Long)currentKey > (Long)entityKey){
currentKey = entityKey;
}
} else {
//skip
System.out.println("skipping "+entityClass);
return null;
}
}
return currentKey;
}
@Override
public void dump(Class entityClass, String subtype, OutputStream outputStream) throws IOException {
EntityManager manager = getManager(entityClass, subtype);
Object minimumKey = getMiniumKey(entityClass, subtype);
Dumper dumper = dumperFactory.create(entityClass, subtype, manager, minimumKey);
if (dumper != null)
dumper.dump(outputStream);
}
}
