67 lines
2.3 KiB
Java
67 lines
2.3 KiB
Java
package com.imaginarycode.minecraft.redisbungee.api.tasks;
|
|
|
|
import com.imaginarycode.minecraft.redisbungee.api.RedisBungeePlugin;
|
|
import redis.clients.jedis.Jedis;
|
|
import redis.clients.jedis.JedisCluster;
|
|
import redis.clients.jedis.exceptions.JedisConnectionException;
|
|
|
|
import java.util.concurrent.TimeUnit;
|
|
import java.util.concurrent.atomic.AtomicInteger;
|
|
|
|
public class HeartbeatTask extends RedisTask<Void>{
|
|
|
|
public final static TimeUnit REPEAT_INTERVAL_TIME_UNIT = TimeUnit.SECONDS;
|
|
public final static int INTERVAL = 1;
|
|
private final AtomicInteger globalPlayerCount;
|
|
|
|
public HeartbeatTask(RedisBungeePlugin<?> plugin, AtomicInteger globalPlayerCount) {
|
|
super(plugin);
|
|
this.globalPlayerCount = globalPlayerCount;
|
|
}
|
|
|
|
@Override
|
|
public Void jedisTask(Jedis jedis) {
|
|
try {
|
|
long redisTime = plugin.getRedisTime(jedis.time());
|
|
jedis.hset("heartbeats", plugin.getConfiguration().getProxyId(), String.valueOf(redisTime));
|
|
} catch (JedisConnectionException e) {
|
|
// Redis server has disappeared!
|
|
plugin.logFatal("Unable to update heartbeat - did your Redis server go away?");
|
|
e.printStackTrace();
|
|
return null;
|
|
}
|
|
try {
|
|
plugin.updateProxyIds();
|
|
globalPlayerCount.set(plugin.getCurrentCount());
|
|
} catch (Throwable e) {
|
|
plugin.logFatal("Unable to update data - did your Redis server go away?");
|
|
e.printStackTrace();
|
|
}
|
|
return null;
|
|
}
|
|
|
|
@Override
|
|
public Void clusterJedisTask(JedisCluster jedisCluster) {
|
|
try {
|
|
long redisTime = plugin.getRedisTime();
|
|
jedisCluster.hset("heartbeats", plugin.getConfiguration().getProxyId(), String.valueOf(redisTime));
|
|
} catch (JedisConnectionException e) {
|
|
// Redis server has disappeared!
|
|
plugin.logFatal("Unable to update heartbeat - did your Redis server go away?");
|
|
e.printStackTrace();
|
|
return null;
|
|
}
|
|
try {
|
|
plugin.updateProxyIds();
|
|
globalPlayerCount.set(plugin.getCurrentCount());
|
|
} catch (Throwable e) {
|
|
plugin.logFatal("Unable to update data - did your Redis server go away?");
|
|
e.printStackTrace();
|
|
}
|
|
return null;
|
|
}
|
|
|
|
|
|
|
|
}
|