Examples with TransientAdapterStore org.locationtech.geowave.core.store.adapter.TransientAdapterStore used on opensource projects

Example 1 with TransientAdapterStore

use of org.locationtech.geowave.core.store.adapter.TransientAdapterStore in project geowave by locationtech.

the class AbstractGeoWaveBasicVectorIT method testStats.

@SuppressWarnings("unchecked")
protected void testStats(final URL[] inputFiles, final boolean multithreaded, final CoordinateReferenceSystem crs, final Index... indices) {
    // In the multithreaded case, only test min/max and count. Stats will be
    // ingested/ in a different order and will not match.
    final LocalFileIngestPlugin<SimpleFeature> localFileIngest = new GeoToolsVectorDataStoreIngestPlugin(Filter.INCLUDE);
    final Map<String, StatisticsCache> statsCache = new HashMap<>();
    final String[] indexNames = Arrays.stream(indices).map(i -> i.getName()).toArray(i -> new String[i]);
    for (final URL inputFile : inputFiles) {
        LOGGER.warn("Calculating stats from file '" + inputFile.getPath() + "' - this may take several minutes...");
        try (final CloseableIterator<GeoWaveData<SimpleFeature>> dataIterator = localFileIngest.toGeoWaveData(inputFile, indexNames)) {
            final TransientAdapterStore adapterCache = new MemoryAdapterStore(localFileIngest.getDataAdapters());
            while (dataIterator.hasNext()) {
                final GeoWaveData<SimpleFeature> data = dataIterator.next();
                final DataTypeAdapter<SimpleFeature> adapter = data.getAdapter(adapterCache);
                // it should be a statistical data adapter
                if (adapter instanceof DefaultStatisticsProvider) {
                    StatisticsCache cachedValues = statsCache.get(adapter.getTypeName());
                    if (cachedValues == null) {
                        cachedValues = new StatisticsCache(adapter, crs);
                        statsCache.put(adapter.getTypeName(), cachedValues);
                    }
                    cachedValues.entryIngested(data.getValue());
                }
            }
        }
    }
    final DataStatisticsStore statsStore = getDataStorePluginOptions().createDataStatisticsStore();
    final PersistentAdapterStore adapterStore = getDataStorePluginOptions().createAdapterStore();
    final InternalDataAdapter<?>[] adapters = adapterStore.getAdapters();
    for (final InternalDataAdapter<?> internalDataAdapter : adapters) {
        final FeatureDataAdapter adapter = (FeatureDataAdapter) internalDataAdapter.getAdapter();
        final StatisticsCache cachedValue = statsCache.get(adapter.getTypeName());
        Assert.assertNotNull(cachedValue);
        final Set<Entry<Statistic<?>, Map<ByteArray, StatisticValue<?>>>> expectedStats = cachedValue.statsCache.entrySet();
        int statsCount = 0;
        try (CloseableIterator<? extends Statistic<? extends StatisticValue<?>>> statsIterator = statsStore.getDataTypeStatistics(adapter, null, null)) {
            while (statsIterator.hasNext()) {
                statsIterator.next();
                statsCount++;
            }
        }
        try (CloseableIterator<? extends Statistic<? extends StatisticValue<?>>> statsIterator = statsStore.getFieldStatistics(adapter, null, null, null)) {
            while (statsIterator.hasNext()) {
                statsIterator.next();
                statsCount++;
            }
        }
        Assert.assertEquals("The number of stats for data adapter '" + adapter.getTypeName() + "' do not match count expected", expectedStats.size(), statsCount);
        for (final Entry<Statistic<?>, Map<ByteArray, StatisticValue<?>>> expectedStat : expectedStats) {
            for (final Entry<ByteArray, StatisticValue<?>> expectedValues : expectedStat.getValue().entrySet()) {
                StatisticValue<Object> actual;
                if (expectedValues.getKey().equals(StatisticValue.NO_BIN)) {
                    actual = statsStore.getStatisticValue((Statistic<StatisticValue<Object>>) expectedStat.getKey());
                } else {
                    actual = statsStore.getStatisticValue((Statistic<StatisticValue<Object>>) expectedStat.getKey(), expectedValues.getKey());
                }
                assertEquals(expectedValues.getValue().getValue(), actual.getValue());
            }
        }
        // finally check the one stat that is more manually calculated -
        // the bounding box
        StatisticQuery<BoundingBoxValue, Envelope> query = StatisticQueryBuilder.newBuilder(BoundingBoxStatistic.STATS_TYPE).fieldName(adapter.getFeatureType().getGeometryDescriptor().getLocalName()).typeName(adapter.getTypeName()).build();
        BoundingBoxValue bboxStat = getDataStorePluginOptions().createDataStore().aggregateStatistics(query);
        validateBBox(bboxStat.getValue(), cachedValue);
        // now make sure it works without giving field name because there is only one geometry field
        // anyways
        query = StatisticQueryBuilder.newBuilder(BoundingBoxStatistic.STATS_TYPE).typeName(adapter.getTypeName()).build();
        bboxStat = getDataStorePluginOptions().createDataStore().aggregateStatistics(query);
        validateBBox(bboxStat.getValue(), cachedValue);
        final StatisticId<BoundingBoxValue> bboxStatId = FieldStatistic.generateStatisticId(adapter.getTypeName(), BoundingBoxStatistic.STATS_TYPE, adapter.getFeatureType().getGeometryDescriptor().getLocalName(), Statistic.INTERNAL_TAG);
        Assert.assertTrue("Unable to remove individual stat", statsStore.removeStatistic(statsStore.getStatisticById(bboxStatId)));
        Assert.assertNull("Individual stat was not successfully removed", statsStore.getStatisticById(bboxStatId));
    }
}

Example 2 with TransientAdapterStore

use of org.locationtech.geowave.core.store.adapter.TransientAdapterStore in project geowave by locationtech.

the class GeoWaveOutputFormat method getRecordWriter.

@Override
public RecordWriter<GeoWaveOutputKey<Object>, Object> getRecordWriter(final TaskAttemptContext context) throws IOException, InterruptedException {
    try {
        final Map<String, String> configOptions = getStoreOptionsMap(context);
        final IndexStore persistentIndexStore = GeoWaveStoreFinder.createIndexStore(configOptions);
        final DataStore dataStore = GeoWaveStoreFinder.createDataStore(configOptions);
        final Index[] indices = JobContextIndexStore.getIndices(context);
        if (LOGGER.isDebugEnabled()) {
            final StringBuilder sbDebug = new StringBuilder();
            sbDebug.append("Config Options: ");
            for (final Map.Entry<String, String> entry : configOptions.entrySet()) {
                sbDebug.append(entry.getKey() + "/" + entry.getValue() + ", ");
            }
            sbDebug.append("\n\tIndices Size: " + indices.length);
            sbDebug.append("\n\tpersistentIndexStore: " + persistentIndexStore);
            final String filename = "/META-INF/services/org.locationtech.geowave.core.store.StoreFactoryFamilySpi";
            final InputStream is = context.getClass().getResourceAsStream(filename);
            if (is == null) {
                sbDebug.append("\n\tStoreFactoryFamilySpi: Unable to open file '" + filename + "'");
            } else {
                sbDebug.append("\n\tStoreFactoryFamilySpi: " + IOUtils.toString(is, "UTF-8"));
                is.close();
            }
            LOGGER.debug(sbDebug.toString());
        }
        for (final Index i : indices) {
            if (!persistentIndexStore.indexExists(i.getName())) {
                dataStore.addIndex(i);
            }
        }
        final TransientAdapterStore jobContextAdapterStore = GeoWaveConfiguratorBase.getJobContextAdapterStore(CLASS, context);
        final IndexStore jobContextIndexStore = new JobContextIndexStore(context, persistentIndexStore);
        return new GeoWaveRecordWriter(dataStore, jobContextIndexStore, jobContextAdapterStore);
    } catch (final Exception e) {
        throw new IOException(e);
    }
}