背景:
这次游戏中台采用lettuce的zset完成游戏内的本服和跨服排行榜,因此写一下案例。
pom.xml
<dependency>
<groupId>io.lettuce</groupId>
<artifactId>lettuce-core</artifactId>
<version>6.2.4.RELEASE</version>
<exclusions>
<exclusion>
<artifactId>netty-common</artifactId>
<groupId>io.netty</groupId>
</exclusion>
<exclusion>
<artifactId>netty-handler</artifactId>
<groupId>io.netty</groupId>
</exclusion>
<exclusion>
<artifactId>netty-transport</artifactId>
<groupId>io.netty</groupId>
</exclusion>
</exclusions>
</dependency>
RedisManager.java
package org.example.testRank.manager;
import io.lettuce.core.RedisClient;
import io.lettuce.core.RedisURI;
import io.lettuce.core.api.StatefulRedisConnection;
import io.lettuce.core.api.async.RedisAsyncCommands;
import io.lettuce.core.api.sync.RedisCommands;
import lombok.Getter;
import lombok.extern.slf4j.Slf4j;
import java.time.Duration;
@Slf4j
public class RedisManager {
private static RedisManager instance = new RedisManager();
private RedisClient redisClient;
private StatefulRedisConnection<String, String> connection;
/*** async */
@Getter
private RedisAsyncCommands<String, String> asyncCommands;
/*** sync*/
@Getter
private RedisCommands<String, String> commands;
public static RedisManager inst() {
return instance;
}
public void init(String host, int port) {
int dbIndex = 15;
int timeout = 10;
try {
RedisURI uri = RedisURI.builder()
.withHost(host)
.withPort(port)
.withDatabase(dbIndex)
.withTimeout(Duration.ofSeconds(timeout)).build();
redisClient = RedisClient.create(uri);
connection = redisClient.connect();
asyncCommands = connection.async();
commands = connection.sync();
} catch (Exception e) {
log.error("redis init error=", e);
}
}
public void close() {
if (connection != null) {
connection.close();
}
if (redisClient != null) {
redisClient.close();
}
}
}
RankManager.java
package org.example.testRank.manager;
import com.google.common.collect.Lists;
import io.lettuce.core.RedisFuture;
import io.lettuce.core.ScoredValue;
import lombok.extern.slf4j.Slf4j;
import org.example.testRank.model.RankInfo;
import org.example.testRank.model.RankItem;
import java.math.BigDecimal;
import java.util.List;
@Slf4j
public class RankManager {
private static RankManager instance = new RankManager();
public static RankManager inst() {
return instance;
}
/**
* 尝试上榜
* @param rankKey 排行榜类型
* @param uid 玩家id
* @param num 得分
* @param increment 是否是增加 false的话直接设置为得分
*/
public void updateRank(String rankKey, long uid, double num, boolean increment) {
RedisFuture<Double> future = RedisManager.inst().getAsyncCommands().zscore(rankKey, uid + "");
future.whenCompleteAsync((v, e) -> {
if (increment && v != null) {
RedisManager.inst().getAsyncCommands().zadd(rankKey, addNumAndGetScoreWithTime(v.doubleValue(), num), String.valueOf(uid));
} else {
RedisManager.inst().getAsyncCommands().zadd(rankKey, getScoreWithTime(num), String.valueOf(uid));
}
});
}
/**
* 获取排行榜列表 + 自己的排名
*/
public RankInfo getRankInfo(String rankKey, int start, int end, long selfUid) {
RankInfo rankInfo = new RankInfo();
List<RankItem> rankItems = Lists.newArrayList();
List<ScoredValue<String>> list = RedisManager.inst().getCommands().zrevrangeWithScores(rankKey, start, end);
int userRank = start;
for (ScoredValue<String> scoredValue : list) {
userRank++;
String uid = scoredValue.getValue();
double score = getRealScore(scoredValue.getScore());
rankItems.add(new RankItem(uid, userRank, (long) score));
}
rankInfo.setRankItems(rankItems);
Long selfRankObj = RedisManager.inst().getCommands().zrevrank(rankKey, selfUid + "");
Double selfScoreObj = RedisManager.inst().getCommands().zscore(rankKey, selfUid + "");
rankInfo.setSelfRankItem(new RankItem(selfUid + "", selfRankObj == null ? 0 : selfRankObj.intValue()+1, selfScoreObj == null ? 0 : selfScoreObj.longValue()));
return rankInfo;
}
private double getScoreWithTime(double score) {
return score + (1 - Double.parseDouble("0." + System.currentTimeMillis()));
}
private double getRealScore(double score) {
BigDecimal bigDecimal = new BigDecimal(score);
String realScore = String.valueOf(bigDecimal).split("\\.")[0];
return Double.parseDouble(realScore);
}
private double addNumAndGetScoreWithTime(double score, double addNum) {
double num = getRealScore(score) + addNum;
return getScoreWithTime(num);
}
}
RankItem.java
package org.example.testRank.model;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.ToString;
@Data
@AllArgsConstructor
@ToString
public class RankItem {
private String uid;
private int rank;
private long score;
}
RankInfo.java
package org.example.testRank.model;
import lombok.Data;
import lombok.ToString;
import java.util.List;
@Data
@ToString
public class RankInfo {
private List<RankItem> rankItems;
private RankItem selfRankItem;
}
Main.java
package org.example.testRank;
import lombok.extern.slf4j.Slf4j;
import org.example.testRank.manager.RankManager;
import org.example.testRank.manager.RedisManager;
import org.example.testRank.model.RankInfo;
@Slf4j
public class Main {
public static String rankKey = "power_rank";
public static void main(String[] args) {
RedisManager.inst().init("localhost", 6379);
// RankManager.inst().updateRank(rankKey, 1002, 10, true);
//
// RankManager.inst().updateRank(rankKey, 1001, 10, true);
// RankManager.inst().updateRank(rankKey, 1003, 100, true);
RankInfo rankInfo = RankManager.inst().getRankInfo(rankKey, 0, -1, 1002);
log.info("{}", rankInfo);
}
}
/*
RankInfo(rankItems=[RankItem(uid=1003, rank=1, score=100), RankItem(uid=1001, rank=2, score=30), RankItem(uid=1002, rank=3, score=10)], selfRankItem=RankItem(uid=1002, rank=3, score=10))
*/
redis中查看下