/**
* Copyright (c) 2012, The National Archives <pronom@nationalarchives.gsi.gov.uk>
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following
* conditions are met:
*
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
*
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* * Neither the name of the The National Archives nor the
* names of its contributors may be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
* NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package uk.gov.nationalarchives.droid.gui.worker;
import java.beans.PropertyChangeEvent;
import java.beans.PropertyChangeListener;
import java.io.IOException;
import java.util.Enumeration;
import java.util.List;
import java.util.concurrent.ExecutionException;
import javax.swing.JOptionPane;
import javax.swing.SwingWorker;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeModel;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import uk.gov.nationalarchives.droid.gui.ProfileForm;
import uk.gov.nationalarchives.droid.gui.util.DroidStringUtils;
import uk.gov.nationalarchives.droid.profile.ProfileManager;
import uk.gov.nationalarchives.droid.profile.ProfileResourceNode;
import uk.gov.nationalarchives.droid.profile.ProfileResultObserver;
import uk.gov.nationalarchives.droid.results.handlers.ProgressObserver;
/**
*
* @author Alok Kumar Dash
*/
public class DroidJob extends SwingWorker<Integer, ProfileResourceNode> {
private static final int RESULT_MAX_LENGTH = 60;
private static final int RESULT_LEFT_MIN = 20;
private final Log log = LogFactory.getLog(getClass());
private ProfileForm profileForm;
private ProfileManager profileManager;
private DefaultTreeModel treeModel;
/**
*
* @param profileManager a profile manager.
*/
public void setProfileManager(ProfileManager profileManager) {
this.profileManager = profileManager;
}
/**
* {@inheritDoc}
*/
@Override
protected void process(List<ProfileResourceNode> chunks) {
for (ProfileResourceNode node : chunks) {
Long parentId = node.getParentId() == null ? -1L : node.getParentId();
DefaultMutableTreeNode parent = profileForm.getInMemoryNodes().get(parentId);
if (parent != null) {
parent.setAllowsChildren(true);
boolean updated = false;
for (Enumeration<DefaultMutableTreeNode> e = parent.children();
e.hasMoreElements() && !updated;) {
DefaultMutableTreeNode childNode = e.nextElement();
if (childNode.getUserObject().equals(node)) {
childNode.setUserObject(node);
childNode.setAllowsChildren(node.allowsChildren());
treeModel.nodeChanged(childNode);
updated = true;
}
}
if (!updated) {
DefaultMutableTreeNode newNode = new DefaultMutableTreeNode(node, node.allowsChildren());
treeModel.insertNodeInto(newNode, parent, parent.getChildCount());
}
treeModel.nodeChanged(parent);
}
}
if (!chunks.isEmpty()) {
String decodedURI = java.net.URLDecoder.decode(chunks.get(0).getUri().toString());
String abbreviatedUri = DroidStringUtils.abbreviate(decodedURI, profileForm.getProfileProgressBar());
profileForm.getProfileProgressBar().setString(abbreviatedUri);
}
}
@Override
protected Integer doInBackground() throws IOException {
treeModel = profileForm.getTreeModel();
ProfileResultObserver myObserver = new ProfileResultObserver() {
@Override
public void onResult(ProfileResourceNode result) {
publish(result);
}
};
String profileUuid = profileForm.getProfile().getUuid();
final ProgressObserver observer = new ProgressObserver() {
@Override
public void onProgress(Integer progress) {
setProgress(progress);
}
};
addPropertyChangeListener(new PropertyChangeListener() {
@Override
public void propertyChange(PropertyChangeEvent evt) {
if ("progress".equals(evt.getPropertyName())) {
Integer value = (Integer) evt.getNewValue();
profileForm.getProfileProgressBar().setValue(value);
}
}
});
profileManager.setResultsObserver(profileUuid, myObserver);
profileManager.setProgressObserver(profileUuid, observer);
try {
profileManager.start(profileUuid).get();
} catch (ExecutionException e) {
log.error(e);
throw new RuntimeException(e.getCause().getMessage(), e.getCause());
} catch (InterruptedException e) {
log.debug(e);
throw new RuntimeException(e.getMessage(), e);
}
return null;
}
@Override
protected void done() {
try {
if (!isCancelled()) {
get();
}
} catch (ExecutionException e) {
Throwable cause = e.getCause();
log.error(cause.getMessage(), e);
profileManager.stop(profileForm.getProfile().getUuid());
JOptionPane.showMessageDialog(profileForm,
"An error occurred during profiling. Your profile has been paused.", "Error",
JOptionPane.ERROR_MESSAGE);
} catch (InterruptedException e) {
log.debug(e);
throw new RuntimeException(e.getMessage(), e);
}
}
/**
* Sets the profile form.
* @param profileForm the profile form to set
*/
public void setProfileForm(ProfileForm profileForm) {
this.profileForm = profileForm;
}
/**
* Starts the job.
*/
public void start() {
execute();
}
}