/* * 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<AsyncAtomicSemaphore, AtomicSemaphoreConfig> { 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()); } }); } }