ImportProgressPage.java

/* Copyright (c) 2001 - 2008 TOPP - www.openplans.org. All rights reserved.
 * This code is licensed under the GPL 2.0 license, available at the root
 * application directory.
 */
package org.geoserver.web.importer;

import org.apache.wicket.AttributeModifier;
import org.apache.wicket.ajax.AbstractAjaxTimerBehavior;
import org.apache.wicket.ajax.AjaxRequestTarget;
import org.apache.wicket.ajax.markup.html.AjaxLink;
import org.apache.wicket.markup.html.WebMarkupContainer;
import org.apache.wicket.markup.html.basic.Label;
import org.apache.wicket.model.Model;
import org.apache.wicket.util.time.Duration;
import org.geoserver.importer.FeatureTypeImporter;
import org.geoserver.importer.ImportSummary;
import org.geoserver.importer.ImporterThreadManager;
import org.geoserver.web.GeoServerSecuredPage;

/**
 * Reports the import progress, the current layer, and allows to end the import mid way
 */
 @SuppressWarnings("serial")
public class ImportProgressPage extends GeoServerSecuredPage {
    String importerId;
    Label bar;
    Label percentage;
    Label currentFile;
    WebMarkupContainer info;
    Model widthModel;
    
    public ImportProgressPage(String importerKey) {
        this.importerId = importerKey;
        
        FeatureTypeImporter importer = getImporter();
        
        // construction
        add(new Label("project", importer.getProject()));
        add(info = new WebMarkupContainer("info"));
        info.setOutputMarkupId(true);
        info.add(bar = new Label("bar", "0"));
        widthModel = new Model("width: 0%;");
        bar.add(new AttributeModifier("style", widthModel));
        info.add(percentage = new Label("percentage", "0"));
        info.add(currentFile = new Label("currentFile", ""));
        info.add(new AjaxLink("cancel") {

            @Override
            public void onClick(AjaxRequestTarget target) {
                FeatureTypeImporter importer = getImporter();
                importer.cancel();
                
                setResponsePage(new ImportSummaryPage(importer.getSummary()));
            }
            
        });

        // comment this out if you need to hack on the HTML of a live page
        info.add(new AbstractAjaxTimerBehavior(Duration.milliseconds(500)) {

            @Override
            protected void onTimer(AjaxRequestTarget target) {
                FeatureTypeImporter importer = getImporter();
                
                 ImportSummary summary = importer.getSummary();
                if(summary != null) {
                    if(summary.isCompleted()) {
                        setResponsePage(new ImportSummaryPage(summary));
                    }
                    
                    long perc = Math.round(100.0 * (summary.getProcessedLayers() + 1) / summary.getTotalLayers());
                    if(perc > 100) {
                        perc = 100;
                    }
                    widthModel.setObject("width: " + perc + "%;");
                    percentage.setModelObject(perc);
                    currentFile.setModelObject(summary.getCurrentLayer());
                    
                    target.addComponent(info);
                }
            }

            
            
        });
    }
    
    FeatureTypeImporter getImporter() {
        ImporterThreadManager manager = (ImporterThreadManager) getGeoServerApplication().getBean("importerPool");
        FeatureTypeImporter importer = manager.getImporter(importerId);
        return importer;
    }
    
    /**
     * We want the indicator always on
     */
    public String getAjaxIndicatorMarkupId() {
        return null;
    }

}