org.geoserver.web.importer.ImportProgressPage.java Source code

Introduction

Here is the source code for org.geoserver.web.importer.ImportProgressPage.java

Source

/* 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: 5%;");
        bar.add(new AttributeModifier("style", widthModel));
        info.add(percentage = new Label("percentage", "5"));
        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));
                    }

                    // we have the percentage go between 5 and 95. This way the user sees something is in progress
                    // at the beginning and does not have to wonder why we keep at 100 for a few seconds (happens
                    // if the last layer requires a costly SRS code lookup)
                    long perc = 5
                            + Math.round(100.0 * (summary.getProcessedLayers() + 1) / summary.getTotalLayers());
                    if (perc > 90) {
                        perc = 90;
                    }
                    widthModel.setObject("width: " + perc + "%;");
                    percentage.setDefaultModelObject(perc);
                    currentFile.setDefaultModelObject(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
     */
    @Override
    public String getAjaxIndicatorMarkupId() {
        return null;
    }

}