Compare commits
56 Commits
master
...
dev/spring
Author | SHA1 | Date | |
---|---|---|---|
c2bcc00741 | |||
d28a8e4d7b | |||
e2c2b62665 | |||
e8b9140103 | |||
1786724c88 | |||
f4771aadf2 | |||
8e2779f2d8 | |||
094d3fc5a2 | |||
9a3314e0e6 | |||
99cb1ea3c8 | |||
46082b5461 | |||
35e4bebb6d | |||
144b99c3d2 | |||
2296a7fb15 | |||
90378c5daf | |||
0df9e254e6 | |||
3debcee7d2 | |||
6995d103f0 | |||
ab7e9536a7 | |||
c135ef8925 | |||
a16e448438 | |||
92ef169b6d | |||
89d474284c | |||
257e840e7e | |||
0bb543d1a9 | |||
a812e2d78d | |||
3f2db60432 | |||
38c023a503 | |||
f7ef96d0f6 | |||
fc1f91af04 | |||
c6940cef2b | |||
ee7154990b | |||
3d0d2f8f81 | |||
895df0e05a | |||
57c216c791 | |||
c4639fcabf | |||
7acacc2ce3 | |||
80fe44d5d7 | |||
9652c60dcd | |||
e5b5c66835 | |||
3c1c38207f | |||
f6c56c62df | |||
91e4e48aa2 | |||
361ef58d98 | |||
69be5ec3d7 | |||
d83dcd37c2 | |||
f4089d4118 | |||
d70d9ad244 | |||
c457e8094d | |||
26a1fc4971 | |||
4d5fb83b3c | |||
5dd58d1fab | |||
a37f1b9564 | |||
10472078b0 | |||
9138790fd2 | |||
1603140c1c |
@ -1,20 +0,0 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
|
||||
<application port="2001">
|
||||
|
||||
<resources>
|
||||
<properties load="redis.properties"></properties>
|
||||
</resources>
|
||||
|
||||
<server protocol="HTTP" port="80">
|
||||
<request>
|
||||
<remoteaddr value="request.headers.X-Real-IP"/>
|
||||
</request>
|
||||
|
||||
<rest autoload="true" path=""/>
|
||||
|
||||
<services autoload="true"/>
|
||||
<servlets path="/platf"/>
|
||||
</server>
|
||||
|
||||
</application>
|
@ -1,19 +0,0 @@
|
||||
# Producer
|
||||
#bootstrap.servers=47.111.150.118:6062
|
||||
#bootstrap.servers=121.196.17.55:6062
|
||||
bootstrap.servers=39.108.56.246:9092
|
||||
#bootstrap.servers=122.112.180.156:6062
|
||||
acks=all
|
||||
retries=0
|
||||
batch.size=16384
|
||||
linger.ms=1
|
||||
buffer.memory=33554432
|
||||
key.serializer=org.apache.kafka.common.serialization.StringSerializer
|
||||
value.serializer=org.apache.kafka.common.serialization.StringSerializer
|
||||
|
||||
# Consumer
|
||||
enable.auto.commit=true
|
||||
auto.commit.interval.ms=1000
|
||||
group.id=
|
||||
key.deserializer=org.apache.kafka.common.serialization.StringDeserializer
|
||||
value.deserializer=org.apache.kafka.common.serialization.StringDeserializer
|
@ -1,4 +0,0 @@
|
||||
# redis
|
||||
redis.host=47.111.150.118
|
||||
redis.password=*Zhong9307!
|
||||
redis.port=6064
|
55
pom.xml
Normal file
55
pom.xml
Normal file
@ -0,0 +1,55 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project xmlns="http://maven.apache.org/POM/4.0.0"
|
||||
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||
<modelVersion>4.0.0</modelVersion>
|
||||
|
||||
<groupId>net.tccn</groupId>
|
||||
<artifactId>zhub-client-spring</artifactId>
|
||||
<version>0.1.2</version>
|
||||
|
||||
<properties>
|
||||
<maven.compiler.source>1.8</maven.compiler.source>
|
||||
<maven.compiler.target>1.8</maven.compiler.target>
|
||||
</properties>
|
||||
|
||||
<dependencies>
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-web</artifactId>
|
||||
<version>3.2.1</version>
|
||||
<scope>compile</scope>
|
||||
</dependency>
|
||||
|
||||
<dependency>
|
||||
<groupId>com.google.code.gson</groupId>
|
||||
<artifactId>gson</artifactId>
|
||||
<version>2.10.1</version>
|
||||
<scope>compile</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>junit</groupId>
|
||||
<artifactId>junit</artifactId>
|
||||
<version>4.13.1</version>
|
||||
<scope>test</scope>
|
||||
</dependency>
|
||||
</dependencies>
|
||||
|
||||
<repositories>
|
||||
<repository>
|
||||
<id>maven-nexus</id>
|
||||
<name>maven-nexus</name>
|
||||
<url>https://nexus.1216.top/repository/maven-public/</url>
|
||||
</repository>
|
||||
</repositories>
|
||||
|
||||
|
||||
<distributionManagement>
|
||||
<repository>
|
||||
<id>mvn-release</id>
|
||||
<name>mvn-release</name>
|
||||
<url>https://nexus.1216.top/repository/maven-releases/</url>
|
||||
</repository>
|
||||
</distributionManagement>
|
||||
|
||||
</project>
|
@ -1,55 +0,0 @@
|
||||
package com.zdemo;
|
||||
|
||||
import org.redkale.convert.json.JsonConvert;
|
||||
|
||||
import java.util.Collection;
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* @author Liang
|
||||
* @data 2020-09-05 23:18
|
||||
*/
|
||||
public abstract class AbstractConsumer implements IConsumer {
|
||||
|
||||
public final Map<String, EventType> eventMap = new HashMap<>();
|
||||
|
||||
public abstract String getGroupid();
|
||||
|
||||
public boolean preInit() {
|
||||
return true;
|
||||
}
|
||||
|
||||
public void addEventType(EventType... eventType) {
|
||||
for (EventType type : eventType) {
|
||||
String[] topics = type.topic.split(",");
|
||||
for (String topic : topics) {
|
||||
if (topic.isEmpty()) {
|
||||
continue;
|
||||
}
|
||||
eventMap.put(topic, type);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public final Collection<String> getSubscribes() {
|
||||
return eventMap.keySet();
|
||||
}
|
||||
|
||||
@Override
|
||||
public final 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 = JsonConvert.root().convertFrom(eventType.typeToken.getType(), value);
|
||||
}
|
||||
|
||||
eventType.accept(data);
|
||||
}
|
||||
|
||||
|
||||
}
|
@ -1,19 +0,0 @@
|
||||
package com.zdemo;
|
||||
|
||||
import org.redkale.util.TypeToken;
|
||||
|
||||
import java.util.Collection;
|
||||
import java.util.logging.Logger;
|
||||
|
||||
public interface IConsumer<T extends Event> {
|
||||
TypeToken<String> TYPE_TOKEN_STRING = new TypeToken<String>() {
|
||||
};
|
||||
TypeToken<Integer> TYPE_TOKEN_INT = new TypeToken<Integer>() {
|
||||
};
|
||||
|
||||
Logger logger = Logger.getLogger(IConsumer.class.getSimpleName());
|
||||
|
||||
Collection<String> getSubscribes();
|
||||
|
||||
<T> void accept(String topic, String record);
|
||||
}
|
@ -1,101 +0,0 @@
|
||||
package com.zdemo.kafak;
|
||||
|
||||
import com.zdemo.AbstractConsumer;
|
||||
import com.zdemo.EventType;
|
||||
import com.zdemo.IConsumer;
|
||||
import org.apache.kafka.clients.consumer.ConsumerRecords;
|
||||
import org.apache.kafka.clients.consumer.KafkaConsumer;
|
||||
import org.apache.kafka.common.errors.WakeupException;
|
||||
import org.redkale.net.http.RestService;
|
||||
import org.redkale.service.Service;
|
||||
import org.redkale.util.AnyValue;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
import java.io.File;
|
||||
import java.io.FileInputStream;
|
||||
import java.io.IOException;
|
||||
import java.time.Duration;
|
||||
import java.util.Properties;
|
||||
import java.util.concurrent.LinkedBlockingQueue;
|
||||
import java.util.logging.Level;
|
||||
|
||||
import static java.util.Arrays.asList;
|
||||
|
||||
/**
|
||||
* 消费
|
||||
*/
|
||||
@RestService
|
||||
public abstract class KafakConsumer extends AbstractConsumer implements IConsumer, Service {
|
||||
|
||||
@Resource(name = "APP_HOME")
|
||||
protected File APP_HOME;
|
||||
|
||||
protected Properties props;
|
||||
|
||||
// 0:none 1:restart -1:stop
|
||||
//private int cmd = -1;
|
||||
|
||||
public abstract String getGroupid();
|
||||
|
||||
private final LinkedBlockingQueue<EventType> queue = new LinkedBlockingQueue<>();
|
||||
|
||||
@Override
|
||||
public void addEventType(EventType... eventTypes) {
|
||||
super.addEventType(eventTypes);
|
||||
try {
|
||||
for (EventType eventType : eventTypes) {
|
||||
queue.put(eventType);
|
||||
}
|
||||
} catch (InterruptedException e) {
|
||||
logger.log(Level.WARNING, "", e);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public final void init(AnyValue config) {
|
||||
if (!preInit()) {
|
||||
return;
|
||||
}
|
||||
try (FileInputStream fis = new FileInputStream(new File(APP_HOME, "conf/kafak.properties"));) {
|
||||
props = new Properties();
|
||||
props.load(fis);
|
||||
|
||||
if (logger.isLoggable(Level.INFO)) logger.info(getGroupid() + " consumer started!");
|
||||
|
||||
new Thread(() -> {
|
||||
try {
|
||||
props.put("group.id", getGroupid());
|
||||
KafkaConsumer<String, String> consumer = new KafkaConsumer<>(props);
|
||||
consumer.subscribe(asList("_"));
|
||||
while (true) {
|
||||
ConsumerRecords<String, String> records = consumer.poll(Duration.ofMillis(1_000));
|
||||
records.forEach(record -> {
|
||||
String topic = record.topic();
|
||||
long offset = record.offset();
|
||||
String value = record.value();
|
||||
try {
|
||||
accept(topic, value);
|
||||
} catch (Exception e) {
|
||||
logger.log(Level.WARNING, String.format("topic[%s] event accept error, offset=%s,value:%s", topic, offset, value), e);
|
||||
}
|
||||
});
|
||||
|
||||
// 动态新增订阅
|
||||
while (!queue.isEmpty()) {
|
||||
queue.clear();
|
||||
consumer.unsubscribe();
|
||||
consumer.subscribe(getSubscribes());
|
||||
}
|
||||
}
|
||||
} catch (WakeupException ex) {
|
||||
System.out.println("WakeupException !!!!");
|
||||
}
|
||||
|
||||
}, "thread-consumer-[" + getGroupid() + "]").start();
|
||||
} catch (IOException e) {
|
||||
logger.log(Level.WARNING, "", e);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
}
|
@ -1,56 +0,0 @@
|
||||
package com.zdemo.kafak;
|
||||
|
||||
import com.zdemo.Event;
|
||||
import com.zdemo.IProducer;
|
||||
import org.apache.kafka.clients.producer.KafkaProducer;
|
||||
import org.apache.kafka.clients.producer.ProducerRecord;
|
||||
import org.redkale.convert.json.JsonConvert;
|
||||
import org.redkale.net.http.RestService;
|
||||
import org.redkale.service.Service;
|
||||
import org.redkale.util.AnyValue;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
import java.io.File;
|
||||
import java.io.FileInputStream;
|
||||
import java.io.IOException;
|
||||
import java.util.Properties;
|
||||
import java.util.logging.Level;
|
||||
|
||||
/**
|
||||
* 生产
|
||||
*
|
||||
* @param <T>
|
||||
*/
|
||||
@RestService
|
||||
public class KafakProducer<T extends Event> implements IProducer<T>, Service {
|
||||
private KafkaProducer<String, String> producer;
|
||||
|
||||
@Resource(name = "APP_HOME")
|
||||
protected File APP_HOME;
|
||||
|
||||
@Override
|
||||
public void init(AnyValue config) {
|
||||
File file = new File(APP_HOME, "conf/kafak.properties");
|
||||
try (FileInputStream fis = new FileInputStream(file)) {
|
||||
Properties props = new Properties();
|
||||
props.load(fis);
|
||||
producer = new KafkaProducer(props);
|
||||
} catch (IOException e) {
|
||||
logger.log(Level.WARNING, "未初始化kafak 生产者,kafak发布消息不可用", e);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void send(T t) {
|
||||
String v = JsonConvert.root().convertTo(t.value);
|
||||
if (v.startsWith("\"") && v.endsWith("\"")) {
|
||||
v = v.substring(1, v.length() - 1);
|
||||
}
|
||||
producer.send(new ProducerRecord(t.topic, v));
|
||||
}
|
||||
|
||||
@Override
|
||||
public void destroy(AnyValue config) {
|
||||
producer.close();
|
||||
}
|
||||
}
|
@ -1,104 +0,0 @@
|
||||
package com.zdemo.redis;
|
||||
|
||||
import com.zdemo.AbstractConsumer;
|
||||
import com.zdemo.EventType;
|
||||
import com.zdemo.IConsumer;
|
||||
import org.redkale.service.Service;
|
||||
import org.redkale.util.AnyValue;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
import java.io.BufferedReader;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStreamReader;
|
||||
import java.io.OutputStreamWriter;
|
||||
import java.net.InetSocketAddress;
|
||||
import java.net.Socket;
|
||||
import java.util.logging.Level;
|
||||
|
||||
public abstract class RedisConsumer extends AbstractConsumer implements IConsumer, Service {
|
||||
|
||||
@Resource(name = "property.redis.host")
|
||||
private String host = "127.0.0.1";
|
||||
@Resource(name = "property.redis.password")
|
||||
private String password = "";
|
||||
@Resource(name = "property.redis.port")
|
||||
private int port = 6379;
|
||||
|
||||
private Socket client;
|
||||
private OutputStreamWriter writer;
|
||||
private BufferedReader reader;
|
||||
|
||||
@Override
|
||||
public void init(AnyValue config) {
|
||||
try {
|
||||
client = new Socket();
|
||||
client.connect(new InetSocketAddress(host, port));
|
||||
client.setKeepAlive(true);
|
||||
|
||||
writer = new OutputStreamWriter(client.getOutputStream());
|
||||
writer.write("AUTH " + password + "\r\n");
|
||||
writer.flush();
|
||||
|
||||
StringBuffer buf = new StringBuffer("SUBSCRIBE");
|
||||
for (String topic : getSubscribes()) {
|
||||
buf.append(" ").append(topic);
|
||||
}
|
||||
buf.append(" _\r\n");
|
||||
writer.write(buf.toString());
|
||||
writer.flush();
|
||||
|
||||
reader = new BufferedReader(new InputStreamReader(client.getInputStream()));
|
||||
} catch (IOException e) {
|
||||
logger.log(Level.WARNING, "Redis Consumer 初始化失败!", e);
|
||||
}
|
||||
|
||||
new Thread(() -> {
|
||||
try {
|
||||
while (true) {
|
||||
String readLine = reader.readLine();
|
||||
String type = "";
|
||||
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
|
||||
|
||||
reader.readLine(); //$n len(value)
|
||||
String value = reader.readLine(); // value
|
||||
try {
|
||||
accept(topic, value);
|
||||
} catch (Exception e) {
|
||||
logger.log(Level.WARNING, "topic[" + topic + "] event accept error :" + value, e);
|
||||
}
|
||||
}
|
||||
}
|
||||
} catch (IOException e) {
|
||||
logger.log(Level.WARNING, "", e);
|
||||
}
|
||||
}).start();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void addEventType(EventType... eventType) {
|
||||
for (EventType type : eventType) {
|
||||
String[] topics = type.topic.split(",");
|
||||
for (String topic : topics) {
|
||||
if (topic.isEmpty()) {
|
||||
continue;
|
||||
}
|
||||
eventMap.put(topic, type);
|
||||
|
||||
//新增订阅
|
||||
try {
|
||||
writer.write("SUBSCRIBE " + topic + "\r\n");
|
||||
writer.flush();
|
||||
} catch (IOException e) {
|
||||
logger.log(Level.WARNING, "", e);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
@ -1,52 +0,0 @@
|
||||
package com.zdemo.redis;
|
||||
|
||||
import com.zdemo.Event;
|
||||
import com.zdemo.IProducer;
|
||||
import org.redkale.convert.json.JsonConvert;
|
||||
import org.redkale.service.Service;
|
||||
import org.redkale.util.AnyValue;
|
||||
|
||||
import javax.annotation.Resource;
|
||||
import java.io.IOException;
|
||||
import java.io.OutputStreamWriter;
|
||||
import java.net.InetSocketAddress;
|
||||
import java.net.Socket;
|
||||
import java.util.logging.Level;
|
||||
|
||||
public class RedisProducer<T extends Event> implements IProducer<T>, Service {
|
||||
|
||||
@Resource(name = "property.redis.host")
|
||||
private String host = "127.0.0.1";
|
||||
@Resource(name = "property.redis.password")
|
||||
private String password = "";
|
||||
@Resource(name = "property.redis.port")
|
||||
private int port = 6379;
|
||||
|
||||
private OutputStreamWriter osw;
|
||||
|
||||
@Override
|
||||
public void init(AnyValue config) {
|
||||
try {
|
||||
Socket client = new Socket();
|
||||
client.connect(new InetSocketAddress(host, port));
|
||||
client.setKeepAlive(true);
|
||||
|
||||
osw = new OutputStreamWriter(client.getOutputStream());
|
||||
osw.write("AUTH " + password + "\r\n");
|
||||
osw.flush();
|
||||
} catch (IOException e) {
|
||||
logger.log(Level.WARNING, "", e);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void send(T t) {
|
||||
try {
|
||||
osw.write("PUBLISH " + t.topic + " '" + JsonConvert.root().convertTo(t.value) + "' \r\n");
|
||||
osw.flush();
|
||||
} catch (IOException e) {
|
||||
logger.log(Level.WARNING, "", e);
|
||||
|
||||
}
|
||||
}
|
||||
}
|
79
src/main/java/tccn/AbstractConsumer.java
Normal file
79
src/main/java/tccn/AbstractConsumer.java
Normal file
@ -0,0 +1,79 @@
|
||||
package tccn;
|
||||
|
||||
import com.google.gson.Gson;
|
||||
import com.google.gson.reflect.TypeToken;
|
||||
import tccn.zhub.Rpc;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.HashSet;
|
||||
import java.util.Map;
|
||||
import java.util.Set;
|
||||
import java.util.function.Consumer;
|
||||
|
||||
/**
|
||||
* @author Liang
|
||||
* @data 2020-09-05 23:18
|
||||
*/
|
||||
public abstract class AbstractConsumer implements IConsumer {
|
||||
|
||||
public Gson gson = Rpc.gson;
|
||||
|
||||
private Map<String, EventType> eventMap = new HashMap<>();
|
||||
|
||||
protected abstract String getGroupid();
|
||||
|
||||
protected boolean preInit() {
|
||||
return true;
|
||||
}
|
||||
|
||||
protected final Set<String> getTopics() {
|
||||
if (!eventMap.isEmpty()) {
|
||||
return eventMap.keySet();
|
||||
}
|
||||
HashSet<String> set = new HashSet<>();
|
||||
set.add("-");
|
||||
|
||||
return set;
|
||||
}
|
||||
|
||||
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 = gson.fromJson(value, eventType.typeToken.getType());
|
||||
}
|
||||
|
||||
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, TYPE_TOKEN_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);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
@ -1,4 +1,4 @@
|
||||
package com.zdemo;
|
||||
package tccn;
|
||||
|
||||
/**
|
||||
* 发布订阅 事件
|
||||
@ -15,8 +15,8 @@ public class Event<V> {
|
||||
this.value = value;
|
||||
}
|
||||
|
||||
public static <V> Event of(String topic, V value) {
|
||||
return new Event<V>(topic, value);
|
||||
public static <V> Event<V> of(String topic, V value) {
|
||||
return new Event<>(topic, value);
|
||||
}
|
||||
|
||||
|
@ -1,6 +1,7 @@
|
||||
package com.zdemo;
|
||||
package tccn;
|
||||
|
||||
import org.redkale.util.TypeToken;
|
||||
|
||||
import com.google.gson.reflect.TypeToken;
|
||||
|
||||
import java.util.function.Consumer;
|
||||
|
||||
@ -9,7 +10,7 @@ public class EventType<T> {
|
||||
public final TypeToken<T> typeToken;
|
||||
private final Consumer<T> consumer;
|
||||
|
||||
private final static TypeToken<String> stringToken = new TypeToken<>() {
|
||||
private final static TypeToken<String> stringToken = new TypeToken<String>() {
|
||||
};
|
||||
|
||||
private EventType(String topic, TypeToken<T> typeToken, Consumer<T> consumer) {
|
37
src/main/java/tccn/IConsumer.java
Normal file
37
src/main/java/tccn/IConsumer.java
Normal file
@ -0,0 +1,37 @@
|
||||
package tccn;
|
||||
|
||||
import com.google.gson.reflect.TypeToken;
|
||||
|
||||
import java.util.function.Consumer;
|
||||
|
||||
public interface IConsumer {
|
||||
TypeToken<String> TYPE_TOKEN_STRING = new TypeToken<String>() {
|
||||
};
|
||||
TypeToken<Integer> TYPE_TOKEN_INT = new TypeToken<Integer>() {
|
||||
};
|
||||
|
||||
/**
|
||||
* 取消订阅
|
||||
*
|
||||
* @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);
|
||||
}
|
@ -1,10 +1,9 @@
|
||||
package com.zdemo;
|
||||
package tccn;
|
||||
|
||||
import java.util.logging.Logger;
|
||||
|
||||
public interface IProducer<T extends Event> {
|
||||
public interface IProducer {
|
||||
Logger logger = Logger.getLogger(IProducer.class.getSimpleName());
|
||||
|
||||
void send(T t);
|
||||
|
||||
boolean publish(String topic, Object v);
|
||||
}
|
21
src/main/java/tccn/IType.java
Normal file
21
src/main/java/tccn/IType.java
Normal file
@ -0,0 +1,21 @@
|
||||
package tccn;
|
||||
|
||||
import com.google.gson.reflect.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<Map<String, String>> MAP = new TypeToken<Map<String, String>>() {
|
||||
};
|
||||
|
||||
TypeToken<List<Map<String, String>>> LMAP = new TypeToken<List<Map<String, String>>>() {
|
||||
};
|
||||
}
|
66
src/main/java/tccn/timer/TimerExecutor.java
Normal file
66
src/main/java/tccn/timer/TimerExecutor.java
Normal file
@ -0,0 +1,66 @@
|
||||
package tccn.timer;
|
||||
|
||||
import tccn.timer.queue.TimerQueue;
|
||||
import 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/tccn/timer/TimerTask.java
Normal file
108
src/main/java/tccn/timer/TimerTask.java
Normal file
@ -0,0 +1,108 @@
|
||||
package tccn.timer;
|
||||
|
||||
import tccn.timer.scheduled.Scheduled;
|
||||
import tccn.timer.task.Job;
|
||||
import 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);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
40
src/main/java/tccn/timer/Timers.java
Normal file
40
src/main/java/tccn/timer/Timers.java
Normal file
@ -0,0 +1,40 @@
|
||||
package tccn.timer;
|
||||
|
||||
import tccn.timer.scheduled.ScheduledCycle;
|
||||
|
||||
import java.util.UUID;
|
||||
import java.util.function.Supplier;
|
||||
|
||||
public class Timers {
|
||||
|
||||
private static final TimerExecutor timerExecutor = new TimerExecutor(1);
|
||||
|
||||
/**
|
||||
* 本地延时重试
|
||||
*/
|
||||
public static void tryDelay(Supplier<Boolean> supplier, long millis, int maxCount) {
|
||||
timerExecutor.add(TimerTask.by("try-delay-task-" + UUID.randomUUID().toString().replaceAll("-", ""), ScheduledCycle.of(0), task -> {
|
||||
if (supplier.get() || task.getExecCount() == maxCount) {
|
||||
task.setComplete(true);
|
||||
}
|
||||
|
||||
if (task.getExecCount() == 1) {
|
||||
task.setScheduled(ScheduledCycle.of(millis));
|
||||
}
|
||||
}));
|
||||
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
* 本地延时:延时时间极短的场景下使用 (如:1分钟内)
|
||||
*/
|
||||
public static void delay(Runnable runnable, long millis) {
|
||||
timerExecutor.add(TimerTask.by("delay-task-" + UUID.randomUUID().toString().replaceAll("-", ""), ScheduledCycle.of(millis), task -> {
|
||||
runnable.run();
|
||||
task.setComplete(true);
|
||||
}));
|
||||
}
|
||||
|
||||
|
||||
}
|
105
src/main/java/tccn/timer/queue/TimerQueue.java
Normal file
105
src/main/java/tccn/timer/queue/TimerQueue.java
Normal file
@ -0,0 +1,105 @@
|
||||
package tccn.timer.queue;
|
||||
|
||||
import 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/tccn/timer/scheduled/Scheduled.java
Normal file
23
src/main/java/tccn/timer/scheduled/Scheduled.java
Normal file
@ -0,0 +1,23 @@
|
||||
package 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/tccn/timer/scheduled/ScheduledCycle.java
Normal file
79
src/main/java/tccn/timer/scheduled/ScheduledCycle.java
Normal file
@ -0,0 +1,79 @@
|
||||
package 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/tccn/timer/scheduled/ScheduledExpres.java
Normal file
499
src/main/java/tccn/timer/scheduled/ScheduledExpres.java
Normal file
@ -0,0 +1,499 @@
|
||||
package 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/tccn/timer/task/Job.java
Normal file
14
src/main/java/tccn/timer/task/Job.java
Normal file
@ -0,0 +1,14 @@
|
||||
package tccn.timer.task;
|
||||
|
||||
/**
|
||||
* @author: liangxianyou at 2018/12/8 17:24.
|
||||
*/
|
||||
@FunctionalInterface
|
||||
public interface Job {
|
||||
|
||||
/**
|
||||
* 任务执行的内容
|
||||
*/
|
||||
void execute(Task task);
|
||||
|
||||
}
|
70
src/main/java/tccn/timer/task/Task.java
Normal file
70
src/main/java/tccn/timer/task/Task.java
Normal file
@ -0,0 +1,70 @@
|
||||
package tccn.timer.task;
|
||||
|
||||
import tccn.timer.TimerExecutor;
|
||||
import 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();
|
||||
}
|
25
src/main/java/tccn/zhub/Lock.java
Normal file
25
src/main/java/tccn/zhub/Lock.java
Normal file
@ -0,0 +1,25 @@
|
||||
package tccn.zhub;
|
||||
|
||||
// ================================================== lock ==================================================
|
||||
public class Lock {
|
||||
protected String name;
|
||||
protected String uuid;
|
||||
protected int duration;
|
||||
protected boolean success;
|
||||
protected 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);
|
||||
}
|
||||
|
||||
public boolean success() {
|
||||
return success;
|
||||
}
|
||||
}
|
93
src/main/java/tccn/zhub/Rpc.java
Normal file
93
src/main/java/tccn/zhub/Rpc.java
Normal file
@ -0,0 +1,93 @@
|
||||
package tccn.zhub;
|
||||
|
||||
import com.google.gson.Gson;
|
||||
import com.google.gson.annotations.Expose;
|
||||
|
||||
public class Rpc<T> {
|
||||
/*public final static Gson gson = new GsonBuilder()
|
||||
.excludeFieldsWithoutExposeAnnotation()
|
||||
.create();*/
|
||||
public final static Gson gson = new Gson();
|
||||
|
||||
private String ruk; // request unique key:
|
||||
private String topic; // call topic
|
||||
private T value; // call paras
|
||||
|
||||
@Expose(deserialize = false, serialize = false)
|
||||
private RpcResult rpcResult;
|
||||
|
||||
public Rpc() {
|
||||
}
|
||||
|
||||
public Rpc(String appname, String ruk, String topic, Object value) {
|
||||
this.ruk = appname + "::" + ruk;
|
||||
this.topic = topic;
|
||||
this.value = (T) gson.toJson(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;
|
||||
}
|
||||
|
||||
|
||||
public RpcResult getRpcResult() {
|
||||
return rpcResult;
|
||||
}
|
||||
|
||||
public void setRpcResult(RpcResult rpcResult) {
|
||||
this.rpcResult = rpcResult;
|
||||
}
|
||||
|
||||
public String getBackTopic() {
|
||||
return ruk.split("::")[0];
|
||||
}
|
||||
|
||||
public <R> RpcResult<R> render() {
|
||||
RpcResult<R> response = new RpcResult<>();
|
||||
response.setRuk(ruk);
|
||||
return response;
|
||||
}
|
||||
|
||||
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/tccn/zhub/RpcResult.java
Normal file
40
src/main/java/tccn/zhub/RpcResult.java
Normal file
@ -0,0 +1,40 @@
|
||||
package 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;
|
||||
}
|
||||
}
|
706
src/main/java/tccn/zhub/ZHubClient.java
Normal file
706
src/main/java/tccn/zhub/ZHubClient.java
Normal file
@ -0,0 +1,706 @@
|
||||
package tccn.zhub;
|
||||
|
||||
import com.google.gson.reflect.TypeToken;
|
||||
import org.springframework.beans.factory.annotation.Value;
|
||||
import org.springframework.stereotype.Component;
|
||||
import tccn.AbstractConsumer;
|
||||
import tccn.Event;
|
||||
import tccn.IConsumer;
|
||||
import tccn.IProducer;
|
||||
import tccn.timer.Timers;
|
||||
|
||||
import javax.annotation.PostConstruct;
|
||||
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.UUID;
|
||||
import java.util.concurrent.*;
|
||||
import java.util.function.Consumer;
|
||||
import java.util.function.Function;
|
||||
import java.util.logging.Level;
|
||||
import java.util.logging.Logger;
|
||||
|
||||
import static java.nio.charset.StandardCharsets.UTF_8;
|
||||
|
||||
@Component
|
||||
public class ZHubClient extends AbstractConsumer implements IConsumer, IProducer {
|
||||
|
||||
public Logger logger = Logger.getLogger(ZHubClient.class.getSimpleName());
|
||||
@Value("${zhub.addr}")
|
||||
private String addr = "127.0.0.1:1216";
|
||||
|
||||
@Value("${zhub.groupid}")
|
||||
private String groupid = "";
|
||||
|
||||
@Value("${zhub.auth}")
|
||||
private String auth = "";
|
||||
|
||||
@Value("${zhub.appid}")
|
||||
protected String appid = "";
|
||||
|
||||
@PostConstruct
|
||||
public void init() {
|
||||
init(null);
|
||||
}
|
||||
|
||||
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 static boolean isFirst = true;
|
||||
private boolean isMain = false;*/
|
||||
private static final Map<String, ZHubClient> mainHub = new HashMap<>(); // 127.0.0.1:1216 - ZHubClient
|
||||
|
||||
public ZHubClient(String addr, String groupid, String appid, String auth) {
|
||||
this.addr = addr;
|
||||
this.groupid = groupid;
|
||||
this.appid = appid;
|
||||
this.auth = auth;
|
||||
init(null);
|
||||
}
|
||||
|
||||
public void init(Map<String, String> config) {
|
||||
if (!preInit()) {
|
||||
return;
|
||||
}
|
||||
|
||||
// 自动注入
|
||||
if (config != null) {
|
||||
addr = config.getOrDefault("addr", addr);
|
||||
groupid = config.getOrDefault("groupid", groupid);
|
||||
appid = config.getOrDefault("appname", appid);
|
||||
}
|
||||
|
||||
// 设置第一个启动的 实例为主实例
|
||||
/*if (isFirst) {
|
||||
isMain = true;
|
||||
isFirst = false;
|
||||
}*/
|
||||
if (!mainHub.containsKey(addr)) { // 确保同步执行此 init 逻辑
|
||||
mainHub.put(addr, this);
|
||||
}
|
||||
|
||||
if (!initSocket(0)) {
|
||||
return;
|
||||
}
|
||||
// 消息 事件接收
|
||||
new Thread(() -> {
|
||||
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)) {
|
||||
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));
|
||||
|
||||
|
||||
// lock msg
|
||||
if ("lock".equals(topic)) {
|
||||
Lock lock = lockTag.get(value);
|
||||
if (lock != null) {
|
||||
synchronized (lock) {
|
||||
lock.success = true;
|
||||
lock.notifyAll();
|
||||
}
|
||||
}
|
||||
continue;
|
||||
}
|
||||
// trylock msg
|
||||
if ("trylock".equals(topic)) {
|
||||
Lock lock = lockTag.get(value);
|
||||
if (lock != null) {
|
||||
synchronized (lock) {
|
||||
lock.success = false;
|
||||
lock.notifyAll();
|
||||
}
|
||||
}
|
||||
continue;
|
||||
}
|
||||
|
||||
// rpc back msg
|
||||
if (appid.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)) {
|
||||
reader.readLine(); // $7 len()
|
||||
|
||||
type = reader.readLine(); // message
|
||||
if (!"timer".equals(type)) {
|
||||
continue;
|
||||
}
|
||||
reader.readLine(); //$n len(key)
|
||||
String topic = reader.readLine(); // name
|
||||
|
||||
timerQueue.add(timerMap.get(topic));
|
||||
continue;
|
||||
}
|
||||
|
||||
logger.finest(readLine);
|
||||
} catch (IOException e) {
|
||||
if (e instanceof SocketException) {
|
||||
initSocket(Integer.MAX_VALUE);
|
||||
}
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}).start();
|
||||
|
||||
// 定时调度事件
|
||||
new Thread(() -> {
|
||||
ExecutorService executor = Executors.newSingleThreadExecutor();
|
||||
while (true) {
|
||||
Timer timer = null;
|
||||
try {
|
||||
timer = timerQueue.take();
|
||||
long start = System.currentTimeMillis();
|
||||
executor.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 | ExecutionException | TimeoutException e) {
|
||||
if (e instanceof TimeoutException) {
|
||||
executor = Executors.newSingleThreadExecutor();
|
||||
logger.log(Level.WARNING, "TimeoutException [" + timer.name + "]", e);
|
||||
} else {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}
|
||||
}).start();
|
||||
|
||||
// topic msg
|
||||
new Thread(() -> {
|
||||
ExecutorService executor = Executors.newSingleThreadExecutor();
|
||||
while (true) {
|
||||
Event<String> event = null;
|
||||
try {
|
||||
event = topicQueue.take();
|
||||
logger.log(Level.FINE, "topic[" + event.topic + "] :" + event.value);
|
||||
|
||||
String topic = event.topic;
|
||||
String value = event.value;
|
||||
executor.submit(() -> accept(topic, value)).get(5, TimeUnit.SECONDS);
|
||||
} catch (InterruptedException | ExecutionException | TimeoutException e) {
|
||||
if (e instanceof TimeoutException) {
|
||||
executor = Executors.newSingleThreadExecutor();
|
||||
logger.log(Level.WARNING, "TimeoutException, topic[" + event.topic + "], value[" + event.value + "]", e);
|
||||
} else if (event != null) {
|
||||
logger.log(Level.WARNING, "topic[" + event.topic + "] event accept error :" + event.value, e);
|
||||
}
|
||||
}
|
||||
}
|
||||
}, "ZHub-topic-accept").start();
|
||||
// rpc back
|
||||
new Thread(() -> {
|
||||
ExecutorService executor = Executors.newSingleThreadExecutor();
|
||||
while (true) {
|
||||
Event<String> event = null;
|
||||
try {
|
||||
event = rpcBackQueue.take();
|
||||
logger.info(String.format("rpc-back:[%s]: %s", event.topic, event.value));
|
||||
rpcAccept(event.value);
|
||||
|
||||
String value = event.value;
|
||||
executor.submit(() -> rpcAccept(value)).get(5, TimeUnit.SECONDS);
|
||||
|
||||
} catch (InterruptedException | ExecutionException | TimeoutException e) {
|
||||
if (e instanceof TimeoutException) {
|
||||
executor = Executors.newSingleThreadExecutor();
|
||||
logger.log(Level.WARNING, "rpc-back TimeoutException, topic[" + event.topic + "], value[" + event.value + "]", e);
|
||||
} else if (event != null) {
|
||||
logger.log(Level.WARNING, "rpc-back[" + event.topic + "] event accept error :" + event.value, e);
|
||||
}
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
}).start();
|
||||
|
||||
// rpc call
|
||||
new Thread(() -> {
|
||||
ExecutorService executor = Executors.newSingleThreadExecutor();
|
||||
while (true) {
|
||||
Event<String> event = null;
|
||||
try {
|
||||
event = rpcCallQueue.take();
|
||||
logger.info(String.format("rpc-call:[%s] %s", event.topic, event.value));
|
||||
|
||||
String topic = event.topic;
|
||||
String value = event.value;
|
||||
executor.submit(() -> accept(topic, value)).get(5, TimeUnit.SECONDS);
|
||||
} catch (InterruptedException | ExecutionException | TimeoutException e) {
|
||||
if (e instanceof TimeoutException) {
|
||||
executor = Executors.newSingleThreadExecutor();
|
||||
logger.log(Level.WARNING, "rpc-call TimeoutException, topic[" + event.topic + "], value[" + event.value + "]", e);
|
||||
} else if (event != null) {
|
||||
logger.log(Level.WARNING, "rpc-call[" + event.topic + "] event accept error :" + event.value, e);
|
||||
}
|
||||
}
|
||||
}
|
||||
}, "ZHub-rpc-call").start();
|
||||
|
||||
// send msg
|
||||
new Thread(() -> {
|
||||
while (true) {
|
||||
String msg = null;
|
||||
try {
|
||||
msg = sendMsgQueue.take();
|
||||
writer.write(msg.getBytes(UTF_8));
|
||||
writer.flush();
|
||||
} catch (InterruptedException | IOException e) {
|
||||
logger.log(Level.WARNING, "send-msg[" + msg + "] event accept error :", e);
|
||||
|
||||
try {
|
||||
Thread.sleep(3000);
|
||||
assert msg != null;
|
||||
writer.write(msg.getBytes(UTF_8));
|
||||
writer.flush();
|
||||
} catch (IOException | InterruptedException | NullPointerException ex) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}
|
||||
}).start();
|
||||
|
||||
}
|
||||
|
||||
// ---------------------
|
||||
// -- 消息发送 --
|
||||
protected boolean send(String... data) {
|
||||
if (data.length == 1) {
|
||||
sendMsgQueue.add(data[0] + "\r\n");
|
||||
} else if (data.length > 1) {
|
||||
StringBuilder buf = new StringBuilder();
|
||||
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 gson.toJson(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]);
|
||||
|
||||
//private ReentrantLock lock = new ReentrantLock();
|
||||
Socket 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 trylock");
|
||||
/*if (isMain) {
|
||||
}*/
|
||||
if (mainHub.containsValue(this)) {
|
||||
buf.append(" ").append(appid);
|
||||
}
|
||||
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 (retry == 0 || 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));
|
||||
}
|
||||
}
|
||||
|
||||
switch (endchar) {
|
||||
case "M":
|
||||
delay *= (1000 * 60 * 60 * 24 * 30);
|
||||
break;
|
||||
case "d":
|
||||
delay *= (1000 * 60 * 60 * 24);
|
||||
break;
|
||||
case "H":
|
||||
delay *= (1000 * 60 * 60);
|
||||
break;
|
||||
case "m":
|
||||
delay *= (1000 * 60);
|
||||
break;
|
||||
case "s":
|
||||
delay *= 1000;
|
||||
break;
|
||||
}
|
||||
|
||||
delay(topic, v, delay);
|
||||
}
|
||||
|
||||
@Override
|
||||
protected String getGroupid() {
|
||||
return groupid;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void subscribe(String topic) {
|
||||
send("subscribe " + topic); //新增订阅
|
||||
}
|
||||
|
||||
// ================================================== lock ==================================================
|
||||
private final Map<String, Lock> lockTag = new ConcurrentHashMap<>();
|
||||
|
||||
/**
|
||||
* 尝试加锁,立即返回,
|
||||
*
|
||||
* @param key
|
||||
* @param duration
|
||||
* @return Lock: lock.success 锁定是否成功标识
|
||||
*/
|
||||
public Lock tryLock(String key, int duration) {
|
||||
return lock("trylock", key, duration);
|
||||
}
|
||||
|
||||
public Lock lock(String key, int duration) {
|
||||
return lock("lock", key, duration);
|
||||
}
|
||||
|
||||
/**
|
||||
* @param cmd lock|trylock
|
||||
* @param key 加锁 key
|
||||
* @param duration 锁定时长
|
||||
* @return
|
||||
*/
|
||||
private Lock lock(String cmd, String key, int duration) {
|
||||
String uuid = UUID.randomUUID().toString().replaceAll("-", "");
|
||||
Lock lock = new Lock(key, uuid, duration, this);
|
||||
lockTag.put(uuid, lock);
|
||||
|
||||
try {
|
||||
// c.send("lock", key, uuid, strconv.Itoa(duration))
|
||||
send(cmd, key, uuid, String.valueOf(duration));
|
||||
synchronized (lock) {
|
||||
lock.wait();
|
||||
}
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
return lock;
|
||||
}
|
||||
|
||||
// ================================================== timer ==================================================
|
||||
private final 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);
|
||||
}
|
||||
|
||||
public void reloadTimer() {
|
||||
send("cmd", "reload-timer");
|
||||
}
|
||||
|
||||
// ================================================== rpc ==================================================
|
||||
// -- 调用端 --
|
||||
private static final Map<String, Rpc> rpcMap = new ConcurrentHashMap<>();
|
||||
private static final Map<String, TypeToken> rpcRetType = new ConcurrentHashMap<>();
|
||||
|
||||
// rpc call
|
||||
public RpcResult<Void> rpc(String topic, Object v) {
|
||||
return rpc(topic, v, null);
|
||||
}
|
||||
|
||||
// rpc call
|
||||
public <T, R> RpcResult<R> rpc(String topic, T v, TypeToken<R> typeToken) {
|
||||
return rpc(topic, v, typeToken, 0);
|
||||
}
|
||||
|
||||
// rpc call
|
||||
public <T, R> RpcResult<R> rpc(String topic, T v, TypeToken<R> typeToken, long timeout) {
|
||||
Rpc rpc = new Rpc<>(appid, UUID.randomUUID().toString().replaceAll("-", ""), 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.retError(505, "请求超时");
|
||||
rpc.setRpcResult(rpcResult);
|
||||
logger.warning("rpc timeout: " + gson.toJson(rpc));
|
||||
rpc.notify();
|
||||
}
|
||||
}, timeout);
|
||||
|
||||
rpc.wait();
|
||||
rpcMap.remove(ruk);
|
||||
}
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
// call error
|
||||
RpcResult rpcResult = rpc.retError(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}
|
||||
// rpc call back consumer
|
||||
private void rpcAccept(String value) {
|
||||
RpcResult resp = gson.fromJson(value, new TypeToken<RpcResult<String>>() {
|
||||
}.getType());
|
||||
|
||||
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())) {
|
||||
result = gson.fromJson((String) resp.getResult(), typeToken.getType());
|
||||
}
|
||||
|
||||
resp.setResult(result);
|
||||
rpc.setRpcResult(resp);
|
||||
synchronized (rpc) {
|
||||
rpc.notify();
|
||||
}
|
||||
}
|
||||
|
||||
// -- 订阅端 --
|
||||
private final HashSet rpcTopics = new HashSet();
|
||||
|
||||
// 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 = gson.fromJson(v, new TypeToken<Rpc<String>>() {
|
||||
}.getType());
|
||||
|
||||
// 参数转换
|
||||
T paras = gson.fromJson((String) rpc.getValue(), typeToken.getType());
|
||||
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);
|
||||
}
|
||||
}
|
7
src/main/resources/application.yml
Normal file
7
src/main/resources/application.yml
Normal file
@ -0,0 +1,7 @@
|
||||
|
||||
# zhub 配置
|
||||
zhub:
|
||||
appid: local_api
|
||||
addr: 127.0.0.1:1216
|
||||
groupid: hub-api
|
||||
auth: token-12345
|
108
test/HelloService.java
Normal file
108
test/HelloService.java
Normal file
@ -0,0 +1,108 @@
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import tccn.IType;
|
||||
import tccn.zhub.Lock;
|
||||
import tccn.zhub.ZHubClient;
|
||||
|
||||
// @RestService(automapping = true)
|
||||
public class HelloService {
|
||||
|
||||
// @Resource(name = "zhub")
|
||||
private ZHubClient zhub;
|
||||
|
||||
@Before
|
||||
public void init() {
|
||||
|
||||
|
||||
|
||||
//zhub = new ZHubClient("127.0.0.1:1216", "g-dev", "DEV-LOCAL", "zchd@123456");
|
||||
zhub = new ZHubClient("127.0.0.1:1216", "g-dev", "DEV-LOCAL", "token-12345");
|
||||
|
||||
zhub.subscribe("tv:test", x -> {
|
||||
System.out.println(x);
|
||||
});
|
||||
|
||||
Lock lock = zhub.tryLock("lock-a", 5);
|
||||
System.out.println("lock-1: " + lock.success());
|
||||
|
||||
//zhub.init(Kv.of("host", "47.111.150.118", "port", "6066", "groupid", "g-dev", "appname", "DEV-LOCAL"));
|
||||
|
||||
// Function<Rpc<T>, RpcResult<R>> fun
|
||||
/*zhub.rpcSubscribe("x", new TypeToken<String>() {
|
||||
}, r -> {
|
||||
return r.buildResp(Map.of("v", r.getValue().toUpperCase() + ": Ok"));
|
||||
});
|
||||
|
||||
zhub.subscribe("sport:reqtime", x -> {
|
||||
//System.out.println(x);
|
||||
});
|
||||
zhub.subscribe("abx", x -> {
|
||||
System.out.println(x);
|
||||
});
|
||||
|
||||
try {
|
||||
Thread.sleep(010);
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
zhub.delay("sport:reqtime", "别✈人家的✦女娃子❤🤞🏻", 0);
|
||||
zhub.delay("sport:reqtime", "别人家的女娃子➾🤞🏻", 0);
|
||||
zhub.delay("sport:reqtime", "❤别人家✉<EFBFBD>的女娃子❤🤞🏻", 0);*/
|
||||
/*zhub.delay("sport:reqtime", "中文特殊符号:『』 $ £ ♀ ‖ 「」\n" +
|
||||
"英文:# + = & ﹉ .. ^ \"\" ·{ } % – ' €\n" +
|
||||
"数学:+× = - ° ± < > ℃ ㎡ ∑ ≥ ∫ ㏄ ⊥ ≯ ∠ ∴ ∈ ∧ ∵ ≮ ∪ ㎝ ㏑ ≌ ㎞ № § ℉ ÷ % ‰ ㎎ ㎏ ㎜ ㏒ ⊙ ∮ ∝ ∞ º ¹ ² ³ ½ ¾ ¼ ≈ ≡ ≠ ≤ ≦ ≧ ∽ ∷ / ∨ ∏ ∩ ⌒ √Ψ ¤ ‖ ¶\n" +
|
||||
"特殊:♤ ♧ ♡ ♢ ♪ ♬ ♭ ✔ ✘ ♞ ♟ ↪ ↣ ♚ ♛ ♝ ☞ ☜ ⇔ ☆ ★ □ ■ ○ ● △ ▲ ▽ ▼ ◇ ◆ ♀ ♂ ※ ↓ ↑ ↔ ↖ ↙ ↗ ↘ ← → ♣ ♠ ♥ ◎ ◣ ◢ ◤ ◥ 卍 ℡ ⊙ ㊣ ® © ™ ㈱ 囍\n" +
|
||||
"序号:①②③④⑤⑥⑦⑧⑨⑩㈠㈡㈢㈣㈤㈥㈦㈧㈨㈩⑴ ⑵ ⑶ ⑷ ⑸ ⑹ ⑺ ⑻ ⑼ ⑽ ⒈ ⒉ ⒊ ⒋ ⒌ ⒍ ⒎ ⒏ ⒐ ⒑ Ⅰ Ⅱ Ⅲ Ⅳ Ⅴ Ⅵ Ⅶ Ⅷ ⅨⅩ\n" +
|
||||
"日文:アイウエオァィゥェォカキクケコガギグゲゴサシスセソザジズゼゾタチツテトダヂヅデドッナニヌネノハヒフヘホバビブベボパピプペポマミムメモャヤュユョラリヨルレロワヰヱヲンヴヵヶヽヾ゛゜ー、。「「あいうえおぁぃぅぇぉかきくけこがぎぐげごさしすせそざじずぜぞたちつてでどっなにぬねのはひふへ」」ほばびぶべぼぱぴぷぺぽまみむめもやゆよゃゅょらりるれろわをんゎ゛゜ー、。「」\n" +
|
||||
"部首:犭 凵 巛 冖 氵 廴 讠 亻 钅 宀 亠 忄 辶 弋 饣 刂 阝 冫 卩 疒 艹 疋 豸 冂 匸 扌 丬 屮衤 礻 勹 彳 彡", 0);*/
|
||||
|
||||
}
|
||||
|
||||
@Test
|
||||
public void rpcTest() {
|
||||
//RpcResult<String> rpc = zhub.rpc("wx:users", Map.of("appId", "wxa554ec3ab3bf1fc7"), IConsumer.TYPE_TOKEN_STRING);
|
||||
//RpcResult<String> rpc = zhub.rpc("a", "fa", IConsumer.TYPE_TOKEN_STRING);
|
||||
zhub.publish("tv:test", "hello ym!");
|
||||
|
||||
zhub.subscribe("tv:abx", x -> {
|
||||
System.out.println(x);
|
||||
});
|
||||
|
||||
zhub.rpcSubscribe("rpc-x", IType.STRING, x -> {
|
||||
return x.render(x.getValue().toUpperCase());
|
||||
});
|
||||
|
||||
Lock lock = zhub.tryLock("lock-a", 5);
|
||||
System.out.println("lock-2: " + lock.success());
|
||||
|
||||
try {
|
||||
Thread.sleep(5 * 1000);
|
||||
} catch (InterruptedException e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
|
||||
Lock lock2 = zhub.tryLock("lock-a", 5);
|
||||
System.out.println("lock-3: " + lock2.success());
|
||||
/*try {
|
||||
Thread.sleep(100);
|
||||
} catch (InterruptedException e) {
|
||||
throw new RuntimeException(e);
|
||||
}*/
|
||||
Lock lock3 = zhub.tryLock("lock-a", 5);
|
||||
System.out.println("lock-4: " + lock3.success());
|
||||
|
||||
try {
|
||||
Thread.sleep(3000 * 30000);
|
||||
} catch (InterruptedException e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public void lockTest() {
|
||||
|
||||
}
|
||||
|
||||
/*RpcResult<FileToken> x = zhub.rpc("rpc:file:up-token", Map.of(), new TypeToken<>() {
|
||||
});*/
|
||||
}
|
@ -1,73 +0,0 @@
|
||||
package com.zdemo.test;
|
||||
|
||||
import com.zdemo.Event;
|
||||
import com.zdemo.EventType;
|
||||
import com.zdemo.kafak.KafakProducer;
|
||||
import org.junit.Test;
|
||||
import org.redkale.boot.Application;
|
||||
import org.redkale.convert.json.JsonConvert;
|
||||
import org.redkale.util.TypeToken;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import static java.util.Arrays.asList;
|
||||
|
||||
/**
|
||||
* 消息发布订阅测试
|
||||
*/
|
||||
public class AppTest {
|
||||
|
||||
@Test
|
||||
public void runConsumer() {
|
||||
try {
|
||||
//启动并开启消费监听
|
||||
MyConsumer consumer = Application.singleton(MyConsumer.class);
|
||||
|
||||
consumer.addEventType(
|
||||
EventType.of("a1", new TypeToken<Float>() {
|
||||
}, r -> {
|
||||
System.out.println("我收到了消息 主题a1 事件:" + JsonConvert.root().convertTo(r));
|
||||
}),
|
||||
|
||||
EventType.of("bx", str -> {
|
||||
System.out.println("我收到了消息 主题bx 事件:" + str);
|
||||
})
|
||||
|
||||
, EventType.of("game-update", str -> {
|
||||
System.out.println("我收到了消息 主题game-update 事件:" + str);
|
||||
})
|
||||
);
|
||||
|
||||
Thread.sleep(60_000 * 60);
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void runProducer() {
|
||||
try {
|
||||
KafakProducer producer = Application.singleton(KafakProducer.class);
|
||||
|
||||
// 发送不同的 事件
|
||||
float v0 = 1f;
|
||||
Map v1 = Map.of("k", "v");
|
||||
List v2 = asList(1, 2, 3);
|
||||
|
||||
//producer.send(Event.of("a1", v0));
|
||||
/*producer.send(Event.of("b1", v1));
|
||||
producer.send(Event.of("c1", v2));*/
|
||||
|
||||
producer.send(Event.of("game-update", 23256));
|
||||
|
||||
try {
|
||||
Thread.sleep(10_000);
|
||||
} catch (InterruptedException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}
|
@ -1,15 +0,0 @@
|
||||
package com.zdemo.test;
|
||||
|
||||
import com.zdemo.kafak.KafakConsumer;
|
||||
|
||||
public class MyConsumer extends KafakConsumer {
|
||||
|
||||
public String getGroupid() {
|
||||
return "group-test"; //消费组名称
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean preInit() {
|
||||
return true;
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user