Examples with ComplexMetricSerde org.apache.druid.segment.serde.ComplexMetricSerde used on opensource projects

Example 6 with ComplexMetricSerde

use of org.apache.druid.segment.serde.ComplexMetricSerde in project druid by druid-io.

the class IndexMergerV9 method setupMetricsWriters.

private ArrayList<GenericColumnSerializer> setupMetricsWriters(final SegmentWriteOutMedium segmentWriteOutMedium, final List<String> mergedMetrics, final Map<String, ValueType> metricsValueTypes, final Map<String, String> metricTypeNames, final IndexSpec indexSpec) throws IOException {
    ArrayList<GenericColumnSerializer> metWriters = Lists.newArrayListWithCapacity(mergedMetrics.size());
    for (String metric : mergedMetrics) {
        ValueType type = metricsValueTypes.get(metric);
        GenericColumnSerializer writer;
        switch(type) {
            case LONG:
                writer = createLongColumnSerializer(segmentWriteOutMedium, metric, indexSpec);
                break;
            case FLOAT:
                writer = createFloatColumnSerializer(segmentWriteOutMedium, metric, indexSpec);
                break;
            case DOUBLE:
                writer = createDoubleColumnSerializer(segmentWriteOutMedium, metric, indexSpec);
                break;
            case COMPLEX:
                final String typeName = metricTypeNames.get(metric);
                ComplexMetricSerde serde = ComplexMetrics.getSerdeForType(typeName);
                if (serde == null) {
                    throw new ISE("Unknown type[%s]", typeName);
                }
                writer = serde.getSerializer(segmentWriteOutMedium, metric);
                break;
            default:
                throw new ISE("Unknown type[%s]", type);
        }
        writer.open();
        // we will close these writers in another method after we added all the metrics
        metWriters.add(writer);
    }
    return metWriters;
}