5 Commits

3 changed files with 92 additions and 91 deletions

30
pom.xml
View File

@@ -6,52 +6,44 @@
<groupId>net.tccn</groupId> <groupId>net.tccn</groupId>
<artifactId>zhub-client-spring</artifactId> <artifactId>zhub-client-spring</artifactId>
<version>0.1.3.dev</version> <version>0.1.2</version>
<parent>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-parent</artifactId>
<version>3.1.10</version>
<relativePath/> <!-- lookup parent from repository -->
</parent>
<properties> <properties>
<maven.compiler.source>17</maven.compiler.source> <maven.compiler.source>1.8</maven.compiler.source>
<maven.compiler.target>17</maven.compiler.target> <maven.compiler.target>1.8</maven.compiler.target>
</properties> </properties>
<dependencies> <dependencies>
<dependency> <dependency>
<groupId>org.springframework.boot</groupId> <groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-web</artifactId> <artifactId>spring-boot-starter-web</artifactId>
<scope>provided</scope> <version>3.2.1</version>
<scope>compile</scope>
</dependency> </dependency>
<dependency> <dependency>
<groupId>com.google.code.gson</groupId> <groupId>com.google.code.gson</groupId>
<artifactId>gson</artifactId> <artifactId>gson</artifactId>
<version>2.10.1</version>
<scope>compile</scope>
</dependency> </dependency>
<dependency> <dependency>
<groupId>junit</groupId> <groupId>junit</groupId>
<artifactId>junit</artifactId> <artifactId>junit</artifactId>
<!--<version>4.13.1</version>--> <version>4.13.1</version>
<scope>test</scope> <scope>test</scope>
</dependency> </dependency>
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
<!--<version>1.18.30</version>-->
<scope>provided</scope>
</dependency>
</dependencies> </dependencies>
<repositories> <repositories>
<repository> <repository>
<id>maven-release</id> <id>maven-nexus</id>
<name>maven-nexus</name> <name>maven-nexus</name>
<url>https://nexus.1216.top/repository/maven-public/</url> <url>https://nexus.1216.top/repository/maven-public/</url>
</repository> </repository>
</repositories> </repositories>
<distributionManagement> <distributionManagement>
<repository> <repository>
<id>mvn-release</id> <id>mvn-release</id>

View File

@@ -15,8 +15,8 @@ public class Event<V> {
this.value = value; this.value = value;
} }
public static <V> Event of(String topic, V value) { public static <V> Event<V> of(String topic, V value) {
return new Event<V>(topic, value); return new Event<>(topic, value);
} }

View File

@@ -1,8 +1,6 @@
package tccn.zhub; package tccn.zhub;
import com.google.gson.reflect.TypeToken; import com.google.gson.reflect.TypeToken;
import jakarta.annotation.PostConstruct;
import lombok.Setter;
import org.springframework.beans.factory.annotation.Value; import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component; import org.springframework.stereotype.Component;
import tccn.AbstractConsumer; import tccn.AbstractConsumer;
@@ -11,6 +9,7 @@ import tccn.IConsumer;
import tccn.IProducer; import tccn.IProducer;
import tccn.timer.Timers; import tccn.timer.Timers;
import javax.annotation.PostConstruct;
import java.io.BufferedReader; import java.io.BufferedReader;
import java.io.IOException; import java.io.IOException;
import java.io.InputStreamReader; import java.io.InputStreamReader;
@@ -22,29 +21,27 @@ import java.util.HashMap;
import java.util.HashSet; import java.util.HashSet;
import java.util.Map; import java.util.Map;
import java.util.UUID; import java.util.UUID;
import java.util.concurrent.CompletableFuture; import java.util.concurrent.*;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.function.BiConsumer;
import java.util.function.Consumer; import java.util.function.Consumer;
import java.util.function.Function; import java.util.function.Function;
import java.util.logging.Level; import java.util.logging.Level;
import java.util.logging.Logger; import java.util.logging.Logger;
import static java.nio.charset.StandardCharsets.UTF_8;
@Component @Component
public class ZHubClient extends AbstractConsumer implements IConsumer, IProducer { public class ZHubClient extends AbstractConsumer implements IConsumer, IProducer {
public Logger logger = Logger.getLogger(ZHubClient.class.getSimpleName()); public Logger logger = Logger.getLogger(ZHubClient.class.getSimpleName());
@Setter
@Value("${zhub.addr}") @Value("${zhub.addr}")
private String addr = "127.0.0.1:1216"; private String addr = "127.0.0.1:1216";
@Setter
@Value("${zhub.groupid}") @Value("${zhub.groupid}")
private String groupid = ""; private String groupid = "";
@Setter
@Value("${zhub.auth}") @Value("${zhub.auth}")
private String auth = ""; private String auth = "";
@Setter
@Value("${zhub.appid}") @Value("${zhub.appid}")
protected String appid = ""; protected String appid = "";
@@ -62,19 +59,10 @@ public class ZHubClient extends AbstractConsumer implements IConsumer, IProducer
private final LinkedBlockingQueue<Event<String>> rpcCallQueue = new LinkedBlockingQueue<>(); // RPC CALL MSG private final LinkedBlockingQueue<Event<String>> rpcCallQueue = new LinkedBlockingQueue<>(); // RPC CALL MSG
private final LinkedBlockingQueue<String> sendMsgQueue = new LinkedBlockingQueue<>(); // SEND MSG private final LinkedBlockingQueue<String> sendMsgQueue = new LinkedBlockingQueue<>(); // SEND MSG
private final BiConsumer<Runnable, Integer> threadBuilder = (r, n) -> {
for (int i = 0; i < n; i++) {
new Thread(() -> r.run()).start();
}
};
/*private static boolean isFirst = true; /*private static boolean isFirst = true;
private boolean isMain = false;*/ private boolean isMain = false;*/
private static final Map<String, ZHubClient> mainHub = new HashMap<>(); // 127.0.0.1:1216 - ZHubClient private static final Map<String, ZHubClient> mainHub = new HashMap<>(); // 127.0.0.1:1216 - ZHubClient
public ZHubClient() {
}
public ZHubClient(String addr, String groupid, String appid, String auth) { public ZHubClient(String addr, String groupid, String appid, String auth) {
this.addr = addr; this.addr = addr;
this.groupid = groupid; this.groupid = groupid;
@@ -217,98 +205,119 @@ public class ZHubClient extends AbstractConsumer implements IConsumer, IProducer
}).start(); }).start();
// 定时调度事件 // 定时调度事件
threadBuilder.accept(() -> { new Thread(() -> {
ExecutorService executor = Executors.newSingleThreadExecutor();
while (true) { while (true) {
Timer timer = null; Timer timer = null;
try { try {
if ((timer = timerQueue.take()) == null) { timer = timerQueue.take();
return;
}
long start = System.currentTimeMillis(); long start = System.currentTimeMillis();
timer.runnable.run(); executor.submit(timer.runnable).get(5, TimeUnit.SECONDS);
long end = System.currentTimeMillis(); long end = System.currentTimeMillis();
logger.finest(String.format("timer [%s] : elapsed time %s ms", timer.name, end - start)); logger.finest(String.format("timer [%s] : elapsed time %s ms", timer.name, end - start));
} catch (InterruptedException e) { } catch (InterruptedException | ExecutionException | TimeoutException e) {
e.printStackTrace(); if (e instanceof TimeoutException) {
} catch (Exception e) { executor = Executors.newSingleThreadExecutor();
logger.log(Level.WARNING, "timer [" + timer.name + "]", e); logger.log(Level.WARNING, "TimeoutException [" + timer.name + "]", e);
} else {
e.printStackTrace();
}
} }
} }
}, 1); }).start();
// topic msg // topic msg
threadBuilder.accept(() -> { new Thread(() -> {
ExecutorService executor = Executors.newSingleThreadExecutor();
while (true) { while (true) {
Event<String> event = null; Event<String> event = null;
try { try {
if ((event = topicQueue.take()) == null) { event = topicQueue.take();
continue;
}
logger.log(Level.FINE, "topic[" + event.topic + "] :" + event.value); logger.log(Level.FINE, "topic[" + event.topic + "] :" + event.value);
accept(event.topic, event.value);
} catch (InterruptedException e) { String topic = event.topic;
e.printStackTrace(); String value = event.value;
} catch (Exception e) { executor.submit(() -> accept(topic, value)).get(5, TimeUnit.SECONDS);
logger.log(Level.WARNING, "topic[" + event.topic + "] event accept error :" + event.value, e); } 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);
}
} }
} }
}, 1); }, "ZHub-topic-accept").start();
// rpc back // rpc back
threadBuilder.accept(() -> { new Thread(() -> {
ExecutorService executor = Executors.newSingleThreadExecutor();
while (true) { while (true) {
Event<String> event = null; Event<String> event = null;
try { try {
if ((event = rpcBackQueue.take()) == null) { event = rpcBackQueue.take();
continue;
}
//if (event)
logger.info(String.format("rpc-back:[%s]: %s", event.topic, event.value)); logger.info(String.format("rpc-back:[%s]: %s", event.topic, event.value));
rpcAccept(event.value); rpcAccept(event.value);
} catch (InterruptedException e) {
e.printStackTrace(); String value = event.value;
} catch (Exception e) { executor.submit(() -> rpcAccept(value)).get(5, TimeUnit.SECONDS);
logger.log(Level.WARNING, "rpc-back[" + event.topic + "] event accept error :" + event.value, e);
} 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);
} }
} }
}, 1); }).start();
// rpc call // rpc call
threadBuilder.accept(() -> { new Thread(() -> {
ExecutorService executor = Executors.newSingleThreadExecutor();
while (true) { while (true) {
Event<String> event = null; Event<String> event = null;
try { try {
if ((event = rpcCallQueue.take()) == null) { event = rpcCallQueue.take();
continue;
}
logger.info(String.format("rpc-call:[%s] %s", event.topic, event.value)); logger.info(String.format("rpc-call:[%s] %s", event.topic, event.value));
accept(event.topic, event.value);
} catch (InterruptedException e) { String topic = event.topic;
e.printStackTrace(); String value = event.value;
} catch (Exception e) { executor.submit(() -> accept(topic, value)).get(5, TimeUnit.SECONDS);
logger.log(Level.WARNING, "rpc-call[" + event.topic + "] event accept error :" + event.value, e); } 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);
}
} }
} }
}, 1); }, "ZHub-rpc-call").start();
// send msg // send msg
threadBuilder.accept(() -> { new Thread(() -> {
while (true) { while (true) {
String msg = null; String msg = null;
try { try {
if ((msg = sendMsgQueue.take()) == null) { msg = sendMsgQueue.take();
continue; writer.write(msg.getBytes(UTF_8));
}
// logger.log(Level.FINEST, "send-msg: [" + msg + "]");
writer.write(msg.getBytes());
writer.flush(); writer.flush();
} catch (InterruptedException e) { } catch (InterruptedException | IOException e) {
e.printStackTrace();
} catch (Exception e) {
logger.log(Level.WARNING, "send-msg[" + msg + "] event accept error :", 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();
}
} }
} }
}, 1); }).start();
} }