com.enonic.xp.lib.node.mapper.AggregationMapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of lib-node Show documentation
lib-node
The newest version!
package com.enonic.xp.lib.node.mapper;

import com.enonic.xp.aggregation.Aggregation;
import com.enonic.xp.aggregation.Aggregations;
import com.enonic.xp.aggregation.Bucket;
import com.enonic.xp.aggregation.BucketAggregation;
import com.enonic.xp.aggregation.Buckets;
import com.enonic.xp.aggregation.DateRangeBucket;
import com.enonic.xp.aggregation.NumericRangeBucket;
import com.enonic.xp.aggregation.SingleValueMetricAggregation;
import com.enonic.xp.aggregation.StatsAggregation;
import com.enonic.xp.script.serializer.MapGenerator;
import com.enonic.xp.script.serializer.MapSerializable;

final class AggregationMapper
    implements MapSerializable
{
    private final Aggregations value;

    AggregationMapper( final Aggregations value )
    {
        this.value = value;
    }

    @Override
    public void serialize( final MapGenerator gen )
    {
        serializeAggregations( gen, this.value );
    }

    private static void serializeAggregations( final MapGenerator gen, final Aggregations value )
    {
        for ( Aggregation aggregation : value )
        {
            gen.map( aggregation.getName() );
            if ( aggregation instanceof BucketAggregation )
            {
                final Buckets buckets = ( (BucketAggregation) aggregation ).getBuckets();
                serializeBuckets( gen, buckets );
            }
            else if ( aggregation instanceof StatsAggregation )
            {
                final StatsAggregation statsAggregation = ( (StatsAggregation) aggregation );
                serializeStatsAggregation( gen, statsAggregation );
            }
            else if ( aggregation instanceof SingleValueMetricAggregation )
            {
                gen.value( "value", ( (SingleValueMetricAggregation) aggregation ).getValue() );
            }
            gen.end();
        }
    }

    private static void serializeBuckets( final MapGenerator gen, final Buckets value )
    {
        gen.array( "buckets" );
        for ( Bucket bucket : value )
        {
            serializeBucket( gen, bucket );
        }
        gen.end();

    }

    private static void serializeBucket( final MapGenerator gen, final Bucket value )
    {
        gen.map();
        gen.value( "key", value.getKey() );
        gen.value( "docCount", value.getDocCount() );
        if ( value instanceof DateRangeBucket )
        {
            serializeDateBucket( gen, (DateRangeBucket) value );
        }
        else if ( value instanceof NumericRangeBucket )
        {
            serializeNumericBucket( gen, (NumericRangeBucket) value );
        }

        final Aggregations subAggregations = value.getSubAggregations();
        if ( subAggregations != null )
        {
            serializeAggregations( gen, subAggregations );
        }
        gen.end();
    }

    private static void serializeNumericBucket( final MapGenerator gen, final NumericRangeBucket value )
    {
        gen.value( "from", value.getFrom() );
        gen.value( "to", value.getTo() );
    }

    private static void serializeDateBucket( final MapGenerator gen, final DateRangeBucket value )
    {
        gen.value( "from", value.getFrom() );
        gen.value( "to", value.getTo() );
    }

    private static void serializeStatsAggregation( final MapGenerator gen, final StatsAggregation value )
    {
        gen.value( "count", value.getCount() );
        gen.value( "min", value.getMin() );
        gen.value( "max", value.getMax() );
        gen.value( "avg", value.getAvg() );
        gen.value( "sum", value.getSum() );
    }
}