376 lines
15 KiB
Java
376 lines
15 KiB
Java
package com.imaginarycode.minecraft.redisbungee.api;
|
|
|
|
import com.google.common.collect.ImmutableList;
|
|
import com.google.common.collect.ImmutableMultimap;
|
|
import com.google.common.collect.ImmutableSet;
|
|
import com.google.common.collect.Multimap;
|
|
import com.imaginarycode.minecraft.redisbungee.RedisBungeeAPI;
|
|
import com.imaginarycode.minecraft.redisbungee.api.config.ConfigLoader;
|
|
import com.imaginarycode.minecraft.redisbungee.api.config.RedisBungeeConfiguration;
|
|
import com.imaginarycode.minecraft.redisbungee.api.summoners.Summoner;
|
|
import com.imaginarycode.minecraft.redisbungee.api.tasks.RedisTask;
|
|
import com.imaginarycode.minecraft.redisbungee.api.util.RedisUtil;
|
|
import com.imaginarycode.minecraft.redisbungee.api.util.payload.PayloadUtils;
|
|
import com.imaginarycode.minecraft.redisbungee.api.util.uuid.UUIDTranslator;
|
|
import redis.clients.jedis.Jedis;
|
|
import redis.clients.jedis.JedisCluster;
|
|
import redis.clients.jedis.exceptions.JedisConnectionException;
|
|
|
|
import java.net.InetAddress;
|
|
import java.util.*;
|
|
import java.util.concurrent.TimeUnit;
|
|
|
|
import static com.google.common.base.Preconditions.checkArgument;
|
|
|
|
|
|
/**
|
|
* This Class has all internal methods needed by every redis bungee plugin, and it can be used to implement another platforms than bungeecord
|
|
*
|
|
* @author Ham1255
|
|
* @since 0.7.0
|
|
*/
|
|
public interface RedisBungeePlugin<P> extends EventsPlatform, ConfigLoader {
|
|
|
|
default void initialize() {
|
|
|
|
}
|
|
|
|
default void stop() {
|
|
|
|
}
|
|
|
|
Summoner<?> getSummoner();
|
|
|
|
RedisBungeeConfiguration getConfiguration();
|
|
|
|
int getCount();
|
|
|
|
default int getCurrentCount() {
|
|
return new RedisTask<Long>(this) {
|
|
@Override
|
|
public Long jedisTask(Jedis jedis) {
|
|
long total = 0;
|
|
long redisTime = getRedisTime(jedis.time());
|
|
Map<String, String> heartBeats = jedis.hgetAll("heartbeats");
|
|
for (Map.Entry<String, String> stringStringEntry : heartBeats.entrySet()) {
|
|
String k = stringStringEntry.getKey();
|
|
String v = stringStringEntry.getValue();
|
|
|
|
long heartbeatTime = Long.parseLong(v);
|
|
if (heartbeatTime + 30 >= redisTime) {
|
|
total = total + jedis.scard("proxy:" + k + ":usersOnline");
|
|
}
|
|
}
|
|
return total;
|
|
}
|
|
|
|
@Override
|
|
public Long clusterJedisTask(JedisCluster jedisCluster) {
|
|
long total = 0;
|
|
long redisTime = getRedisTime();
|
|
Map<String, String> heartBeats = jedisCluster.hgetAll("heartbeats");
|
|
for (Map.Entry<String, String> stringStringEntry : heartBeats.entrySet()) {
|
|
String k = stringStringEntry.getKey();
|
|
String v = stringStringEntry.getValue();
|
|
|
|
long heartbeatTime = Long.parseLong(v);
|
|
if (heartbeatTime + 30 >= redisTime) {
|
|
total = total + jedisCluster.scard("proxy:" + k + ":usersOnline");
|
|
}
|
|
}
|
|
return total;
|
|
}
|
|
}.execute().intValue();
|
|
}
|
|
|
|
Set<String> getLocalPlayersAsUuidStrings();
|
|
|
|
AbstractDataManager<P, ?, ?, ?> getDataManager();
|
|
|
|
default Set<UUID> getPlayers() {
|
|
return new RedisTask<Set<UUID>>(this) {
|
|
@Override
|
|
public Set<UUID> jedisTask(Jedis jedis) {
|
|
ImmutableSet.Builder<UUID> setBuilder = ImmutableSet.builder();
|
|
try {
|
|
List<String> keys = new ArrayList<>();
|
|
for (String i : getProxiesIds()) {
|
|
keys.add("proxy:" + i + ":usersOnline");
|
|
}
|
|
if (!keys.isEmpty()) {
|
|
Set<String> users = jedis.sunion(keys.toArray(new String[0]));
|
|
if (users != null && !users.isEmpty()) {
|
|
for (String user : users) {
|
|
try {
|
|
setBuilder = setBuilder.add(UUID.fromString(user));
|
|
} catch (IllegalArgumentException ignored) {
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} catch (JedisConnectionException e) {
|
|
// Redis server has disappeared!
|
|
logFatal("Unable to get connection from pool - did your Redis server go away?");
|
|
throw new RuntimeException("Unable to get all players online", e);
|
|
}
|
|
return setBuilder.build();
|
|
}
|
|
|
|
@Override
|
|
public Set<UUID> clusterJedisTask(JedisCluster jedisCluster) {
|
|
ImmutableSet.Builder<UUID> setBuilder = ImmutableSet.builder();
|
|
try {
|
|
List<String> keys = new ArrayList<>();
|
|
for (String i : getProxiesIds()) {
|
|
keys.add("proxy:" + i + ":usersOnline");
|
|
}
|
|
if (!keys.isEmpty()) {
|
|
Set<String> users = jedisCluster.sunion(keys.toArray(new String[0]));
|
|
if (users != null && !users.isEmpty()) {
|
|
for (String user : users) {
|
|
try {
|
|
setBuilder = setBuilder.add(UUID.fromString(user));
|
|
} catch (IllegalArgumentException ignored) {
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} catch (JedisConnectionException e) {
|
|
// Redis server has disappeared!
|
|
logFatal("Unable to get connection from pool - did your Redis server go away?");
|
|
throw new RuntimeException("Unable to get all players online", e);
|
|
}
|
|
return setBuilder.build();
|
|
}
|
|
}.execute();
|
|
}
|
|
|
|
RedisBungeeAPI getApi();
|
|
|
|
UUIDTranslator getUuidTranslator();
|
|
|
|
Multimap<String, UUID> serverToPlayersCache();
|
|
|
|
default Multimap<String, UUID> serversToPlayers() {
|
|
return new RedisTask<Multimap<String, UUID>>(this) {
|
|
@Override
|
|
public Multimap<String, UUID> jedisTask(Jedis jedis) {
|
|
ImmutableMultimap.Builder<String, UUID> builder = ImmutableMultimap.builder();
|
|
for (String serverId : getProxiesIds()) {
|
|
Set<String> players = jedis.smembers("proxy:" + serverId + ":usersOnline");
|
|
for (String player : players) {
|
|
String playerServer = jedis.hget("player:" + player, "server");
|
|
if (playerServer == null) {
|
|
continue;
|
|
}
|
|
builder.put(playerServer, UUID.fromString(player));
|
|
}
|
|
}
|
|
return builder.build();
|
|
}
|
|
|
|
@Override
|
|
public Multimap<String, UUID> clusterJedisTask(JedisCluster jedisCluster) {
|
|
ImmutableMultimap.Builder<String, UUID> builder = ImmutableMultimap.builder();
|
|
for (String serverId : getProxiesIds()) {
|
|
Set<String> players = jedisCluster.smembers("proxy:" + serverId + ":usersOnline");
|
|
for (String player : players) {
|
|
String playerServer = jedisCluster.hget("player:" + player, "server");
|
|
if (playerServer == null) {
|
|
continue;
|
|
}
|
|
builder.put(playerServer, UUID.fromString(player));
|
|
}
|
|
}
|
|
return builder.build();
|
|
}
|
|
}.execute();
|
|
}
|
|
|
|
default Set<UUID> getPlayersOnProxy(String proxyId) {
|
|
checkArgument(getProxiesIds().contains(proxyId), proxyId + " is not a valid proxy ID");
|
|
return new RedisTask<Set<UUID>>(this) {
|
|
@Override
|
|
public Set<UUID> jedisTask(Jedis jedis) {
|
|
Set<String> users = jedis.smembers("proxy:" + proxyId + ":usersOnline");
|
|
ImmutableSet.Builder<UUID> builder = ImmutableSet.builder();
|
|
for (String user : users) {
|
|
builder.add(UUID.fromString(user));
|
|
}
|
|
return builder.build();
|
|
}
|
|
|
|
@Override
|
|
public Set<UUID> clusterJedisTask(JedisCluster jedisCluster) {
|
|
Set<String> users = jedisCluster.smembers("proxy:" + proxyId + ":usersOnline");
|
|
ImmutableSet.Builder<UUID> builder = ImmutableSet.builder();
|
|
for (String user : users) {
|
|
builder.add(UUID.fromString(user));
|
|
}
|
|
return builder.build();
|
|
}
|
|
}.execute();
|
|
}
|
|
|
|
default void sendProxyCommand(String proxyId, String command) {
|
|
checkArgument(getProxiesIds().contains(proxyId) || proxyId.equals("allservers"), "proxyId is invalid");
|
|
sendChannelMessage("redisbungee-" + proxyId, command);
|
|
}
|
|
|
|
List<String> getProxiesIds();
|
|
|
|
default List<String> getCurrentProxiesIds(boolean lagged) {
|
|
return new RedisTask<List<String>>(this) {
|
|
@Override
|
|
public List<String> jedisTask(Jedis jedis) {
|
|
try {
|
|
long time = getRedisTime(jedis.time());
|
|
ImmutableList.Builder<String> servers = ImmutableList.builder();
|
|
Map<String, String> heartbeats = jedis.hgetAll("heartbeats");
|
|
for (Map.Entry<String, String> entry : heartbeats.entrySet()) {
|
|
try {
|
|
long stamp = Long.parseLong(entry.getValue());
|
|
if (lagged ? time >= stamp + RedisUtil.PROXY_TIMEOUT : time <= stamp + RedisUtil.PROXY_TIMEOUT) {
|
|
servers.add(entry.getKey());
|
|
} else if (time > stamp + RedisUtil.PROXY_TIMEOUT) {
|
|
logWarn(entry.getKey() + " is " + (time - stamp) + " seconds behind! (Time not synchronized or server down?) and was removed from heartbeat.");
|
|
jedis.hdel("heartbeats", entry.getKey());
|
|
}
|
|
} catch (NumberFormatException ignored) {
|
|
}
|
|
}
|
|
return servers.build();
|
|
} catch (JedisConnectionException e) {
|
|
logFatal("Unable to fetch server IDs");
|
|
e.printStackTrace();
|
|
return Collections.singletonList(getConfiguration().getProxyId());
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public List<String> clusterJedisTask(JedisCluster jedisCluster) {
|
|
try {
|
|
long time = getRedisTime();
|
|
ImmutableList.Builder<String> servers = ImmutableList.builder();
|
|
Map<String, String> heartbeats = jedisCluster.hgetAll("heartbeats");
|
|
for (Map.Entry<String, String> entry : heartbeats.entrySet()) {
|
|
try {
|
|
long stamp = Long.parseLong(entry.getValue());
|
|
if (lagged ? time >= stamp + RedisUtil.PROXY_TIMEOUT : time <= stamp + RedisUtil.PROXY_TIMEOUT) {
|
|
servers.add(entry.getKey());
|
|
} else if (time > stamp + RedisUtil.PROXY_TIMEOUT) {
|
|
logWarn(entry.getKey() + " is " + (time - stamp) + " seconds behind! (Time not synchronized or server down?) and was removed from heartbeat.");
|
|
jedisCluster.hdel("heartbeats", entry.getKey());
|
|
}
|
|
} catch (NumberFormatException ignored) {
|
|
}
|
|
}
|
|
return servers.build();
|
|
} catch (JedisConnectionException e) {
|
|
logFatal("Unable to fetch server IDs");
|
|
e.printStackTrace();
|
|
return Collections.singletonList(getConfiguration().getProxyId());
|
|
}
|
|
}
|
|
}.execute();
|
|
}
|
|
|
|
PubSubListener getPubSubListener();
|
|
|
|
default void sendChannelMessage(String channel, String message) {
|
|
new RedisTask<Void>(this) {
|
|
@Override
|
|
public Void jedisTask(Jedis jedis) {
|
|
try {
|
|
jedis.publish(channel, message);
|
|
} catch (JedisConnectionException e) {
|
|
// Redis server has disappeared!
|
|
logFatal("Unable to get connection from pool - did your Redis server go away?");
|
|
throw new RuntimeException("Unable to publish channel message", e);
|
|
}
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public Void clusterJedisTask(JedisCluster jedisCluster) {
|
|
try {
|
|
jedisCluster.publish(channel, message);
|
|
} catch (JedisConnectionException e) {
|
|
// Redis server has disappeared!
|
|
logFatal("Unable to get connection from pool - did your Redis server go away?");
|
|
throw new RuntimeException("Unable to publish channel message", e);
|
|
}
|
|
return null;
|
|
}
|
|
}.execute();
|
|
}
|
|
|
|
void executeAsync(Runnable runnable);
|
|
|
|
void executeAsyncAfter(Runnable runnable, TimeUnit timeUnit, int time);
|
|
|
|
void callEvent(Object event);
|
|
|
|
boolean isOnlineMode();
|
|
|
|
void logInfo(String msg);
|
|
|
|
void logWarn(String msg);
|
|
|
|
void logFatal(String msg);
|
|
|
|
P getPlayer(UUID uuid);
|
|
|
|
P getPlayer(String name);
|
|
|
|
UUID getPlayerUUID(String player);
|
|
|
|
String getPlayerName(UUID player);
|
|
|
|
String getPlayerServerName(P player);
|
|
|
|
boolean isPlayerOnAServer(P player);
|
|
|
|
InetAddress getPlayerIp(P player);
|
|
|
|
default void sendProxyCommand(String cmd) {
|
|
sendProxyCommand(getConfiguration().getProxyId(), cmd);
|
|
}
|
|
|
|
default long getRedisTime(List<String> timeRes) {
|
|
return Long.parseLong(timeRes.get(0));
|
|
}
|
|
|
|
default void kickPlayer(UUID playerUniqueId, String message) {
|
|
// first handle on origin proxy if player not found publish the payload
|
|
if (!getDataManager().handleKick(playerUniqueId, message)) {
|
|
new RedisTask<Void>(this) {
|
|
@Override
|
|
public Void jedisTask(Jedis jedis) {
|
|
PayloadUtils.kickPlayerPayload(playerUniqueId, message, jedis);
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public Void clusterJedisTask(JedisCluster jedisCluster) {
|
|
PayloadUtils.kickPlayerPayload(playerUniqueId, message, jedisCluster);
|
|
return null;
|
|
}
|
|
}.execute();
|
|
}
|
|
}
|
|
|
|
default void kickPlayer(String playerName, String message) {
|
|
// fetch the uuid from name
|
|
UUID playerUUID = getUuidTranslator().getTranslatedUuid(playerName, true);
|
|
kickPlayer(playerUUID, message);
|
|
}
|
|
|
|
RedisBungeeMode getRedisBungeeMode();
|
|
|
|
Long getRedisTime();
|
|
|
|
void updateProxyIds();
|
|
|
|
}
|