use of org.locationtech.geowave.core.store.api.DataTypeStatistic in project geowave by locationtech.
the class ListStatTypesCommand method listAllRegisteredStatistics.
private void listAllRegisteredStatistics(final Console console) {
final List<Statistic<?>> indexStats = Lists.newLinkedList();
final List<Statistic<?>> adapterStats = Lists.newLinkedList();
final List<Statistic<?>> fieldStats = Lists.newLinkedList();
final List<? extends Statistic<? extends StatisticValue<?>>> allStats = StatisticsRegistry.instance().getAllRegisteredStatistics();
Collections.sort(allStats, (s1, s2) -> s1.getStatisticType().getString().compareTo(s2.getStatisticType().getString()));
for (final Statistic<?> statistic : allStats) {
if (statistic instanceof IndexStatistic) {
indexStats.add(statistic);
} else if (statistic instanceof DataTypeStatistic) {
adapterStats.add(statistic);
} else if (statistic instanceof FieldStatistic) {
fieldStats.add(statistic);
}
}
final ConsoleTablePrinter printer = new ConsoleTablePrinter(0, Integer.MAX_VALUE, console);
displayStatList(printer, indexStats, "Registered Index Statistics");
displayStatList(printer, adapterStats, "Registered Adapter Statistics");
displayStatList(printer, fieldStats, "Registered Field Statistics");
displayBinningStrategies(printer);
}
use of org.locationtech.geowave.core.store.api.DataTypeStatistic in project geowave by locationtech.
the class StatsCommandLineOptions method resolveMatchingStatistics.
@SuppressWarnings({ "rawtypes", "unchecked" })
public List<Statistic<? extends StatisticValue<?>>> resolveMatchingStatistics(final DataStore dataStore, final DataStatisticsStore statsStore, final IndexStore indexStore) {
final List<Statistic<? extends StatisticValue<?>>> matching = Lists.newArrayList();
if ((indexName != null) && ((typeName != null) || (fieldName != null))) {
throw new ParameterException("Unable to process index statistics for a single type. Specify either an index name or a type name.");
}
StatisticType statisticType = null;
if (statType != null) {
statisticType = StatisticsRegistry.instance().getStatisticType(statType);
if (statisticType == null) {
throw new ParameterException("Unrecognized statistic type: " + statType);
}
}
if (statisticType != null) {
if (statisticType instanceof IndexStatisticType) {
if (indexName == null) {
throw new ParameterException("An index name must be supplied when specifying an index statistic type.");
}
final Index index = indexStore.getIndex(indexName);
if (index == null) {
throw new ParameterException("Unable to find an index named: " + indexName);
}
try (CloseableIterator<? extends Statistic<? extends StatisticValue<?>>> stats = statsStore.getIndexStatistics(index, statisticType, tag)) {
stats.forEachRemaining(stat -> matching.add(stat));
}
} else if (statisticType instanceof DataTypeStatisticType) {
if (typeName == null) {
throw new ParameterException("A type name must be supplied when specifying a data type statistic type.");
}
final DataTypeAdapter<?> adapter = dataStore.getType(typeName);
if (adapter == null) {
throw new ParameterException("Unable to find an type named: " + typeName);
}
try (CloseableIterator<? extends Statistic<? extends StatisticValue<?>>> stats = statsStore.getDataTypeStatistics(adapter, statisticType, tag)) {
stats.forEachRemaining(stat -> matching.add(stat));
}
} else if (statisticType instanceof FieldStatisticType) {
if (typeName == null) {
throw new ParameterException("A type name must be supplied when specifying a field statistic type.");
}
final DataTypeAdapter<?> adapter = dataStore.getType(typeName);
if (adapter == null) {
throw new ParameterException("Unable to find an type named: " + typeName);
}
if (fieldName == null) {
throw new ParameterException("A field name must be supplied when specifying a field statistic type.");
}
boolean fieldFound = false;
final FieldDescriptor[] fields = adapter.getFieldDescriptors();
for (int i = 0; i < fields.length; i++) {
if (fields[i].fieldName().equals(fieldName)) {
fieldFound = true;
break;
}
}
if (!fieldFound) {
throw new ParameterException("Unable to find a field named '" + fieldName + "' on type '" + typeName + "'.");
}
try (CloseableIterator<? extends Statistic<? extends StatisticValue<?>>> stats = statsStore.getFieldStatistics(adapter, statisticType, fieldName, tag)) {
stats.forEachRemaining(stat -> matching.add(stat));
}
}
} else {
try (CloseableIterator<? extends Statistic<? extends StatisticValue<?>>> stats = statsStore.getAllStatistics(null)) {
stats.forEachRemaining(stat -> {
// This could all be optimized to one giant check, but it's split for readability
if ((tag != null) && !tag.equals(stat.getTag())) {
return;
}
if ((indexName != null) && (!(stat instanceof IndexStatistic) || !indexName.equals(((IndexStatistic) stat).getIndexName()))) {
return;
}
if (typeName != null) {
if (stat instanceof IndexStatistic) {
return;
}
if ((stat instanceof DataTypeStatistic) && !typeName.equals(((DataTypeStatistic) stat).getTypeName())) {
return;
}
if ((stat instanceof FieldStatistic) && !typeName.equals(((FieldStatistic) stat).getTypeName())) {
return;
}
}
if ((fieldName != null) && (!(stat instanceof FieldStatistic) || !fieldName.equals(((FieldStatistic) stat).getFieldName()))) {
return;
}
matching.add(stat);
});
}
}
return matching;
}
use of org.locationtech.geowave.core.store.api.DataTypeStatistic in project geowave by locationtech.
the class BaseDataStore method groupAndValidateStats.
private Pair<Map<Index, List<IndexStatistic<?>>>, Map<InternalDataAdapter<?>, List<Statistic<? extends StatisticValue<?>>>>> groupAndValidateStats(final Statistic<? extends StatisticValue<?>>[] statistics, final boolean allowExisting) {
final Map<Index, List<IndexStatistic<?>>> indexStatsToAdd = Maps.newHashMap();
final Map<InternalDataAdapter<?>, List<Statistic<? extends StatisticValue<?>>>> otherStatsToAdd = Maps.newHashMap();
for (final Statistic<? extends StatisticValue<?>> statistic : statistics) {
if (!allowExisting && statisticsStore.exists(statistic)) {
throw new IllegalArgumentException("The statistic already exists. If adding it is still desirable, use a 'tag' to make the statistic unique.");
}
if (statistic instanceof IndexStatistic) {
final IndexStatistic<?> indexStat = (IndexStatistic<?>) statistic;
if (indexStat.getIndexName() == null) {
throw new IllegalArgumentException("No index specified.");
}
final Index index = indexStore.getIndex(indexStat.getIndexName());
if (index == null) {
throw new IllegalArgumentException("No index named " + indexStat.getIndexName() + ".");
}
if (!indexStatsToAdd.containsKey(index)) {
indexStatsToAdd.put(index, Lists.newArrayList());
}
indexStatsToAdd.get(index).add(indexStat);
} else if (statistic instanceof DataTypeStatistic) {
final DataTypeStatistic<?> adapterStat = (DataTypeStatistic<?>) statistic;
if (adapterStat.getTypeName() == null) {
throw new IllegalArgumentException("No type specified.");
}
final InternalDataAdapter<?> adapter = getInternalAdapter(adapterStat.getTypeName());
if (adapter == null) {
throw new IllegalArgumentException("No type named " + adapterStat.getTypeName() + ".");
}
if (!otherStatsToAdd.containsKey(adapter)) {
otherStatsToAdd.put(adapter, Lists.newArrayList());
}
otherStatsToAdd.get(adapter).add(adapterStat);
} else if (statistic instanceof FieldStatistic) {
final FieldStatistic<?> fieldStat = (FieldStatistic<?>) statistic;
if (fieldStat.getTypeName() == null) {
throw new IllegalArgumentException("No type specified.");
}
final InternalDataAdapter<?> adapter = getInternalAdapter(fieldStat.getTypeName());
if (adapter == null) {
throw new IllegalArgumentException("No type named " + fieldStat.getTypeName() + ".");
}
if (fieldStat.getFieldName() == null) {
throw new IllegalArgumentException("No field specified.");
}
boolean foundMatch = false;
final FieldDescriptor<?>[] fields = adapter.getFieldDescriptors();
for (int i = 0; i < fields.length; i++) {
if (fieldStat.getFieldName().equals(fields[i].fieldName())) {
foundMatch = true;
break;
}
}
if (!foundMatch) {
throw new IllegalArgumentException("No field named " + fieldStat.getFieldName() + " was found on the type " + fieldStat.getTypeName() + ".");
}
if (!otherStatsToAdd.containsKey(adapter)) {
otherStatsToAdd.put(adapter, Lists.newArrayList());
}
otherStatsToAdd.get(adapter).add(fieldStat);
} else {
throw new IllegalArgumentException("Unrecognized statistic type.");
}
}
return Pair.of(indexStatsToAdd, otherStatsToAdd);
}
Aggregations