private void groupDistinct(Database database, int dataType) { if (distinctValues == null) { return; } count = 0; for (Value v : distinctValues.keys()) { add(database, dataType, false, v); } }
@Override void add(Database database, int dataType, boolean distinct, Value v) { if (distinctValues == null) { distinctValues = ValueHashMap.newInstance(); } AggregateDataHistogram a = distinctValues.get(v); if (a == null) { if (distinctValues.size() < Constants.SELECTIVITY_DISTINCT_COUNT) { a = new AggregateDataHistogram(); distinctValues.put(v, a); } } if (a != null) { a.count++; } }
@Override Value getValue(Database database, int dataType, boolean distinct) { if (distinct) { count = 0; groupDistinct(database, dataType); } ValueArray[] values = new ValueArray[distinctValues.size()]; int i = 0; for (Value dv : distinctValues.keys()) { AggregateDataHistogram d = distinctValues.get(dv); values[i] = ValueArray.get(new Value[]{dv, ValueLong.get(d.count)}); i++; } final CompareMode compareMode = database.getCompareMode(); Arrays.sort(values, new Comparator<ValueArray>() { @Override public int compare(ValueArray v1, ValueArray v2) { Value a1 = v1.getList()[0]; Value a2 = v2.getList()[0]; return a1.compareTo(a2, compareMode); } }); Value v = ValueArray.get(values); return v.convertTo(dataType); }
/** * Create an AggregateData object of the correct sub-type. * * @param aggregateType the type of the aggregate operation * @return the aggregate data object of the specified type */ static AggregateData create(int aggregateType) { if (aggregateType == Aggregate.SELECTIVITY) { return new AggregateDataSelectivity(); } else if (aggregateType == Aggregate.GROUP_CONCAT) { return new AggregateDataGroupConcat(); } else if (aggregateType == Aggregate.COUNT_ALL) { return new AggregateDataCountAll(); } else if (aggregateType == Aggregate.COUNT) { return new AggregateDataCount(); } else if (aggregateType == Aggregate.HISTOGRAM) { return new AggregateDataHistogram(); } else { return new AggregateDataDefault(aggregateType); } }