org.redisson.transaction.RedissonTransactionalBucket Maven / Gradle / Ivy
/**
* Copyright 2018 Nikita Koksharov
*
* 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 org.redisson.transaction;
import java.util.Date;
import java.util.List;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicBoolean;
import org.redisson.RedissonBucket;
import org.redisson.api.RFuture;
import org.redisson.api.RLock;
import org.redisson.client.codec.Codec;
import org.redisson.command.CommandAsyncExecutor;
import org.redisson.misc.RPromise;
import org.redisson.misc.RedissonPromise;
import org.redisson.transaction.operation.DeleteOperation;
import org.redisson.transaction.operation.TouchOperation;
import org.redisson.transaction.operation.TransactionalOperation;
import org.redisson.transaction.operation.UnlinkOperation;
import org.redisson.transaction.operation.bucket.BucketCompareAndSetOperation;
import org.redisson.transaction.operation.bucket.BucketGetAndDeleteOperation;
import org.redisson.transaction.operation.bucket.BucketGetAndSetOperation;
import org.redisson.transaction.operation.bucket.BucketSetOperation;
import org.redisson.transaction.operation.bucket.BucketTrySetOperation;
import io.netty.buffer.ByteBuf;
import io.netty.util.concurrent.Future;
import io.netty.util.concurrent.FutureListener;
/**
*
* @author Nikita Koksharov
*
* @param value type
*/
public class RedissonTransactionalBucket extends RedissonBucket {
static final Object NULL = new Object();
private long timeout;
private final AtomicBoolean executed;
private final List operations;
private Object state;
private final String transactionId;
public RedissonTransactionalBucket(CommandAsyncExecutor commandExecutor, String name, List operations, AtomicBoolean executed, String transactionId) {
super(commandExecutor, name);
this.operations = operations;
this.executed = executed;
this.transactionId = transactionId;
}
public RedissonTransactionalBucket(Codec codec, CommandAsyncExecutor commandExecutor, String name, List operations, AtomicBoolean executed, String transactionId) {
super(codec, commandExecutor, name);
this.operations = operations;
this.executed = executed;
this.transactionId = transactionId;
}
@Override
public RFuture expireAsync(long timeToLive, TimeUnit timeUnit) {
throw new UnsupportedOperationException("expire method is not supported in transaction");
}
@Override
public RFuture expireAtAsync(Date timestamp) {
throw new UnsupportedOperationException("expireAt method is not supported in transaction");
}
@Override
public RFuture expireAtAsync(long timestamp) {
throw new UnsupportedOperationException("expireAt method is not supported in transaction");
}
@Override
public RFuture clearExpireAsync() {
throw new UnsupportedOperationException("clearExpire method is not supported in transaction");
}
@Override
public RFuture moveAsync(int database) {
throw new UnsupportedOperationException("moveAsync method is not supported in transaction");
}
@Override
public RFuture migrateAsync(String host, int port, int database, long timeout) {
throw new UnsupportedOperationException("migrateAsync method is not supported in transaction");
}
@Override
public RFuture sizeAsync() {
checkState();
if (state != null) {
if (state == NULL) {
return RedissonPromise.newSucceededFuture(0L);
} else {
ByteBuf buf = encode(state);
long size = buf.readableBytes();
buf.release();
return RedissonPromise.newSucceededFuture(size);
}
}
return super.sizeAsync();
}
@Override
public RFuture isExistsAsync() {
checkState();
if (state != null) {
if (state == NULL) {
return RedissonPromise.newSucceededFuture(null);
} else {
return RedissonPromise.newSucceededFuture(true);
}
}
return super.isExistsAsync();
}
@Override
public RFuture touchAsync() {
checkState();
final RPromise result = new RedissonPromise();
executeLocked(result, new Runnable() {
@Override
public void run() {
if (state != null) {
operations.add(new TouchOperation(getName(), getLockName()));
result.trySuccess(state != NULL);
return;
}
isExistsAsync().addListener(new FutureListener() {
@Override
public void operationComplete(Future future) throws Exception {
if (!future.isSuccess()) {
result.tryFailure(future.cause());
return;
}
operations.add(new TouchOperation(getName(), getLockName()));
result.trySuccess(future.getNow());
}
});
}
});
return result;
}
@Override
public RFuture unlinkAsync() {
checkState();
final RPromise result = new RedissonPromise();
executeLocked(result, new Runnable() {
@Override
public void run() {
if (state != null) {
operations.add(new UnlinkOperation(getName(), getLockName()));
if (state == NULL) {
result.trySuccess(false);
} else {
state = NULL;
result.trySuccess(true);
}
return;
}
isExistsAsync().addListener(new FutureListener() {
@Override
public void operationComplete(Future future) throws Exception {
if (!future.isSuccess()) {
result.tryFailure(future.cause());
return;
}
operations.add(new UnlinkOperation(getName(), getLockName()));
state = NULL;
result.trySuccess(future.getNow());
}
});
}
});
return result;
}
@Override
public RFuture deleteAsync() {
checkState();
final RPromise result = new RedissonPromise();
executeLocked(result, new Runnable() {
@Override
public void run() {
if (state != null) {
operations.add(new DeleteOperation(getName(), getLockName()));
if (state == NULL) {
result.trySuccess(false);
} else {
state = NULL;
result.trySuccess(true);
}
return;
}
isExistsAsync().addListener(new FutureListener() {
@Override
public void operationComplete(Future future) throws Exception {
if (!future.isSuccess()) {
result.tryFailure(future.cause());
return;
}
operations.add(new DeleteOperation(getName(), getLockName()));
state = NULL;
result.trySuccess(future.getNow());
}
});
}
});
return result;
}
@Override
@SuppressWarnings("unchecked")
public RFuture getAsync() {
checkState();
if (state != null) {
if (state == NULL) {
return RedissonPromise.newSucceededFuture(null);
} else {
return RedissonPromise.newSucceededFuture((V)state);
}
}
return super.getAsync();
}
@Override
public RFuture compareAndSetAsync(final V expect, final V update) {
checkState();
final RPromise result = new RedissonPromise();
executeLocked(result, new Runnable() {
@Override
public void run() {
if (state != null) {
operations.add(new BucketCompareAndSetOperation(getName(), getLockName(), getCodec(), expect, update));
if ((state == NULL && expect == null)
|| isEquals(state, expect)) {
if (update == null) {
state = NULL;
} else {
state = update;
}
result.trySuccess(true);
} else {
result.trySuccess(false);
}
return;
}
getAsync().addListener(new FutureListener() {
@Override
public void operationComplete(Future future) throws Exception {
if (!future.isSuccess()) {
result.tryFailure(future.cause());
return;
}
operations.add(new BucketCompareAndSetOperation(getName(), getLockName(), getCodec(), expect, update));
if ((future.getNow() == null && expect == null)
|| isEquals(future.getNow(), expect)) {
if (update == null) {
state = NULL;
} else {
state = update;
}
result.trySuccess(true);
} else {
result.trySuccess(false);
}
}
});
}
});
return result;
}
@Override
@SuppressWarnings("unchecked")
public RFuture getAndSetAsync(final V newValue) {
checkState();
final RPromise result = new RedissonPromise();
executeLocked(result, new Runnable() {
@Override
public void run() {
if (state != null) {
Object prevValue;
if (state == NULL) {
prevValue = null;
} else {
prevValue = state;
}
operations.add(new BucketGetAndSetOperation(getName(), getLockName(), getCodec(), newValue));
if (newValue == null) {
state = NULL;
} else {
state = newValue;
}
result.trySuccess((V) prevValue);
return;
}
getAsync().addListener(new FutureListener() {
@Override
public void operationComplete(Future future) throws Exception {
if (!future.isSuccess()) {
result.tryFailure(future.cause());
return;
}
if (newValue == null) {
state = NULL;
} else {
state = newValue;
}
operations.add(new BucketGetAndSetOperation(getName(), getLockName(), getCodec(), newValue));
result.trySuccess(future.getNow());
}
});
}
});
return result;
}
@Override
@SuppressWarnings("unchecked")
public RFuture getAndDeleteAsync() {
checkState();
final RPromise result = new RedissonPromise();
executeLocked(result, new Runnable() {
@Override
public void run() {
if (state != null) {
Object prevValue;
if (state == NULL) {
prevValue = null;
} else {
prevValue = state;
}
operations.add(new BucketGetAndDeleteOperation(getName(), getLockName(), getCodec()));
state = NULL;
result.trySuccess((V) prevValue);
return;
}
getAsync().addListener(new FutureListener() {
@Override
public void operationComplete(Future future) throws Exception {
if (!future.isSuccess()) {
result.tryFailure(future.cause());
return;
}
state = NULL;
operations.add(new BucketGetAndDeleteOperation(getName(), getLockName(), getCodec()));
result.trySuccess(future.getNow());
}
});
}
});
return result;
}
@Override
public RFuture setAsync(V newValue) {
return setAsync(newValue, new BucketSetOperation(getName(), getLockName(), getCodec(), newValue));
}
private RFuture setAsync(final V newValue, final TransactionalOperation operation) {
checkState();
final RPromise result = new RedissonPromise();
executeLocked(result, new Runnable() {
@Override
public void run() {
operations.add(operation);
if (newValue == null) {
state = NULL;
} else {
state = newValue;
}
result.trySuccess(null);
}
});
return result;
}
@Override
public RFuture setAsync(V value, long timeToLive, TimeUnit timeUnit) {
return setAsync(value, new BucketSetOperation(getName(), getLockName(), getCodec(), value, timeToLive, timeUnit));
}
@Override
public RFuture trySetAsync(V newValue) {
return trySet(newValue, new BucketTrySetOperation(getName(), getLockName(), getCodec(), newValue));
}
@Override
public RFuture trySetAsync(V value, long timeToLive, TimeUnit timeUnit) {
return trySet(value, new BucketTrySetOperation(getName(), getLockName(), getCodec(), value, timeToLive, timeUnit));
}
private RFuture trySet(final V newValue, final TransactionalOperation operation) {
checkState();
final RPromise result = new RedissonPromise();
executeLocked(result, new Runnable() {
@Override
public void run() {
if (state != null) {
operations.add(operation);
if (state == NULL) {
if (newValue == null) {
state = NULL;
} else {
state = newValue;
}
result.trySuccess(true);
} else {
result.trySuccess(false);
}
return;
}
getAsync().addListener(new FutureListener() {
@Override
public void operationComplete(Future future) throws Exception {
if (!future.isSuccess()) {
result.tryFailure(future.cause());
return;
}
operations.add(operation);
if (future.getNow() == null) {
if (newValue == null) {
state = NULL;
} else {
state = newValue;
}
result.trySuccess(true);
} else {
result.trySuccess(false);
}
}
});
}
});
return result;
}
private boolean isEquals(Object value, Object oldValue) {
ByteBuf valueBuf = encode(value);
ByteBuf oldValueBuf = encode(oldValue);
try {
return valueBuf.equals(oldValueBuf);
} finally {
valueBuf.readableBytes();
oldValueBuf.readableBytes();
}
}
protected void executeLocked(final RPromise promise, final Runnable runnable) {
RLock lock = getLock();
lock.lockAsync(timeout, TimeUnit.MILLISECONDS).addListener(new FutureListener() {
@Override
public void operationComplete(Future future) throws Exception {
if (future.isSuccess()) {
runnable.run();
} else {
promise.tryFailure(future.cause());
}
}
});
}
private RLock getLock() {
return new RedissonTransactionalLock(commandExecutor, getLockName(), transactionId);
}
private String getLockName() {
return getName() + ":transaction_lock";
}
protected void checkState() {
if (executed.get()) {
throw new IllegalStateException("Unable to execute operation. Transaction is in finished state!");
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy