org.lumongo.server.rest.IndexResource Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of lumongo-cluster Show documentation

$project.description

The newest version!

package org.lumongo.server.rest;

import com.cedarsoftware.util.io.JsonWriter;
import com.google.protobuf.util.JsonFormat;
import org.lumongo.LumongoConstants;
import org.lumongo.server.config.IndexConfig;
import org.lumongo.server.index.LumongoIndexManager;

import javax.ws.rs.GET;
import javax.ws.rs.Path;
import javax.ws.rs.Produces;
import javax.ws.rs.QueryParam;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;

@Path(LumongoConstants.INDEX_URL)
public class IndexResource {

	private LumongoIndexManager indexManager;

	public IndexResource(LumongoIndexManager indexManager) {
		this.indexManager = indexManager;
	}

	@GET
	@Produces({ MediaType.APPLICATION_JSON + ";charset=utf-8" })
	public Response get(@Context Response response, @QueryParam(LumongoConstants.INDEX) String index, @QueryParam(LumongoConstants.PRETTY) boolean pretty) {

		try {
			StringBuilder responseBuilder = new StringBuilder();

			IndexConfig indexConfig = indexManager.getIndexConfig(index);

			responseBuilder.append("{");
			responseBuilder.append("\"indexName\": ");
			responseBuilder.append("\"");
			responseBuilder.append(indexConfig.getIndexName());
			responseBuilder.append("\"");
			responseBuilder.append(",");
			responseBuilder.append("\"numberOfSegments\": ");
			responseBuilder.append(indexConfig.getNumberOfSegments());
			responseBuilder.append(",");
			responseBuilder.append("\"indexSettings\": ");
			JsonFormat.Printer printer = JsonFormat.printer();
			responseBuilder.append(printer.print(indexConfig.getIndexSettings()));
			responseBuilder.append("}");

			String docString = responseBuilder.toString();

			if (pretty) {
				docString = JsonWriter.formatJson(docString);
			}

			return Response.status(LumongoConstants.SUCCESS).entity(docString).build();

		}
		catch (Exception e) {
			return Response.status(LumongoConstants.INTERNAL_ERROR).entity("Failed to get index names: " + e.getMessage()).build();
		}

	}

}