github-forks
/
redisson
mirror of https://github.com/redisson/redisson.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

refactoring

Browse Source
pull/5099/head
Nikita Koksharov 2 years ago
parent cd4c603de5
commit 4663fcdba8
1 changed files with 68 additions and 44 deletions
Show Stats Download Patch File Download Diff File

80
redisson/src/main/java/org/redisson/RedissonScoredSortedSet.java
Unescape Escape View File

@ -122,14 +122,18 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
}
private <T> RFuture<T> poll(int from, int to, RedisCommand<?> command) {
return commandExecutor.evalWriteAsync(getRawName(), codec, command,
return poll(getRawName(), from, to, command);
}
protected <T> RFuture<T> poll(String name, int from, int to, RedisCommand<?> command) {
return commandExecutor.evalWriteAsync(name, codec, command,
"local v = redis.call('zrange', KEYS[1], ARGV[1], ARGV[2]); "
+ "if #v > 0 then "
+ "redis.call('zremrangebyrank', KEYS[1], ARGV[1], ARGV[2]); "
+ "return v; "
+ "end "
+ "return v;",
Collections.<Object>singletonList(getRawName()), from, to);
Collections.<Object>singletonList(name), from, to);
}
@Override
@ -180,14 +184,18 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
}
private <T> RFuture<T> pollEntry(int from, int to, RedisCommand<?> command) {
return commandExecutor.evalWriteAsync(getRawName(), codec, command,
return pollEntry(getRawName(), from, to, command);
}
protected final <T> RFuture<T> pollEntry(String name, int from, int to, RedisCommand<?> command) {
return commandExecutor.evalWriteAsync(name, codec, command,
"local v = redis.call('zrange', KEYS[1], ARGV[1], ARGV[2], 'withscores'); "
+ "if #v > 0 then "
+ "redis.call('zremrangebyrank', KEYS[1], ARGV[1], ARGV[2]); "
+ "return v; "
+ "end "
+ "return v;",
Collections.singletonList(getRawName()), from, to);
Collections.singletonList(name), from, to);
}
@Override
@ -456,10 +464,11 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
@Override
public RFuture<Integer> addAndGetRankAsync(double score, V object) {
return commandExecutor.evalWriteAsync(getRawName(), LongCodec.INSTANCE, RedisCommands.EVAL_INTEGER,
String name = getRawName(object);
return commandExecutor.evalWriteAsync(name, LongCodec.INSTANCE, RedisCommands.EVAL_INTEGER,
"redis.call('zadd', KEYS[1], ARGV[1], ARGV[2]);" +
"return redis.call('zrank', KEYS[1], ARGV[2]); ",
Collections.<Object>singletonList(getRawName()), new BigDecimal(score).toPlainString(), encode(object));
Collections.<Object>singletonList(name), new BigDecimal(score).toPlainString(), encode(object));
}
@Override
@ -474,10 +483,11 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
@Override
public RFuture<Integer> addAndGetRevRankAsync(double score, V object) {
return commandExecutor.evalWriteAsync(getRawName(), LongCodec.INSTANCE, RedisCommands.EVAL_INTEGER,
String name = getRawName(object);
return commandExecutor.evalWriteAsync(name, LongCodec.INSTANCE, RedisCommands.EVAL_INTEGER,
"redis.call('zadd', KEYS[1], ARGV[1], ARGV[2]);" +
"return redis.call('zrevrank', KEYS[1], ARGV[2]); ",
Collections.<Object>singletonList(getRawName()), new BigDecimal(score).toPlainString(), encode(object));
Collections.<Object>singletonList(name), new BigDecimal(score).toPlainString(), encode(object));
}
@Override
@ -522,7 +532,8 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
@Override
public RFuture<Boolean> addIfExistsAsync(double score, V object) {
return commandExecutor.writeAsync(getRawName(), codec, RedisCommands.ZADD_BOOL, getRawName(), "XX", "CH", BigDecimal.valueOf(score).toPlainString(), encode(object));
String name = getRawName(object);
return commandExecutor.writeAsync(name, codec, RedisCommands.ZADD_BOOL, name, "XX", "CH", BigDecimal.valueOf(score).toPlainString(), encode(object));
}
@Override
@ -537,14 +548,16 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
@Override
public RFuture<Boolean> addIfLessAsync(double score, V object) {
return commandExecutor.writeAsync(getRawName(), codec, RedisCommands.ZADD_BOOL,
getRawName(), "LT", "CH", BigDecimal.valueOf(score).toPlainString(), encode(object));
String name = getRawName(object);
return commandExecutor.writeAsync(name, codec, RedisCommands.ZADD_BOOL,
name, "LT", "CH", BigDecimal.valueOf(score).toPlainString(), encode(object));
}
@Override
public RFuture<Boolean> addIfGreaterAsync(double score, V object) {
return commandExecutor.writeAsync(getRawName(), codec, RedisCommands.ZADD_BOOL,
getRawName(), "GT", "CH", BigDecimal.valueOf(score).toPlainString(), encode(object));
String name = getRawName(object);
return commandExecutor.writeAsync(name, codec, RedisCommands.ZADD_BOOL,
name, "GT", "CH", BigDecimal.valueOf(score).toPlainString(), encode(object));
}
@Override
@ -610,7 +623,8 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
@Override
public RFuture<Boolean> addAsync(double score, V object) {
return commandExecutor.writeAsync(getRawName(), codec, RedisCommands.ZADD_BOOL, getRawName(), BigDecimal.valueOf(score).toPlainString(), encode(object));
String name = getRawName(object);
return commandExecutor.writeAsync(name, codec, RedisCommands.ZADD_BOOL, name, BigDecimal.valueOf(score).toPlainString(), encode(object));
}
@Override
@ -664,7 +678,7 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
if (objects.isEmpty()) {
return new CompletableFutureWrapper<>(0);
}
List<Object> params = new ArrayList<>(objects.size()*2+1);
List<Object> params = new ArrayList<>(objects.size()*2+3);
params.add(getRawName());
params.add("XX");
params.add("CH");
@ -722,7 +736,8 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
@Override
public RFuture<Boolean> tryAddAsync(double score, V object) {
return commandExecutor.writeAsync(getRawName(), codec, RedisCommands.ZADD_BOOL, getRawName(), "NX", BigDecimal.valueOf(score).toPlainString(), encode(object));
String name = getRawName(object);
return commandExecutor.writeAsync(name, codec, RedisCommands.ZADD_BOOL, name, "NX", BigDecimal.valueOf(score).toPlainString(), encode(object));
}
@Override
@ -732,7 +747,8 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
@Override
public RFuture<Boolean> addIfAbsentAsync(double score, V object) {
return commandExecutor.writeAsync(getRawName(), codec, RedisCommands.ZADD_BOOL, getRawName(), "NX", BigDecimal.valueOf(score).toPlainString(), encode(object));
String name = getRawName(object);
return commandExecutor.writeAsync(name, codec, RedisCommands.ZADD_BOOL, name, "NX", BigDecimal.valueOf(score).toPlainString(), encode(object));
}
@Override
@ -786,7 +802,8 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
@Override
public RFuture<Boolean> removeAsync(Object object) {
return commandExecutor.writeAsync(getRawName(), codec, RedisCommands.ZREM, getRawName(), encode(object));
String name = getRawName(object);
return commandExecutor.writeAsync(name, codec, RedisCommands.ZREM, name, encode(object));
}
@Override
@ -829,7 +846,8 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
@Override
public RFuture<Boolean> containsAsync(Object o) {
return commandExecutor.readAsync(getRawName(), StringCodec.INSTANCE, RedisCommands.ZSCORE_CONTAINS, getRawName(), encode(o));
String name = getRawName(o);
return commandExecutor.readAsync(name, StringCodec.INSTANCE, RedisCommands.ZSCORE_CONTAINS, name, encode(o));
}
@Override
@ -844,7 +862,8 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
@Override
public RFuture<Double> getScoreAsync(V o) {
return commandExecutor.readAsync(getRawName(), StringCodec.INSTANCE, RedisCommands.ZSCORE, getRawName(), encode(o));
String name = getRawName(o);
return commandExecutor.readAsync(name, StringCodec.INSTANCE, RedisCommands.ZSCORE, name, encode(o));
}
@Override
@ -865,7 +884,8 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
@Override
public RFuture<Integer> rankAsync(V o) {
return commandExecutor.readAsync(getRawName(), codec, RedisCommands.ZRANK_INT, getRawName(), encode(o));
String name = getRawName(o);
return commandExecutor.readAsync(name, codec, RedisCommands.ZRANK_INT, name, encode(o));
}
@Override
@ -875,7 +895,8 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
@Override
public RFuture<RankedEntry<V>> rankEntryAsync(V o) {
return commandExecutor.readAsync(getRawName(), codec, RedisCommands.ZRANK_ENTRY, getRawName(), encode(o), "WITHSCORE");
String name = getRawName(o);
return commandExecutor.readAsync(name, codec, RedisCommands.ZRANK_ENTRY, name, encode(o), "WITHSCORE");
}
private ScanResult<Object> scanIterator(RedisClient client, long startPos, String pattern, int count) {
@ -1125,8 +1146,9 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
@Override
public RFuture<Double> addScoreAsync(V object, Number value) {
return commandExecutor.writeAsync(getRawName(), DoubleCodec.INSTANCE, RedisCommands.ZINCRBY,
getRawName(), new BigDecimal(value.toString()).toPlainString(), encode(object));
String name = getRawName(object);
return commandExecutor.writeAsync(name, DoubleCodec.INSTANCE, RedisCommands.ZINCRBY,
name, new BigDecimal(value.toString()).toPlainString(), encode(object));
}
@Override
@ -1136,10 +1158,11 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
@Override
public RFuture<Integer> addScoreAndGetRankAsync(V object, Number value) {
return commandExecutor.evalWriteAsync(getRawName(), LongCodec.INSTANCE, RedisCommands.EVAL_INTEGER,
String name = getRawName(object);
return commandExecutor.evalWriteAsync(name, LongCodec.INSTANCE, RedisCommands.EVAL_INTEGER,
"redis.call('zincrby', KEYS[1], ARGV[1], ARGV[2]); "
+"return redis.call('zrank', KEYS[1], ARGV[2]); ",
Collections.singletonList(getRawName()), new BigDecimal(value.toString()).toPlainString(), encode(object));
Collections.singletonList(name), new BigDecimal(value.toString()).toPlainString(), encode(object));
}
@Override
@ -1149,10 +1172,11 @@ public class RedissonScoredSortedSet<V> extends RedissonExpirable implements RSc
@Override
public RFuture<Integer> addScoreAndGetRevRankAsync(V object, Number value) {
return commandExecutor.evalWriteAsync(getRawName(), LongCodec.INSTANCE, RedisCommands.EVAL_INTEGER,
String name = getRawName(object);
return commandExecutor.evalWriteAsync(name, LongCodec.INSTANCE, RedisCommands.EVAL_INTEGER,
"redis.call('zincrby', KEYS[1], ARGV[1], ARGV[2]); "
+"return redis.call('zrevrank', KEYS[1], ARGV[2]); ",
Collections.singletonList(getRawName()), new BigDecimal(value.toString()).toPlainString(), encode(object));
Collections.singletonList(name), new BigDecimal(value.toString()).toPlainString(), encode(object));
}
@Override

Write Preview
Loading…
Cancel
Save