修改:工程结构变更为 maven 工程
This commit is contained in:
@@ -0,0 +1 @@
|
||||
net.tccn.ZhubProvider
|
@@ -0,0 +1 @@
|
||||
org.redkalex.cache.redis.RedisCacheSourceProvider
|
80
src/main/java/net/tccn/AbstractConsumer.java
Normal file
80
src/main/java/net/tccn/AbstractConsumer.java
Normal file
@@ -0,0 +1,80 @@
|
||||
package net.tccn;
|
||||
|
||||
import org.redkale.convert.json.JsonConvert;
|
||||
import org.redkale.util.Resourcable;
|
||||
import org.redkale.util.TypeToken;
|
||||
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
import java.util.concurrent.ConcurrentHashMap;
|
||||
import java.util.function.Consumer;
|
||||
|
||||
/**
|
||||
* @author Liang
|
||||
* @data 2020-09-05 23:18
|
||||
*/
|
||||
public abstract class AbstractConsumer extends ZhubAgentProvider implements IConsumer, Resourcable {
|
||||
|
||||
protected JsonConvert convert = JsonConvert.root();
|
||||
|
||||
protected static String APP_NAME = "";
|
||||
|
||||
private Map<String, EventType> eventMap = new ConcurrentHashMap<>();
|
||||
|
||||
protected abstract String getGroupid();
|
||||
|
||||
protected final Set<String> getTopics() {
|
||||
if (!eventMap.isEmpty()) {
|
||||
return eventMap.keySet();
|
||||
}
|
||||
|
||||
return Set.of("-");
|
||||
}
|
||||
|
||||
protected void accept(String topic, String value) {
|
||||
EventType eventType = eventMap.get(topic);
|
||||
|
||||
Object data = null;
|
||||
if ("java.lang.String".equals(eventType.typeToken.getType().getTypeName())) {
|
||||
data = value;
|
||||
} else {
|
||||
data = convert.convertFrom(eventType.typeToken.getType(), value);
|
||||
}
|
||||
|
||||
eventType.accept(data);
|
||||
}
|
||||
|
||||
protected final void removeEventType(String topic) {
|
||||
eventMap.remove(topic);
|
||||
}
|
||||
|
||||
/**
|
||||
* 不同组件的订阅事件 发送
|
||||
*
|
||||
* @param topic
|
||||
*/
|
||||
protected abstract void subscribe(String topic);
|
||||
|
||||
public void subscribe(String topic, Consumer<String> consumer) {
|
||||
subscribe(topic, IType.STRING, consumer);
|
||||
}
|
||||
|
||||
@Override
|
||||
public <T> void subscribe(String topic, TypeToken<T> typeToken, Consumer<T> consumer) {
|
||||
if (topic.contains(",")) {
|
||||
for (String x : topic.split(",")) {
|
||||
subscribe(x, typeToken, consumer);
|
||||
}
|
||||
} else {
|
||||
eventMap.put(topic, EventType.of(topic, typeToken, consumer));
|
||||
subscribe(topic);
|
||||
}
|
||||
}
|
||||
|
||||
// --------------
|
||||
|
||||
@Override
|
||||
public String resourceName() {
|
||||
return super.getName();
|
||||
}
|
||||
}
|
23
src/main/java/net/tccn/Event.java
Normal file
23
src/main/java/net/tccn/Event.java
Normal file
@@ -0,0 +1,23 @@
|
||||
package net.tccn;
|
||||
|
||||
/**
|
||||
* 发布订阅 事件
|
||||
*
|
||||
* @param <V>
|
||||
*/
|
||||
public class Event<V> {
|
||||
public final String topic;
|
||||
//public final String key;
|
||||
public final V value;
|
||||
|
||||
private Event(String topic, V value) {
|
||||
this.topic = topic;
|
||||
this.value = value;
|
||||
}
|
||||
|
||||
public static <V> Event of(String topic, V value) {
|
||||
return new Event<V>(topic, value);
|
||||
}
|
||||
|
||||
|
||||
}
|
32
src/main/java/net/tccn/EventType.java
Normal file
32
src/main/java/net/tccn/EventType.java
Normal file
@@ -0,0 +1,32 @@
|
||||
package net.tccn;
|
||||
|
||||
import org.redkale.util.TypeToken;
|
||||
|
||||
import java.util.function.Consumer;
|
||||
|
||||
public class EventType<T> {
|
||||
public final String topic;
|
||||
public final TypeToken<T> typeToken;
|
||||
private final Consumer<T> consumer;
|
||||
|
||||
private final static TypeToken<String> stringToken = new TypeToken<>() {
|
||||
};
|
||||
|
||||
private EventType(String topic, TypeToken<T> typeToken, Consumer<T> consumer) {
|
||||
this.topic = topic;
|
||||
this.typeToken = typeToken;
|
||||
this.consumer = consumer;
|
||||
}
|
||||
|
||||
public static <T> EventType of(String topic, TypeToken<T> typeToken, Consumer<T> consumer) {
|
||||
return new EventType<>(topic, typeToken, consumer);
|
||||
}
|
||||
|
||||
public static EventType of(String topic, Consumer<String> consumer) {
|
||||
return new EventType(topic, stringToken, consumer);
|
||||
}
|
||||
|
||||
public void accept(T t) {
|
||||
consumer.accept(t);
|
||||
}
|
||||
}
|
33
src/main/java/net/tccn/IConsumer.java
Normal file
33
src/main/java/net/tccn/IConsumer.java
Normal file
@@ -0,0 +1,33 @@
|
||||
package net.tccn;
|
||||
|
||||
import org.redkale.util.TypeToken;
|
||||
|
||||
import java.util.function.Consumer;
|
||||
|
||||
public interface IConsumer {
|
||||
|
||||
/**
|
||||
* 取消订阅
|
||||
*
|
||||
* @param topic
|
||||
*/
|
||||
void unsubscribe(String topic);
|
||||
|
||||
/**
|
||||
* 订阅, 接收数据类型 String
|
||||
*
|
||||
* @param topic
|
||||
* @param consumer
|
||||
*/
|
||||
void subscribe(String topic, Consumer<String> consumer);
|
||||
|
||||
/**
|
||||
* 订阅,接收类型为 <T>
|
||||
*
|
||||
* @param topic
|
||||
* @param typeToken
|
||||
* @param consumer
|
||||
* @param <T>
|
||||
*/
|
||||
<T> void subscribe(String topic, TypeToken<T> typeToken, Consumer<T> consumer);
|
||||
}
|
9
src/main/java/net/tccn/IProducer.java
Normal file
9
src/main/java/net/tccn/IProducer.java
Normal file
@@ -0,0 +1,9 @@
|
||||
package net.tccn;
|
||||
|
||||
import java.util.logging.Logger;
|
||||
|
||||
public interface IProducer {
|
||||
Logger logger = Logger.getLogger(IProducer.class.getSimpleName());
|
||||
|
||||
boolean publish(String topic, Object v);
|
||||
}
|
33
src/main/java/net/tccn/IType.java
Normal file
33
src/main/java/net/tccn/IType.java
Normal file
@@ -0,0 +1,33 @@
|
||||
package net.tccn;
|
||||
|
||||
import org.redkale.util.TypeToken;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
public interface IType {
|
||||
|
||||
TypeToken<String> STRING = new TypeToken<String>() {
|
||||
};
|
||||
|
||||
TypeToken<Integer> INT = new TypeToken<Integer>() {
|
||||
};
|
||||
|
||||
TypeToken<Short> SHORT = new TypeToken<Short>() {
|
||||
};
|
||||
|
||||
TypeToken<Long> LONG = new TypeToken<Long>() {
|
||||
};
|
||||
|
||||
TypeToken<Map<String, String>> MAP = new TypeToken<Map<String, String>>() {
|
||||
};
|
||||
|
||||
TypeToken<List<Map<String, String>>> LMAP = new TypeToken<List<Map<String, String>>>() {
|
||||
};
|
||||
|
||||
TypeToken<List<String>> LSTRING = new TypeToken<List<String>>() {
|
||||
};
|
||||
|
||||
TypeToken<List<Integer>> LINT = new TypeToken<List<Integer>>() {
|
||||
};
|
||||
}
|
403
src/main/java/net/tccn/RedisTest.java
Normal file
403
src/main/java/net/tccn/RedisTest.java
Normal file
@@ -0,0 +1,403 @@
|
||||
package net.tccn;
|
||||
|
||||
import org.redkale.net.AsyncIOGroup;
|
||||
import org.redkale.util.AnyValue;
|
||||
import org.redkale.util.ResourceFactory;
|
||||
import org.redkalex.cache.redis.MyRedisCacheSource;
|
||||
|
||||
import static org.redkale.boot.Application.RESNAME_APP_CLIENT_ASYNCGROUP;
|
||||
import static org.redkale.source.AbstractCacheSource.*;
|
||||
|
||||
public class RedisTest {
|
||||
|
||||
static MyRedisCacheSource source = new MyRedisCacheSource();
|
||||
|
||||
static { // redis://:*Zhong9307!@47.111.150.118:6064?db=2
|
||||
AnyValue.DefaultAnyValue conf = new AnyValue.DefaultAnyValue().addValue(CACHE_SOURCE_MAXCONNS, "1");
|
||||
conf.addValue(CACHE_SOURCE_NODE, new AnyValue.DefaultAnyValue().addValue(CACHE_SOURCE_URL, "redis://:123456@127.0.0.1:6379?db=0"));
|
||||
final ResourceFactory factory = ResourceFactory.create();
|
||||
final AsyncIOGroup asyncGroup = new AsyncIOGroup(8192, 16);
|
||||
asyncGroup.start();
|
||||
factory.register(RESNAME_APP_CLIENT_ASYNCGROUP, asyncGroup);
|
||||
factory.inject(source);
|
||||
//source.defaultConvert = JsonFactory.root().getConvert();
|
||||
source.init(conf);
|
||||
|
||||
System.out.println(source.get("a"));
|
||||
|
||||
//--------------------- bit ------------------------------
|
||||
/*boolean ax = source.getBit("ax", 6);
|
||||
System.out.println("ax:"+ ax);
|
||||
source.setBit("ax", 6, true);
|
||||
|
||||
ax = source.getBit("ax", 6);
|
||||
System.out.println("ax:"+ ax);
|
||||
|
||||
source.setBit("ax", 6, false);
|
||||
ax = source.getBit("ax", 6);
|
||||
System.out.println("ax:"+ ax);*/
|
||||
//--------------------- bit ------------------------------
|
||||
|
||||
//--------------------- bit ------------------------------
|
||||
|
||||
/*
|
||||
source.lock("lockx", 5000);
|
||||
*/
|
||||
|
||||
|
||||
source.keysStartsWith("more-hot").forEach(x -> {
|
||||
System.out.println(x);
|
||||
source.del(x);
|
||||
|
||||
int i = (short) 3;
|
||||
});
|
||||
|
||||
source.keysStartsWith("districtbeans").forEach(x -> {
|
||||
System.out.println(x);
|
||||
source.del(x);
|
||||
});
|
||||
source.keysStartsWith("oss-blind-users").forEach(x -> {
|
||||
System.out.println(x);
|
||||
source.del(x);
|
||||
});
|
||||
source.keysStartsWith("venue:drama-product-schedule-fee:rt-lbeuai84:20221217").forEach(x -> {
|
||||
System.out.println(x);
|
||||
source.del(x);
|
||||
});
|
||||
source.keysStartsWith("venue:site-hour-fee:54f6e9b74cd7416db8d605366c1f49c2:20220729").forEach(x -> {
|
||||
System.out.println(x);
|
||||
source.del(x);
|
||||
});
|
||||
source.keysStartsWith("rainbow").forEach(x -> {
|
||||
System.out.println(x);
|
||||
source.del(x);
|
||||
});
|
||||
source.keysStartsWith("run").forEach(x -> {
|
||||
System.out.println(x);
|
||||
source.del(x);
|
||||
});
|
||||
source.keysStartsWith("today").forEach(x -> {
|
||||
System.out.println(x);
|
||||
source.del(x);
|
||||
});
|
||||
source.keysStartsWith("user").forEach(x -> {
|
||||
System.out.println(x);
|
||||
source.del(x);
|
||||
});
|
||||
source.keysStartsWith("im:").forEach(x -> {
|
||||
System.out.println(x);
|
||||
source.del(x);
|
||||
});
|
||||
source.keysStartsWith("ios-").forEach(x -> {
|
||||
System.out.println(x);
|
||||
source.del(x);
|
||||
});
|
||||
|
||||
|
||||
//--------------------- set ------------------------------
|
||||
/*source.del("setx");
|
||||
*//*
|
||||
int[] ints = {1, 2, 3};
|
||||
source.sadd("setx", ints);
|
||||
*//*
|
||||
|
||||
//source.sadd("setx", list.toArray(Integer[]::new));
|
||||
List<Integer> list = List.of(2, 3, 5);
|
||||
// source.sadd("setx", list.toArray(Integer[]::new));
|
||||
source.sadd("setx", list.toArray(Integer[]::new));
|
||||
source.sadd("setx", 12, 2312, 213);
|
||||
source.sadd("setx", List.of(1011, 10222));
|
||||
|
||||
source.keys("setx*").forEach(x -> {
|
||||
System.out.println(x);
|
||||
});
|
||||
|
||||
source.srem("setx", 213, 2312);
|
||||
|
||||
|
||||
Collection<String> setx1 = source.getCollection("setx", String.class);
|
||||
|
||||
System.out.println(setx1);
|
||||
|
||||
|
||||
//source.getexLong()
|
||||
|
||||
source.setHms("hmx", Map.of("a", "5", "b", "51", "c", "ads"));
|
||||
|
||||
List<Serializable> hmget = source.hmget("hmx", int.class, "a");
|
||||
|
||||
System.out.println(hmget);
|
||||
|
||||
Integer hm = source.getHm("hmx", int.class, "ads");
|
||||
System.out.println(hm);
|
||||
|
||||
Map<String, String> hms = source.getHms("hmx", "a", "b");
|
||||
System.out.println("hmx:" + hms);
|
||||
|
||||
*//*System.out.println("======================================================");
|
||||
System.out.println(source.incrHm("hmx", "a", -6.0));
|
||||
hms = source.getHms("hmx", "a", "b");
|
||||
System.out.println("hmx:a+1后的结果 " + hms);*//*
|
||||
System.out.println("======================================================");
|
||||
source.setHm("hmx", "c", 12);
|
||||
hms = source.getHms("hmx", "a", "b", "c", "d", "a");
|
||||
System.out.println("hmx:设置 c=12 后的结果 " + hms);
|
||||
System.out.println("======================================================");
|
||||
Double c = source.getHm("hmx", double.class, "c");
|
||||
System.out.println("hmx 中 c 值:" + c);*/
|
||||
|
||||
/*Map<String, Object> hmx = source.getHmall("hmx");
|
||||
System.out.println("Hmall:" + hmx);*/
|
||||
|
||||
|
||||
|
||||
|
||||
/*AnyValue.DefaultAnyValue conf = new AnyValue.DefaultAnyValue();
|
||||
conf.addValue("node", new AnyValue.DefaultAnyValue().addValue("addr", "47.111.150.118").addValue("port", "6064").addValue("password", "*Zhong9307!").addValue("db", 2));
|
||||
|
||||
source.defaultConvert = JsonFactory.root().getConvert();
|
||||
source.initValueType(String.class); //value用String类型
|
||||
source.init(conf);*/
|
||||
}
|
||||
|
||||
public static void main(String[] args) {
|
||||
|
||||
|
||||
//source.setLong("a", 125);
|
||||
|
||||
/*long a = source.getLong("a", 0);
|
||||
System.out.println(a);
|
||||
|
||||
List<String> keys = source.keys("farm*");
|
||||
keys.forEach(x -> System.out.println(x));
|
||||
|
||||
try {
|
||||
Thread.sleep(1000);
|
||||
} catch (InterruptedException e) {
|
||||
throw new RuntimeException(e);
|
||||
}*/
|
||||
|
||||
|
||||
// ===========================================
|
||||
//System.out.println(source.remove("a", "b"));
|
||||
|
||||
// bit
|
||||
/*source.initValueType(Integer.class);
|
||||
source.remove("a");
|
||||
boolean a = source.getBit("a", 1);
|
||||
System.out.println(a);
|
||||
|
||||
source.setBit("a", 1, true);
|
||||
a = source.getBit("a", 1);
|
||||
System.out.println("bit-a-1: " + a);
|
||||
|
||||
source.setBit("a", 1, false);
|
||||
a = source.getBit("a", 1);
|
||||
System.out.println("bit-a-1: " + a);*/
|
||||
|
||||
/*source.remove("a");
|
||||
|
||||
// setnx
|
||||
System.out.println(source.setnx("a", 1));
|
||||
source.remove("a");
|
||||
System.out.println(source.setnx("a", 1));
|
||||
|
||||
// set
|
||||
source.remove("abx1");
|
||||
source.appendSetItems("abx1", "a", "b", "c");
|
||||
List<String> list = source.srandomItems("abx1", 2);
|
||||
String str = source.srandomItem("abx1"); //r
|
||||
System.out.println(list);//[r1, r2] */
|
||||
|
||||
/*int[] arr = {0};
|
||||
ExecutorService executor = Executors.newFixedThreadPool(10);
|
||||
CountDownLatch latch = new CountDownLatch(10_0000);
|
||||
for (int i = 0; i < 10_0000; i++) {
|
||||
executor.submit(() -> {
|
||||
try {
|
||||
source.lock("a", 50000);
|
||||
arr[0]++;
|
||||
System.out.println("Thread: " + Thread.currentThread().getName());
|
||||
Thread.sleep(2000);
|
||||
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
} finally {
|
||||
source.unlock("a");
|
||||
latch.countDown();
|
||||
}
|
||||
});
|
||||
}
|
||||
try {
|
||||
latch.await();
|
||||
System.out.println("n=" + arr[0]);
|
||||
executor.shutdown();
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
}*/
|
||||
|
||||
|
||||
/*List<String> list = (List) source.getCollection("gamerank-comment-stat");
|
||||
System.out.println(list);*/
|
||||
|
||||
/*for (int i = 0; i < 10; i++) {
|
||||
String brpop = source.brpop("z", 2);
|
||||
System.out.println(brpop);
|
||||
}*/
|
||||
|
||||
// key 测试
|
||||
/*source.set("a", "123321");
|
||||
System.out.println(source.get("a")); // 123321
|
||||
System.out.println(source.getTtl("a")); // -1
|
||||
System.out.println(source.getPttl("a")); // -1
|
||||
System.out.println(source.getPttl("x")); // -2*/
|
||||
|
||||
// hashmap 测试
|
||||
/*source.remove("sk");
|
||||
source.setHm("sk", "a", "1");
|
||||
source.setHm("sk", "b", "2");
|
||||
System.out.println(source.getHm("sk", "a")); // 1
|
||||
source.remove("sk");
|
||||
|
||||
source.setHms("sk", Map.of("b", "5", "c", "3", "a", "1"));
|
||||
source.hdel("sk", "a");
|
||||
|
||||
Map map = source.getHms("sk", "a", "x", "b", "c", "f"); // {b=5, c=3}
|
||||
System.out.println(map);
|
||||
System.out.println(source.getHmall("sk")); //{b=5, c=3}
|
||||
System.out.println(source.incrHm("sk", "b", 1.1d)); // b = 6.1
|
||||
System.out.println(source.incrHm("sk", "c", 1)); // c = 4
|
||||
System.out.println(source.getHmall("sk")); //{b=6.1, c=4}
|
||||
|
||||
System.out.println("--------------");
|
||||
System.out.println(source.hexists("sk", "b")); // true
|
||||
System.out.println(source.getCollectionSize("sk")); // 2*/
|
||||
|
||||
|
||||
/*Map<String, String> hms = source.getHms("supportusers", "5-kfeu0f", "xxxx", "3-0kbt7u8t", "95q- ");
|
||||
hms.forEach((k, v) -> {
|
||||
System.out.println(k + " : " + v);
|
||||
});*/
|
||||
|
||||
|
||||
/*MyRedisCacheSource<String> source2 = new MyRedisCacheSource();
|
||||
source2.defaultConvert = JsonFactory.root().getConvert();
|
||||
source2.initValueType(String.class); //value用String类型
|
||||
source2.init(conf);*/
|
||||
|
||||
/*Map<String, String> gcMap = source.getHmall("hot-gamecomment");
|
||||
gcMap.forEach((k,v) -> {
|
||||
System.out.println(k + " : " + v);
|
||||
});*/
|
||||
|
||||
|
||||
//Map<String, String> gameinfo = source.getHms("gameinfo", "22759", "22838", "10097", "22751", "22632", "22711", "22195", "15821", "10099", "16313", "11345", "10534", "22768", "22647", "22924", "18461", "15871", "17099", "22640", "22644", "10744", "10264", "18032", "22815", "13584", "10031", "22818", "22452", "22810", "10513", "10557", "15848", "11923", "15920", "22808", "20073", "22809", "15840", "12332", "15803", "10597", "22624", "17113", "19578", "22664", "22621", "20722", "16226", "10523", "12304", "10597","11923","10031");
|
||||
//Map<String, String> gameinfo = source.getHms("gameinfo", "22759","22838","10097","22751","22632","22711","22195","15821","10099","16313","11345","10534","22768","22647","22924","18461","15871","17099","22363","22640","22644","10744","10264","18032","22815","13584","22818","22452","22810","10513","10557","15848","15920","22808","20073","22809","15840","12332","15803","10597","22624","17113","19578","22627","22664","22621","20722","16226","10523","12304");
|
||||
|
||||
/*gameinfo.forEach((k,v ) -> {
|
||||
System.out.println(v);
|
||||
});*/
|
||||
|
||||
|
||||
/*source.queryKeysStartsWith("articlebean:").forEach(x -> {
|
||||
System.out.println(x);
|
||||
//source.remove(x);
|
||||
//System.out.println(source.getHmall(x));
|
||||
});*/
|
||||
|
||||
// list 测试
|
||||
/*MyRedisCacheSource<Integer> sourceInt = new MyRedisCacheSource();
|
||||
sourceInt.defaultConvert = JsonFactory.root().getConvert();
|
||||
sourceInt.initValueType(Integer.class); //value用String类型
|
||||
sourceInt.init(conf);
|
||||
sourceInt.remove("list");
|
||||
Collection<Integer> list = sourceInt.getCollection("list");
|
||||
System.out.println(list);
|
||||
for (int i = 1; i <= 10; i++) {
|
||||
sourceInt.appendListItem("list", i);
|
||||
}
|
||||
System.out.println(sourceInt.getCollection("list")); // [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
|
||||
|
||||
sourceInt.appendListItems("list", 11, 12, 13);
|
||||
System.out.println(sourceInt.getCollection("list")); // [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13]
|
||||
System.out.println(sourceInt.getCollection("list", 0, 5)); // [1, 2, 3, 4, 5]
|
||||
System.out.println(sourceInt.getCollectionSize("list")); // 13
|
||||
|
||||
List<Integer> ids = new ArrayList<>(100);
|
||||
for (int i = 0; i < 5000; i++) {
|
||||
ids.add(i);
|
||||
}
|
||||
sourceInt.remove("abx");
|
||||
sourceInt.appendListItems("abx", ids.toArray(Integer[]::new));
|
||||
System.out.println(sourceInt.getCollection("abx"));*/
|
||||
|
||||
/*System.out.println(sourceInt.getCollectionSize("recommend-user-quality"));
|
||||
Collection<Integer> uid = sourceInt.getCollection("recommend-user-quality");
|
||||
System.out.println(uid);*/
|
||||
|
||||
// zset 测试
|
||||
/*source.initValueType(String.class); //value用Integer类型
|
||||
source.remove("zx");
|
||||
source.zadd("zx", Map.of("a", 1, "b", 2));
|
||||
|
||||
source.zadd("zx", Map.of("a", 1, "c", 5L));
|
||||
source.zadd("zx", Map.of("x", 20, "j", 3.5));
|
||||
source.zadd("zx", Map.of("f", System.currentTimeMillis(), "c", 5L));
|
||||
source.zadd("zx", Map.of("a", 1, "c", 5L));
|
||||
|
||||
System.out.println(source.zincr("zx", "a", 1.34)); // 2.34
|
||||
System.out.println(source.getZsetDoubleScore("zx")); // {f=1592924555704, x=20, c=5, j=3, b=2.34, a=1}
|
||||
source.zrem("zx", "b", "c", "e", "x");
|
||||
System.out.println(source.getZsetLongScore("zx")); // {f=1592924555704, j=3, a=2}
|
||||
|
||||
System.out.println("--------------");
|
||||
System.out.println(source.getZsetLongScore("zx", "f"));
|
||||
|
||||
System.out.println(source.getZrevrank("zx", "f")); // 0
|
||||
System.out.println(source.getZrank("zx", "f")); // 2
|
||||
System.out.println(source.getZrank("zx", "Y")); // -1
|
||||
System.out.println(source.getCollectionSize("zx")); // 3
|
||||
|
||||
System.out.println(source.getZset("zx"));
|
||||
System.out.println(source.zexists("zx", "f", "x", "a"));*/
|
||||
|
||||
/*LocalDate date = LocalDate.of(2019, 12, 31);
|
||||
for (int i = 0; i < 60; i++) {
|
||||
LocalDate localDate = date.plusDays(-i);
|
||||
String day = localDate.format(DateTimeFormatter.ISO_LOCAL_DATE);
|
||||
System.out.println(String.format("mkdir %s; mv *%s*.zip %s", day, day, day));
|
||||
}*/
|
||||
|
||||
|
||||
/*MyRedisCacheSource<UserDetail> source = new MyRedisCacheSource();
|
||||
source.defaultConvert = JsonFactory.root().getConvert();
|
||||
source.initValueType(UserDetail.class);
|
||||
source.init(conf);
|
||||
|
||||
|
||||
Map<String, UserDetail> map = source.getHmall("user-detail");
|
||||
|
||||
Integer[] array = map.values().stream().map(x -> x.getUserid()).toArray(Integer[]::new);
|
||||
|
||||
System.out.println(JsonConvert.root().convertTo(array));
|
||||
|
||||
Map<Integer, UserDetail> hms = source.getHms("user-detail", 11746, 11988, 11504, 11987, 11745, 11503, 11748, 11506, 11747, 11989, 11505, 11508, 11507, 11509, 11980, 11740, 11982, 11981, 11984, 11742, 11500, 11983, 11741, 11502, 11744, 11986, 11985, 11501, 11743, 11999, 11757, 11515, 1, 11514, 11998, 11756, 2, 11517, 11516, 11758, 3, 11519, 4, 5, 11518, 6, 7, 11991, 8, 11990, 9, 11993, 11751, 11750, 11992, 11753, 11511, 11995, 11994, 11510, 11752, 11755, 11513, 11997, 11512, 11996, 11754, 11724, 11966, 11965, 11723, 11968, 11726, 11967, 11725, 11728, 11969, 11727, 11729, 11960, 11720, 11962, 11961, 11722, 11964, 11721);
|
||||
|
||||
System.out.println(hms.size());*/
|
||||
|
||||
/*source.getCollection("article-comment-list", 19, 1).forEach(x -> System.out.println(x));
|
||||
|
||||
|
||||
while (true) {
|
||||
System.out.println("---" + Utility.now() + "---");
|
||||
source.getHmall("ck").forEach((k, v) -> {
|
||||
System.out.println(k + ":" + v);
|
||||
});
|
||||
try {
|
||||
Thread.sleep(60 * 1000);
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}*/
|
||||
}
|
||||
}
|
69
src/main/java/net/tccn/ZhubAgentProvider.java
Normal file
69
src/main/java/net/tccn/ZhubAgentProvider.java
Normal file
@@ -0,0 +1,69 @@
|
||||
package net.tccn;
|
||||
|
||||
import org.redkale.boot.Application;
|
||||
import org.redkale.boot.NodeServer;
|
||||
import org.redkale.cluster.CacheClusterAgent;
|
||||
import org.redkale.cluster.ClusterAgent;
|
||||
import org.redkale.service.Service;
|
||||
import org.redkale.util.ResourceEvent;
|
||||
|
||||
import java.net.InetSocketAddress;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
import java.util.concurrent.CompletableFuture;
|
||||
|
||||
public abstract class ZhubAgentProvider extends ClusterAgent {
|
||||
|
||||
@Override
|
||||
public void onResourceChange(ResourceEvent[] events) {
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
public void register(Application application) {
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
public void deregister(Application application) {
|
||||
|
||||
}
|
||||
|
||||
@Override
|
||||
public CompletableFuture<Map<String, Set<InetSocketAddress>>> queryMqtpAddress(String protocol, String module, String resname) {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public CompletableFuture<Set<InetSocketAddress>> queryHttpAddress(String protocol, String module, String resname) {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public CompletableFuture<Set<InetSocketAddress>> querySncpAddress(String protocol, String restype, String resname) {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected CompletableFuture<Set<InetSocketAddress>> queryAddress(ClusterEntry entry) {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected ClusterEntry register(NodeServer ns, String protocol, Service service) {
|
||||
deregister(ns, protocol, service);
|
||||
ClusterEntry clusterEntry = new ClusterEntry(ns, protocol, service);
|
||||
CacheClusterAgent.AddressEntry entry = new CacheClusterAgent.AddressEntry();
|
||||
entry.addr = clusterEntry.address;
|
||||
entry.resname = clusterEntry.resourceName;
|
||||
entry.nodeid = this.nodeid;
|
||||
entry.time = System.currentTimeMillis();
|
||||
//source.hset(clusterEntry.serviceName, clusterEntry.serviceid, CacheClusterAgent.AddressEntry.class, entry);
|
||||
return clusterEntry;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void deregister(NodeServer ns, String protocol, Service service) {
|
||||
|
||||
}
|
||||
}
|
21
src/main/java/net/tccn/ZhubProvider.java
Normal file
21
src/main/java/net/tccn/ZhubProvider.java
Normal file
@@ -0,0 +1,21 @@
|
||||
package net.tccn;
|
||||
|
||||
import net.tccn.zhub.ZHubClient;
|
||||
import org.redkale.annotation.Priority;
|
||||
import org.redkale.cluster.ClusterAgent;
|
||||
import org.redkale.cluster.ClusterAgentProvider;
|
||||
import org.redkale.util.AnyValue;
|
||||
|
||||
@Priority(1)
|
||||
public class ZhubProvider implements ClusterAgentProvider {
|
||||
|
||||
@Override
|
||||
public boolean acceptsConf(AnyValue config) {
|
||||
return new ZHubClient().acceptsConf(config);
|
||||
}
|
||||
|
||||
@Override
|
||||
public ClusterAgent createInstance() {
|
||||
return new ZHubClient();
|
||||
}
|
||||
}
|
66
src/main/java/net/tccn/timer/TimerExecutor.java
Normal file
66
src/main/java/net/tccn/timer/TimerExecutor.java
Normal file
@@ -0,0 +1,66 @@
|
||||
package net.tccn.timer;
|
||||
|
||||
import net.tccn.timer.queue.TimerQueue;
|
||||
import net.tccn.timer.task.Task;
|
||||
|
||||
import java.util.concurrent.ExecutorService;
|
||||
import java.util.concurrent.Executors;
|
||||
import java.util.logging.Logger;
|
||||
|
||||
/**
|
||||
* @author: liangxianyou
|
||||
*/
|
||||
public class TimerExecutor {
|
||||
private Logger logger = Logger.getLogger(this.getClass().getSimpleName());
|
||||
private TimerQueue queue = new TimerQueue();
|
||||
private ExecutorService executor;
|
||||
|
||||
public TimerExecutor(int n) {
|
||||
executor = Executors.newFixedThreadPool(n);
|
||||
start();
|
||||
}
|
||||
|
||||
public void add(Task... task) {
|
||||
for (Task t : task) {
|
||||
t.setTimerExecutor(this);
|
||||
queue.push(t);
|
||||
logger.finest("add new task : " + t.getName());
|
||||
}
|
||||
}
|
||||
|
||||
protected void add(Task task, boolean upTime) {
|
||||
task.setTimerExecutor(this);
|
||||
if (upTime) task.nextTime();
|
||||
queue.push(task);
|
||||
}
|
||||
|
||||
public Task remove(String name) {
|
||||
return queue.remove(name);
|
||||
}
|
||||
|
||||
public Task get(String name) {
|
||||
return queue.get(name);
|
||||
}
|
||||
|
||||
|
||||
public void start() {
|
||||
new Thread(() -> {
|
||||
while (true) {
|
||||
try {
|
||||
Task take = null;
|
||||
try {
|
||||
take = queue.take();
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
//执行调度
|
||||
executor.execute(take);
|
||||
//add(take, true); //继续添加任务到 队列
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}, "Thread-Redtimer-0").start();
|
||||
}
|
||||
}
|
108
src/main/java/net/tccn/timer/TimerTask.java
Normal file
108
src/main/java/net/tccn/timer/TimerTask.java
Normal file
@@ -0,0 +1,108 @@
|
||||
package net.tccn.timer;
|
||||
|
||||
import net.tccn.timer.scheduled.Scheduled;
|
||||
import net.tccn.timer.task.Job;
|
||||
import net.tccn.timer.task.Task;
|
||||
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.ZoneId;
|
||||
import java.util.Date;
|
||||
import java.util.concurrent.atomic.AtomicInteger;
|
||||
import java.util.logging.Logger;
|
||||
|
||||
/**
|
||||
* Created by liangxianyou at 2018/7/23 14:33.
|
||||
*/
|
||||
public class TimerTask implements Task {
|
||||
private Logger logger = Logger.getLogger(this.getClass().getSimpleName());
|
||||
private long startTime = System.currentTimeMillis();
|
||||
private AtomicInteger execCount = new AtomicInteger();
|
||||
protected String name;
|
||||
private long theTime;
|
||||
private Scheduled scheduled;
|
||||
private boolean isComplete;
|
||||
|
||||
private TimerExecutor timerExecutor;
|
||||
private Job job;
|
||||
|
||||
public static Task by(String name, Scheduled scheduled, Job job) {
|
||||
TimerTask task = new TimerTask();
|
||||
task.name = name;
|
||||
task.scheduled = scheduled;
|
||||
task.job = job;
|
||||
return task;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getName() {
|
||||
return name;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void setScheduled(Scheduled scheduled) {
|
||||
this.scheduled = scheduled;
|
||||
//this.theTime = Date.from(scheduled.theTime().atZone(ZoneId.systemDefault()).toInstant()).getTime();
|
||||
}
|
||||
|
||||
@Override
|
||||
public long nextTime() {
|
||||
LocalDateTime next = scheduled.nextTime();
|
||||
this.theTime = Date.from(next.atZone(ZoneId.systemDefault()).toInstant()).getTime();
|
||||
|
||||
/*SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
|
||||
System.out.println("下次执行:"+ sdf.format(next.toInstant(ZoneOffset.of("+8")).toEpochMilli()));*/
|
||||
return theTime;
|
||||
}
|
||||
|
||||
@Override
|
||||
public long theTime() {
|
||||
LocalDateTime next = scheduled.theTime();
|
||||
this.theTime = Date.from(next.atZone(ZoneId.systemDefault()).toInstant()).getTime();
|
||||
return theTime;
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean isComplete() {
|
||||
return isComplete;
|
||||
}
|
||||
|
||||
public void setComplete(boolean complete) {
|
||||
if (isComplete = complete)
|
||||
timerExecutor.remove(name);
|
||||
}
|
||||
|
||||
public int getExecCount() {
|
||||
return execCount.get();
|
||||
}
|
||||
|
||||
public TimerExecutor getTimerExecutor() {
|
||||
return timerExecutor;
|
||||
}
|
||||
|
||||
public void setTimerExecutor(TimerExecutor timerExecutor) {
|
||||
this.timerExecutor = timerExecutor;
|
||||
}
|
||||
|
||||
public long startTime() {
|
||||
return startTime;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void run() {
|
||||
//没有完成任务,继续执行
|
||||
if (!isComplete) {
|
||||
int count = execCount.incrementAndGet(); // 执行次数+1
|
||||
|
||||
long start = System.currentTimeMillis();
|
||||
job.execute(this);
|
||||
long end = System.currentTimeMillis();
|
||||
logger.finest(String.format("task [%s] : not complete -> %s, time: %s ms, exec count: %s.", getName(), isComplete ? "had complete" : "not complete", end - start, count));
|
||||
|
||||
if (!isComplete) {
|
||||
timerExecutor.add(this, true);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
45
src/main/java/net/tccn/timer/Timers.java
Normal file
45
src/main/java/net/tccn/timer/Timers.java
Normal file
@@ -0,0 +1,45 @@
|
||||
package net.tccn.timer;
|
||||
|
||||
import net.tccn.timer.scheduled.ScheduledCycle;
|
||||
import org.redkale.util.Utility;
|
||||
|
||||
import java.util.function.Supplier;
|
||||
|
||||
public class Timers {
|
||||
|
||||
private static TimerExecutor timerExecutor = new TimerExecutor(1);
|
||||
|
||||
/**
|
||||
* 本地延时重试
|
||||
* @param supplier
|
||||
* @param millis
|
||||
* @param maxCount
|
||||
*/
|
||||
public static void tryDelay(Supplier<Boolean> supplier, long millis, int maxCount) {
|
||||
timerExecutor.add(TimerTask.by("try-delay-task-" + Utility.uuid(), ScheduledCycle.of(0), task -> {
|
||||
if (supplier.get() || task.getExecCount() == maxCount) {
|
||||
task.setComplete(true);
|
||||
}
|
||||
|
||||
if (task.getExecCount() == 1) {
|
||||
task.setScheduled(ScheduledCycle.of(millis));
|
||||
}
|
||||
}));
|
||||
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 本地延时:延时时间极短的场景下使用 (如:1分钟内)
|
||||
* @param runnable
|
||||
* @param millis
|
||||
*/
|
||||
public static void delay(Runnable runnable, long millis) {
|
||||
timerExecutor.add(TimerTask.by("delay-task-" + Utility.uuid(), ScheduledCycle.of(millis), task -> {
|
||||
runnable.run();
|
||||
task.setComplete(true);
|
||||
}));
|
||||
}
|
||||
|
||||
|
||||
}
|
105
src/main/java/net/tccn/timer/queue/TimerQueue.java
Normal file
105
src/main/java/net/tccn/timer/queue/TimerQueue.java
Normal file
@@ -0,0 +1,105 @@
|
||||
package net.tccn.timer.queue;
|
||||
|
||||
import net.tccn.timer.task.Task;
|
||||
|
||||
import java.util.LinkedList;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
import java.util.concurrent.locks.Condition;
|
||||
import java.util.concurrent.locks.ReentrantLock;
|
||||
|
||||
/**
|
||||
* Created by liangxianyou at 2018/7/23 14:07.
|
||||
*/
|
||||
public class TimerQueue {
|
||||
private ReentrantLock lock = new ReentrantLock();
|
||||
private Condition isEmpty = lock.newCondition();
|
||||
private LinkedList<Task> queue = new LinkedList();
|
||||
|
||||
/**
|
||||
* 新加调度任务
|
||||
*
|
||||
* @param task
|
||||
*/
|
||||
public void push(Task task) {
|
||||
try {
|
||||
lock.lock();
|
||||
remove(task.getName());
|
||||
int inx = queue.size();//目标坐标
|
||||
while (inx > 0 && queue.get(inx - 1).theTime() > task.theTime()) {
|
||||
inx--;
|
||||
}
|
||||
|
||||
queue.add(inx, task);
|
||||
isEmpty.signal();
|
||||
} finally {
|
||||
lock.unlock();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 调度等待执行的任务
|
||||
*
|
||||
* @return
|
||||
* @throws InterruptedException
|
||||
*/
|
||||
public Task take() throws InterruptedException {
|
||||
try {
|
||||
lock.lock();
|
||||
while (queue.size() == 0) {
|
||||
isEmpty.await();
|
||||
}
|
||||
|
||||
long currentTime = System.currentTimeMillis();
|
||||
long nextTime = queue.getFirst().theTime();
|
||||
|
||||
if (currentTime >= nextTime) {
|
||||
return queue.removeFirst();
|
||||
} else {
|
||||
isEmpty.await(nextTime - currentTime, TimeUnit.MILLISECONDS);
|
||||
return take();
|
||||
}
|
||||
} finally {
|
||||
lock.unlock();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 删除指定名称的任务
|
||||
*
|
||||
* @param name
|
||||
* @return
|
||||
*/
|
||||
public Task remove(String name) {
|
||||
return get(name, true);
|
||||
}
|
||||
|
||||
/**
|
||||
* 返回指定名称的任务
|
||||
*
|
||||
* @param name
|
||||
* @return
|
||||
*/
|
||||
public Task get(String name) {
|
||||
return get(name, false);
|
||||
}
|
||||
|
||||
private Task get(String name, boolean remove) {
|
||||
try {
|
||||
lock.lock();
|
||||
Task take = null;
|
||||
for (int i = 0; i < queue.size(); i++) {
|
||||
if (name.equals(queue.get(i).getName())) {
|
||||
take = queue.get(i);
|
||||
}
|
||||
}
|
||||
if (remove && take != null) {
|
||||
queue.remove(take);
|
||||
}
|
||||
|
||||
isEmpty.signal();
|
||||
return take;
|
||||
} finally {
|
||||
lock.unlock();
|
||||
}
|
||||
}
|
||||
}
|
23
src/main/java/net/tccn/timer/scheduled/Scheduled.java
Normal file
23
src/main/java/net/tccn/timer/scheduled/Scheduled.java
Normal file
@@ -0,0 +1,23 @@
|
||||
package net.tccn.timer.scheduled;
|
||||
|
||||
import java.time.LocalDateTime;
|
||||
|
||||
/**
|
||||
* @author: liangxianyou at 2018/8/5 17:35.
|
||||
*/
|
||||
public interface Scheduled {
|
||||
|
||||
/**
|
||||
* 下次执行时间
|
||||
*
|
||||
* @return
|
||||
*/
|
||||
LocalDateTime nextTime();
|
||||
|
||||
/**
|
||||
* 当前执行时间
|
||||
*
|
||||
* @return
|
||||
*/
|
||||
LocalDateTime theTime();
|
||||
}
|
79
src/main/java/net/tccn/timer/scheduled/ScheduledCycle.java
Normal file
79
src/main/java/net/tccn/timer/scheduled/ScheduledCycle.java
Normal file
@@ -0,0 +1,79 @@
|
||||
package net.tccn.timer.scheduled;
|
||||
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.temporal.ChronoUnit;
|
||||
import java.time.temporal.TemporalUnit;
|
||||
|
||||
/**
|
||||
* @author: liangxianyou at 2018/8/5 18:05.
|
||||
*/
|
||||
public class ScheduledCycle implements Scheduled {
|
||||
|
||||
private LocalDateTime theTime;
|
||||
private long period;
|
||||
private TemporalUnit unit = ChronoUnit.MILLIS;
|
||||
|
||||
private ScheduledCycle() {
|
||||
}
|
||||
|
||||
public static Scheduled of(String periodCfg) {
|
||||
TemporalUnit unit = ChronoUnit.MILLIS;
|
||||
String endchar = "";
|
||||
long period;
|
||||
|
||||
if (periodCfg.matches("^\\d+[y,M,d,H,m,s]$")) {
|
||||
endchar = periodCfg.substring(periodCfg.length() - 1);
|
||||
period = Long.parseLong(periodCfg.substring(0, periodCfg.length() - 1));
|
||||
} else if (periodCfg.matches("^\\d+$")) {
|
||||
period = Long.parseLong(periodCfg);
|
||||
if (period <= 0) {
|
||||
throw new IllegalArgumentException(String.format("ScheduledCycle period config error: [%s]", periodCfg));
|
||||
}
|
||||
} else {
|
||||
throw new IllegalArgumentException(String.format("ScheduledCycle period config error: [%s]", periodCfg));
|
||||
}
|
||||
|
||||
if ("y".equals(endchar)) unit = ChronoUnit.YEARS;
|
||||
else if ("M".equals(endchar)) unit = ChronoUnit.MONTHS;
|
||||
else if ("d".equals(endchar)) unit = ChronoUnit.DAYS;
|
||||
else if ("H".equals(endchar)) unit = ChronoUnit.HOURS;
|
||||
else if ("m".equals(endchar)) unit = ChronoUnit.MINUTES;
|
||||
else if ("s".equals(endchar)) unit = ChronoUnit.SECONDS;
|
||||
|
||||
return of(period, unit);
|
||||
}
|
||||
|
||||
public static Scheduled of(long period) {
|
||||
return of(period, ChronoUnit.MILLIS);
|
||||
}
|
||||
|
||||
public static Scheduled of(long period, TemporalUnit unit) {
|
||||
LocalDateTime theTime = LocalDateTime.now().plus(period, unit);
|
||||
return of(theTime, period, unit);
|
||||
}
|
||||
|
||||
public static Scheduled of(LocalDateTime startTime, long period) {
|
||||
return of(startTime, period, ChronoUnit.MILLIS);
|
||||
}
|
||||
|
||||
public static Scheduled of(LocalDateTime startTime, long period, TemporalUnit unit) {
|
||||
ScheduledCycle scheduled = new ScheduledCycle();
|
||||
scheduled.theTime = startTime;
|
||||
scheduled.period = period;
|
||||
scheduled.unit = unit;
|
||||
return scheduled;
|
||||
}
|
||||
|
||||
@Override
|
||||
public LocalDateTime nextTime() {
|
||||
if (theTime.isAfter(LocalDateTime.now())) {
|
||||
return theTime;
|
||||
}
|
||||
return theTime = theTime.plus(period, unit);
|
||||
}
|
||||
|
||||
@Override
|
||||
public LocalDateTime theTime() {
|
||||
return theTime;
|
||||
}
|
||||
}
|
499
src/main/java/net/tccn/timer/scheduled/ScheduledExpres.java
Normal file
499
src/main/java/net/tccn/timer/scheduled/ScheduledExpres.java
Normal file
@@ -0,0 +1,499 @@
|
||||
package net.tccn.timer.scheduled;
|
||||
|
||||
import java.time.LocalDate;
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
* 时间解析器
|
||||
*
|
||||
* @author: liangxianyou
|
||||
*/
|
||||
@SuppressWarnings("Duplicates")
|
||||
public class ScheduledExpres implements Scheduled {
|
||||
private int year;
|
||||
private int month;
|
||||
private int[] minutes;
|
||||
private int[] hours;
|
||||
private int[] days;
|
||||
private int[] monthes;
|
||||
private int[] weeks;
|
||||
|
||||
private String cfg;
|
||||
private String[] cfgArr;
|
||||
private LocalDateTime theTime;
|
||||
private int _y, _M, _d, _H, _m;
|
||||
|
||||
@Deprecated
|
||||
private ScheduledExpres(String cfg) {
|
||||
this.cfg = cfg;
|
||||
this.theTime = LocalDateTime.now();
|
||||
initTheTime();
|
||||
}
|
||||
|
||||
@Deprecated
|
||||
private ScheduledExpres(final LocalDateTime startTime, String cfg) {
|
||||
LocalDateTime now = LocalDateTime.now();
|
||||
this.theTime = now.isAfter(startTime) ? now : startTime;
|
||||
this.cfg = cfg;
|
||||
initTheTime();
|
||||
}
|
||||
|
||||
public static Scheduled of(String cfg) {
|
||||
return new ScheduledExpres(cfg);
|
||||
}
|
||||
|
||||
public static Scheduled of(final LocalDateTime startTime, String cfg) {
|
||||
return new ScheduledExpres(startTime, cfg);
|
||||
}
|
||||
|
||||
//寻找初始合法时间
|
||||
public void initTheTime() {
|
||||
year = theTime.getYear();
|
||||
month = theTime.getMonthValue();
|
||||
cfgArr = cfg.split(" ");
|
||||
|
||||
setWeeks();
|
||||
setMonthes();
|
||||
setDays();
|
||||
setHours();
|
||||
setMinutes();
|
||||
|
||||
_y = theTime.getYear();
|
||||
_M = theTime.getMonthValue();
|
||||
_d = theTime.getDayOfMonth();
|
||||
_H = theTime.getHour();
|
||||
_m = theTime.getMinute();
|
||||
|
||||
String cmd = "";//y M d H m
|
||||
if (days.length == 0) cmd = "M";
|
||||
do {
|
||||
carry(cmd);
|
||||
int inx;
|
||||
if ((inx = nowOk(monthes, _M)) < 0) {
|
||||
cmd = "y";
|
||||
continue;
|
||||
}
|
||||
_M = monthes[inx];
|
||||
|
||||
if ((inx = nowOk(days, _d)) < 0) {
|
||||
cmd = "M";
|
||||
continue;
|
||||
}
|
||||
_d = days[inx];
|
||||
|
||||
if ((inx = nowOk(hours, _H)) < 0) {
|
||||
cmd = "d";
|
||||
continue;
|
||||
}
|
||||
_H = hours[inx];
|
||||
|
||||
if ((inx = nowOk(minutes, _m)) < 0) {
|
||||
cmd = "H";
|
||||
continue;
|
||||
}
|
||||
_m = minutes[inx];
|
||||
break;
|
||||
} while (true);
|
||||
|
||||
theTime = LocalDateTime.of(_y, _M, _d, _H, _m);
|
||||
LocalDateTime now = LocalDateTime.now();
|
||||
while (theTime.isBefore(now)) {
|
||||
theTime = carry("m");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 下一次执行的时间
|
||||
*
|
||||
* @return
|
||||
*/
|
||||
@Override
|
||||
public LocalDateTime nextTime() {
|
||||
if (theTime.isAfter(LocalDateTime.now())) {
|
||||
return theTime;
|
||||
}
|
||||
return theTime = carry("m");
|
||||
}
|
||||
|
||||
@Override
|
||||
public LocalDateTime theTime() {
|
||||
return theTime;
|
||||
}
|
||||
|
||||
/**
|
||||
* 通过发送指令进行进位
|
||||
*
|
||||
* @param cmd 进位指令
|
||||
*/
|
||||
private LocalDateTime carry(String cmd) {
|
||||
int inx;
|
||||
while (!"".equals(cmd)) {
|
||||
switch (cmd) {
|
||||
case "y":
|
||||
_y = this.year = ++_y;
|
||||
_M = this.month = monthes[0];
|
||||
setDays();
|
||||
if (days.length == 0) {
|
||||
cmd = "M";
|
||||
continue;
|
||||
}
|
||||
_d = days[0];
|
||||
_H = hours[0];
|
||||
_m = minutes[0];
|
||||
break;
|
||||
case "M":
|
||||
if (_M < monthes[0]) {
|
||||
_M = monthes[0];
|
||||
break;
|
||||
}
|
||||
inx = Arrays.binarySearch(monthes, _M);
|
||||
if (inx < 0 || inx >= monthes.length - 1) {
|
||||
cmd = "y";
|
||||
continue;
|
||||
}
|
||||
_M = this.month = monthes[inx + 1];
|
||||
setDays();
|
||||
if (days.length == 0) {
|
||||
cmd = "M";
|
||||
continue;
|
||||
}
|
||||
_d = days[0];
|
||||
_H = hours[0];
|
||||
_m = minutes[0];
|
||||
break;
|
||||
case "d":
|
||||
if (_d < days[0]) {
|
||||
_d = days[0];
|
||||
break;
|
||||
}
|
||||
inx = Arrays.binarySearch(days, _d);
|
||||
if (inx < 0 || inx >= days.length - 1) {
|
||||
cmd = "M";
|
||||
continue;
|
||||
}
|
||||
_d = days[inx + 1];
|
||||
_H = hours[0];
|
||||
_m = minutes[0];
|
||||
break;
|
||||
case "H":
|
||||
if (_H < hours[0]) {
|
||||
_H = hours[0];
|
||||
break;
|
||||
}
|
||||
inx = Arrays.binarySearch(hours, _H);
|
||||
if (inx < 0 || inx >= hours.length - 1) {
|
||||
cmd = "d";
|
||||
continue;
|
||||
}
|
||||
_H = hours[inx + 1];
|
||||
_m = minutes[0];
|
||||
break;
|
||||
case "m":
|
||||
if (_m < minutes[0]) {
|
||||
_m = minutes[0];
|
||||
break;
|
||||
}
|
||||
inx = Arrays.binarySearch(minutes, _m);
|
||||
if (inx < 0 || inx >= minutes.length - 1) {
|
||||
cmd = "H";
|
||||
continue;
|
||||
}
|
||||
_m = minutes[inx + 1];
|
||||
break;
|
||||
}
|
||||
cmd = "";
|
||||
}
|
||||
return LocalDateTime.of(_y, _M, _d, _H, _m);
|
||||
}
|
||||
|
||||
/**
|
||||
* 得到初始合法时间的索引
|
||||
*
|
||||
* @param arr 合法时间序列
|
||||
* @param n 初始选中值
|
||||
* @return 合法时间的索引
|
||||
*/
|
||||
private int nowOk(int[] arr, int n) {
|
||||
if (arr == null || arr.length == 0) return -1;
|
||||
if (arr[0] > n)
|
||||
return 0;
|
||||
if (arr[arr.length - 1] < n)
|
||||
return 0;
|
||||
if (arr[arr.length - 1] == n)
|
||||
return arr.length - 1;
|
||||
|
||||
for (int i = 0; i < arr.length - 1; i++) {
|
||||
if ((arr[i] < n && arr[i + 1] > n) || arr[i] == n) {
|
||||
return i;
|
||||
}
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
|
||||
/**
|
||||
* 以下为 初始化合法时间 weeks,monthes,days,hour,minutes 序列
|
||||
*/
|
||||
private void setMinutes() {
|
||||
String cfg = cfgArr[0];
|
||||
if ("*".equals(cfg)) {//*
|
||||
minutes = new int[60];
|
||||
for (int i = 0; i < 60; i++) {
|
||||
minutes[i] = i;
|
||||
}
|
||||
} else if (cfg.matches("^[0-5]??[0-9]??$")) {//n
|
||||
minutes = new int[1];
|
||||
minutes[0] = Integer.parseInt(cfg);
|
||||
} else if (cfg.matches("^[*]/[0-9]+$")) {// */5
|
||||
String[] strArr = cfg.split("/");
|
||||
int p = Integer.parseInt(strArr[1]);
|
||||
minutes = new int[60 / p];
|
||||
for (int i = 0; i < minutes.length; i++) {
|
||||
minutes[i] = i * p;
|
||||
}
|
||||
} else if (cfg.matches("^([0-5]??[0-9]??,)+([0-5]??[0-9]??)?$")) {//1,3
|
||||
String[] strings = cfg.split(",");
|
||||
minutes = new int[strings.length];
|
||||
for (int i = 0; i < strings.length; i++) {
|
||||
minutes[i] = Integer.parseInt(strings[i]);
|
||||
}
|
||||
} else if (cfg.matches("^[0-5]??[0-9]??\\-[0-5]??[0-9]??$")) {//1-3
|
||||
String[] split = cfg.split("-");
|
||||
int s = Integer.parseInt(split[0]);
|
||||
int e = Integer.parseInt(split[1]);
|
||||
|
||||
minutes = new int[e - s + 1];
|
||||
for (int i = 0; i < minutes.length; i++) {
|
||||
minutes[i] = s + i;
|
||||
}
|
||||
} else if (cfg.matches("^[0-5]??[0-9]??\\-[0-5]??[0-9]??/[0-5]??[0-9]??$")) {//3-18/5
|
||||
String[] strArr = cfg.split("/");
|
||||
String[] str2Arr = strArr[0].split("-");
|
||||
int s = Integer.parseInt(str2Arr[0]);
|
||||
int e = Integer.parseInt(str2Arr[1]);
|
||||
int p = Integer.parseInt(strArr[1]);
|
||||
|
||||
minutes = new int[(e - s) / p];
|
||||
for (int i = 0; i < minutes.length; i++) {
|
||||
minutes[i] = s + i * p;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void setHours() {
|
||||
String cfg = cfgArr[1];
|
||||
if ("*".equals(cfg)) {//*
|
||||
hours = new int[24];
|
||||
for (int i = 0; i < hours.length; i++) {
|
||||
hours[i] = i;
|
||||
}
|
||||
} else if (cfg.matches("^[0-5]??[0-9]??$")) {//n
|
||||
hours = new int[1];
|
||||
hours[0] = Integer.parseInt(cfg);
|
||||
} else if (cfg.matches("^[*]/[0-9]+$")) {// */5
|
||||
String[] strArr = cfg.split("/");
|
||||
int p = Integer.parseInt(strArr[1]);
|
||||
hours = new int[24 / p];
|
||||
for (int i = 0; i < hours.length; i++) {
|
||||
hours[i] = i * p;
|
||||
}
|
||||
} else if (cfg.matches("^([0-5]??[0-9]??,)+([0-5]??[0-9]??)?$")) {//1,3
|
||||
String[] strArr = cfg.split(",");
|
||||
hours = new int[strArr.length];
|
||||
for (int i = 0; i < strArr.length; i++) {
|
||||
hours[i] = Integer.parseInt(strArr[i]);
|
||||
}
|
||||
} else if (cfg.matches("^[0-5]??[0-9]??\\-[0-5]??[0-9]??$")) {//1-3
|
||||
String[] split = cfg.split("-");
|
||||
int s = Integer.parseInt(split[0]);
|
||||
int e = Integer.parseInt(split[1]);
|
||||
|
||||
hours = new int[e - s + 1];
|
||||
for (int i = 0; i < hours.length; i++) {
|
||||
hours[i] = s + i;
|
||||
}
|
||||
} else if (cfg.matches("^[0-5]??[0-9]??\\-[0-5]??[0-9]??/[0-5]??[0-9]??$")) {//3-18/5
|
||||
String[] strArr = cfg.split("/");
|
||||
String[] str2Arr = strArr[0].split("-");
|
||||
int s = Integer.parseInt(str2Arr[0]);
|
||||
int e = Integer.parseInt(str2Arr[1]);
|
||||
int p = Integer.parseInt(strArr[1]);
|
||||
|
||||
hours = new int[(e - s) / p];
|
||||
for (int i = 0; i < hours.length; i++) {
|
||||
hours[i] = s + i * p;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void setWeeks() {
|
||||
String cfg = cfgArr[4];
|
||||
if ("*".equals(cfg)) {//*
|
||||
weeks = new int[7];
|
||||
for (int i = 0; i < weeks.length; i++) {
|
||||
weeks[i] = i + 1;
|
||||
}
|
||||
} else if (cfg.matches("^[0-5]??[0-9]??$")) {//n
|
||||
weeks = new int[1];
|
||||
weeks[0] = Integer.parseInt(cfg);
|
||||
} else if (cfg.matches("^[*]/[0-9]+$")) {// */5
|
||||
String[] strArr = cfg.split("/");
|
||||
int p = Integer.parseInt(strArr[1]);
|
||||
weeks = new int[7 / p];
|
||||
for (int i = 0; i < weeks.length; i++) {
|
||||
weeks[i] = i * p;
|
||||
}
|
||||
} else if (cfg.matches("^([0-5]??[0-9]??,)+([0-5]??[0-9]??)?$")) {//1,3
|
||||
String[] strArr = cfg.split(",");
|
||||
weeks = new int[strArr.length];
|
||||
for (int i = 0; i < strArr.length; i++) {
|
||||
weeks[i] = Integer.parseInt(strArr[i]);
|
||||
}
|
||||
} else if (cfg.matches("^[0-5]??[0-9]??\\-[0-5]??[0-9]??$")) {//1-3
|
||||
String[] split = cfg.split("-");
|
||||
int s = Integer.parseInt(split[0]);
|
||||
int e = Integer.parseInt(split[1]);
|
||||
|
||||
weeks = new int[e - s + 1];
|
||||
for (int i = 0; i < weeks.length; i++) {
|
||||
weeks[i] = s + i;
|
||||
}
|
||||
} else if (cfg.matches("^[0-5]??[0-9]??\\-[0-5]??[0-9]??/[0-5]??[0-9]??$")) {//3-18/5
|
||||
String[] strArr = cfg.split("/");
|
||||
String[] str2Arr = strArr[0].split("-");
|
||||
int s = Integer.parseInt(str2Arr[0]);
|
||||
int e = Integer.parseInt(str2Arr[1]);
|
||||
int p = Integer.parseInt(strArr[1]);
|
||||
|
||||
weeks = new int[(e - s) / p];
|
||||
for (int i = 0; i < weeks.length; i++) {
|
||||
weeks[i] = s + i * p;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void setMonthes() {
|
||||
String cfg = cfgArr[3];
|
||||
if ("*".equals(cfg)) {//*
|
||||
monthes = new int[12];
|
||||
for (int i = 0; i < monthes.length; i++) {
|
||||
monthes[i] = i + 1;
|
||||
}
|
||||
} else if (cfg.matches("^[0-5]??[0-9]??$")) {//n
|
||||
monthes = new int[1];
|
||||
monthes[0] = Integer.parseInt(cfg);
|
||||
} else if (cfg.matches("^[*]/[0-9]+$")) {// */5
|
||||
String[] strArr = cfg.split("/");
|
||||
int p = Integer.parseInt(strArr[1]);
|
||||
monthes = new int[12 / p];
|
||||
for (int i = 0; i < monthes.length; i++) {
|
||||
monthes[i] = i * p;
|
||||
}
|
||||
} else if (cfg.matches("^([0-5]??[0-9]??,)+([0-5]??[0-9]??)?$")) {//1,3
|
||||
String[] strArr = cfg.split(",");
|
||||
monthes = new int[strArr.length];
|
||||
for (int i = 0; i < strArr.length; i++) {
|
||||
monthes[i] = Integer.parseInt(strArr[i]);
|
||||
}
|
||||
} else if (cfg.matches("^[0-5]??[0-9]??\\-[0-5]??[0-9]??$")) {//1-3
|
||||
String[] split = cfg.split("-");
|
||||
int s = Integer.parseInt(split[0]);
|
||||
int e = Integer.parseInt(split[1]);
|
||||
|
||||
monthes = new int[e - s + 1];
|
||||
for (int i = 0; i < monthes.length; i++) {
|
||||
monthes[i] = s + i;
|
||||
}
|
||||
} else if (cfg.matches("^[0-5]??[0-9]??\\-[0-5]??[0-9]??/[0-5]??[0-9]??$")) {//3-18/5
|
||||
String[] strArr = cfg.split("/");
|
||||
String[] str2Arr = strArr[0].split("-");
|
||||
int s = Integer.parseInt(str2Arr[0]);
|
||||
int e = Integer.parseInt(str2Arr[1]);
|
||||
int p = Integer.parseInt(strArr[1]);
|
||||
|
||||
monthes = new int[(e - s) / p];
|
||||
for (int i = 0; i < monthes.length; i++) {
|
||||
monthes[i] = s + i * p;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void setDays() {
|
||||
String cfg = cfgArr[2];
|
||||
//当前月份总天数,
|
||||
LocalDate firstDay = LocalDate.of(year, month, 1);
|
||||
int lengthOfMonth = firstDay.lengthOfMonth();
|
||||
|
||||
if ("*".equals(cfg)) {//*
|
||||
days = new int[lengthOfMonth];
|
||||
for (int i = 0; i < days.length; i++) {
|
||||
days[i] = i + 1;
|
||||
}
|
||||
} else if (cfg.matches("^[0-5]??[0-9]??$")) {//n
|
||||
days = new int[1];
|
||||
days[0] = Integer.parseInt(cfg);
|
||||
} else if (cfg.matches("^[*]/[0-9]+$")) {// */5
|
||||
String[] strArr = cfg.split("/");
|
||||
int p = Integer.parseInt(strArr[1]);
|
||||
days = new int[lengthOfMonth / p];
|
||||
for (int i = 0; i < days.length; i++) {
|
||||
days[i] = i * p;
|
||||
}
|
||||
} else if (cfg.matches("^([0-5]??[0-9]??,)+([0-5]??[0-9]??)?$")) {//1,3
|
||||
String[] strArr = cfg.split(",");
|
||||
days = new int[strArr.length];
|
||||
for (int i = 0; i < strArr.length; i++) {
|
||||
days[i] = Integer.parseInt(strArr[i]);
|
||||
}
|
||||
} else if (cfg.matches("^[0-5]??[0-9]??\\-[0-5]??[0-9]??$")) {//1-3
|
||||
String[] split = cfg.split("-");
|
||||
int s = Integer.parseInt(split[0]);
|
||||
int e = Integer.parseInt(split[1]);
|
||||
|
||||
days = new int[e - s + 1];
|
||||
for (int i = 0; i < days.length; i++) {
|
||||
days[i] = s + i;
|
||||
}
|
||||
} else if (cfg.matches("^[0-5]??[0-9]??\\-[0-5]??[0-9]??/[0-5]??[0-9]??$")) {//3-18/5
|
||||
String[] strArr = cfg.split("/");
|
||||
String[] str2Arr = strArr[0].split("-");
|
||||
int s = Integer.parseInt(str2Arr[0]);
|
||||
int e = Integer.parseInt(str2Arr[1]);
|
||||
int p = Integer.parseInt(strArr[1]);
|
||||
|
||||
days = new int[(e - s) / p];
|
||||
for (int i = 0; i < days.length; i++) {
|
||||
days[i] = s + i * p;
|
||||
}
|
||||
}
|
||||
|
||||
int firstWeek = firstDay.getDayOfWeek().getValue();
|
||||
List<Integer> allDay = new ArrayList<>();
|
||||
for (int i = 0; i < days.length; i++) {
|
||||
//int week = 7 - Math.abs(i - firstWeek) % 7;//当前星期X
|
||||
int week;
|
||||
int d = days[i];
|
||||
if (d + firstWeek <= 8) {
|
||||
week = firstWeek + d - 1;
|
||||
} else {
|
||||
week = (d - (8 - firstWeek)) % 7;
|
||||
if (week == 0) week = 7;
|
||||
}
|
||||
|
||||
//System.out.printf("M:%s,d:%s,w:%s%n", month, d, week);
|
||||
|
||||
if (Arrays.binarySearch(weeks, week) > -1) {
|
||||
allDay.add(d);//加入日期
|
||||
}
|
||||
}
|
||||
|
||||
days = new int[allDay.size()];
|
||||
for (int i = 0; i < allDay.size(); i++) {
|
||||
days[i] = allDay.get(i);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
14
src/main/java/net/tccn/timer/task/Job.java
Normal file
14
src/main/java/net/tccn/timer/task/Job.java
Normal file
@@ -0,0 +1,14 @@
|
||||
package net.tccn.timer.task;
|
||||
|
||||
/**
|
||||
* @author: liangxianyou at 2018/12/8 17:24.
|
||||
*/
|
||||
@FunctionalInterface
|
||||
public interface Job {
|
||||
|
||||
/**
|
||||
* 任务执行的内容
|
||||
*/
|
||||
void execute(Task task);
|
||||
|
||||
}
|
69
src/main/java/net/tccn/timer/task/Task.java
Normal file
69
src/main/java/net/tccn/timer/task/Task.java
Normal file
@@ -0,0 +1,69 @@
|
||||
package net.tccn.timer.task;
|
||||
|
||||
import net.tccn.timer.TimerExecutor;
|
||||
import net.tccn.timer.scheduled.Scheduled;
|
||||
|
||||
/**
|
||||
* @author: liangxianyou at 2018/8/5 19:32.
|
||||
*/
|
||||
public interface Task extends Runnable {
|
||||
|
||||
/**
|
||||
* 得到任务名称
|
||||
*
|
||||
* @return
|
||||
*/
|
||||
String getName();
|
||||
|
||||
/**
|
||||
* 设置任务执行计划
|
||||
*
|
||||
* @param scheduled
|
||||
*/
|
||||
void setScheduled(Scheduled scheduled);
|
||||
|
||||
/**
|
||||
* 得到下一次执行计划的时间,并设置thenTime
|
||||
*
|
||||
* @return
|
||||
*/
|
||||
long nextTime();
|
||||
|
||||
/**
|
||||
* 任务即将执行的时间点
|
||||
*
|
||||
* @return
|
||||
*/
|
||||
long theTime();
|
||||
|
||||
/**
|
||||
* 是否完成
|
||||
*
|
||||
* @return
|
||||
*/
|
||||
boolean isComplete();
|
||||
|
||||
/**
|
||||
* 完成任务(结束标记)
|
||||
*
|
||||
* @param complete
|
||||
*/
|
||||
void setComplete(boolean complete);
|
||||
|
||||
/**
|
||||
* 开始时间(创建时间)
|
||||
*
|
||||
* @return
|
||||
*/
|
||||
long startTime();
|
||||
|
||||
TimerExecutor getTimerExecutor();
|
||||
|
||||
void setTimerExecutor(TimerExecutor timerExecutor);
|
||||
|
||||
/**
|
||||
* 得到总执行次数
|
||||
* @return
|
||||
*/
|
||||
int getExecCount();
|
||||
}
|
20
src/main/java/net/tccn/zhub/Lock.java
Normal file
20
src/main/java/net/tccn/zhub/Lock.java
Normal file
@@ -0,0 +1,20 @@
|
||||
package net.tccn.zhub;
|
||||
|
||||
// ================================================== lock ==================================================
|
||||
public class Lock {
|
||||
private String name;
|
||||
private String uuid;
|
||||
private int duration;
|
||||
private ZHubClient hubClient;
|
||||
|
||||
protected Lock(String name, String uuid, int duration, ZHubClient hubClient) {
|
||||
this.name = name;
|
||||
this.uuid = uuid;
|
||||
this.duration = duration;
|
||||
this.hubClient = hubClient;
|
||||
}
|
||||
|
||||
public void unLock() {
|
||||
hubClient.send("unlock", name, uuid);
|
||||
}
|
||||
}
|
107
src/main/java/net/tccn/zhub/Rpc.java
Normal file
107
src/main/java/net/tccn/zhub/Rpc.java
Normal file
@@ -0,0 +1,107 @@
|
||||
package net.tccn.zhub;
|
||||
|
||||
import org.redkale.convert.ConvertColumn;
|
||||
import org.redkale.convert.json.JsonConvert;
|
||||
import org.redkale.service.RetResult;
|
||||
|
||||
public class Rpc<T> {
|
||||
private String ruk; // request unique key:
|
||||
private String topic; // call topic
|
||||
private T value; // call paras
|
||||
|
||||
private RpcResult rpcResult;
|
||||
|
||||
public Rpc() {
|
||||
}
|
||||
|
||||
protected Rpc(String appname, String ruk, String topic, Object value) {
|
||||
this.ruk = appname + "::" + ruk;
|
||||
this.topic = topic;
|
||||
this.value = (T) JsonConvert.root().convertTo(value);
|
||||
}
|
||||
|
||||
public String getRuk() {
|
||||
return ruk;
|
||||
}
|
||||
|
||||
public void setRuk(String ruk) {
|
||||
this.ruk = ruk;
|
||||
}
|
||||
|
||||
public String getTopic() {
|
||||
return topic;
|
||||
}
|
||||
|
||||
public void setTopic(String topic) {
|
||||
this.topic = topic;
|
||||
}
|
||||
|
||||
public T getValue() {
|
||||
return value;
|
||||
}
|
||||
|
||||
public void setValue(T value) {
|
||||
this.value = value;
|
||||
}
|
||||
|
||||
@ConvertColumn(ignore = true)
|
||||
public RpcResult getRpcResult() {
|
||||
return rpcResult;
|
||||
}
|
||||
|
||||
public void setRpcResult(RpcResult rpcResult) {
|
||||
this.rpcResult = rpcResult;
|
||||
}
|
||||
|
||||
@ConvertColumn(ignore = true)
|
||||
public String getBackTopic() {
|
||||
return ruk.split("::")[0];
|
||||
}
|
||||
|
||||
public <R> RpcResult<R> render(int retcode, String retinfo) {
|
||||
RpcResult<R> response = new RpcResult<>();
|
||||
response.setRuk(ruk);
|
||||
response.setRetcode(retcode);
|
||||
response.setRetinfo(retinfo);
|
||||
return response;
|
||||
}
|
||||
|
||||
public <R> RpcResult<R> render() {
|
||||
RpcResult<R> response = new RpcResult<>();
|
||||
response.setRuk(ruk);
|
||||
return response;
|
||||
}
|
||||
|
||||
public RpcResult render(RetResult result) {
|
||||
RpcResult resp = new RpcResult<>();
|
||||
resp.setRuk(ruk);
|
||||
resp.setRetcode(result.getRetcode());
|
||||
resp.setRetinfo(result.getRetinfo());
|
||||
resp.setResult(result.getResult());
|
||||
return resp;
|
||||
}
|
||||
|
||||
public <R> RpcResult<R> render(R result) {
|
||||
RpcResult<R> response = new RpcResult<>();
|
||||
response.setRuk(ruk);
|
||||
response.setResult(result);
|
||||
return response;
|
||||
}
|
||||
|
||||
public <R> RpcResult<R> retError(String retinfo) {
|
||||
RpcResult<R> response = new RpcResult<>();
|
||||
response.setRuk(ruk);
|
||||
response.setRetcode(100);
|
||||
response.setRetinfo(retinfo);
|
||||
return response;
|
||||
}
|
||||
|
||||
public <R> RpcResult<R> retError(int retcode, String retinfo) {
|
||||
RpcResult<R> response = new RpcResult<>();
|
||||
response.setRuk(ruk);
|
||||
response.setRetcode(retcode);
|
||||
response.setRetinfo(retinfo);
|
||||
return response;
|
||||
}
|
||||
|
||||
}
|
40
src/main/java/net/tccn/zhub/RpcResult.java
Normal file
40
src/main/java/net/tccn/zhub/RpcResult.java
Normal file
@@ -0,0 +1,40 @@
|
||||
package net.tccn.zhub;
|
||||
|
||||
public class RpcResult<R> {
|
||||
private String ruk;
|
||||
private int retcode;
|
||||
private String retinfo;
|
||||
private R result;
|
||||
|
||||
public String getRuk() {
|
||||
return ruk;
|
||||
}
|
||||
|
||||
public void setRuk(String ruk) {
|
||||
this.ruk = ruk;
|
||||
}
|
||||
|
||||
public int getRetcode() {
|
||||
return retcode;
|
||||
}
|
||||
|
||||
public void setRetcode(int retcode) {
|
||||
this.retcode = retcode;
|
||||
}
|
||||
|
||||
public String getRetinfo() {
|
||||
return retinfo;
|
||||
}
|
||||
|
||||
public void setRetinfo(String retinfo) {
|
||||
this.retinfo = retinfo;
|
||||
}
|
||||
|
||||
public R getResult() {
|
||||
return result;
|
||||
}
|
||||
|
||||
public void setResult(R result) {
|
||||
this.result = result;
|
||||
}
|
||||
}
|
728
src/main/java/net/tccn/zhub/ZHubClient.java
Normal file
728
src/main/java/net/tccn/zhub/ZHubClient.java
Normal file
@@ -0,0 +1,728 @@
|
||||
package net.tccn.zhub;
|
||||
|
||||
import net.tccn.*;
|
||||
import net.tccn.timer.Timers;
|
||||
import org.redkale.annotation.AutoLoad;
|
||||
import org.redkale.annotation.ResourceType;
|
||||
import org.redkale.service.Local;
|
||||
import org.redkale.service.Service;
|
||||
import org.redkale.util.AnyValue;
|
||||
import org.redkale.util.Comment;
|
||||
import org.redkale.util.TypeToken;
|
||||
import org.redkale.util.Utility;
|
||||
|
||||
import java.io.BufferedReader;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStreamReader;
|
||||
import java.io.OutputStream;
|
||||
import java.net.InetSocketAddress;
|
||||
import java.net.Socket;
|
||||
import java.net.SocketException;
|
||||
import java.util.HashMap;
|
||||
import java.util.HashSet;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
import java.util.concurrent.*;
|
||||
import java.util.function.Consumer;
|
||||
import java.util.function.Function;
|
||||
import java.util.logging.Level;
|
||||
import java.util.logging.Logger;
|
||||
|
||||
@Local
|
||||
@AutoLoad(false)
|
||||
@ResourceType(ZHubClient.class)
|
||||
public class ZHubClient extends AbstractConsumer implements IConsumer, IProducer, Service {
|
||||
|
||||
public Logger logger = Logger.getLogger(ZHubClient.class.getSimpleName());
|
||||
private String addr = "127.0.0.1:1216";
|
||||
private String auth = "";
|
||||
private String groupid = "";
|
||||
|
||||
//private ReentrantLock lock = new ReentrantLock();
|
||||
private Socket client;
|
||||
private OutputStream writer;
|
||||
private BufferedReader reader;
|
||||
|
||||
private final LinkedBlockingQueue<Timer> timerQueue = new LinkedBlockingQueue<>();
|
||||
private final LinkedBlockingQueue<Event<String>> topicQueue = new LinkedBlockingQueue<>();
|
||||
private final LinkedBlockingQueue<Event<String>> rpcBackQueue = new LinkedBlockingQueue<>(); // RPC BACK MSG
|
||||
private final LinkedBlockingQueue<Event<String>> rpcCallQueue = new LinkedBlockingQueue<>(); // RPC CALL MSG
|
||||
private final LinkedBlockingQueue<String> sendMsgQueue = new LinkedBlockingQueue<>(); // SEND MSG
|
||||
|
||||
/*private BiConsumer<Runnable, Integer> threadBuilder = (r, n) -> {
|
||||
for (int i = 0; i < n; i++) {
|
||||
new Thread(() -> r.run()).start();
|
||||
}
|
||||
};*/
|
||||
|
||||
private static Map<String, ZHubClient> mainHub = new HashMap<>(); // 127.0.0.1:1216 - ZHubClient
|
||||
|
||||
public ZHubClient() {
|
||||
|
||||
}
|
||||
|
||||
public ZHubClient(String name, Map<String, String> attr) {
|
||||
this.APP_NAME = name;
|
||||
this.addr = attr.get("addr");
|
||||
this.groupid = attr.get("groupid");
|
||||
this.auth = attr.get("auth");
|
||||
|
||||
this.initClient(null);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void init(AnyValue config) {
|
||||
APP_NAME = application.getName();
|
||||
/*if (!preInit()) {
|
||||
return;
|
||||
}*/
|
||||
|
||||
if (config == null) {
|
||||
initClient(null);
|
||||
} else {
|
||||
Map<String, AnyValue> nodes = getNodes(config);
|
||||
for (String rsName : nodes.keySet()) {
|
||||
ZHubClient client = new ZHubClient().initClient(nodes.get(rsName));
|
||||
application.getResourceFactory().register(rsName, client);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private ZHubClient initClient(AnyValue config) {
|
||||
// 自动注入
|
||||
if (config != null) {
|
||||
addr = config.getValue("addr", addr);
|
||||
|
||||
// 合并 addr = host:port, 做历史兼容
|
||||
int port = config.getIntValue("port", 0);
|
||||
if (port != 0 && !addr.contains(":")) {
|
||||
addr = addr + ":" + port;
|
||||
}
|
||||
|
||||
auth = config.getOrDefault("auth", auth);
|
||||
groupid = config.getValue("groupid", groupid);
|
||||
}
|
||||
|
||||
// 设置第一个启动的 实例为主实例
|
||||
if (!mainHub.containsKey(addr)) { // 确保同步执行此 init 逻辑
|
||||
mainHub.put(addr, this);
|
||||
}
|
||||
|
||||
CompletableFuture.runAsync(() -> {
|
||||
// 消息 事件接收
|
||||
new Thread(() -> {
|
||||
if (!initSocket(0)) {
|
||||
return;
|
||||
}
|
||||
|
||||
while (true) {
|
||||
try {
|
||||
String readLine = reader.readLine();
|
||||
if (readLine == null && initSocket(Integer.MAX_VALUE)) { // 连接中断 处理
|
||||
continue;
|
||||
}
|
||||
|
||||
String type = "";
|
||||
|
||||
// +ping
|
||||
if ("+ping".equals(readLine)) {
|
||||
send("+pong");
|
||||
continue;
|
||||
}
|
||||
|
||||
// 主题订阅消息
|
||||
if ("*3".equals(readLine)) {
|
||||
readLine = reader.readLine(); // $7 len()
|
||||
type = reader.readLine(); // message
|
||||
if (!"message".equals(type)) {
|
||||
continue;
|
||||
}
|
||||
reader.readLine(); //$n len(key)
|
||||
String topic = reader.readLine(); // topic
|
||||
|
||||
String lenStr = reader.readLine();//$n len(value)
|
||||
int clen = 0;
|
||||
if (lenStr.startsWith("$")) {
|
||||
clen = Integer.parseInt(lenStr.replace("$", ""));
|
||||
}
|
||||
|
||||
String value = "";
|
||||
do {
|
||||
if (value.length() > 0) {
|
||||
value += "\r\n";
|
||||
}
|
||||
String s = reader.readLine();
|
||||
value += s; // value
|
||||
} while (clen > 0 && clen > strLength(value));
|
||||
|
||||
logger.finest("topic[" + topic + "]: " + value);
|
||||
|
||||
// lock msg
|
||||
if ("lock".equals(topic)) {
|
||||
Lock lock = lockTag.get(value);
|
||||
if (lock != null) {
|
||||
synchronized (lock) {
|
||||
lock.notifyAll();
|
||||
}
|
||||
}
|
||||
continue;
|
||||
}
|
||||
// rpc back msg
|
||||
if (APP_NAME.equals(topic)) {
|
||||
rpcBackQueue.add(Event.of(topic, value));
|
||||
continue;
|
||||
}
|
||||
|
||||
// rpc call msg
|
||||
if (rpcTopics.contains(topic)) {
|
||||
rpcCallQueue.add(Event.of(topic, value));
|
||||
continue;
|
||||
}
|
||||
|
||||
// oth msg
|
||||
topicQueue.add(Event.of(topic, value));
|
||||
continue;
|
||||
}
|
||||
|
||||
// timer 消息
|
||||
if ("*2".equals(readLine)) {
|
||||
readLine = reader.readLine(); // $7 len()
|
||||
type = reader.readLine(); // message
|
||||
if (!"timer".equals(type)) {
|
||||
continue;
|
||||
}
|
||||
reader.readLine(); //$n len(key)
|
||||
String topic = reader.readLine(); // name
|
||||
|
||||
logger.finest("timer[" + topic + "]: ");
|
||||
timerQueue.add(timerMap.get(topic));
|
||||
continue;
|
||||
}
|
||||
|
||||
logger.finest(readLine);
|
||||
} catch (IOException e) {
|
||||
if (e instanceof SocketException) {
|
||||
initSocket(Integer.MAX_VALUE);
|
||||
}
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}).start();
|
||||
}).thenAcceptAsync(x -> {
|
||||
// 定时调度事件,已加入耗时监控
|
||||
new Thread(() -> {
|
||||
ExecutorService pool = Executors.newFixedThreadPool(1);
|
||||
while (true) {
|
||||
Timer timer = null;
|
||||
try {
|
||||
if ((timer = timerQueue.take()) == null) {
|
||||
return;
|
||||
}
|
||||
long start = System.currentTimeMillis();
|
||||
pool.submit(timer.runnable).get(5, TimeUnit.SECONDS);
|
||||
long end = System.currentTimeMillis();
|
||||
logger.finest(String.format("timer [%s] : elapsed time %s ms", timer.name, end - start));
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
} catch (TimeoutException e) {
|
||||
logger.log(Level.SEVERE, "timer [" + timer.name + "] time out: " + 5 + " S", e);
|
||||
pool = Executors.newFixedThreadPool(1);
|
||||
} catch (Exception e) {
|
||||
logger.log(Level.WARNING, "timer [" + timer.name + "]", e);
|
||||
}
|
||||
}
|
||||
}).start();
|
||||
|
||||
// topic msg,已加入耗时监控
|
||||
new Thread(() -> {
|
||||
ExecutorService pool = Executors.newFixedThreadPool(1);
|
||||
while (true) {
|
||||
Event<String> event = null;
|
||||
try {
|
||||
if ((event = topicQueue.take()) == null) {
|
||||
continue;
|
||||
}
|
||||
|
||||
String topic = event.topic;
|
||||
String value = event.value;
|
||||
pool.submit(() -> accept(topic, value)).get(5, TimeUnit.SECONDS);
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
} catch (TimeoutException e) {
|
||||
logger.log(Level.SEVERE, "topic[" + event.topic + "] event deal time out: " + 5 + " S, value: " + event.value, e);
|
||||
pool = Executors.newFixedThreadPool(1);
|
||||
} catch (Exception e) {
|
||||
logger.log(Level.WARNING, "topic[" + event.topic + "] event accept error :" + event.value, e);
|
||||
}
|
||||
}
|
||||
}).start();
|
||||
|
||||
// rpc back ,仅做数据解析,暂无耗时监控
|
||||
new Thread(() -> {
|
||||
while (true) {
|
||||
Event<String> event = null;
|
||||
try {
|
||||
if ((event = rpcBackQueue.take()) == null) {
|
||||
continue;
|
||||
}
|
||||
//if (event)
|
||||
logger.finest(String.format("rpc-back:[%s]: %s", event.topic, event.value));
|
||||
rpcAccept(event.value);
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
} catch (Exception e) {
|
||||
logger.log(Level.WARNING, "rpc-back[" + event.topic + "] event accept error :" + event.value, e);
|
||||
}
|
||||
}
|
||||
}).start();
|
||||
|
||||
// rpc call,已加入耗时监控
|
||||
new Thread(() -> {
|
||||
ExecutorService pool = Executors.newFixedThreadPool(1);
|
||||
while (true) {
|
||||
Event<String> event = null;
|
||||
try {
|
||||
if ((event = rpcCallQueue.take()) == null) {
|
||||
continue;
|
||||
}
|
||||
logger.finest(String.format("rpc-call:[%s] %s", event.topic, event.value));
|
||||
String topic = event.topic;
|
||||
String value = event.value;
|
||||
pool.submit(() -> accept(topic, value)).get(5, TimeUnit.SECONDS);
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
} catch (TimeoutException e) {
|
||||
logger.log(Level.SEVERE, "topic[" + event.topic + "] event deal time out: " + 5 + " S, value: " + event.value, e);
|
||||
pool = Executors.newFixedThreadPool(1);
|
||||
} catch (Exception e) {
|
||||
logger.log(Level.WARNING, "rpc-call[" + event.topic + "] event accept error :" + event.value, e);
|
||||
}
|
||||
}
|
||||
}).start();
|
||||
|
||||
// send msg
|
||||
new Thread(() -> {
|
||||
while (true) {
|
||||
String msg = null;
|
||||
try {
|
||||
if ((msg = sendMsgQueue.take()) == null) {
|
||||
continue;
|
||||
}
|
||||
// logger.log(Level.FINEST, "send-msg: [" + msg + "]");
|
||||
writer.write(msg.getBytes());
|
||||
writer.flush();
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
} catch (Exception e) {
|
||||
logger.log(Level.WARNING, "send-msg[" + msg + "] event accept error :", e);
|
||||
}
|
||||
}
|
||||
}).start();
|
||||
});
|
||||
|
||||
return this;
|
||||
}
|
||||
|
||||
public boolean acceptsConf(AnyValue config) {
|
||||
if (config == null) {
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!getNodes(config).isEmpty()) {
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
private HashMap<String, AnyValue> getNodes(AnyValue config) {
|
||||
AnyValue[] zhubs = config.getAnyValues("zhub");
|
||||
HashMap<String, AnyValue> confMap = new HashMap<>();
|
||||
|
||||
for (AnyValue zhub : zhubs) {
|
||||
String[] names = zhub.getNames();
|
||||
for (String name : names) {
|
||||
confMap.put(name, zhub.getAnyValue(name));
|
||||
}
|
||||
}
|
||||
|
||||
return confMap;
|
||||
}
|
||||
|
||||
// ---------------------
|
||||
// -- 消息发送 --
|
||||
protected boolean send(String... data) {
|
||||
if (data.length == 1) {
|
||||
sendMsgQueue.add(data[0] + "\r\n");
|
||||
} else if (data.length > 1) {
|
||||
StringBuffer buf = new StringBuffer();
|
||||
buf.append("*" + data.length + "\r\n");
|
||||
for (String d : data) {
|
||||
buf.append("$" + strLength(d) + "\r\n");
|
||||
buf.append(d + "\r\n");
|
||||
}
|
||||
sendMsgQueue.add(buf.toString());
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/*protected boolean send(String... data) {
|
||||
try {
|
||||
lock.lock();
|
||||
if (data.length == 1) {
|
||||
writer.write((data[0] + "\r\n").getBytes());
|
||||
} else if (data.length > 1) {
|
||||
writer.write(("*" + data.length + "\r\n").getBytes());
|
||||
for (String d : data) {
|
||||
writer.write(("$" + d.length() + "\r\n").getBytes());
|
||||
writer.write((d + "\r\n").getBytes());
|
||||
}
|
||||
}
|
||||
writer.flush();
|
||||
return true;
|
||||
} catch (IOException e) {
|
||||
logger.log(Level.WARNING, "", e);
|
||||
} finally {
|
||||
lock.unlock();
|
||||
}
|
||||
return false;
|
||||
}*/
|
||||
|
||||
private int strLength(String str) {
|
||||
str = str.replaceAll("[^\\x00-\\xff]", "*");
|
||||
return str.length();
|
||||
}
|
||||
|
||||
private String toStr(Object v) {
|
||||
if (v instanceof String) {
|
||||
return (String) v;
|
||||
} else if (v == null) {
|
||||
return null;
|
||||
}
|
||||
return convert.convertTo(v);
|
||||
}
|
||||
|
||||
protected boolean initSocket(int retry) {
|
||||
for (int i = 0; i <= retry; i++) {
|
||||
try {
|
||||
String[] hostPort = addr.split(":");
|
||||
String host = hostPort[0];
|
||||
int port = Integer.parseInt(hostPort[1]);
|
||||
|
||||
client = new Socket();
|
||||
client.connect(new InetSocketAddress(host, port));
|
||||
client.setKeepAlive(true);
|
||||
|
||||
writer = client.getOutputStream();
|
||||
reader = new BufferedReader(new InputStreamReader(client.getInputStream()));
|
||||
|
||||
String groupid = getGroupid();
|
||||
if (groupid == null || groupid.isEmpty()) {
|
||||
throw new RuntimeException("ZHubClient groupid can not is empty");
|
||||
}
|
||||
send("auth", auth);
|
||||
send("groupid " + groupid);
|
||||
|
||||
StringBuffer buf = new StringBuffer("subscribe lock");
|
||||
if (mainHub.containsValue(this)) {
|
||||
buf.append(" " + APP_NAME);
|
||||
}
|
||||
for (String topic : getTopics()) {
|
||||
buf.append(" ").append(topic);
|
||||
}
|
||||
send(buf.toString());
|
||||
|
||||
// 重连 timer 订阅
|
||||
timerMap.forEach((name, timer) -> {
|
||||
send("timer", name);
|
||||
});
|
||||
if (retry > 0) {
|
||||
logger.warning(String.format("ZHubClient[%s][%s] %s Succeed!", getGroupid(), i + 1, retry > 0 ? "reconnection" : "init"));
|
||||
} else {
|
||||
logger.fine(String.format("ZHubClient[%s] %s Succeed!", getGroupid(), retry > 0 ? "reconnection" : "init"));
|
||||
}
|
||||
return true;
|
||||
} catch (Exception e) {
|
||||
if (i == 0) {
|
||||
logger.log(Level.WARNING, String.format("ZHubClient[%s] %s Failed 初始化失败!", getGroupid(), retry == 0 ? "init" : "reconnection"), e);
|
||||
try {
|
||||
Thread.sleep(1000);
|
||||
} catch (InterruptedException interruptedException) {
|
||||
interruptedException.printStackTrace();
|
||||
}
|
||||
} else {
|
||||
logger.log(Level.WARNING, String.format("ZHubClient[%s][%s] reconnection Failed!", getGroupid(), i + 1));
|
||||
try {
|
||||
Thread.sleep(1000 * 5);
|
||||
} catch (InterruptedException interruptedException) {
|
||||
interruptedException.printStackTrace();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void unsubscribe(String topic) {
|
||||
send("unsubscribe " + topic);
|
||||
super.removeEventType(topic);
|
||||
}
|
||||
|
||||
public boolean publish(String topic, Object v) {
|
||||
return send("publish", topic, toStr(v));
|
||||
}
|
||||
|
||||
public void broadcast(String topic, Object v) {
|
||||
send("broadcast", topic, toStr(v));
|
||||
}
|
||||
|
||||
// 发送 publish 主题消息,若多次发送的 topic + "-" + value 相同,将会做延时重置
|
||||
public void delay(String topic, Object v, long millis) {
|
||||
send("delay", topic, toStr(v), String.valueOf(millis));
|
||||
}
|
||||
|
||||
// 表达式支持:d+[d,H,m,s]
|
||||
public void delay(String topic, Object v, String delayExpr) {
|
||||
String endchar = "";
|
||||
int delay;
|
||||
if (delayExpr.matches("^\\d+[d,H,m,s]$")) {
|
||||
endchar = delayExpr.substring(delayExpr.length() - 1);
|
||||
delay = Integer.parseInt(delayExpr.substring(0, delayExpr.length() - 1));
|
||||
} else {
|
||||
if (!delayExpr.matches("^\\d+$")) {
|
||||
throw new IllegalArgumentException(String.format("ScheduledCycle period config error: [%s]", delayExpr));
|
||||
}
|
||||
|
||||
delay = Integer.parseInt(delayExpr);
|
||||
if (delay <= 0L) {
|
||||
throw new IllegalArgumentException(String.format("ScheduledCycle period config error: [%s]", delayExpr));
|
||||
}
|
||||
}
|
||||
|
||||
if ("M".equals(endchar)) {
|
||||
delay *= (1000 * 60 * 60 * 24 * 30);
|
||||
} else if ("d".equals(endchar)) {
|
||||
delay *= (1000 * 60 * 60 * 24);
|
||||
} else if ("H".equals(endchar)) {
|
||||
delay *= (1000 * 60 * 60);
|
||||
} else if ("m".equals(endchar)) {
|
||||
delay *= (1000 * 60);
|
||||
} else if ("s".equals(endchar)) {
|
||||
delay *= 1000;
|
||||
}
|
||||
|
||||
delay(topic, v, delay);
|
||||
}
|
||||
|
||||
@Override
|
||||
protected String getGroupid() {
|
||||
return groupid;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void subscribe(String topic) {
|
||||
send("subscribe " + topic); //新增订阅
|
||||
}
|
||||
|
||||
// ================================================== lock ==================================================
|
||||
private Map<String, Lock> lockTag = new ConcurrentHashMap<>();
|
||||
|
||||
public Lock tryLock(String key, int duration) {
|
||||
String uuid = Utility.uuid();
|
||||
Lock lock = new Lock(key, uuid, duration, this);
|
||||
lockTag.put(uuid, lock);
|
||||
|
||||
try {
|
||||
// c.send("lock", key, uuid, strconv.Itoa(duration))
|
||||
send("lock", key, uuid, String.valueOf(duration));
|
||||
synchronized (lock) {
|
||||
lock.wait();
|
||||
}
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
return lock;
|
||||
}
|
||||
|
||||
// 为替换 tryLock 方法做过度准确
|
||||
public Lock lock(String key, int duration) {
|
||||
String uuid = Utility.uuid();
|
||||
Lock lock = new Lock(key, uuid, duration, this);
|
||||
lockTag.put(uuid, lock);
|
||||
|
||||
try {
|
||||
// c.send("lock", key, uuid, strconv.Itoa(duration))
|
||||
send("lock", key, uuid, String.valueOf(duration));
|
||||
synchronized (lock) {
|
||||
lock.wait();
|
||||
}
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
return lock;
|
||||
}
|
||||
|
||||
// ================================================== timer ==================================================
|
||||
private ConcurrentHashMap<String, Timer> timerMap = new ConcurrentHashMap();
|
||||
|
||||
class Timer {
|
||||
String name;
|
||||
//String expr;
|
||||
Runnable runnable;
|
||||
//boolean single;
|
||||
|
||||
public String getName() {
|
||||
return name;
|
||||
}
|
||||
|
||||
|
||||
public Runnable getRunnable() {
|
||||
return runnable;
|
||||
}
|
||||
|
||||
public Timer(String name, Runnable runnable) {
|
||||
this.name = name;
|
||||
this.runnable = runnable;
|
||||
}
|
||||
}
|
||||
|
||||
public void timer(String name, Runnable run) {
|
||||
timerMap.put(name, new Timer(name, run));
|
||||
send("timer", name);
|
||||
}
|
||||
|
||||
@Deprecated
|
||||
public void reloadTimer() {
|
||||
send("cmd", "reload-timer");
|
||||
}
|
||||
|
||||
// ================================================== rpc ==================================================
|
||||
// -- 调用端 --
|
||||
private static Map<String, Rpc> rpcMap = new ConcurrentHashMap<>();
|
||||
private static Map<String, TypeToken> rpcRetType = new ConcurrentHashMap<>();
|
||||
|
||||
@Comment("rpc call")
|
||||
public RpcResult<Void> rpc(String topic, Object v) {
|
||||
return rpc(topic, v, null);
|
||||
}
|
||||
|
||||
@Comment("rpc call")
|
||||
public <T, R> RpcResult<R> rpc(String topic, T v, TypeToken<R> typeToken) {
|
||||
return rpc(topic, v, typeToken, 0);
|
||||
}
|
||||
|
||||
@Comment("rpc call")
|
||||
public <T, R> RpcResult<R> rpc(String topic, T v, TypeToken<R> typeToken, long timeout) {
|
||||
Rpc rpc = new Rpc<>(APP_NAME, Utility.uuid(), topic, v);
|
||||
String ruk = rpc.getRuk();
|
||||
rpcMap.put(ruk, rpc);
|
||||
if (typeToken != null) {
|
||||
rpcRetType.put(ruk, typeToken);
|
||||
}
|
||||
try {
|
||||
publish(topic, rpc); // send("rpc", topic, toStr(rpc));
|
||||
synchronized (rpc) {
|
||||
if (timeout <= 0) {
|
||||
timeout = 1000 * 15;
|
||||
}
|
||||
// call timeout default: 15s
|
||||
Timers.delay(() -> {
|
||||
synchronized (rpc) {
|
||||
Rpc rpc1 = rpcMap.get(ruk);
|
||||
if (rpc1 == null) {
|
||||
return;
|
||||
}
|
||||
|
||||
RpcResult rpcResult = rpc.render(505, "请求超时");
|
||||
rpc.setRpcResult(rpcResult);
|
||||
logger.warning("rpc timeout: " + convert.convertTo(rpc));
|
||||
rpc.notify();
|
||||
}
|
||||
}, timeout);
|
||||
|
||||
rpc.wait();
|
||||
rpcMap.remove(ruk);
|
||||
}
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
// call error
|
||||
RpcResult rpcResult = rpc.render(501, "请求失败");
|
||||
rpc.setRpcResult(rpcResult);
|
||||
}
|
||||
return rpc.getRpcResult();
|
||||
}
|
||||
|
||||
public <T, R> CompletableFuture<RpcResult<R>> rpcAsync(String topic, T v) {
|
||||
return CompletableFuture.supplyAsync(() -> rpc(topic, v, null));
|
||||
}
|
||||
|
||||
public <T, R> CompletableFuture<RpcResult<R>> rpcAsync(String topic, T v, TypeToken<R> typeToken) {
|
||||
return CompletableFuture.supplyAsync(() -> rpc(topic, v, typeToken));
|
||||
}
|
||||
|
||||
public <T, R> CompletableFuture<RpcResult<R>> rpcAsync(String topic, T v, long timeout) {
|
||||
return CompletableFuture.supplyAsync(() -> rpc(topic, v, null, timeout));
|
||||
}
|
||||
|
||||
public <T, R> CompletableFuture<RpcResult<R>> rpcAsync(String topic, T v, TypeToken<R> typeToken, long timeout) {
|
||||
return CompletableFuture.supplyAsync(() -> rpc(topic, v, typeToken, timeout));
|
||||
}
|
||||
|
||||
// RpcResult: {ruk:xxx-xxxx, retcode:0}
|
||||
@Comment("rpc call back consumer")
|
||||
private void rpcAccept(String value) {
|
||||
RpcResult resp = convert.convertFrom(new TypeToken<RpcResult<String>>() {
|
||||
}.getType(), value);
|
||||
|
||||
String ruk = resp.getRuk();
|
||||
Rpc rpc = rpcMap.remove(ruk);
|
||||
if (rpc == null) {
|
||||
return;
|
||||
}
|
||||
TypeToken typeToken = rpcRetType.get(ruk);
|
||||
|
||||
Object result = resp.getResult();
|
||||
if (result != null && typeToken != null && !"java.lang.String".equals(typeToken.getType().getTypeName()) && !"java.lang.Void".equals(typeToken.getType().getTypeName())) {
|
||||
result = convert.convertFrom(typeToken.getType(), (String) resp.getResult());
|
||||
}
|
||||
|
||||
resp.setResult(result);
|
||||
rpc.setRpcResult(resp);
|
||||
synchronized (rpc) {
|
||||
rpc.notify();
|
||||
}
|
||||
}
|
||||
|
||||
// -- 订阅端 --
|
||||
private Set<String> rpcTopics = new HashSet();
|
||||
|
||||
@Comment("rpc call consumer")
|
||||
public <R> void rpcSubscribe(String topic, Function<Rpc<String>, RpcResult<R>> fun) {
|
||||
rpcSubscribe(topic, IType.STRING, fun);
|
||||
}
|
||||
|
||||
@Comment("rpc call consumer")
|
||||
public <T, R> void rpcSubscribe(String topic, TypeToken<T> typeToken, Function<Rpc<T>, RpcResult<R>> fun) {
|
||||
Consumer<String> consumer = v -> {
|
||||
Rpc<T> rpc = null;
|
||||
try {
|
||||
rpc = convert.convertFrom(new TypeToken<Rpc<String>>() {
|
||||
}.getType(), v);
|
||||
|
||||
// 参数转换
|
||||
T paras = convert.convertFrom(typeToken.getType(), (String) rpc.getValue());
|
||||
rpc.setValue(paras);
|
||||
RpcResult result = fun.apply(rpc);
|
||||
result.setResult(toStr(result.getResult()));
|
||||
publish(rpc.getBackTopic(), result);
|
||||
} catch (Exception e) {
|
||||
logger.log(Level.WARNING, "rpc call consumer error: " + v, e);
|
||||
publish(rpc.getBackTopic(), rpc.retError("服务调用失败!"));
|
||||
}
|
||||
// back
|
||||
};
|
||||
|
||||
rpcTopics.add(topic);
|
||||
subscribe(topic, consumer);
|
||||
}
|
||||
}
|
131
src/main/java/org/redkalex/cache/redis/AbstractRedisSource.java
vendored
Normal file
131
src/main/java/org/redkalex/cache/redis/AbstractRedisSource.java
vendored
Normal file
@@ -0,0 +1,131 @@
|
||||
/*
|
||||
* Click nbfs://nbhost/SystemFileSystem/Templates/Licenses/license-default.txt to change this license
|
||||
* Click nbfs://nbhost/SystemFileSystem/Templates/Classes/Class.java to edit this template
|
||||
*/
|
||||
package org.redkalex.cache.redis;
|
||||
|
||||
import org.redkale.annotation.Resource;
|
||||
import org.redkale.convert.Convert;
|
||||
import org.redkale.convert.json.JsonConvert;
|
||||
import org.redkale.source.AbstractCacheSource;
|
||||
import org.redkale.util.AnyValue;
|
||||
import org.redkale.util.RedkaleClassLoader;
|
||||
import org.redkale.util.RedkaleException;
|
||||
import org.redkale.util.ResourceFactory;
|
||||
|
||||
import java.lang.reflect.Type;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
|
||||
/**
|
||||
* @author zhangjx
|
||||
* @since 2.8.0
|
||||
*/
|
||||
public abstract class AbstractRedisSource extends AbstractCacheSource {
|
||||
|
||||
public static final String CACHE_SOURCE_CRYPTOR = "cryptor";
|
||||
|
||||
protected String name;
|
||||
|
||||
@Resource(required = false)
|
||||
protected ResourceFactory resourceFactory;
|
||||
|
||||
@Resource(required = false)
|
||||
protected JsonConvert defaultConvert;
|
||||
|
||||
@Resource(name = "$_convert", required = false)
|
||||
protected JsonConvert convert;
|
||||
|
||||
protected int db;
|
||||
|
||||
protected RedisCryptor cryptor;
|
||||
|
||||
protected AnyValue config;
|
||||
|
||||
@Override
|
||||
public void init(AnyValue conf) {
|
||||
this.config = conf;
|
||||
super.init(conf);
|
||||
this.name = conf.getValue("name", "");
|
||||
if (this.convert == null) this.convert = this.defaultConvert;
|
||||
if (conf != null) {
|
||||
String cryptStr = conf.getValue(CACHE_SOURCE_CRYPTOR, "").trim();
|
||||
if (!cryptStr.isEmpty()) {
|
||||
try {
|
||||
Class<RedisCryptor> cryptClass = (Class) getClass().getClassLoader().loadClass(cryptStr);
|
||||
RedkaleClassLoader.putReflectionPublicConstructors(cryptClass, cryptClass.getName());
|
||||
this.cryptor = cryptClass.getConstructor().newInstance();
|
||||
} catch (ReflectiveOperationException e) {
|
||||
throw new RedkaleException(e);
|
||||
}
|
||||
}
|
||||
}
|
||||
if (cryptor != null) {
|
||||
if (resourceFactory != null) {
|
||||
resourceFactory.inject(cryptor);
|
||||
}
|
||||
cryptor.init(conf);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void destroy(AnyValue conf) {
|
||||
super.destroy(conf);
|
||||
if (cryptor != null) {
|
||||
cryptor.destroy(conf);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void close() throws Exception { //在 Application 关闭时调用
|
||||
destroy(null);
|
||||
}
|
||||
|
||||
@Override
|
||||
public String resourceName() {
|
||||
return name;
|
||||
}
|
||||
|
||||
protected String decryptValue(String key, RedisCryptor cryptor, String value) {
|
||||
return cryptor != null ? cryptor.decrypt(key, value) : value;
|
||||
}
|
||||
|
||||
protected <T> T decryptValue(String key, RedisCryptor cryptor, Type type, byte[] bs) {
|
||||
return decryptValue(key, cryptor, convert, type, bs);
|
||||
}
|
||||
|
||||
protected <T> T decryptValue(String key, RedisCryptor cryptor, Convert c, Type type, byte[] bs) {
|
||||
if (bs == null) return null;
|
||||
if (type == byte[].class) return (T) bs;
|
||||
if (cryptor == null || (type instanceof Class && (((Class) type).isPrimitive() || Number.class.isAssignableFrom((Class) type)))) {
|
||||
return (T) (c == null ? this.convert : c).convertFrom(type, bs);
|
||||
}
|
||||
String deval = cryptor.decrypt(key, new String(bs, StandardCharsets.UTF_8));
|
||||
return deval == null ? null : (T) (c == null ? this.convert : c).convertFrom(type, deval.getBytes(StandardCharsets.UTF_8));
|
||||
}
|
||||
|
||||
protected String encryptValue(String key, RedisCryptor cryptor, String value) {
|
||||
return cryptor != null ? cryptor.encrypt(key, value) : value;
|
||||
}
|
||||
|
||||
protected <T> byte[] encryptValue(String key, RedisCryptor cryptor, Convert c, T value) {
|
||||
return encryptValue(key, cryptor, null, c, value);
|
||||
}
|
||||
|
||||
protected <T> byte[] encryptValue(String key, RedisCryptor cryptor, Type type, Convert c, T value) {
|
||||
if (value == null) return null;
|
||||
Type t = type == null ? value.getClass() : type;
|
||||
if (cryptor == null && type == String.class) {
|
||||
return value.toString().getBytes(StandardCharsets.UTF_8);
|
||||
}
|
||||
return encryptValue(key, cryptor, t, (c == null ? this.convert : c).convertToBytes(t, value));
|
||||
}
|
||||
|
||||
protected byte[] encryptValue(String key, RedisCryptor cryptor, Type type, byte[] bs) {
|
||||
if (bs == null) return null;
|
||||
if (cryptor == null || (type instanceof Class && (((Class) type).isPrimitive() || Number.class.isAssignableFrom((Class) type)))) {
|
||||
return bs;
|
||||
}
|
||||
String enval = cryptor.encrypt(key, new String(bs, StandardCharsets.UTF_8));
|
||||
return enval == null ? null : enval.getBytes(StandardCharsets.UTF_8);
|
||||
}
|
||||
}
|
343
src/main/java/org/redkalex/cache/redis/MyRedisCacheSource.java
vendored
Normal file
343
src/main/java/org/redkalex/cache/redis/MyRedisCacheSource.java
vendored
Normal file
@@ -0,0 +1,343 @@
|
||||
package org.redkalex.cache.redis;
|
||||
|
||||
import org.redkale.annotation.AutoLoad;
|
||||
import org.redkale.annotation.ResourceType;
|
||||
import org.redkale.service.Local;
|
||||
import org.redkale.source.CacheSource;
|
||||
import org.redkale.util.AnyValue;
|
||||
|
||||
import java.io.Serializable;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.util.*;
|
||||
import java.util.concurrent.CompletableFuture;
|
||||
|
||||
@Local
|
||||
@AutoLoad(false)
|
||||
@ResourceType(CacheSource.class)
|
||||
public class MyRedisCacheSource extends RedisCacheSource {
|
||||
|
||||
@Override
|
||||
public void init(AnyValue conf) {
|
||||
super.init(conf);
|
||||
}
|
||||
/*
|
||||
//--------------------- zset ------------------------------
|
||||
|
||||
public int getZrank(String key, V v) {
|
||||
byte[][] bytes = Stream.of(key, v).map(x -> String.valueOf(x).getBytes(StandardCharsets.UTF_8)).toArray(byte[][]::new);
|
||||
Long t = (Long) send("ZRANK", CacheEntryType.OBJECT, (Type) null, key, bytes).join();
|
||||
|
||||
return t == null ? -1 : (int) (long) t;
|
||||
}
|
||||
|
||||
public int getZrevrank(String key, V v) {
|
||||
byte[][] bytes = Stream.of(key, v).map(x -> String.valueOf(x).getBytes(StandardCharsets.UTF_8)).toArray(byte[][]::new);
|
||||
Long t = (Long) send("ZREVRANK", CacheEntryType.OBJECT, (Type) null, key, bytes).join();
|
||||
|
||||
return t == null ? -1 : (int) (long) t;
|
||||
}
|
||||
|
||||
//ZRANGE/ZREVRANGE key start stop
|
||||
public List<V> getZset(String key) {
|
||||
byte[][] bytes = Stream.of(key, 0, -1).map(x -> String.valueOf(x).getBytes(StandardCharsets.UTF_8)).toArray(byte[][]::new);
|
||||
List<V> vs = (List<V>) send("ZREVRANGE", CacheEntryType.OBJECT, (Type) null, key, bytes).join();
|
||||
List<V> vs2 = new ArrayList(vs.size());
|
||||
|
||||
for (int i = 0; i < vs.size(); ++i) {
|
||||
if (i % 2 == 1) {
|
||||
vs2.add(this.convert.convertFrom(this.objValueType, String.valueOf(vs.get(i))));
|
||||
} else {
|
||||
vs2.add(vs.get(i));
|
||||
}
|
||||
}
|
||||
|
||||
return vs2;
|
||||
}
|
||||
|
||||
public List<V> getZset(String key, int offset, int limit) {
|
||||
byte[][] bytes = Stream.of(key, offset, offset + limit - 1).map(x -> String.valueOf(x).getBytes(StandardCharsets.UTF_8)).toArray(byte[][]::new);
|
||||
List<V> vs = (List<V>) send("ZREVRANGE", CacheEntryType.OBJECT, (Type) null, key, bytes).join();
|
||||
List<V> vs2 = new ArrayList(vs.size());
|
||||
|
||||
for (int i = 0; i < vs.size(); ++i) {
|
||||
if (i % 2 == 1) {
|
||||
vs2.add(this.convert.convertFrom(this.objValueType, String.valueOf(vs.get(i))));
|
||||
} else {
|
||||
vs2.add(vs.get(i));
|
||||
}
|
||||
}
|
||||
|
||||
return vs2;
|
||||
}
|
||||
|
||||
public LinkedHashMap<V, Long> getZsetLongScore(String key) {
|
||||
LinkedHashMap<V, Double> map = getZsetDoubleScore(key);
|
||||
if (map.isEmpty()) {
|
||||
return new LinkedHashMap<>();
|
||||
}
|
||||
|
||||
LinkedHashMap<V, Long> map2 = new LinkedHashMap<>(map.size());
|
||||
map.forEach((k, v) -> map2.put(k, (long) (double) v));
|
||||
return map2;
|
||||
}
|
||||
|
||||
public LinkedHashMap<V, Long> getZsetItemsLongScore(String key) {
|
||||
byte[][] bytes = Stream.of(key, 0, -1, "WITHSCORES").map(x -> String.valueOf(x).getBytes(StandardCharsets.UTF_8)).toArray(byte[][]::new);
|
||||
List vs = (List) send("ZRANGE", CacheEntryType.OBJECT, (Type) null, key, bytes).join();
|
||||
|
||||
LinkedHashMap<V, Long> map = new LinkedHashMap<>();
|
||||
for (int i = 0; i < vs.size(); i += 2) {
|
||||
map.put((V) vs.get(i), (long) Double.parseDouble((String) vs.get(i + 1)));
|
||||
}
|
||||
return map;
|
||||
}
|
||||
|
||||
public Long getZsetLongScore(String key, V v) {
|
||||
Double score = getZsetDoubleScore(key, v);
|
||||
if (score == null) {
|
||||
return null;
|
||||
}
|
||||
return (long) (double) score;
|
||||
}
|
||||
|
||||
public LinkedHashMap<V, Double> getZsetDoubleScore(String key) {
|
||||
byte[][] bytes = Stream.of(key, 0, -1, "WITHSCORES").map(x -> String.valueOf(x).getBytes(StandardCharsets.UTF_8)).toArray(byte[][]::new);
|
||||
List vs = (List) send("ZREVRANGE", CacheEntryType.OBJECT, (Type) null, key, bytes).join();
|
||||
|
||||
LinkedHashMap<V, Double> map = new LinkedHashMap<>();
|
||||
for (int i = 0; i < vs.size(); i += 2) {
|
||||
map.put((V) vs.get(i), Double.parseDouble((String) vs.get(i + 1)));
|
||||
}
|
||||
return map;
|
||||
}
|
||||
|
||||
public Double getZsetDoubleScore(String key, V v) {
|
||||
byte[][] bytes = Stream.of(key, v).map(x -> String.valueOf(x).getBytes(StandardCharsets.UTF_8)).toArray(byte[][]::new);
|
||||
Serializable zscore = send("ZSCORE", CacheEntryType.OBJECT, (Type) null, key, bytes).join();
|
||||
if (zscore == null) {
|
||||
return null;
|
||||
}
|
||||
|
||||
return Double.parseDouble(String.valueOf(zscore));
|
||||
}
|
||||
|
||||
public LinkedHashMap<V, Long> getZsetLongScore(String key, int offset, int limit) {
|
||||
byte[][] bytes = Stream.of(key, offset, offset + limit - 1, "WITHSCORES").map(x -> String.valueOf(x).getBytes(StandardCharsets.UTF_8)).toArray(byte[][]::new);
|
||||
List vs = (List) send("ZREVRANGE", CacheEntryType.OBJECT, (Type) null, key, bytes).join();
|
||||
|
||||
LinkedHashMap<V, Long> map = new LinkedHashMap<>();
|
||||
for (int i = 0; i < vs.size(); i += 2) {
|
||||
map.put((V) vs.get(i), (long) Double.parseDouble((String) vs.get(i + 1)));
|
||||
}
|
||||
return map;
|
||||
}
|
||||
|
||||
public LinkedHashMap<V, Double> getZsetDoubleScore(String key, int offset, int limit) {
|
||||
byte[][] bytes = Stream.of(key, offset, offset + limit - 1, "WITHSCORES").map(x -> String.valueOf(x).getBytes(StandardCharsets.UTF_8)).toArray(byte[][]::new);
|
||||
List vs = (List) send("ZREVRANGE", CacheEntryType.OBJECT, (Type) null, key, bytes).join();
|
||||
|
||||
LinkedHashMap<V, Double> map = new LinkedHashMap<>();
|
||||
for (int i = 0; i < vs.size(); i += 2) {
|
||||
map.put((V) vs.get(i), Double.parseDouble(vs.get(i + 1) + ""));
|
||||
}
|
||||
return map;
|
||||
}
|
||||
* */
|
||||
|
||||
// --------------------
|
||||
public <N extends Number> void zadd(String key, Map<Serializable, N> kv) {
|
||||
if (kv == null || kv.isEmpty()) {
|
||||
return;
|
||||
}
|
||||
List<Serializable> args = new ArrayList();
|
||||
kv.forEach((k, v) -> {
|
||||
args.add(k);
|
||||
args.add(v);
|
||||
});
|
||||
|
||||
sendAsync("ZADD", key, args.toArray(Serializable[]::new)).join();
|
||||
}
|
||||
|
||||
public <N extends Number> double zincr(String key, Serializable number, N n) {
|
||||
return sendAsync("ZINCRBY", key, number, n).thenApply(x -> x.getDoubleValue(0d)).join();
|
||||
}
|
||||
|
||||
public void zrem(String key, Serializable... vs) {
|
||||
sendAsync("ZREM", key, vs).join();
|
||||
}
|
||||
|
||||
/*public <T> List<T> zexists(String key, T... fields) {
|
||||
if (fields == null || fields.length == 0) {
|
||||
return new ArrayList<>();
|
||||
}
|
||||
List<String> para = new ArrayList<>();
|
||||
para.add("" +
|
||||
" local key = KEYS[1];" +
|
||||
" local args = ARGV;" +
|
||||
" local result = {};" +
|
||||
" for i,v in ipairs(args) do" +
|
||||
" local inx = redis.call('ZREVRANK', key, v);" +
|
||||
" if(inx) then" +
|
||||
" table.insert(result,1,v);" +
|
||||
" end" +
|
||||
" end" +
|
||||
" return result;");
|
||||
para.add("1");
|
||||
para.add(key);
|
||||
for (Object field : fields) {
|
||||
para.add(String.valueOf(field));
|
||||
}
|
||||
|
||||
// todo:
|
||||
//sendAsync("EVAL", null, para.toArray(Serializable[]::new)).thenApply(x -> x.).join();
|
||||
|
||||
return null;
|
||||
}*/
|
||||
|
||||
//--------------------- bit ------------------------------
|
||||
public boolean getBit(String key, int offset) {
|
||||
return sendAsync("GETBIT", key, key.getBytes(StandardCharsets.UTF_8), String.valueOf(offset).getBytes(StandardCharsets.UTF_8)).thenApply(v -> v.getIntValue(0) > 0).join();
|
||||
}
|
||||
|
||||
public void setBit(String key, int offset, boolean bool) {
|
||||
sendAsync("SETBIT", key, offset, bool ? 1 : 0).join();
|
||||
}
|
||||
//--------------------- bit ------------------------------
|
||||
|
||||
//--------------------- lock ------------------------------
|
||||
// 尝试加锁,成功返回0,否则返回上一锁剩余毫秒值
|
||||
public long tryLock(String key, int millis) {
|
||||
Serializable[] obj = {"" +
|
||||
"if (redis.call('EXISTS',KEYS[1]) == 0) then " +
|
||||
"redis.call('PSETEX',KEYS[1],ARGV[1],1); " +
|
||||
"return 0; " +
|
||||
"else " +
|
||||
"return redis.call('PTTL',KEYS[1]); " +
|
||||
"end;", 1, key, millis
|
||||
};
|
||||
|
||||
return sendAsync("EVAL", null, obj).thenApply(v -> v.getIntValue(1)).join();
|
||||
}
|
||||
|
||||
// 加锁
|
||||
public void lock(String key, int millis) {
|
||||
long i;
|
||||
do {
|
||||
i = tryLock(key, millis);
|
||||
try {
|
||||
Thread.sleep(10);
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
} while (i > 0);
|
||||
}
|
||||
|
||||
// 解锁
|
||||
public void unlock(String key) {
|
||||
remove(key);
|
||||
}
|
||||
|
||||
//--------------------- key ------------------------------
|
||||
|
||||
public String get(String key) {
|
||||
return get(key, String.class);
|
||||
}
|
||||
|
||||
public void set(String key, Serializable value) {
|
||||
sendAsync("SET", key, value).join();
|
||||
}
|
||||
|
||||
//--------------------- set ------------------------------
|
||||
public <T> void sadd(String key, Collection<T> args) {
|
||||
saddAsync(key, args.toArray(Serializable[]::new)).join();
|
||||
}
|
||||
|
||||
public void sadd(String key, Serializable... args) {
|
||||
saddAsync(key, Arrays.stream(args).toArray(Serializable[]::new)).join();
|
||||
}
|
||||
|
||||
public void srem(String key, Serializable... args) {
|
||||
sremAsync(key, args).join();
|
||||
}
|
||||
|
||||
public CompletableFuture<RedisCacheResult> saddAsync(String key, Serializable... args) {
|
||||
return sendAsync("SADD", key, args);
|
||||
}
|
||||
|
||||
public CompletableFuture<RedisCacheResult> sremAsync(String key, Serializable... args) {
|
||||
return sendAsync("SREM", key, args);
|
||||
}
|
||||
|
||||
//--------------------- hm ------------------------------
|
||||
|
||||
/*public Long incrHm(String key, String field, int value) {
|
||||
return sendAsync("HINCRBY", key, field, value).thenApply(x -> x.getLongValue(0l)).join();
|
||||
}
|
||||
|
||||
public Double incrHm(String key, String field, double value) {
|
||||
return sendAsync("HINCRBYFLOAT", key, field, value).thenApply(x -> x.getDoubleValue(0d)).join();
|
||||
}*/
|
||||
|
||||
public void setHm(String key, String field, Serializable value) {
|
||||
setHmsAsync(key, Map.of(field, value)).join();
|
||||
}
|
||||
|
||||
public void setHms(String key, Map kv) {
|
||||
setHmsAsync(key, kv).join();
|
||||
}
|
||||
|
||||
public CompletableFuture<RedisCacheResult> setHmsAsync(String key, Map<Serializable, Serializable> kv) {
|
||||
List<Serializable> args = new ArrayList();
|
||||
kv.forEach((k, v) -> {
|
||||
args.add(k);
|
||||
args.add(v);
|
||||
});
|
||||
|
||||
return sendAsync("HMSET", key, args.toArray(Serializable[]::new));
|
||||
}
|
||||
|
||||
public String getHm(String key, String field) {
|
||||
return getHm(key, String.class, field);
|
||||
}
|
||||
|
||||
public <T extends Serializable> T getHm(String key, Class<T> type, String field) {
|
||||
List<Serializable> list = super.hmget(key, type, field);
|
||||
if (list == null && list.isEmpty()) {
|
||||
return null;
|
||||
}
|
||||
return (T) list.get(0);
|
||||
}
|
||||
|
||||
public Map<String, String> getHms(String key, String... field) {
|
||||
return getHms(key, String.class, field);
|
||||
}
|
||||
|
||||
public <T extends Serializable> Map<String, T> getHms(String key, Class<T> type, String... field) {
|
||||
List<Serializable> list = super.hmget(key, type, field);
|
||||
if (list == null && list.isEmpty()) {
|
||||
return null;
|
||||
}
|
||||
Map<String, T> map = new HashMap<>(field.length);
|
||||
|
||||
for (int i = 0; i < field.length; i++) {
|
||||
if (list.get(i) == null) {
|
||||
continue;
|
||||
}
|
||||
map.put(field[i], (T) list.get(i));
|
||||
}
|
||||
return map;
|
||||
}
|
||||
|
||||
/*public Map<String, Object> getHmall(String key) {
|
||||
List<String> list = null; // TODO:
|
||||
Map<String, Object> map = new HashMap<>();
|
||||
if (list.isEmpty()) {
|
||||
return map;
|
||||
}
|
||||
|
||||
for (int i = 0; i + 1 < list.size(); i += 2) {
|
||||
map.put((String) list.get(i), list.get(i + 1));
|
||||
}
|
||||
return map;
|
||||
}*/
|
||||
}
|
78
src/main/java/org/redkalex/cache/redis/RedisCRC16.java
vendored
Normal file
78
src/main/java/org/redkalex/cache/redis/RedisCRC16.java
vendored
Normal file
@@ -0,0 +1,78 @@
|
||||
/*
|
||||
*
|
||||
*/
|
||||
package org.redkalex.cache.redis;
|
||||
|
||||
import org.redkale.util.Utility;
|
||||
|
||||
import java.util.Arrays;
|
||||
|
||||
/**
|
||||
* @author zhangjx
|
||||
*/
|
||||
public class RedisCRC16 {
|
||||
|
||||
private static final int[] LOOKUP_TABLE = {0x0000, 0x1021, 0x2042, 0x3063, 0x4084, 0x50A5, 0x60C6,
|
||||
0x70E7, 0x8108, 0x9129, 0xA14A, 0xB16B, 0xC18C, 0xD1AD, 0xE1CE, 0xF1EF, 0x1231, 0x0210, 0x3273,
|
||||
0x2252, 0x52B5, 0x4294, 0x72F7, 0x62D6, 0x9339, 0x8318, 0xB37B, 0xA35A, 0xD3BD, 0xC39C, 0xF3FF,
|
||||
0xE3DE, 0x2462, 0x3443, 0x0420, 0x1401, 0x64E6, 0x74C7, 0x44A4, 0x5485, 0xA56A, 0xB54B, 0x8528,
|
||||
0x9509, 0xE5EE, 0xF5CF, 0xC5AC, 0xD58D, 0x3653, 0x2672, 0x1611, 0x0630, 0x76D7, 0x66F6, 0x5695,
|
||||
0x46B4, 0xB75B, 0xA77A, 0x9719, 0x8738, 0xF7DF, 0xE7FE, 0xD79D, 0xC7BC, 0x48C4, 0x58E5, 0x6886,
|
||||
0x78A7, 0x0840, 0x1861, 0x2802, 0x3823, 0xC9CC, 0xD9ED, 0xE98E, 0xF9AF, 0x8948, 0x9969, 0xA90A,
|
||||
0xB92B, 0x5AF5, 0x4AD4, 0x7AB7, 0x6A96, 0x1A71, 0x0A50, 0x3A33, 0x2A12, 0xDBFD, 0xCBDC, 0xFBBF,
|
||||
0xEB9E, 0x9B79, 0x8B58, 0xBB3B, 0xAB1A, 0x6CA6, 0x7C87, 0x4CE4, 0x5CC5, 0x2C22, 0x3C03, 0x0C60,
|
||||
0x1C41, 0xEDAE, 0xFD8F, 0xCDEC, 0xDDCD, 0xAD2A, 0xBD0B, 0x8D68, 0x9D49, 0x7E97, 0x6EB6, 0x5ED5,
|
||||
0x4EF4, 0x3E13, 0x2E32, 0x1E51, 0x0E70, 0xFF9F, 0xEFBE, 0xDFDD, 0xCFFC, 0xBF1B, 0xAF3A, 0x9F59,
|
||||
0x8F78, 0x9188, 0x81A9, 0xB1CA, 0xA1EB, 0xD10C, 0xC12D, 0xF14E, 0xE16F, 0x1080, 0x00A1, 0x30C2,
|
||||
0x20E3, 0x5004, 0x4025, 0x7046, 0x6067, 0x83B9, 0x9398, 0xA3FB, 0xB3DA, 0xC33D, 0xD31C, 0xE37F,
|
||||
0xF35E, 0x02B1, 0x1290, 0x22F3, 0x32D2, 0x4235, 0x5214, 0x6277, 0x7256, 0xB5EA, 0xA5CB, 0x95A8,
|
||||
0x8589, 0xF56E, 0xE54F, 0xD52C, 0xC50D, 0x34E2, 0x24C3, 0x14A0, 0x0481, 0x7466, 0x6447, 0x5424,
|
||||
0x4405, 0xA7DB, 0xB7FA, 0x8799, 0x97B8, 0xE75F, 0xF77E, 0xC71D, 0xD73C, 0x26D3, 0x36F2, 0x0691,
|
||||
0x16B0, 0x6657, 0x7676, 0x4615, 0x5634, 0xD94C, 0xC96D, 0xF90E, 0xE92F, 0x99C8, 0x89E9, 0xB98A,
|
||||
0xA9AB, 0x5844, 0x4865, 0x7806, 0x6827, 0x18C0, 0x08E1, 0x3882, 0x28A3, 0xCB7D, 0xDB5C, 0xEB3F,
|
||||
0xFB1E, 0x8BF9, 0x9BD8, 0xABBB, 0xBB9A, 0x4A75, 0x5A54, 0x6A37, 0x7A16, 0x0AF1, 0x1AD0, 0x2AB3,
|
||||
0x3A92, 0xFD2E, 0xED0F, 0xDD6C, 0xCD4D, 0xBDAA, 0xAD8B, 0x9DE8, 0x8DC9, 0x7C26, 0x6C07, 0x5C64,
|
||||
0x4C45, 0x3CA2, 0x2C83, 0x1CE0, 0x0CC1, 0xEF1F, 0xFF3E, 0xCF5D, 0xDF7C, 0xAF9B, 0xBFBA, 0x8FD9,
|
||||
0x9FF8, 0x6E17, 0x7E36, 0x4E55, 0x5E74, 0x2E93, 0x3EB2, 0x0ED1, 0x1EF0};
|
||||
|
||||
private RedisCRC16() {
|
||||
}
|
||||
|
||||
public static int crc16(byte[] bytes) {
|
||||
int crc = 0x0000;
|
||||
for (byte b : bytes) {
|
||||
crc = (crc << 8) ^ LOOKUP_TABLE[((crc >>> 8) ^ (b & 0xFF)) & 0xFF];
|
||||
}
|
||||
return crc & 0xFFFF;
|
||||
}
|
||||
|
||||
public static int calcSlot(int maxSlot, byte[] key) {
|
||||
if (key == null) {
|
||||
return 0;
|
||||
}
|
||||
int start = Utility.indexOf(key, (byte) '{');
|
||||
if (start != -1) {
|
||||
int end = Utility.indexOf(key, start + 1, (byte) '}');
|
||||
if (end != -1) {
|
||||
key = Arrays.copyOfRange(key, start + 1, end);
|
||||
}
|
||||
}
|
||||
int result = crc16(key) % maxSlot;
|
||||
return result;
|
||||
}
|
||||
|
||||
public static int calcSlot(int maxSlot, String key) {
|
||||
if (key == null) {
|
||||
return 0;
|
||||
}
|
||||
int start = key.indexOf('{');
|
||||
if (start != -1) {
|
||||
int end = key.indexOf('}');
|
||||
if (end != -1 && start + 1 < end) {
|
||||
key = key.substring(start + 1, end);
|
||||
}
|
||||
}
|
||||
int result = crc16(key.getBytes()) % maxSlot;
|
||||
return result;
|
||||
}
|
||||
}
|
36
src/main/java/org/redkalex/cache/redis/RedisCacheClient.java
vendored
Normal file
36
src/main/java/org/redkalex/cache/redis/RedisCacheClient.java
vendored
Normal file
@@ -0,0 +1,36 @@
|
||||
/*
|
||||
* To change this license header, choose License Headers in Project Properties.
|
||||
* To change this template file, choose Tools | Templates
|
||||
* and open the template in the editor.
|
||||
*/
|
||||
package org.redkalex.cache.redis;
|
||||
|
||||
import org.redkale.net.AsyncConnection;
|
||||
import org.redkale.net.AsyncGroup;
|
||||
import org.redkale.net.client.Client;
|
||||
import org.redkale.net.client.ClientAddress;
|
||||
|
||||
/**
|
||||
* @author zhangjx
|
||||
*/
|
||||
public class RedisCacheClient extends Client<RedisCacheConnection, RedisCacheRequest, RedisCacheResult> {
|
||||
|
||||
public RedisCacheClient(String name, AsyncGroup group, String key, ClientAddress address, int maxConns, int maxPipelines, RedisCacheReqAuth authReq, RedisCacheReqDB dbReq) {
|
||||
super(name, group, true, address, maxConns, maxPipelines, () -> new RedisCacheReqPing(), () -> new RedisCacheReqClose(), null); //maxConns
|
||||
if (authReq != null || dbReq != null) {
|
||||
if (authReq != null && dbReq != null) {
|
||||
this.authenticate = conn -> writeChannel(conn, authReq).thenCompose(v -> writeChannel(conn, dbReq)).thenApply(v -> conn);
|
||||
} else if (authReq != null) {
|
||||
this.authenticate = conn -> writeChannel(conn, authReq).thenApply(v -> conn);
|
||||
} else {
|
||||
this.authenticate = conn -> writeChannel(conn, dbReq).thenApply(v -> conn);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
protected RedisCacheConnection createClientConnection(final int index, AsyncConnection channel) {
|
||||
return new RedisCacheConnection(this, index, channel);
|
||||
}
|
||||
|
||||
}
|
249
src/main/java/org/redkalex/cache/redis/RedisCacheCodec.java
vendored
Normal file
249
src/main/java/org/redkalex/cache/redis/RedisCacheCodec.java
vendored
Normal file
@@ -0,0 +1,249 @@
|
||||
/*
|
||||
* To change this license header, choose License Headers in Project Properties.
|
||||
* To change this template file, choose Tools | Templates
|
||||
* and open the template in the editor.
|
||||
*/
|
||||
package org.redkalex.cache.redis;
|
||||
|
||||
import org.redkale.net.client.ClientCodec;
|
||||
import org.redkale.net.client.ClientConnection;
|
||||
import org.redkale.util.ByteArray;
|
||||
|
||||
import java.nio.ByteBuffer;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.logging.Logger;
|
||||
|
||||
/**
|
||||
* @author zhangjx
|
||||
*/
|
||||
public class RedisCacheCodec extends ClientCodec<RedisCacheRequest, RedisCacheResult> {
|
||||
|
||||
protected static final byte TYPE_STRING = '+'; //简单字符串(不包含CRLF)类型
|
||||
|
||||
protected static final byte TYPE_ERROR = '-'; //错误(不包含CRLF)类型
|
||||
|
||||
protected static final byte TYPE_NUMBER = ':'; //整型
|
||||
|
||||
protected static final byte TYPE_BULK = '$'; //块字符串
|
||||
|
||||
protected static final byte TYPE_ARRAY = '*'; //数组
|
||||
|
||||
protected static final Logger logger = Logger.getLogger(RedisCacheCodec.class.getSimpleName());
|
||||
|
||||
protected byte halfFrameCmd;
|
||||
|
||||
protected int halfFrameBulkLength = -10;
|
||||
|
||||
protected int halfFrameArraySize = -10;
|
||||
|
||||
protected int halfFrameArrayIndex; //从0开始
|
||||
|
||||
protected int halfFrameArrayItemLength = -10;
|
||||
|
||||
protected ByteArray halfFrameBytes;
|
||||
|
||||
protected byte frameType;
|
||||
|
||||
protected byte[] frameValue; //(不包含CRLF)
|
||||
|
||||
protected List<byte[]> frameList; //(不包含CRLF)
|
||||
|
||||
private ByteArray recyclableArray;
|
||||
|
||||
public RedisCacheCodec(ClientConnection connection) {
|
||||
super(connection);
|
||||
}
|
||||
|
||||
protected ByteArray pollArray(ByteArray array) {
|
||||
if (recyclableArray == null) {
|
||||
recyclableArray = new ByteArray();
|
||||
} else {
|
||||
recyclableArray.clear();
|
||||
}
|
||||
recyclableArray.clear();
|
||||
if (array != null) {
|
||||
recyclableArray.put(array, 0, array.length());
|
||||
}
|
||||
return recyclableArray;
|
||||
}
|
||||
|
||||
private boolean checkBytesFrame(RedisCacheConnection conn, ByteBuffer buffer, ByteArray array) {
|
||||
// byte[] dbs = new byte[buffer.remaining()];
|
||||
// for (int i = 0; i < dbs.length; i++) {
|
||||
// dbs[i] = buffer.get(buffer.position() + i);
|
||||
// }
|
||||
// ArrayDeque<ClientFuture> deque = (ArrayDeque) responseQueue(conn);
|
||||
// logger.log(Level.FINEST, "[" + Utility.nowMillis() + "] [" + Thread.currentThread().getName() + "]: " + conn + ", 原始数据: " + new String(dbs).replace("\r\n", " ") + ", req=" + deque.getFirst().getRequest());
|
||||
|
||||
array.clear();
|
||||
byte type = halfFrameCmd == 0 ? buffer.get() : halfFrameCmd;
|
||||
if (halfFrameBytes != null) {
|
||||
array.put(halfFrameBytes, 0, halfFrameBytes.length());
|
||||
}
|
||||
frameType = type;
|
||||
if (type == TYPE_STRING || type == TYPE_ERROR || type == TYPE_NUMBER) {
|
||||
if (readComplete(buffer, array)) {
|
||||
frameValue = array.getBytes();
|
||||
} else {
|
||||
halfFrameCmd = type;
|
||||
halfFrameBytes = pollArray(array);
|
||||
return false;
|
||||
}
|
||||
} else if (type == TYPE_BULK) {
|
||||
int bulkLength = halfFrameBulkLength;
|
||||
if (bulkLength < -2) {
|
||||
if (!readComplete(buffer, array)) { //没有读到bulkLength
|
||||
halfFrameCmd = type;
|
||||
halfFrameBulkLength = -10;
|
||||
halfFrameBytes = pollArray(array);
|
||||
return false;
|
||||
}
|
||||
bulkLength = Integer.parseInt(array.toString(StandardCharsets.UTF_8));
|
||||
array.clear();
|
||||
}
|
||||
if (bulkLength == -1) {
|
||||
frameValue = null;
|
||||
} else if (readComplete(buffer, array)) {
|
||||
frameValue = array.getBytes();
|
||||
} else {
|
||||
halfFrameCmd = type;
|
||||
halfFrameBulkLength = bulkLength;
|
||||
halfFrameBytes = pollArray(array);
|
||||
return false;
|
||||
}
|
||||
} else if (type == TYPE_ARRAY) {
|
||||
int arraySize = halfFrameArraySize;
|
||||
if (arraySize < -2) {
|
||||
if (!readComplete(buffer, array)) { //没有读到arraySize
|
||||
halfFrameCmd = type;
|
||||
halfFrameArraySize = -10;
|
||||
halfFrameArrayIndex = 0;
|
||||
halfFrameArrayItemLength = -10;
|
||||
halfFrameBytes = pollArray(array);
|
||||
return false;
|
||||
}
|
||||
arraySize = Integer.parseInt(array.toString(StandardCharsets.UTF_8));
|
||||
array.clear();
|
||||
}
|
||||
int arrayIndex = halfFrameArrayIndex;
|
||||
for (int i = arrayIndex; i < arraySize; i++) {
|
||||
int itemLength = halfFrameArrayItemLength;
|
||||
halfFrameArrayItemLength = -10;
|
||||
if (itemLength < -2) {
|
||||
if (!readComplete(buffer, array)) { //没有读到bulkLength
|
||||
halfFrameCmd = type;
|
||||
halfFrameArraySize = arraySize;
|
||||
halfFrameArrayIndex = i;
|
||||
halfFrameArrayItemLength = -10;
|
||||
halfFrameBytes = pollArray(array);
|
||||
return false;
|
||||
}
|
||||
byte sign = array.get(0);
|
||||
itemLength = Integer.parseInt(array.toString(1, StandardCharsets.UTF_8));
|
||||
array.clear();
|
||||
if (sign == TYPE_ARRAY) { //数组中嵌套数组,目前有 HSCAN
|
||||
frameValue = null;
|
||||
if (frameList != null) {
|
||||
frameList.clear();
|
||||
}
|
||||
clearHalfFrame();
|
||||
if (itemLength == 0) {
|
||||
return true;
|
||||
}
|
||||
halfFrameCmd = sign;
|
||||
halfFrameArraySize = itemLength;
|
||||
if (!buffer.hasRemaining()) {
|
||||
return false;
|
||||
}
|
||||
return checkBytesFrame(conn, buffer, array);
|
||||
}
|
||||
}
|
||||
int cha = itemLength - array.length();
|
||||
if (itemLength == -1) {
|
||||
if (frameList == null) {
|
||||
frameList = new ArrayList<>();
|
||||
}
|
||||
frameList.add(null);
|
||||
array.clear();
|
||||
} else if (buffer.remaining() >= cha + 2) {
|
||||
for (int j = 0; j < cha; j++) array.put(buffer.get());
|
||||
buffer.get(); //\r
|
||||
buffer.get(); //\n
|
||||
if (frameList == null) {
|
||||
frameList = new ArrayList<>();
|
||||
}
|
||||
frameList.add(array.getBytes());
|
||||
array.clear();
|
||||
} else {
|
||||
while (buffer.hasRemaining()) array.put(buffer.get());
|
||||
halfFrameCmd = type;
|
||||
halfFrameArraySize = arraySize;
|
||||
halfFrameArrayIndex = i;
|
||||
halfFrameArrayItemLength = itemLength;
|
||||
halfFrameBytes = pollArray(array);
|
||||
return false;
|
||||
}
|
||||
}
|
||||
}
|
||||
clearHalfFrame();
|
||||
return true;
|
||||
}
|
||||
|
||||
protected void clearHalfFrame() {
|
||||
halfFrameCmd = 0;
|
||||
halfFrameBulkLength = -10;
|
||||
halfFrameArraySize = -10;
|
||||
halfFrameArrayIndex = 0;
|
||||
halfFrameArrayItemLength = -10;
|
||||
halfFrameBytes = null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void decodeMessages(ByteBuffer realbuf, ByteArray array) {
|
||||
RedisCacheConnection conn = (RedisCacheConnection) connection;
|
||||
if (!realbuf.hasRemaining()) {
|
||||
return;
|
||||
}
|
||||
ByteBuffer buffer = realbuf;
|
||||
if (!checkBytesFrame(conn, buffer, array)) {
|
||||
return;
|
||||
}
|
||||
//buffer必然包含一个完整的frame数据
|
||||
boolean first = true;
|
||||
RedisCacheRequest request = null;
|
||||
while (first || buffer.hasRemaining()) {
|
||||
if (request == null) {
|
||||
request = nextRequest();
|
||||
}
|
||||
if (!first && !checkBytesFrame(conn, buffer, array)) {
|
||||
break;
|
||||
}
|
||||
if (frameType == TYPE_ERROR) {
|
||||
addMessage(request, new RuntimeException(new String(frameValue, StandardCharsets.UTF_8)));
|
||||
} else {
|
||||
addMessage(request, conn.pollResultSet(request).prepare(frameType, frameValue, frameList));
|
||||
}
|
||||
frameType = 0;
|
||||
frameValue = null;
|
||||
frameList = null;
|
||||
halfFrameCmd = 0;
|
||||
halfFrameBytes = null;
|
||||
first = false;
|
||||
buffer = realbuf;
|
||||
}
|
||||
}
|
||||
|
||||
protected boolean readComplete(ByteBuffer buffer, ByteArray array) {
|
||||
while (buffer.hasRemaining()) {
|
||||
byte b = buffer.get();
|
||||
if (b == '\n') {
|
||||
array.removeLastByte(); //移除 \r
|
||||
return true;
|
||||
}
|
||||
array.put(b);
|
||||
}
|
||||
return false;
|
||||
}
|
||||
}
|
49
src/main/java/org/redkalex/cache/redis/RedisCacheConnection.java
vendored
Normal file
49
src/main/java/org/redkalex/cache/redis/RedisCacheConnection.java
vendored
Normal file
@@ -0,0 +1,49 @@
|
||||
/*
|
||||
* To change this license header, choose License Headers in Project Properties.
|
||||
* To change this template file, choose Tools | Templates
|
||||
* and open the template in the editor.
|
||||
*/
|
||||
package org.redkalex.cache.redis;
|
||||
|
||||
import org.redkale.net.AsyncConnection;
|
||||
import org.redkale.net.WorkThread;
|
||||
import org.redkale.net.client.Client;
|
||||
import org.redkale.net.client.ClientCodec;
|
||||
import org.redkale.net.client.ClientConnection;
|
||||
|
||||
import java.util.concurrent.CompletableFuture;
|
||||
import java.util.function.Function;
|
||||
|
||||
/**
|
||||
* @author zhangjx
|
||||
*/
|
||||
public class RedisCacheConnection extends ClientConnection<RedisCacheRequest, RedisCacheResult> {
|
||||
|
||||
public RedisCacheConnection(Client client, int index, AsyncConnection channel) {
|
||||
super(client, index, channel);
|
||||
}
|
||||
|
||||
@Override
|
||||
protected ClientCodec createCodec() {
|
||||
return new RedisCacheCodec(this);
|
||||
}
|
||||
|
||||
protected CompletableFuture<RedisCacheResult> writeRequest(RedisCacheRequest request) {
|
||||
return super.writeChannel(request);
|
||||
}
|
||||
|
||||
protected <T> CompletableFuture<T> writeRequest(RedisCacheRequest request, Function<RedisCacheResult, T> respTransfer) {
|
||||
return super.writeChannel(request, respTransfer);
|
||||
}
|
||||
|
||||
public RedisCacheResult pollResultSet(RedisCacheRequest request) {
|
||||
RedisCacheResult rs = new RedisCacheResult();
|
||||
return rs;
|
||||
}
|
||||
|
||||
public RedisCacheRequest pollRequest(WorkThread workThread) {
|
||||
RedisCacheRequest rs = new RedisCacheRequest().currThread(workThread);
|
||||
return rs;
|
||||
}
|
||||
|
||||
}
|
45
src/main/java/org/redkalex/cache/redis/RedisCacheReqAuth.java
vendored
Normal file
45
src/main/java/org/redkalex/cache/redis/RedisCacheReqAuth.java
vendored
Normal file
@@ -0,0 +1,45 @@
|
||||
/*
|
||||
* To change this license header, choose License Headers in Project Properties.
|
||||
* To change this template file, choose Tools | Templates
|
||||
* and open the template in the editor.
|
||||
*/
|
||||
package org.redkalex.cache.redis;
|
||||
|
||||
import org.redkale.net.client.ClientConnection;
|
||||
import org.redkale.util.ByteArray;
|
||||
|
||||
import java.nio.charset.StandardCharsets;
|
||||
|
||||
/**
|
||||
* @author zhangjx
|
||||
*/
|
||||
public class RedisCacheReqAuth extends RedisCacheRequest {
|
||||
|
||||
private static final byte[] PS = "AUTH".getBytes(StandardCharsets.UTF_8);
|
||||
|
||||
protected String password;
|
||||
|
||||
public RedisCacheReqAuth(String password) {
|
||||
this.password = password;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void writeTo(ClientConnection conn, ByteArray writer) {
|
||||
byte[] pwd = password.getBytes();
|
||||
writer.put((byte) '*');
|
||||
writer.put((byte) '2');
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
writer.put((byte) '$');
|
||||
writer.put((byte) '4');
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
writer.put(PS);
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
|
||||
writer.put((byte) '$');
|
||||
writer.put(String.valueOf(pwd.length).getBytes(StandardCharsets.UTF_8));
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
writer.put(pwd);
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
|
||||
}
|
||||
}
|
36
src/main/java/org/redkalex/cache/redis/RedisCacheReqClose.java
vendored
Normal file
36
src/main/java/org/redkalex/cache/redis/RedisCacheReqClose.java
vendored
Normal file
@@ -0,0 +1,36 @@
|
||||
/*
|
||||
* To change this license header, choose License Headers in Project Properties.
|
||||
* To change this template file, choose Tools | Templates
|
||||
* and open the template in the editor.
|
||||
*/
|
||||
package org.redkalex.cache.redis;
|
||||
|
||||
import org.redkale.net.client.ClientConnection;
|
||||
import org.redkale.util.ByteArray;
|
||||
|
||||
import java.nio.charset.StandardCharsets;
|
||||
|
||||
/**
|
||||
* @author zhangjx
|
||||
*/
|
||||
public class RedisCacheReqClose extends RedisCacheRequest {
|
||||
|
||||
private static final byte[] PS = "QUIT".getBytes(StandardCharsets.UTF_8);
|
||||
|
||||
@Override
|
||||
public final boolean isCloseType() {
|
||||
return true;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void writeTo(ClientConnection conn, ByteArray writer) {
|
||||
writer.put((byte) '*');
|
||||
writer.put((byte) '1');
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
writer.put((byte) '$');
|
||||
writer.put((byte) '4');
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
writer.put(PS);
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
}
|
||||
}
|
43
src/main/java/org/redkalex/cache/redis/RedisCacheReqDB.java
vendored
Normal file
43
src/main/java/org/redkalex/cache/redis/RedisCacheReqDB.java
vendored
Normal file
@@ -0,0 +1,43 @@
|
||||
/*
|
||||
* To change this license header, choose License Headers in Project Properties.
|
||||
* To change this template file, choose Tools | Templates
|
||||
* and open the template in the editor.
|
||||
*/
|
||||
package org.redkalex.cache.redis;
|
||||
|
||||
import org.redkale.net.client.ClientConnection;
|
||||
import org.redkale.util.ByteArray;
|
||||
|
||||
import java.nio.charset.StandardCharsets;
|
||||
|
||||
/**
|
||||
* @author zhangjx
|
||||
*/
|
||||
public class RedisCacheReqDB extends RedisCacheRequest {
|
||||
|
||||
protected int db;
|
||||
|
||||
public RedisCacheReqDB(int db) {
|
||||
this.db = db;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void writeTo(ClientConnection conn, ByteArray writer) {
|
||||
writer.put((byte) '*');
|
||||
writer.put((byte) '2');
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
writer.put((byte) '$');
|
||||
writer.put((byte) '6');
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
writer.put("SELECT".getBytes(StandardCharsets.UTF_8));
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
|
||||
byte[] dbs = String.valueOf(db).getBytes(StandardCharsets.UTF_8);
|
||||
writer.put((byte) '$');
|
||||
writer.put(String.valueOf(dbs.length).getBytes(StandardCharsets.UTF_8));
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
writer.put(dbs);
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
|
||||
}
|
||||
}
|
31
src/main/java/org/redkalex/cache/redis/RedisCacheReqPing.java
vendored
Normal file
31
src/main/java/org/redkalex/cache/redis/RedisCacheReqPing.java
vendored
Normal file
@@ -0,0 +1,31 @@
|
||||
/*
|
||||
* To change this license header, choose License Headers in Project Properties.
|
||||
* To change this template file, choose Tools | Templates
|
||||
* and open the template in the editor.
|
||||
*/
|
||||
package org.redkalex.cache.redis;
|
||||
|
||||
import org.redkale.net.client.ClientConnection;
|
||||
import org.redkale.util.ByteArray;
|
||||
|
||||
import java.nio.charset.StandardCharsets;
|
||||
|
||||
/**
|
||||
* @author zhangjx
|
||||
*/
|
||||
public class RedisCacheReqPing extends RedisCacheRequest {
|
||||
|
||||
private static final byte[] PS = "PING".getBytes(StandardCharsets.UTF_8);
|
||||
|
||||
@Override
|
||||
public void writeTo(ClientConnection conn, ByteArray writer) {
|
||||
writer.put((byte) '*');
|
||||
writer.put((byte) '1');
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
writer.put((byte) '$');
|
||||
writer.put((byte) '4');
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
writer.put(PS);
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
}
|
||||
}
|
61
src/main/java/org/redkalex/cache/redis/RedisCacheRequest.java
vendored
Normal file
61
src/main/java/org/redkalex/cache/redis/RedisCacheRequest.java
vendored
Normal file
@@ -0,0 +1,61 @@
|
||||
/*
|
||||
* To change this license header, choose License Headers in Project Properties.
|
||||
* To change this template file, choose Tools | Templates
|
||||
* and open the template in the editor.
|
||||
*/
|
||||
package org.redkalex.cache.redis;
|
||||
|
||||
import org.redkale.net.client.ClientConnection;
|
||||
import org.redkale.net.client.ClientRequest;
|
||||
import org.redkale.util.ByteArray;
|
||||
|
||||
import java.nio.charset.StandardCharsets;
|
||||
|
||||
/**
|
||||
* @author zhangjx
|
||||
*/
|
||||
public class RedisCacheRequest extends ClientRequest {
|
||||
|
||||
static final byte[] TRUE = new byte[]{'t'};
|
||||
|
||||
static final byte[] FALSE = new byte[]{'f'};
|
||||
|
||||
protected String key;
|
||||
|
||||
protected String command;
|
||||
|
||||
protected byte[][] args;
|
||||
|
||||
public <T> RedisCacheRequest prepare(String command, String key, byte[]... args) {
|
||||
super.prepare();
|
||||
this.command = command;
|
||||
this.key = key;
|
||||
this.args = args;
|
||||
return this;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void writeTo(ClientConnection conn, ByteArray writer) {
|
||||
writer.put((byte) '*');
|
||||
writer.put(String.valueOf(args.length + 1).getBytes(StandardCharsets.UTF_8));
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
writer.put((byte) '$');
|
||||
writer.put(String.valueOf(command.length()).getBytes(StandardCharsets.UTF_8));
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
writer.put(command.getBytes(StandardCharsets.UTF_8));
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
|
||||
for (final byte[] arg : args) {
|
||||
writer.put((byte) '$');
|
||||
writer.put(String.valueOf(arg.length).getBytes(StandardCharsets.UTF_8));
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
writer.put(arg);
|
||||
writer.put((byte) '\r', (byte) '\n');
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public String toString() {
|
||||
return getClass().getSimpleName() + "{command=" + command + ", key=" + key + "}";
|
||||
}
|
||||
}
|
151
src/main/java/org/redkalex/cache/redis/RedisCacheResult.java
vendored
Normal file
151
src/main/java/org/redkalex/cache/redis/RedisCacheResult.java
vendored
Normal file
@@ -0,0 +1,151 @@
|
||||
/*
|
||||
* To change this license header, choose License Headers in Project Properties.
|
||||
* To change this template file, choose Tools | Templates
|
||||
* and open the template in the editor.
|
||||
*/
|
||||
package org.redkalex.cache.redis;
|
||||
|
||||
import org.redkale.convert.json.JsonConvert;
|
||||
|
||||
import java.lang.reflect.Type;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.util.*;
|
||||
|
||||
/**
|
||||
* @author zhangjx
|
||||
*/
|
||||
public class RedisCacheResult {
|
||||
|
||||
//+ 简单字符串类型 (不包含CRLF)
|
||||
//- 错误类型 (不包含CRLF)
|
||||
//': 整型
|
||||
//$ 块字符串
|
||||
//* 数组
|
||||
protected byte frameType;
|
||||
|
||||
protected byte[] frameValue; //(不包含CRLF)
|
||||
|
||||
protected List<byte[]> frameList; //(不包含CRLF)
|
||||
|
||||
public RedisCacheResult prepare(byte byteType, byte[] val, List<byte[]> bytesList) {
|
||||
this.frameType = byteType;
|
||||
this.frameValue = val;
|
||||
this.frameList = bytesList;
|
||||
return this;
|
||||
}
|
||||
|
||||
public Void getVoidValue() {
|
||||
return null;
|
||||
}
|
||||
|
||||
public byte[] getFrameValue() {
|
||||
return frameValue;
|
||||
}
|
||||
|
||||
public Boolean getBoolValue() {
|
||||
if (frameValue == null) {
|
||||
return false;
|
||||
}
|
||||
String val = new String(frameValue, StandardCharsets.UTF_8);
|
||||
if ("OK".equals(val)) {
|
||||
return true;
|
||||
}
|
||||
return Integer.parseInt(val) > 0;
|
||||
}
|
||||
|
||||
public String getStringValue(String key, RedisCryptor cryptor) {
|
||||
if (frameValue == null) {
|
||||
return null;
|
||||
}
|
||||
String val = new String(frameValue, StandardCharsets.UTF_8);
|
||||
if (cryptor != null) {
|
||||
val = cryptor.decrypt(key, val);
|
||||
}
|
||||
return val;
|
||||
}
|
||||
|
||||
public Double getDoubleValue(Double defvalue) {
|
||||
return frameValue == null ? defvalue : Double.parseDouble(new String(frameValue, StandardCharsets.UTF_8));
|
||||
}
|
||||
|
||||
public Long getLongValue(Long defvalue) {
|
||||
return frameValue == null ? defvalue : Long.parseLong(new String(frameValue, StandardCharsets.UTF_8));
|
||||
}
|
||||
|
||||
public Integer getIntValue(Integer defvalue) {
|
||||
return frameValue == null ? defvalue : Integer.parseInt(new String(frameValue, StandardCharsets.UTF_8));
|
||||
}
|
||||
|
||||
public <T> T getObjectValue(String key, RedisCryptor cryptor, Type type) {
|
||||
return formatValue(key, cryptor, frameValue, type);
|
||||
}
|
||||
|
||||
protected <T> Set<T> getSetValue(String key, RedisCryptor cryptor, Type type) {
|
||||
if (frameList == null || frameList.isEmpty()) {
|
||||
return new LinkedHashSet<>();
|
||||
}
|
||||
Set<T> set = new LinkedHashSet<>();
|
||||
for (byte[] bs : frameList) {
|
||||
set.add(formatValue(key, cryptor, bs, type));
|
||||
}
|
||||
return set;
|
||||
}
|
||||
|
||||
protected <T> List<T> getListValue(String key, RedisCryptor cryptor, Type type) {
|
||||
if (frameList == null || frameList.isEmpty()) {
|
||||
return new ArrayList<>();
|
||||
}
|
||||
List<T> list = new ArrayList<>();
|
||||
for (byte[] bs : frameList) {
|
||||
list.add(formatValue(key, cryptor, bs, type));
|
||||
}
|
||||
return list;
|
||||
}
|
||||
|
||||
protected <T> Map<String, T> getMapValue(String key, RedisCryptor cryptor, Type type) {
|
||||
if (frameList == null || frameList.isEmpty()) {
|
||||
return new LinkedHashMap<>();
|
||||
}
|
||||
Map<String, T> map = new LinkedHashMap<>();
|
||||
for (int i = 0; i < frameList.size(); i += 2) {
|
||||
byte[] bs1 = frameList.get(i);
|
||||
byte[] bs2 = frameList.get(i + 1);
|
||||
T val = formatValue(key, cryptor, bs2, type);
|
||||
if (val != null) {
|
||||
map.put(formatValue(key, cryptor, bs1, String.class).toString(), val);
|
||||
}
|
||||
}
|
||||
return map;
|
||||
}
|
||||
|
||||
protected static <T> T formatValue(String key, RedisCryptor cryptor, byte[] frames, Type type) {
|
||||
if (frames == null) {
|
||||
return null;
|
||||
}
|
||||
if (type == byte[].class) {
|
||||
return (T) frames;
|
||||
}
|
||||
if (type == String.class) {
|
||||
String val = new String(frames, StandardCharsets.UTF_8);
|
||||
if (cryptor != null) {
|
||||
val = cryptor.decrypt(key, val);
|
||||
}
|
||||
return (T) val;
|
||||
}
|
||||
if (type == boolean.class || type == Boolean.class) {
|
||||
return (T) (Boolean) "t".equalsIgnoreCase(new String(frames, StandardCharsets.UTF_8));
|
||||
}
|
||||
if (type == long.class || type == Long.class) {
|
||||
return (T) (Long) Long.parseLong(new String(frames, StandardCharsets.UTF_8));
|
||||
}
|
||||
if (type == double.class || type == Double.class) {
|
||||
return (T) (Double) Double.parseDouble(new String(frames, StandardCharsets.UTF_8));
|
||||
}
|
||||
if (cryptor != null) {
|
||||
String val = cryptor.decrypt(key, new String(frames, StandardCharsets.UTF_8));
|
||||
return (T) JsonConvert.root().convertFrom(type, val);
|
||||
}
|
||||
return (T) JsonConvert.root().convertFrom(type, frames);
|
||||
}
|
||||
|
||||
}
|
1801
src/main/java/org/redkalex/cache/redis/RedisCacheSource.java
vendored
Normal file
1801
src/main/java/org/redkalex/cache/redis/RedisCacheSource.java
vendored
Normal file
File diff suppressed because it is too large
Load Diff
29
src/main/java/org/redkalex/cache/redis/RedisCacheSourceProvider.java
vendored
Normal file
29
src/main/java/org/redkalex/cache/redis/RedisCacheSourceProvider.java
vendored
Normal file
@@ -0,0 +1,29 @@
|
||||
/*
|
||||
* To change this license header, choose License Headers in Project Properties.
|
||||
* To change this template file, choose Tools | Templates
|
||||
* and open the template in the editor.
|
||||
*/
|
||||
package org.redkalex.cache.redis;
|
||||
|
||||
import org.redkale.annotation.Priority;
|
||||
import org.redkale.source.CacheSource;
|
||||
import org.redkale.source.CacheSourceProvider;
|
||||
import org.redkale.util.AnyValue;
|
||||
|
||||
/**
|
||||
* @author zhangjx
|
||||
*/
|
||||
@Priority(1)
|
||||
public class RedisCacheSourceProvider implements CacheSourceProvider {
|
||||
|
||||
@Override
|
||||
public boolean acceptsConf(AnyValue config) {
|
||||
return new MyRedisCacheSource().acceptsConf(config);
|
||||
}
|
||||
|
||||
@Override
|
||||
public CacheSource createInstance() {
|
||||
return new MyRedisCacheSource();
|
||||
}
|
||||
|
||||
}
|
46
src/main/java/org/redkalex/cache/redis/RedisCryptor.java
vendored
Normal file
46
src/main/java/org/redkalex/cache/redis/RedisCryptor.java
vendored
Normal file
@@ -0,0 +1,46 @@
|
||||
/*
|
||||
* Click nbfs://nbhost/SystemFileSystem/Templates/Licenses/license-default.txt to change this license
|
||||
* Click nbfs://nbhost/SystemFileSystem/Templates/Classes/Class.java to edit this template
|
||||
*/
|
||||
package org.redkalex.cache.redis;
|
||||
|
||||
import org.redkale.util.AnyValue;
|
||||
|
||||
/**
|
||||
* @author zhangjx
|
||||
*/
|
||||
public interface RedisCryptor {
|
||||
|
||||
/**
|
||||
* 初始化
|
||||
*
|
||||
* @param conf 配置
|
||||
*/
|
||||
public void init(AnyValue conf);
|
||||
|
||||
/**
|
||||
* 加密, 无需加密的key对应的值需要直接返回value
|
||||
*
|
||||
* @param key key
|
||||
* @param value 明文
|
||||
* @return 密文
|
||||
*/
|
||||
public String encrypt(String key, String value);
|
||||
|
||||
/**
|
||||
* 解密, 无需解密的key对应的值需要直接返回value
|
||||
*
|
||||
* @param key key
|
||||
* @param value 密文
|
||||
* @return 明文
|
||||
*/
|
||||
public String decrypt(String key, String value);
|
||||
|
||||
/**
|
||||
* 销毁
|
||||
*
|
||||
* @param conf 配置
|
||||
*/
|
||||
public void destroy(AnyValue conf);
|
||||
|
||||
}
|
Reference in New Issue
Block a user