package org.xbib.elasticsearch.rest.action.termlist;
import org.elasticsearch.client.Client;
import org.elasticsearch.common.Strings;
import org.elasticsearch.common.inject.Inject;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.common.xcontent.XContentBuilder;
import org.elasticsearch.rest.BaseRestHandler;
import org.elasticsearch.rest.BytesRestResponse;
import org.elasticsearch.rest.RestChannel;
import org.elasticsearch.rest.RestController;
import org.elasticsearch.rest.RestRequest;
import org.elasticsearch.rest.RestResponse;
import org.elasticsearch.rest.RestStatus;
import org.elasticsearch.rest.action.support.RestBuilderListener;
import org.xbib.elasticsearch.action.termlist.TermInfo;
import org.xbib.elasticsearch.action.termlist.TermlistAction;
import org.xbib.elasticsearch.action.termlist.TermlistRequest;
import org.xbib.elasticsearch.action.termlist.TermlistResponse;
import java.util.Map;
import static org.elasticsearch.rest.RestRequest.Method.GET;
import static org.elasticsearch.rest.action.support.RestActions.buildBroadcastShardsHeader;
public class RestTermlistAction extends BaseRestHandler {
@Inject
public RestTermlistAction(Settings settings, Client client, RestController controller) {
super(settings, client);
controller.registerHandler(GET, "/_termlist", this);
controller.registerHandler(GET, "/{index}/_termlist", this);
controller.registerHandler(GET, "/{index}/{field}/_termlist", this);
}
public void handleRequest(final RestRequest request, final RestChannel channel, final Client client) {
TermlistRequest termlistRequest = new TermlistRequest(
Strings.splitStringByCommaToArray(request.param("index")));
termlistRequest.setField(request.param("field"));
termlistRequest.setTerm(request.param("term"));
termlistRequest.setFrom(request.paramAsInt("from", 0));
termlistRequest.setSize(request.paramAsInt("size", 0));
termlistRequest.setWithDocFreq(request.paramAsBoolean("docfreqs", false));
termlistRequest.setWithTotalFreq(request.paramAsBoolean("totalfreqs", false));
termlistRequest.sortByDocFreq(request.paramAsBoolean("sortbydocfreqs", false));
termlistRequest.sortByTotalFreq(request.paramAsBoolean("sortbytotalfreqs", false));
termlistRequest.sortByTerm(request.paramAsBoolean("sortbyterms", false));
client.execute(TermlistAction.INSTANCE, termlistRequest, new RestBuilderListener<TermlistResponse>(channel) {
@Override
public RestResponse buildResponse(TermlistResponse response, XContentBuilder builder) throws Exception {
builder.startObject();
buildBroadcastShardsHeader(builder, response);
builder.field("total", response.getSize());
builder.startArray("terms");
for (Map.Entry<String, TermInfo> t : response.getTermlist().entrySet()) {
builder.startObject().field("name", t.getKey());
if (t.getValue().getDocFreq() != null) {
builder.field("docfreq", t.getValue().getDocFreq());
}
if (t.getValue().getTotalFreq() != null) {
builder.field("totalfreq", t.getValue().getTotalFreq());
}
builder.endObject();
}
builder.endArray();
builder.endObject();
return new BytesRestResponse(RestStatus.OK, builder);
}
});
}
}