/*
* eXist Open Source Native XML Database
* Copyright (C) 2001-06 The eXist Project
* http://exist-db.org
*
* 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 2
* 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, write to the Free Software
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*
* $Id$
*/
package org.exist.performance.actions;
import org.w3c.dom.Element;
import org.exist.EXistException;
import org.exist.storage.TextSearchEngine;
import org.exist.xmldb.EXistResource;
import org.exist.util.DirectoryScanner;
import org.exist.util.MimeTable;
import org.exist.util.MimeType;
import org.exist.util.ProgressIndicator;
import org.exist.performance.Connection;
import org.exist.performance.Runner;
import org.exist.performance.AbstractAction;
import org.xmldb.api.base.XMLDBException;
import org.xmldb.api.base.Collection;
import org.xmldb.api.base.Resource;
import org.xmldb.api.modules.CollectionManagementService;
import java.io.File;
import java.util.StringTokenizer;
import java.util.Observer;
import java.util.Observable;
public class StoreFromFile extends AbstractAction {
private String collectionPath;
private String dir;
private String includes;
private String mimeType = "application/xml";
private boolean overwrite = true;
public void configure(Runner runner, Action parent, Element config) throws EXistException {
super.configure(runner, parent, config);
if (!config.hasAttribute("collection"))
throw new EXistException(StoreFromFile.class.getName() + " requires an attribute 'collection'");
collectionPath = config.getAttribute("collection");
if (!config.hasAttribute("dir"))
throw new EXistException(StoreFromFile.class.getName() + " requires an attribute 'dir'");
dir = config.getAttribute("dir");
includes = config.getAttribute("includes");
if (config.hasAttribute("mime-type"))
mimeType = config.getAttribute("mime-type");
overwrite = getBooleanValue(config, "overwrite", true);
}
public void execute(Connection connection) throws XMLDBException, EXistException {
Collection collection = connection.getCollection(collectionPath);
if (collection == null)
throw new EXistException("collection " + collectionPath + " not found");
ProgressObserver observer = new ProgressObserver();
if (collection instanceof Observable)
((Observable)collection).addObserver(observer);
String resourceType = getResourceType();
File baseDir = new File(dir);
File[] files = DirectoryScanner.scanDir(baseDir, includes);
Collection col = collection;
String relDir, prevDir = null;
for(int j = 0; j < files.length; j++) {
String relPath = files[j].toString().substring( baseDir.toString().length() );
int p = relPath.lastIndexOf( File.separatorChar );
relDir = relPath.substring( 0, p );
relDir = relDir.replace(File.separatorChar, '/');
if (prevDir == null || ( !relDir.equals(prevDir) ) ) {
col = makeColl(collection, relDir);
if (col instanceof Observable)
((Observable)col).addObserver(observer);
prevDir = relDir;
}
if (col.getResource(files[j].getName()) == null || overwrite) {
//TODO : these probably need to be encoded and check mime via MimeTable
Resource resource =
col.createResource(files[j].getName(), resourceType);
resource.setContent(files[j]);
((EXistResource)resource).setMimeType(mimeType);
LOG.debug("Storing " + col.getName() + "/" + resource.getId());
col.storeResource(resource);
}
}
}
private String getResourceType() {
String resourceType = "XMLResource";
MimeType mime = MimeTable.getInstance().getContentType(mimeType);
if (mime != null)
resourceType = mime.isXMLType() ? "XMLResource" : "BinaryResource";
return resourceType;
}
private Collection makeColl(Collection parentColl, String relPath)
throws XMLDBException {
CollectionManagementService mgtService;
Collection current = parentColl, c;
String token;
StringTokenizer tok = new StringTokenizer(relPath, "/");
while (tok.hasMoreTokens()) {
token = tok.nextToken();
c = current.getChildCollection(token);
if (c == null) {
mgtService = (CollectionManagementService) current.getService("CollectionManagementService", "1.0");
current = mgtService.createCollection(token);
} else
current = c;
}
return current;
}
private class ProgressObserver implements Observer {
private long timestamp = System.currentTimeMillis();
public void update(Observable o, Object arg) {
if (System.currentTimeMillis() - timestamp > 20000) {
ProgressIndicator ind = (ProgressIndicator) arg;
if (!(o instanceof org.exist.storage.ElementIndex || o instanceof TextSearchEngine)) {
LOG.debug("Stored: " + (int)((ind.getValue() / ind.getMax()) * 100) + " %");
}
timestamp = System.currentTimeMillis();
}
}
}
}