io.antmedia.console.rest.ClusterRestServiceV2 Maven / Gradle / Ivy

package io.antmedia.console.rest;

import java.util.ArrayList;
import java.util.List;

import javax.servlet.ServletContext;
import javax.ws.rs.GET;
import javax.ws.rs.Path;
import javax.ws.rs.DELETE;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.web.context.support.WebApplicationContextUtils;

import io.antmedia.cluster.ClusterNode;
import io.antmedia.cluster.IClusterNotifier;
import io.antmedia.cluster.IClusterStore;
import io.antmedia.rest.BroadcastRestService.SimpleStat;
import io.antmedia.rest.model.Result;

@Component
@Path("/v2/cluster")
public class ClusterRestServiceV2 {
	protected static Logger logger = LoggerFactory.getLogger(ClusterRestServiceV2.class);
	
	@Context
	private ServletContext servletContext;
	
	public IClusterStore getClusterStore() 
	{
		IClusterStore clusterStore = null;
		WebApplicationContext ctxt = WebApplicationContextUtils.getWebApplicationContext(servletContext);
		if (ctxt != null) {
			IClusterNotifier clusterNotifier = (IClusterNotifier) ctxt.getBean(IClusterNotifier.BEAN_NAME);
			clusterStore = clusterNotifier.getClusterStore();
		}
		
		return clusterStore;
	}
	
	@GET
	@Path("/node-count")
	public SimpleStat getNodeCount() {
		IClusterStore clusterStore = getClusterStore();
		long nodeCount = -1;
		if (clusterStore != null) {
			nodeCount = clusterStore.getNodeCount();
		}
		return new SimpleStat(nodeCount);
	}
	
	@GET
	@Path("/nodes/{offset}/{size}")
	@Produces(MediaType.APPLICATION_JSON)
	public List getNodeList(@PathParam("offset") int offset, @PathParam("size") int size) 
	{
		IClusterStore clusterStore = getClusterStore();
		List nodeList = null;
		if (clusterStore != null) 
		{
			nodeList = clusterStore.getClusterNodes(offset, size);
		}
		else {
			nodeList = new ArrayList<>();
		}
		return nodeList;
	}	
	
	@DELETE
	@Path("/node/{id}")
	@Produces(MediaType.APPLICATION_JSON)
	public Result deleteNode(@PathParam("id") String nodeId) 
	{
		boolean result = false;
		IClusterStore clusterStore = getClusterStore();
		if (clusterStore != null) 
		{
			result = clusterStore.deleteNode(nodeId);
		}
		return new Result(result);
	}
}