org.elasticsearch.action.suggest.SuggestResponse Maven / Gradle / Ivy

Go to download
package org.elasticsearch.action.suggest;

import java.io.IOException;
import java.util.List;

import org.elasticsearch.action.ShardOperationFailedException;
import org.elasticsearch.action.support.broadcast.BroadcastOperationResponse;
import org.elasticsearch.common.collect.Lists;
import org.elasticsearch.common.io.stream.StreamInput;
import org.elasticsearch.common.io.stream.StreamOutput;

public class SuggestResponse extends BroadcastOperationResponse {

    private List suggestions;

    public SuggestResponse() {
    }

    public SuggestResponse(List suggestions, int totalShards, int successfulShards, int failedShards, List shardFailures) {
        super(totalShards, successfulShards, failedShards, shardFailures);
        this.suggestions = suggestions;
    }

    public List suggestions() {
        return Lists.newArrayList(suggestions);
    }

    public List  getSuggestions() {
        return Lists.newArrayList(suggestions);
    }

    @Override public void readFrom(StreamInput in) throws IOException {
        super.readFrom(in);
        int size = in.readVInt();
        suggestions = Lists.newArrayList();
        for (int i = 0; i < size; i++) {
            suggestions.add(in.readString());
        }
    }

    @Override public void writeTo(StreamOutput out) throws IOException {
        super.writeTo(out);
        out.writeVInt(suggestions.size());
        for (String suggestion: suggestions) {
            out.writeString(suggestion);
        }
    }
}