Improve security

[lgpl/argeo-commons.git] / gis / runtime / org.argeo.gis.geotools / src / main / java / org / argeo / geotools / GeoToolsUtils.java

diff --git a/gis/runtime/org.argeo.gis.geotools/src/main/java/org/argeo/geotools/GeoToolsUtils.java b/gis/runtime/org.argeo.gis.geotools/src/main/java/org/argeo/geotools/GeoToolsUtils.java
new file mode 100644 (file)
index 0000000..88ab849
--- /dev/null
+++ b/gis/runtime/org.argeo.gis.geotools/src/main/java/org/argeo/geotools/GeoToolsUtils.java
@@ -0,0 +1,48 @@
+package org.argeo.geotools;
+
+import java.io.IOException;
+
+import org.argeo.ArgeoException;
+import org.geotools.data.DataStore;
+import org.geotools.data.FeatureSource;
+import org.geotools.data.FeatureStore;
+import org.opengis.feature.simple.SimpleFeature;
+import org.opengis.feature.simple.SimpleFeatureType;
+import org.opengis.feature.type.Name;
+
+/** Utilities related to the GeoTools framework */
+public class GeoToolsUtils {
+
+       /** Opens a read/write feature store */
+       public static FeatureStore<SimpleFeatureType, SimpleFeature> getFeatureStore(
+                       DataStore dataStore, Name name) {
+               FeatureSource<SimpleFeatureType, SimpleFeature> featureSource;
+               try {
+                       featureSource = dataStore.getFeatureSource(name);
+               } catch (IOException e) {
+                       throw new ArgeoException("Cannot open feature source " + name
+                                       + " in data store " + dataStore, e);
+               }
+               if (!(featureSource instanceof FeatureStore)) {
+                       throw new ArgeoException("Feature source " + name
+                                       + " is not writable.");
+               }
+               return (FeatureStore<SimpleFeatureType, SimpleFeature>) featureSource;
+       }
+
+       /** Creates the provided schema in the data store. */
+       public static void createSchemaIfNeeded(DataStore dataStore,
+                       SimpleFeatureType featureType) {
+               try {
+                       dataStore.getSchema(featureType.getName());
+               } catch (IOException e) {
+                       // assume it does not exist
+                       try {
+                               dataStore.createSchema(featureType);
+                       } catch (IOException e1) {
+                               throw new ArgeoException("Cannot create schema " + featureType,
+                                               e);
+                       }
+               }
+       }
+}