ShapefileResource.java

/* Copyright (c) 2001 - 2013 OpenPlans - 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.wps.resource;

import java.io.File;
import java.io.IOException;

import org.apache.commons.io.FileUtils;
import org.geotools.data.shapefile.ShapefileDataStore;

/**
 * Tracks and cleans up a shapefile store contained in its own private directory
 * 
 * @author Andrea Aime - OpenGeo
 * 
 */
public class ShapefileResource implements WPSResource {
    File directory;

    ShapefileDataStore store;

    String name;

    public ShapefileResource(ShapefileDataStore store, File directory) throws IOException {
        this.directory = directory;
        this.name = directory.getPath() + store.getTypeNames()[0] + ".shp";
        this.store = store;
    }

    public void delete() throws Exception {
        try {
            store.dispose();
        } finally {
            FileUtils.deleteDirectory(directory);
        }
    }

    public String getName() {
        return name;
    }

}