org.locationtech.geowave.examples.query.CQLQueryExample Maven / Gradle / Ivy
/**
* Copyright (c) 2013-2022 Contributors to the Eclipse Foundation
*
* See the NOTICE file distributed with this work for additional information regarding copyright
* ownership. All rights reserved. This program and the accompanying materials are made available
* under the terms of the Apache License, Version 2.0 which accompanies this distribution and is
* available at http://www.apache.org/licenses/LICENSE-2.0.txt
*/
package org.locationtech.geowave.examples.query;
import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import org.geotools.feature.AttributeTypeBuilder;
import org.geotools.feature.simple.SimpleFeatureBuilder;
import org.geotools.feature.simple.SimpleFeatureTypeBuilder;
import org.geotools.filter.text.cql2.CQLException;
import org.locationtech.geowave.adapter.vector.FeatureDataAdapter;
import org.locationtech.geowave.core.geotime.index.SpatialDimensionalityTypeProvider;
import org.locationtech.geowave.core.geotime.index.SpatialOptions;
import org.locationtech.geowave.core.geotime.store.query.api.VectorQueryBuilder;
import org.locationtech.geowave.core.geotime.util.GeometryUtils;
import org.locationtech.geowave.core.store.CloseableIterator;
import org.locationtech.geowave.core.store.api.DataStore;
import org.locationtech.geowave.core.store.api.DataStoreFactory;
import org.locationtech.geowave.core.store.api.Index;
import org.locationtech.geowave.core.store.api.Writer;
import org.locationtech.geowave.core.store.memory.MemoryRequiredOptions;
import org.locationtech.jts.geom.Coordinate;
import org.locationtech.jts.geom.Geometry;
import org.opengis.feature.simple.SimpleFeature;
import org.opengis.feature.simple.SimpleFeatureType;
/**
* This class is intended to provide a self-contained, easy-to-follow example of a few GeoTools
* queries against GeoWave. For simplicity, a MiniAccumuloCluster is spun up and a few points from
* the DC area are ingested (Washington Monument, White House, FedEx Field). Two queries are
* executed against this data set.
*/
public class CQLQueryExample {
private static DataStore dataStore;
private static final Index index =
SpatialDimensionalityTypeProvider.createIndexFromOptions(new SpatialOptions());
// Points (to be ingested into GeoWave Data Store)
private static final Coordinate washingtonMonument = new Coordinate(-77.0352, 38.8895);
private static final Coordinate whiteHouse = new Coordinate(-77.0366, 38.8977);
private static final Coordinate fedexField = new Coordinate(-76.8644, 38.9078);
private static final Coordinate bayBridgeAirport = new Coordinate(-76.350677, 38.9641511);
private static final Coordinate wideWater = new Coordinate(-77.3384112, 38.416091);
private static final Map cannedData = new HashMap<>();
static {
cannedData.put("Washington Monument", washingtonMonument);
cannedData.put("White House", whiteHouse);
cannedData.put("FedEx Field", fedexField);
cannedData.put("Bay Bridge Airport", bayBridgeAirport);
cannedData.put("Wide Water Beach", wideWater);
}
static final FeatureDataAdapter ADAPTER = new FeatureDataAdapter(getPointSimpleFeatureType());
public static void main(final String[] args) throws IOException, CQLException {
dataStore = DataStoreFactory.createDataStore(new MemoryRequiredOptions());
// ingest 3 points represented as SimpleFeatures: Washington Monument,
// White House, FedEx Field
ingestCannedData();
// execute a query for a bounding box
executeCQLQuery();
}
private static void executeCQLQuery() throws IOException, CQLException {
System.out.println("Executing query, expecting to match two points...");
final VectorQueryBuilder bldr = VectorQueryBuilder.newBuilder();
try (final CloseableIterator iterator =
dataStore.query(
bldr.indexName(index.getName()).addTypeName(ADAPTER.getTypeName()).constraints(
bldr.constraintsFactory().cqlConstraints(
"BBOX(geometry,-77.6167,38.6833,-76.6,38.9200) and locationName like 'W%'")).build())) {
while (iterator.hasNext()) {
System.out.println("Query match: " + iterator.next().getID());
}
}
}
private static void ingestCannedData() throws IOException {
final List points = new ArrayList<>();
System.out.println("Building SimpleFeatures from canned data set...");
for (final Entry entry : cannedData.entrySet()) {
System.out.println("Added point: " + entry.getKey());
points.add(buildSimpleFeature(entry.getKey(), entry.getValue()));
}
System.out.println("Ingesting canned data...");
dataStore.addType(ADAPTER, index);
try (Writer indexWriter = dataStore.createWriter(ADAPTER.getTypeName())) {
for (final SimpleFeature sf : points) {
//
indexWriter.write(sf);
}
}
System.out.println("Ingest complete.");
}
private static SimpleFeatureType getPointSimpleFeatureType() {
final String NAME = "PointSimpleFeatureType";
final SimpleFeatureTypeBuilder sftBuilder = new SimpleFeatureTypeBuilder();
final AttributeTypeBuilder atBuilder = new AttributeTypeBuilder();
sftBuilder.setName(NAME);
sftBuilder.add(atBuilder.binding(String.class).nillable(false).buildDescriptor("locationName"));
sftBuilder.add(atBuilder.binding(Geometry.class).nillable(false).buildDescriptor("geometry"));
return sftBuilder.buildFeatureType();
}
private static SimpleFeature buildSimpleFeature(
final String locationName,
final Coordinate coordinate) {
final SimpleFeatureBuilder builder = new SimpleFeatureBuilder(getPointSimpleFeatureType());
builder.set("locationName", locationName);
builder.set("geometry", GeometryUtils.GEOMETRY_FACTORY.createPoint(coordinate));
return builder.buildFeature(locationName);
}
}