729 lines
26 KiB
Java
729 lines
26 KiB
Java
package net.tccn.zhub;
|
||
|
||
import net.tccn.*;
|
||
import net.tccn.timer.Timers;
|
||
import org.redkale.annotation.AutoLoad;
|
||
import org.redkale.annotation.ResourceType;
|
||
import org.redkale.service.Local;
|
||
import org.redkale.service.Service;
|
||
import org.redkale.util.AnyValue;
|
||
import org.redkale.util.Comment;
|
||
import org.redkale.util.TypeToken;
|
||
import org.redkale.util.Utility;
|
||
|
||
import java.io.BufferedReader;
|
||
import java.io.IOException;
|
||
import java.io.InputStreamReader;
|
||
import java.io.OutputStream;
|
||
import java.net.InetSocketAddress;
|
||
import java.net.Socket;
|
||
import java.net.SocketException;
|
||
import java.util.HashMap;
|
||
import java.util.HashSet;
|
||
import java.util.Map;
|
||
import java.util.Set;
|
||
import java.util.concurrent.*;
|
||
import java.util.function.Consumer;
|
||
import java.util.function.Function;
|
||
import java.util.logging.Level;
|
||
import java.util.logging.Logger;
|
||
|
||
@Local
|
||
@AutoLoad(false)
|
||
@ResourceType(ZHubClient.class)
|
||
public class ZHubClient extends AbstractConsumer implements IConsumer, IProducer, Service {
|
||
|
||
public Logger logger = Logger.getLogger(ZHubClient.class.getSimpleName());
|
||
private String addr = "127.0.0.1:1216";
|
||
private String auth = "";
|
||
private String groupid = "";
|
||
|
||
//private ReentrantLock lock = new ReentrantLock();
|
||
private Socket client;
|
||
private OutputStream writer;
|
||
private BufferedReader reader;
|
||
|
||
private final LinkedBlockingQueue<Timer> timerQueue = new LinkedBlockingQueue<>();
|
||
private final LinkedBlockingQueue<Event<String>> topicQueue = new LinkedBlockingQueue<>();
|
||
private final LinkedBlockingQueue<Event<String>> rpcBackQueue = new LinkedBlockingQueue<>(); // RPC BACK MSG
|
||
private final LinkedBlockingQueue<Event<String>> rpcCallQueue = new LinkedBlockingQueue<>(); // RPC CALL MSG
|
||
private final LinkedBlockingQueue<String> sendMsgQueue = new LinkedBlockingQueue<>(); // SEND MSG
|
||
|
||
/*private BiConsumer<Runnable, Integer> threadBuilder = (r, n) -> {
|
||
for (int i = 0; i < n; i++) {
|
||
new Thread(() -> r.run()).start();
|
||
}
|
||
};*/
|
||
|
||
private static Map<String, ZHubClient> mainHub = new HashMap<>(); // 127.0.0.1:1216 - ZHubClient
|
||
|
||
public ZHubClient() {
|
||
|
||
}
|
||
|
||
public ZHubClient(String name, Map<String, String> attr) {
|
||
this.APP_NAME = name;
|
||
this.addr = attr.get("addr");
|
||
this.groupid = attr.get("groupid");
|
||
this.auth = attr.get("auth");
|
||
|
||
this.initClient(null);
|
||
}
|
||
|
||
@Override
|
||
public void init(AnyValue config) {
|
||
APP_NAME = application.getName();
|
||
/*if (!preInit()) {
|
||
return;
|
||
}*/
|
||
|
||
if (config == null) {
|
||
initClient(null);
|
||
} else {
|
||
Map<String, AnyValue> nodes = getNodes(config);
|
||
for (String rsName : nodes.keySet()) {
|
||
ZHubClient client = new ZHubClient().initClient(nodes.get(rsName));
|
||
application.getResourceFactory().register(rsName, client);
|
||
}
|
||
}
|
||
}
|
||
|
||
private ZHubClient initClient(AnyValue config) {
|
||
// 自动注入
|
||
if (config != null) {
|
||
addr = config.getValue("addr", addr);
|
||
|
||
// 合并 addr = host:port, 做历史兼容
|
||
int port = config.getIntValue("port", 0);
|
||
if (port != 0 && !addr.contains(":")) {
|
||
addr = addr + ":" + port;
|
||
}
|
||
|
||
auth = config.getOrDefault("auth", auth);
|
||
groupid = config.getValue("groupid", groupid);
|
||
}
|
||
|
||
// 设置第一个启动的 实例为主实例
|
||
if (!mainHub.containsKey(addr)) { // 确保同步执行此 init 逻辑
|
||
mainHub.put(addr, this);
|
||
}
|
||
|
||
CompletableFuture.runAsync(() -> {
|
||
// 消息 事件接收
|
||
new Thread(() -> {
|
||
if (!initSocket(0)) {
|
||
return;
|
||
}
|
||
|
||
while (true) {
|
||
try {
|
||
String readLine = reader.readLine();
|
||
if (readLine == null && initSocket(Integer.MAX_VALUE)) { // 连接中断 处理
|
||
continue;
|
||
}
|
||
|
||
String type = "";
|
||
|
||
// +ping
|
||
if ("+ping".equals(readLine)) {
|
||
send("+pong");
|
||
continue;
|
||
}
|
||
|
||
// 主题订阅消息
|
||
if ("*3".equals(readLine)) {
|
||
readLine = reader.readLine(); // $7 len()
|
||
type = reader.readLine(); // message
|
||
if (!"message".equals(type)) {
|
||
continue;
|
||
}
|
||
reader.readLine(); //$n len(key)
|
||
String topic = reader.readLine(); // topic
|
||
|
||
String lenStr = reader.readLine();//$n len(value)
|
||
int clen = 0;
|
||
if (lenStr.startsWith("$")) {
|
||
clen = Integer.parseInt(lenStr.replace("$", ""));
|
||
}
|
||
|
||
String value = "";
|
||
do {
|
||
if (value.length() > 0) {
|
||
value += "\r\n";
|
||
}
|
||
String s = reader.readLine();
|
||
value += s; // value
|
||
} while (clen > 0 && clen > strLength(value));
|
||
|
||
logger.finest("topic[" + topic + "]: " + value);
|
||
|
||
// lock msg
|
||
if ("lock".equals(topic)) {
|
||
Lock lock = lockTag.get(value);
|
||
if (lock != null) {
|
||
synchronized (lock) {
|
||
lock.notifyAll();
|
||
}
|
||
}
|
||
continue;
|
||
}
|
||
// rpc back msg
|
||
if (APP_NAME.equals(topic)) {
|
||
rpcBackQueue.add(Event.of(topic, value));
|
||
continue;
|
||
}
|
||
|
||
// rpc call msg
|
||
if (rpcTopics.contains(topic)) {
|
||
rpcCallQueue.add(Event.of(topic, value));
|
||
continue;
|
||
}
|
||
|
||
// oth msg
|
||
topicQueue.add(Event.of(topic, value));
|
||
continue;
|
||
}
|
||
|
||
// timer 消息
|
||
if ("*2".equals(readLine)) {
|
||
readLine = reader.readLine(); // $7 len()
|
||
type = reader.readLine(); // message
|
||
if (!"timer".equals(type)) {
|
||
continue;
|
||
}
|
||
reader.readLine(); //$n len(key)
|
||
String topic = reader.readLine(); // name
|
||
|
||
logger.finest("timer[" + topic + "]: ");
|
||
timerQueue.add(timerMap.get(topic));
|
||
continue;
|
||
}
|
||
|
||
logger.finest(readLine);
|
||
} catch (IOException e) {
|
||
if (e instanceof SocketException) {
|
||
initSocket(Integer.MAX_VALUE);
|
||
}
|
||
e.printStackTrace();
|
||
}
|
||
}
|
||
}).start();
|
||
}).thenAcceptAsync(x -> {
|
||
// 定时调度事件，已加入耗时监控
|
||
new Thread(() -> {
|
||
ExecutorService pool = Executors.newFixedThreadPool(1);
|
||
while (true) {
|
||
Timer timer = null;
|
||
try {
|
||
if ((timer = timerQueue.take()) == null) {
|
||
return;
|
||
}
|
||
long start = System.currentTimeMillis();
|
||
pool.submit(timer.runnable).get(5, TimeUnit.SECONDS);
|
||
long end = System.currentTimeMillis();
|
||
logger.finest(String.format("timer [%s] : elapsed time %s ms", timer.name, end - start));
|
||
} catch (InterruptedException e) {
|
||
e.printStackTrace();
|
||
} catch (TimeoutException e) {
|
||
logger.log(Level.SEVERE, "timer [" + timer.name + "] time out: " + 5 + " S", e);
|
||
pool = Executors.newFixedThreadPool(1);
|
||
} catch (Exception e) {
|
||
logger.log(Level.WARNING, "timer [" + timer.name + "]", e);
|
||
}
|
||
}
|
||
}).start();
|
||
|
||
// topic msg，已加入耗时监控
|
||
new Thread(() -> {
|
||
ExecutorService pool = Executors.newFixedThreadPool(1);
|
||
while (true) {
|
||
Event<String> event = null;
|
||
try {
|
||
if ((event = topicQueue.take()) == null) {
|
||
continue;
|
||
}
|
||
|
||
String topic = event.topic;
|
||
String value = event.value;
|
||
pool.submit(() -> accept(topic, value)).get(5, TimeUnit.SECONDS);
|
||
} catch (InterruptedException e) {
|
||
e.printStackTrace();
|
||
} catch (TimeoutException e) {
|
||
logger.log(Level.SEVERE, "topic[" + event.topic + "] event deal time out: " + 5 + " S, value: " + event.value, e);
|
||
pool = Executors.newFixedThreadPool(1);
|
||
} catch (Exception e) {
|
||
logger.log(Level.WARNING, "topic[" + event.topic + "] event accept error :" + event.value, e);
|
||
}
|
||
}
|
||
}).start();
|
||
|
||
// rpc back ,仅做数据解析，暂无耗时监控
|
||
new Thread(() -> {
|
||
while (true) {
|
||
Event<String> event = null;
|
||
try {
|
||
if ((event = rpcBackQueue.take()) == null) {
|
||
continue;
|
||
}
|
||
//if (event)
|
||
logger.finest(String.format("rpc-back:[%s]: %s", event.topic, event.value));
|
||
rpcAccept(event.value);
|
||
} catch (InterruptedException e) {
|
||
e.printStackTrace();
|
||
} catch (Exception e) {
|
||
logger.log(Level.WARNING, "rpc-back[" + event.topic + "] event accept error :" + event.value, e);
|
||
}
|
||
}
|
||
}).start();
|
||
|
||
// rpc call，已加入耗时监控
|
||
new Thread(() -> {
|
||
ExecutorService pool = Executors.newFixedThreadPool(1);
|
||
while (true) {
|
||
Event<String> event = null;
|
||
try {
|
||
if ((event = rpcCallQueue.take()) == null) {
|
||
continue;
|
||
}
|
||
logger.finest(String.format("rpc-call:[%s] %s", event.topic, event.value));
|
||
String topic = event.topic;
|
||
String value = event.value;
|
||
pool.submit(() -> accept(topic, value)).get(5, TimeUnit.SECONDS);
|
||
} catch (InterruptedException e) {
|
||
e.printStackTrace();
|
||
} catch (TimeoutException e) {
|
||
logger.log(Level.SEVERE, "topic[" + event.topic + "] event deal time out: " + 5 + " S, value: " + event.value, e);
|
||
pool = Executors.newFixedThreadPool(1);
|
||
} catch (Exception e) {
|
||
logger.log(Level.WARNING, "rpc-call[" + event.topic + "] event accept error :" + event.value, e);
|
||
}
|
||
}
|
||
}).start();
|
||
|
||
// send msg
|
||
new Thread(() -> {
|
||
while (true) {
|
||
String msg = null;
|
||
try {
|
||
if ((msg = sendMsgQueue.take()) == null) {
|
||
continue;
|
||
}
|
||
// logger.log(Level.FINEST, "send-msg: [" + msg + "]");
|
||
writer.write(msg.getBytes());
|
||
writer.flush();
|
||
} catch (InterruptedException e) {
|
||
e.printStackTrace();
|
||
} catch (Exception e) {
|
||
logger.log(Level.WARNING, "send-msg[" + msg + "] event accept error :", e);
|
||
}
|
||
}
|
||
}).start();
|
||
});
|
||
|
||
return this;
|
||
}
|
||
|
||
public boolean acceptsConf(AnyValue config) {
|
||
if (config == null) {
|
||
return false;
|
||
}
|
||
|
||
if (!getNodes(config).isEmpty()) {
|
||
return true;
|
||
}
|
||
return false;
|
||
}
|
||
|
||
private HashMap<String, AnyValue> getNodes(AnyValue config) {
|
||
AnyValue[] zhubs = config.getAnyValues("zhub");
|
||
HashMap<String, AnyValue> confMap = new HashMap<>();
|
||
|
||
for (AnyValue zhub : zhubs) {
|
||
String[] names = zhub.getNames();
|
||
for (String name : names) {
|
||
confMap.put(name, zhub.getAnyValue(name));
|
||
}
|
||
}
|
||
|
||
return confMap;
|
||
}
|
||
|
||
// ---------------------
|
||
// -- 消息发送 --
|
||
protected boolean send(String... data) {
|
||
if (data.length == 1) {
|
||
sendMsgQueue.add(data[0] + "\r\n");
|
||
} else if (data.length > 1) {
|
||
StringBuffer buf = new StringBuffer();
|
||
buf.append("*" + data.length + "\r\n");
|
||
for (String d : data) {
|
||
buf.append("$" + strLength(d) + "\r\n");
|
||
buf.append(d + "\r\n");
|
||
}
|
||
sendMsgQueue.add(buf.toString());
|
||
}
|
||
return true;
|
||
}
|
||
|
||
/*protected boolean send(String... data) {
|
||
try {
|
||
lock.lock();
|
||
if (data.length == 1) {
|
||
writer.write((data[0] + "\r\n").getBytes());
|
||
} else if (data.length > 1) {
|
||
writer.write(("*" + data.length + "\r\n").getBytes());
|
||
for (String d : data) {
|
||
writer.write(("$" + d.length() + "\r\n").getBytes());
|
||
writer.write((d + "\r\n").getBytes());
|
||
}
|
||
}
|
||
writer.flush();
|
||
return true;
|
||
} catch (IOException e) {
|
||
logger.log(Level.WARNING, "", e);
|
||
} finally {
|
||
lock.unlock();
|
||
}
|
||
return false;
|
||
}*/
|
||
|
||
private int strLength(String str) {
|
||
str = str.replaceAll("[^\\x00-\\xff]", "*");
|
||
return str.length();
|
||
}
|
||
|
||
private String toStr(Object v) {
|
||
if (v instanceof String) {
|
||
return (String) v;
|
||
} else if (v == null) {
|
||
return null;
|
||
}
|
||
return convert.convertTo(v);
|
||
}
|
||
|
||
protected boolean initSocket(int retry) {
|
||
for (int i = 0; i <= retry; i++) {
|
||
try {
|
||
String[] hostPort = addr.split(":");
|
||
String host = hostPort[0];
|
||
int port = Integer.parseInt(hostPort[1]);
|
||
|
||
client = new Socket();
|
||
client.connect(new InetSocketAddress(host, port));
|
||
client.setKeepAlive(true);
|
||
|
||
writer = client.getOutputStream();
|
||
reader = new BufferedReader(new InputStreamReader(client.getInputStream()));
|
||
|
||
String groupid = getGroupid();
|
||
if (groupid == null || groupid.isEmpty()) {
|
||
throw new RuntimeException("ZHubClient groupid can not is empty");
|
||
}
|
||
send("auth", auth);
|
||
send("groupid " + groupid);
|
||
|
||
StringBuffer buf = new StringBuffer("subscribe lock");
|
||
if (mainHub.containsValue(this)) {
|
||
buf.append(" " + APP_NAME);
|
||
}
|
||
for (String topic : getTopics()) {
|
||
buf.append(" ").append(topic);
|
||
}
|
||
send(buf.toString());
|
||
|
||
// 重连 timer 订阅
|
||
timerMap.forEach((name, timer) -> {
|
||
send("timer", name);
|
||
});
|
||
if (retry > 0) {
|
||
logger.warning(String.format("ZHubClient[%s][%s] %s Succeed！", getGroupid(), i + 1, retry > 0 ? "reconnection" : "init"));
|
||
} else {
|
||
logger.fine(String.format("ZHubClient[%s] %s Succeed！", getGroupid(), retry > 0 ? "reconnection" : "init"));
|
||
}
|
||
return true;
|
||
} catch (Exception e) {
|
||
if (i == 0) {
|
||
logger.log(Level.WARNING, String.format("ZHubClient[%s] %s Failed 初始化失败！", getGroupid(), retry == 0 ? "init" : "reconnection"), e);
|
||
try {
|
||
Thread.sleep(1000);
|
||
} catch (InterruptedException interruptedException) {
|
||
interruptedException.printStackTrace();
|
||
}
|
||
} else {
|
||
logger.log(Level.WARNING, String.format("ZHubClient[%s][%s] reconnection Failed！", getGroupid(), i + 1));
|
||
try {
|
||
Thread.sleep(1000 * 5);
|
||
} catch (InterruptedException interruptedException) {
|
||
interruptedException.printStackTrace();
|
||
}
|
||
}
|
||
}
|
||
}
|
||
return false;
|
||
}
|
||
|
||
@Override
|
||
public void unsubscribe(String topic) {
|
||
send("unsubscribe " + topic);
|
||
super.removeEventType(topic);
|
||
}
|
||
|
||
public boolean publish(String topic, Object v) {
|
||
return send("publish", topic, toStr(v));
|
||
}
|
||
|
||
public void broadcast(String topic, Object v) {
|
||
send("broadcast", topic, toStr(v));
|
||
}
|
||
|
||
// 发送 publish 主题消息，若多次发送的 topic + "-" + value 相同，将会做延时重置
|
||
public void delay(String topic, Object v, long millis) {
|
||
send("delay", topic, toStr(v), String.valueOf(millis));
|
||
}
|
||
|
||
// 表达式支持：d+[d,H,m,s]
|
||
public void delay(String topic, Object v, String delayExpr) {
|
||
String endchar = "";
|
||
int delay;
|
||
if (delayExpr.matches("^\\d+[d,H,m,s]$")) {
|
||
endchar = delayExpr.substring(delayExpr.length() - 1);
|
||
delay = Integer.parseInt(delayExpr.substring(0, delayExpr.length() - 1));
|
||
} else {
|
||
if (!delayExpr.matches("^\\d+$")) {
|
||
throw new IllegalArgumentException(String.format("ScheduledCycle period config error: [%s]", delayExpr));
|
||
}
|
||
|
||
delay = Integer.parseInt(delayExpr);
|
||
if (delay <= 0L) {
|
||
throw new IllegalArgumentException(String.format("ScheduledCycle period config error: [%s]", delayExpr));
|
||
}
|
||
}
|
||
|
||
if ("M".equals(endchar)) {
|
||
delay *= (1000 * 60 * 60 * 24 * 30);
|
||
} else if ("d".equals(endchar)) {
|
||
delay *= (1000 * 60 * 60 * 24);
|
||
} else if ("H".equals(endchar)) {
|
||
delay *= (1000 * 60 * 60);
|
||
} else if ("m".equals(endchar)) {
|
||
delay *= (1000 * 60);
|
||
} else if ("s".equals(endchar)) {
|
||
delay *= 1000;
|
||
}
|
||
|
||
delay(topic, v, delay);
|
||
}
|
||
|
||
@Override
|
||
protected String getGroupid() {
|
||
return groupid;
|
||
}
|
||
|
||
@Override
|
||
protected void subscribe(String topic) {
|
||
send("subscribe " + topic); //新增订阅
|
||
}
|
||
|
||
// ================================================== lock ==================================================
|
||
private Map<String, Lock> lockTag = new ConcurrentHashMap<>();
|
||
|
||
public Lock tryLock(String key, int duration) {
|
||
String uuid = Utility.uuid();
|
||
Lock lock = new Lock(key, uuid, duration, this);
|
||
lockTag.put(uuid, lock);
|
||
|
||
try {
|
||
// c.send("lock", key, uuid, strconv.Itoa(duration))
|
||
send("lock", key, uuid, String.valueOf(duration));
|
||
synchronized (lock) {
|
||
lock.wait();
|
||
}
|
||
} catch (InterruptedException e) {
|
||
e.printStackTrace();
|
||
}
|
||
return lock;
|
||
}
|
||
|
||
// 为替换 tryLock 方法做过度准确
|
||
public Lock lock(String key, int duration) {
|
||
String uuid = Utility.uuid();
|
||
Lock lock = new Lock(key, uuid, duration, this);
|
||
lockTag.put(uuid, lock);
|
||
|
||
try {
|
||
// c.send("lock", key, uuid, strconv.Itoa(duration))
|
||
send("lock", key, uuid, String.valueOf(duration));
|
||
synchronized (lock) {
|
||
lock.wait();
|
||
}
|
||
} catch (InterruptedException e) {
|
||
e.printStackTrace();
|
||
}
|
||
return lock;
|
||
}
|
||
|
||
// ================================================== timer ==================================================
|
||
private ConcurrentHashMap<String, Timer> timerMap = new ConcurrentHashMap();
|
||
|
||
class Timer {
|
||
String name;
|
||
//String expr;
|
||
Runnable runnable;
|
||
//boolean single;
|
||
|
||
public String getName() {
|
||
return name;
|
||
}
|
||
|
||
|
||
public Runnable getRunnable() {
|
||
return runnable;
|
||
}
|
||
|
||
public Timer(String name, Runnable runnable) {
|
||
this.name = name;
|
||
this.runnable = runnable;
|
||
}
|
||
}
|
||
|
||
public void timer(String name, Runnable run) {
|
||
timerMap.put(name, new Timer(name, run));
|
||
send("timer", name);
|
||
}
|
||
|
||
@Deprecated
|
||
public void reloadTimer() {
|
||
send("cmd", "reload-timer");
|
||
}
|
||
|
||
// ================================================== rpc ==================================================
|
||
// -- 调用端 --
|
||
private static Map<String, Rpc> rpcMap = new ConcurrentHashMap<>();
|
||
private static Map<String, TypeToken> rpcRetType = new ConcurrentHashMap<>();
|
||
|
||
@Comment("rpc call")
|
||
public RpcResult<Void> rpc(String topic, Object v) {
|
||
return rpc(topic, v, null);
|
||
}
|
||
|
||
@Comment("rpc call")
|
||
public <T, R> RpcResult<R> rpc(String topic, T v, TypeToken<R> typeToken) {
|
||
return rpc(topic, v, typeToken, 0);
|
||
}
|
||
|
||
@Comment("rpc call")
|
||
public <T, R> RpcResult<R> rpc(String topic, T v, TypeToken<R> typeToken, long timeout) {
|
||
Rpc rpc = new Rpc<>(APP_NAME, Utility.uuid(), topic, v);
|
||
String ruk = rpc.getRuk();
|
||
rpcMap.put(ruk, rpc);
|
||
if (typeToken != null) {
|
||
rpcRetType.put(ruk, typeToken);
|
||
}
|
||
try {
|
||
publish(topic, rpc); // send("rpc", topic, toStr(rpc));
|
||
synchronized (rpc) {
|
||
if (timeout <= 0) {
|
||
timeout = 1000 * 15;
|
||
}
|
||
// call timeout default: 15s
|
||
Timers.delay(() -> {
|
||
synchronized (rpc) {
|
||
Rpc rpc1 = rpcMap.get(ruk);
|
||
if (rpc1 == null) {
|
||
return;
|
||
}
|
||
|
||
RpcResult rpcResult = rpc.render(505, "请求超时");
|
||
rpc.setRpcResult(rpcResult);
|
||
logger.warning("rpc timeout: " + convert.convertTo(rpc));
|
||
rpc.notify();
|
||
}
|
||
}, timeout);
|
||
|
||
rpc.wait();
|
||
rpcMap.remove(ruk);
|
||
}
|
||
} catch (InterruptedException e) {
|
||
e.printStackTrace();
|
||
// call error
|
||
RpcResult rpcResult = rpc.render(501, "请求失败");
|
||
rpc.setRpcResult(rpcResult);
|
||
}
|
||
return rpc.getRpcResult();
|
||
}
|
||
|
||
public <T, R> CompletableFuture<RpcResult<R>> rpcAsync(String topic, T v) {
|
||
return CompletableFuture.supplyAsync(() -> rpc(topic, v, null));
|
||
}
|
||
|
||
public <T, R> CompletableFuture<RpcResult<R>> rpcAsync(String topic, T v, TypeToken<R> typeToken) {
|
||
return CompletableFuture.supplyAsync(() -> rpc(topic, v, typeToken));
|
||
}
|
||
|
||
public <T, R> CompletableFuture<RpcResult<R>> rpcAsync(String topic, T v, long timeout) {
|
||
return CompletableFuture.supplyAsync(() -> rpc(topic, v, null, timeout));
|
||
}
|
||
|
||
public <T, R> CompletableFuture<RpcResult<R>> rpcAsync(String topic, T v, TypeToken<R> typeToken, long timeout) {
|
||
return CompletableFuture.supplyAsync(() -> rpc(topic, v, typeToken, timeout));
|
||
}
|
||
|
||
// RpcResult: {ruk:xxx-xxxx, retcode:0}
|
||
@Comment("rpc call back consumer")
|
||
private void rpcAccept(String value) {
|
||
RpcResult resp = convert.convertFrom(new TypeToken<RpcResult<String>>() {
|
||
}.getType(), value);
|
||
|
||
String ruk = resp.getRuk();
|
||
Rpc rpc = rpcMap.remove(ruk);
|
||
if (rpc == null) {
|
||
return;
|
||
}
|
||
TypeToken typeToken = rpcRetType.get(ruk);
|
||
|
||
Object result = resp.getResult();
|
||
if (result != null && typeToken != null && !"java.lang.String".equals(typeToken.getType().getTypeName()) && !"java.lang.Void".equals(typeToken.getType().getTypeName())) {
|
||
result = convert.convertFrom(typeToken.getType(), (String) resp.getResult());
|
||
}
|
||
|
||
resp.setResult(result);
|
||
rpc.setRpcResult(resp);
|
||
synchronized (rpc) {
|
||
rpc.notify();
|
||
}
|
||
}
|
||
|
||
// -- 订阅端 --
|
||
private Set<String> rpcTopics = new HashSet();
|
||
|
||
@Comment("rpc call consumer")
|
||
public <R> void rpcSubscribe(String topic, Function<Rpc<String>, RpcResult<R>> fun) {
|
||
rpcSubscribe(topic, IType.STRING, fun);
|
||
}
|
||
|
||
@Comment("rpc call consumer")
|
||
public <T, R> void rpcSubscribe(String topic, TypeToken<T> typeToken, Function<Rpc<T>, RpcResult<R>> fun) {
|
||
Consumer<String> consumer = v -> {
|
||
Rpc<T> rpc = null;
|
||
try {
|
||
rpc = convert.convertFrom(new TypeToken<Rpc<String>>() {
|
||
}.getType(), v);
|
||
|
||
// 参数转换
|
||
T paras = convert.convertFrom(typeToken.getType(), (String) rpc.getValue());
|
||
rpc.setValue(paras);
|
||
RpcResult result = fun.apply(rpc);
|
||
result.setResult(toStr(result.getResult()));
|
||
publish(rpc.getBackTopic(), result);
|
||
} catch (Exception e) {
|
||
logger.log(Level.WARNING, "rpc call consumer error: " + v, e);
|
||
publish(rpc.getBackTopic(), rpc.retError("服务调用失败！"));
|
||
}
|
||
// back
|
||
};
|
||
|
||
rpcTopics.add(topic);
|
||
subscribe(topic, consumer);
|
||
}
|
||
}
|