package io.vertx.ext.cluster.infinispan.impl;
import io.vertx.core.AsyncResult;
import io.vertx.core.Handler;
import io.vertx.core.Vertx;
import io.vertx.core.shareddata.Counter;
import org.infinispan.counter.api.SyncStrongCounter;
import java.util.Objects;
public class InfinispanCounter implements Counter {
private final Vertx vertx;
private final SyncStrongCounter strongCounter;
public InfinispanCounter(Vertx vertx, SyncStrongCounter strongCounter) {
this.vertx = vertx;
this.strongCounter = strongCounter;
}
@Override
public void get(Handler<AsyncResult<Long>> resultHandler) {
Objects.requireNonNull(resultHandler, "resultHandler");
vertx.executeBlocking(future -> {
future.complete(strongCounter.getValue());
}, false, resultHandler);
}
@Override
public void incrementAndGet(Handler<AsyncResult<Long>> resultHandler) {
Objects.requireNonNull(resultHandler, "resultHandler");
addAndGet(1L, resultHandler);
}
@Override
public void getAndIncrement(Handler<AsyncResult<Long>> resultHandler) {
Objects.requireNonNull(resultHandler, "resultHandler");
getAndAdd(1L, resultHandler);
}
@Override
public void decrementAndGet(Handler<AsyncResult<Long>> resultHandler) {
Objects.requireNonNull(resultHandler, "resultHandler");
addAndGet(-1L, resultHandler);
}
@Override
public void addAndGet(long value, Handler<AsyncResult<Long>> resultHandler) {
Objects.requireNonNull(resultHandler, "resultHandler");
vertx.executeBlocking(future -> {
future.complete(strongCounter.addAndGet(value));
}, false, resultHandler);
}
@Override
public void getAndAdd(long value, Handler<AsyncResult<Long>> resultHandler) {
Objects.requireNonNull(resultHandler, "resultHandler");
vertx.executeBlocking(future -> {
future.complete(strongCounter.addAndGet(value) - value);
}, false, resultHandler);
}
@Override
public void compareAndSet(long expected, long value, Handler<AsyncResult<Boolean>> resultHandler) {
Objects.requireNonNull(resultHandler, "resultHandler");
vertx.executeBlocking(future -> {
future.complete(strongCounter.compareAndSet(expected, value));
}, false, resultHandler);
}
}