org.redisson.spring.data.connection.RedissonReactiveStringCommands Maven / Gradle / Ivy
/**
* Copyright (c) 2013-2024 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.spring.data.connection;
import org.reactivestreams.Publisher;
import org.redisson.client.codec.ByteArrayCodec;
import org.redisson.client.codec.StringCodec;
import org.redisson.client.protocol.RedisCommand;
import org.redisson.client.protocol.RedisCommands;
import org.redisson.client.protocol.RedisStrictCommand;
import org.redisson.client.protocol.convertor.BooleanReplayConvertor;
import org.redisson.reactive.CommandReactiveExecutor;
import org.springframework.data.domain.Range;
import org.springframework.data.redis.connection.BitFieldSubCommands;
import org.springframework.data.redis.connection.ReactiveRedisConnection.*;
import org.springframework.data.redis.connection.ReactiveStringCommands;
import org.springframework.data.redis.connection.RedisStringCommands.BitOperation;
import org.springframework.data.redis.connection.RedisStringCommands.SetOption;
import org.springframework.util.Assert;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import java.nio.ByteBuffer;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
/**
*
* @author Nikita Koksharov
*
*/
public class RedissonReactiveStringCommands extends RedissonBaseReactive implements ReactiveStringCommands {
RedissonReactiveStringCommands(CommandReactiveExecutor executorService) {
super(executorService);
}
private static final RedisCommand SET = new RedisCommand("SET", new BooleanReplayConvertor());
@Override
public Flux> set(Publisher commands) {
return execute(commands, command -> {
Assert.notNull(command.getKey(), "Key must not be null!");
Assert.notNull(command.getValue(), "Value must not be null!");
byte[] key = toByteArray(command.getKey());
byte[] value = toByteArray(command.getValue());
Mono m = Mono.empty();
if (!command.getExpiration().isPresent()) {
m = write(key, StringCodec.INSTANCE, SET, key, value);
} else if (command.getExpiration().get().isPersistent()) {
if (!command.getOption().isPresent() || command.getOption().get() == SetOption.UPSERT) {
m = write(key, StringCodec.INSTANCE, SET, key, value);
}
if (command.getOption().get() == SetOption.SET_IF_ABSENT) {
m = write(key, StringCodec.INSTANCE, SET, key, value, "NX");
}
if (command.getOption().get() == SetOption.SET_IF_PRESENT) {
m = write(key, StringCodec.INSTANCE, SET, key, value, "XX");
}
} else {
if (!command.getOption().isPresent() || command.getOption().get() == SetOption.UPSERT) {
m = write(key, StringCodec.INSTANCE, SET, key, value, "PX", command.getExpiration().get().getExpirationTimeInMilliseconds());
}
if (command.getOption().get() == SetOption.SET_IF_ABSENT) {
m = write(key, StringCodec.INSTANCE, SET, key, value, "PX", command.getExpiration().get().getExpirationTimeInMilliseconds(), "NX");
}
if (command.getOption().get() == SetOption.SET_IF_PRESENT) {
m = write(key, StringCodec.INSTANCE, SET, key, value, "PX", command.getExpiration().get().getExpirationTimeInMilliseconds(), "XX");
}
}
return m.map(v -> new BooleanResponse<>(command, v))
.switchIfEmpty(Mono.just(new BooleanResponse<>(command, Boolean.FALSE)));
});
}
@Override
public Flux> get(Publisher keys) {
return execute(keys, command -> {
Assert.notNull(command.getKey(), "Key must not be null!");
byte[] keyBuf = toByteArray(command.getKey());
Mono m = read(keyBuf, ByteArrayCodec.INSTANCE, RedisCommands.GET, keyBuf);
return m.map(v -> new ByteBufferResponse<>(command, ByteBuffer.wrap(v)))
.defaultIfEmpty(new AbsentByteBufferResponse<>(command));
});
}
@Override
public Flux> getSet(Publisher commands) {
return execute(commands, command -> {
Assert.notNull(command.getKey(), "Key must not be null!");
Assert.notNull(command.getValue(), "Value must not be null!");
if (command.getExpiration().isPresent() || command.getOption().isPresent()) {
throw new IllegalArgumentException("Command must not define expiration nor option for GETSET.");
}
byte[] keyBuf = toByteArray(command.getKey());
byte[] valueBuf = toByteArray(command.getValue());
Mono m = write(keyBuf, ByteArrayCodec.INSTANCE, RedisCommands.GETSET, keyBuf, valueBuf);
return m.map(v -> new ByteBufferResponse<>(command, ByteBuffer.wrap(v)));
});
}
@Override
public Flux, ByteBuffer>> mGet(Publisher> keysets) {
return execute(keysets, coll -> {
Assert.notNull(coll, "List must not be null!");
Object[] params = coll.stream().map(buf -> toByteArray(buf)).toArray(Object[]::new);
Mono> m = read(null, ByteArrayCodec.INSTANCE, RedisCommands.MGET, params);
return m.map(v -> {
List values = v.stream().map(array -> {
if (array == null) {
return ByteBuffer.allocate(0);
}
return ByteBuffer.wrap(array);
}).collect(Collectors.toList());
return new MultiValueResponse<>(coll, values);
});
});
}
@Override
public Flux> setNX(Publisher values) {
return execute(values, command -> {
Assert.notNull(command.getKey(), "Key must not be null!");
Assert.notNull(command.getValue(), "Value must not be null!");
if (command.getExpiration().isPresent() || command.getOption().isPresent()) {
throw new IllegalArgumentException("Command must not define expiration nor option for GETSET.");
}
byte[] keyBuf = toByteArray(command.getKey());
byte[] valueBuf = toByteArray(command.getValue());
Mono m = write(keyBuf, StringCodec.INSTANCE, RedisCommands.SETNX, keyBuf, valueBuf);
return m.map(v -> new BooleanResponse<>(command, v));
});
}
private static final RedisCommand SETEX = new RedisCommand("SETEX", new BooleanReplayConvertor());
@Override
public Flux> setEX(Publisher commands) {
return execute(commands, command -> {
Assert.notNull(command.getKey(), "Key must not be null!");
Assert.notNull(command.getValue(), "Value must not be null!");
if (!command.getExpiration().isPresent()) {
throw new IllegalArgumentException("Expiration must not be null!");
}
byte[] keyBuf = toByteArray(command.getKey());
byte[] valueBuf = toByteArray(command.getValue());
Mono m = write(keyBuf, StringCodec.INSTANCE, SETEX,
keyBuf, command.getExpiration().get().getExpirationTimeInSeconds(), valueBuf);
return m.map(v -> new BooleanResponse<>(command, v));
});
}
private static final RedisCommand PSETEX = new RedisCommand("PSETEX");
@Override
public Flux> pSetEX(Publisher commands) {
return execute(commands, command -> {
Assert.notNull(command.getKey(), "Key must not be null!");
Assert.notNull(command.getValue(), "Value must not be null!");
if (!command.getExpiration().isPresent()) {
throw new IllegalArgumentException("Expiration must not be null!");
}
byte[] keyBuf = toByteArray(command.getKey());
byte[] valueBuf = toByteArray(command.getValue());
Mono m = write(keyBuf, StringCodec.INSTANCE, PSETEX,
keyBuf, command.getExpiration().get().getExpirationTimeInMilliseconds(), valueBuf);
return m.map(v -> new BooleanResponse<>(command, true));
});
}
private static final RedisCommand MSET = new RedisCommand("MSET", new BooleanReplayConvertor());
@Override
public Flux> mSet(Publisher commands) {
return execute(commands, command -> {
Assert.notNull(command.getKeyValuePairs(), "KeyValuePairs must not be null!");
List params = convert(command);
Mono m = write(params.get(0), StringCodec.INSTANCE, MSET, params.toArray());
return m.map(v -> new BooleanResponse<>(command, v));
});
}
protected List convert(MSetCommand command) {
List params = new ArrayList(command.getKeyValuePairs().size());
command.getKeyValuePairs().entrySet().forEach(e -> {
byte[] keyBuf = toByteArray(e.getKey());
byte[] valueBuf = toByteArray(e.getValue());
params.add(keyBuf);
params.add(valueBuf);
});
return params;
}
@Override
public Flux> mSetNX(Publisher source) {
return execute(source, command -> {
Assert.notNull(command.getKeyValuePairs(), "KeyValuePairs must not be null!");
List params = convert(command);
Mono m = write(params.get(0), StringCodec.INSTANCE, RedisCommands.MSETNX, params.toArray());
return m.map(v -> new BooleanResponse<>(command, v));
});
}
private static final RedisStrictCommand APPEND = new RedisStrictCommand("APPEND");
@Override
public Flux> append(Publisher commands) {
return execute(commands, command -> {
Assert.notNull(command.getKey(), "Key must not be null!");
Assert.notNull(command.getValue(), "Value must not be null!");
byte[] keyBuf = toByteArray(command.getKey());
byte[] valueBuf = toByteArray(command.getValue());
Mono m = write(keyBuf, StringCodec.INSTANCE, APPEND, keyBuf, valueBuf);
return m.map(v -> new NumericResponse<>(command, v));
});
}
private static final RedisCommand
© 2015 - 2024 Weber Informatics LLC | Privacy Policy