92 lines
2.9 KiB
Java
92 lines
2.9 KiB
Java
package com.imaginarycode.minecraft.redisbungee.api.util.lua;
|
|
|
|
import com.imaginarycode.minecraft.redisbungee.api.RedisBungeePlugin;
|
|
import com.imaginarycode.minecraft.redisbungee.api.tasks.RedisTask;
|
|
import redis.clients.jedis.Jedis;
|
|
import redis.clients.jedis.JedisCluster;
|
|
import redis.clients.jedis.exceptions.JedisDataException;
|
|
|
|
import java.util.List;
|
|
|
|
public class LuaManager {
|
|
private final RedisBungeePlugin<?> plugin;
|
|
|
|
public LuaManager(RedisBungeePlugin<?> plugin) {
|
|
this.plugin = plugin;
|
|
}
|
|
|
|
|
|
|
|
public Script createScript(String script) {
|
|
RedisTask<Script> scriptRedisTask = new RedisTask<Script>(plugin.getApi()) {
|
|
@Override
|
|
public Script jedisTask(Jedis jedis) {
|
|
String hash = jedis.scriptLoad(script);
|
|
return new Script(script, hash);
|
|
}
|
|
|
|
@Override
|
|
public Script clusterJedisTask(JedisCluster jedisCluster) {
|
|
String hash = jedisCluster.scriptLoad(script, "0");
|
|
return new Script(script, hash);
|
|
}
|
|
};
|
|
return scriptRedisTask.execute();
|
|
}
|
|
|
|
public class Script {
|
|
private final String script;
|
|
private final String hashed;
|
|
|
|
public Script(String script, String hashed) {
|
|
this.script = script;
|
|
this.hashed = hashed;
|
|
}
|
|
|
|
public String getScript() {
|
|
return script;
|
|
}
|
|
|
|
public String getHashed() {
|
|
return hashed;
|
|
}
|
|
|
|
public Object eval(List<String> keys, List<String> args) {
|
|
RedisTask<Object> objectRedisTask = new RedisTask<Object>(plugin.getApi()) {
|
|
@Override
|
|
public Object jedisTask(Jedis jedis) {
|
|
Object data;
|
|
try {
|
|
data = jedis.evalsha(hashed, keys, args);
|
|
} catch (JedisDataException e) {
|
|
if (e.getMessage().startsWith("NOSCRIPT")) {
|
|
data = jedis.eval(script, keys, args);
|
|
} else {
|
|
throw e;
|
|
}
|
|
}
|
|
return data;
|
|
}
|
|
|
|
@Override
|
|
public Object clusterJedisTask(JedisCluster jedisCluster) {
|
|
Object data;
|
|
try {
|
|
data = jedisCluster.evalsha(hashed, keys, args);
|
|
} catch (JedisDataException e) {
|
|
if (e.getMessage().startsWith("NOSCRIPT")) {
|
|
data = jedisCluster.eval(script, keys, args);
|
|
} else {
|
|
throw e;
|
|
}
|
|
}
|
|
return data;
|
|
}
|
|
};
|
|
|
|
|
|
return objectRedisTask.execute();
|
|
}
|
|
}
|
|
}
|