package er.indexing;
import java.util.Enumeration;
import org.apache.log4j.Logger;
import com.webobjects.eoaccess.EOEntity;
import com.webobjects.eoaccess.EOModel;
import com.webobjects.eoaccess.EOModelGroup;
import com.webobjects.eocontrol.EOEditingContext;
import com.webobjects.eocontrol.EOFetchSpecification;
import com.webobjects.foundation.NSArray;
import com.webobjects.foundation.NSMutableArray;
import er.extensions.eof.ERXEC;
import er.extensions.eof.ERXFetchSpecificationBatchIterator;
/**
* Utility class to reindex all objects of a model group, a model or an entity.
* @author ak
*
*/
public class ERIndexer {
private NSArray<ERAutoIndex> _indices;
private Logger log = Logger.getLogger(ERIndexer.class);
public ERIndexer(NSArray<ERAutoIndex> indices) {
_indices = indices;
}
public void clear() {
for(Enumeration i = _indices.objectEnumerator(); i.hasMoreElements(); ) {
ERIndex index = (ERIndex) i.nextElement();
index.clear();
}
}
public NSArray indicesForEntity(String entityName) {
NSMutableArray result = new NSMutableArray();
for(Enumeration i = _indices.objectEnumerator(); i.hasMoreElements(); ) {
ERAutoIndex index = (ERAutoIndex) i.nextElement();
if(index.handlesEntity(entityName)) {
result.addObject(index);
}
}
return result;
}
public void indexAllObjects(EOEntity entity) {
NSArray<ERIndex> incides = indicesForEntity(entity.name());
if(incides.count() > 0) {
long start = System.currentTimeMillis();
int treshhold = 10;
EOEditingContext ec = ERXEC.newEditingContext();
ec.lock();
try {
EOFetchSpecification fs = new EOFetchSpecification(entity.name(), null, null);
ERXFetchSpecificationBatchIterator iterator = new ERXFetchSpecificationBatchIterator(fs);
iterator.setEditingContext(ec);
while(iterator.hasNextBatch()) {
NSArray objects = iterator.nextBatch();
if(iterator.currentBatchIndex() % treshhold == 0) {
ec.unlock();
// ec.dispose();
ec = ERXEC.newEditingContext();
ec.lock();
iterator.setEditingContext(ec);
}
for(Enumeration i = incides.objectEnumerator(); i.hasMoreElements(); ) {
ERIndex index = (ERIndex) i.nextElement();
index.addObjectsToIndex(ec, objects);
}
}
} finally {
ec.unlock();
}
log.info("Indexing " + entity.name() + " took: " + (System.currentTimeMillis() - start) + " ms");
}
}
public void indexAllObjects(EOModel model) {
for (Enumeration entities = model.entities().objectEnumerator(); entities.hasMoreElements();) {
EOEntity entity = (EOEntity) entities.nextElement();
indexAllObjects(entity);
}
}
public void indexAllObjects(EOModelGroup group) {
for (Enumeration models = group.models().objectEnumerator(); models.hasMoreElements();) {
EOModel model = (EOModel) models.nextElement();
indexAllObjects(model);
}
}
}