com.enonic.xp.lib.node.mapper.NodeMultiRepoQueryResultMapper 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.Aggregations;
import com.enonic.xp.node.FindNodesByMultiRepoQueryResult;
import com.enonic.xp.node.MultiRepoNodeHit;
import com.enonic.xp.node.MultiRepoNodeHits;
import com.enonic.xp.script.serializer.MapGenerator;
import com.enonic.xp.suggester.Suggestions;

public final class NodeMultiRepoQueryResultMapper
    extends AbstractQueryResultMapper
{
    private final MultiRepoNodeHits nodeHits;

    private final long total;

    private final Aggregations aggregations;

    private final Suggestions suggestions;

    public NodeMultiRepoQueryResultMapper( final FindNodesByMultiRepoQueryResult result )
    {
        this.nodeHits = result.getNodeHits();
        this.total = result.getTotalHits();
        this.aggregations = result.getAggregations();
        this.suggestions = result.getSuggestions();
    }

    @Override
    public void serialize( final MapGenerator gen )
    {
        gen.value( "total", this.total );
        gen.value( "count", this.nodeHits.getSize() );
        serialize( gen, this.nodeHits );
        serialize( gen, aggregations );
        serialize( gen, suggestions );
    }

    private void serialize( final MapGenerator gen, final MultiRepoNodeHits nodeHits )
    {
        gen.array( "hits" );
        for ( MultiRepoNodeHit nodeHit : nodeHits )
        {
            gen.map();
            gen.value( "id", nodeHit.getNodeId() );
            gen.value( "score", Float.isNaN( nodeHit.getScore() ) ? 0.0 : nodeHit.getScore() );
            gen.value( "repoId", nodeHit.getRepositoryId().toString() );
            gen.value( "branch", nodeHit.getBranch().getValue() );
            serialize( gen, nodeHit.getExplanation() );
            serialize( gen, nodeHit.getHighlight() );
            gen.end();
        }
        gen.end();
    }

    private void serialize( final MapGenerator gen, final Suggestions suggestions )
    {
        if ( suggestions != null && suggestions.isNotEmpty() )
        {
            gen.map( "suggestions" );
            new SuggestionsMapper( suggestions ).serialize( gen );
            gen.end();
        }
    }
}