package org.geotools.data.geojson;
/*
* GeoTools - The Open Source Java GIS Toolkit
* http://geotools.org
*
* (C) 2015, Open Source Geospatial Foundation (OSGeo)
*
* This library 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;
* version 2.1 of the License.
*
* This library 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.
*/
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.Collections;
import java.util.List;
import org.apache.commons.io.FilenameUtils;
import org.geotools.data.Query;
import org.geotools.data.Transaction;
import org.geotools.data.store.ContentEntry;
import org.geotools.data.store.ContentFeatureSource;
import org.geotools.feature.NameImpl;
import org.opengis.feature.simple.SimpleFeatureType;
import org.opengis.feature.type.Name;
public class GeoJSONDataStore extends org.geotools.data.store.ContentDataStore {
URL url;
SimpleFeatureType schema;
protected Name typeName;
public GeoJSONDataStore(URL url) throws IOException {
this.url = url;
}
@Override
protected List<Name> createTypeNames() throws IOException {
String name = FilenameUtils.getBaseName(url.toExternalForm());
// could hard code features in here?
typeName = new NameImpl(name);
return Collections.singletonList(typeName);
}
GeoJSONReader read() {
GeoJSONReader reader = null;
reader = new GeoJSONReader(url);
return reader;
}
@Override
protected ContentFeatureSource createFeatureSource(ContentEntry entry) throws IOException {
// We can only really write to local files
String scheme = url.getProtocol();
String host = url.getHost();
if ("file".equalsIgnoreCase(scheme) && (host == null || host.isEmpty())) {
GeoJSONFeatureStore geoJSONFeatureStore = new GeoJSONFeatureStore(entry, Query.ALL);
return geoJSONFeatureStore;
} else {
GeoJSONFeatureSource geoJSONFeatureSource = new GeoJSONFeatureSource(entry, Query.ALL);
return geoJSONFeatureSource;
}
}
@Override
public void createSchema(SimpleFeatureType featureType) throws IOException {
this.schema = featureType;
}
}