io.atomix.rest.resources.AtomicSemaphoreResource Maven / Gradle / Ivy
/*
* Copyright 2018-present Open Networking Foundation
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.atomix.rest.resources;
import io.atomix.core.semaphore.AsyncAtomicSemaphore;
import io.atomix.core.semaphore.AtomicSemaphoreConfig;
import io.atomix.core.semaphore.AtomicSemaphoreType;
import io.atomix.rest.AtomixResource;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.ws.rs.Consumes;
import javax.ws.rs.GET;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.container.AsyncResponse;
import javax.ws.rs.container.Suspended;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
/**
* Distributed semaphore resource.
*/
@AtomixResource
@Path("/atomic-semaphore")
public class AtomicSemaphoreResource extends PrimitiveResource {
private static final Logger LOGGER = LoggerFactory.getLogger(AtomicSemaphoreResource.class);
public AtomicSemaphoreResource() {
super(AtomicSemaphoreType.instance());
}
@POST
@Path("/{name}/acquire")
@Consumes(MediaType.APPLICATION_JSON)
@Produces(MediaType.APPLICATION_JSON)
public void acquire(
@PathParam("name") String name,
Integer permits,
@Suspended AsyncResponse response) {
getPrimitive(name).thenCompose(semaphore -> semaphore.acquire(permits != null ? permits : 1)).whenComplete((result, error) -> {
if (error == null) {
response.resume(Response.ok(result.value()).build());
} else {
LOGGER.warn("An error occurred", error);
response.resume(Response.serverError());
}
});
}
@POST
@Path("/{name}/release")
@Consumes(MediaType.APPLICATION_JSON)
@Produces(MediaType.APPLICATION_JSON)
public void release(
@PathParam("name") String name,
Integer permits,
@Suspended AsyncResponse response) {
getPrimitive(name).thenCompose(semaphore -> semaphore.release(permits != null ? permits : 1)).whenComplete((result, error) -> {
if (error == null) {
response.resume(Response.ok().build());
} else {
LOGGER.warn("An error occurred", error);
response.resume(Response.serverError());
}
});
}
@POST
@Path("/{name}/increase")
@Consumes(MediaType.APPLICATION_JSON)
@Produces(MediaType.APPLICATION_JSON)
public void increase(
@PathParam("name") String name,
Integer permits,
@Suspended AsyncResponse response) {
getPrimitive(name).thenCompose(semaphore -> semaphore.increasePermits(permits != null ? permits : 1)).whenComplete((result, error) -> {
if (error == null) {
response.resume(Response.ok(result).build());
} else {
LOGGER.warn("An error occurred", error);
response.resume(Response.serverError());
}
});
}
@POST
@Path("/{name}/reduce")
@Consumes(MediaType.APPLICATION_JSON)
@Produces(MediaType.APPLICATION_JSON)
public void reduce(
@PathParam("name") String name,
Integer permits,
@Suspended AsyncResponse response) {
getPrimitive(name).thenCompose(semaphore -> semaphore.reducePermits(permits != null ? permits : 1)).whenComplete((result, error) -> {
if (error == null) {
response.resume(Response.ok(result).build());
} else {
LOGGER.warn("An error occurred", error);
response.resume(Response.serverError());
}
});
}
@GET
@Path("/{name}/permits")
@Produces(MediaType.APPLICATION_JSON)
public void availablePermits(
@PathParam("name") String name,
@Suspended AsyncResponse response) {
getPrimitive(name).thenCompose(semaphore -> semaphore.availablePermits()).whenComplete((result, error) -> {
if (error == null) {
response.resume(Response.ok(result).build());
} else {
LOGGER.warn("An error occurred", error);
response.resume(Response.serverError());
}
});
}
}