3 changed files with 162 additions and 8 deletions
@ -0,0 +1,150 @@ |
|||||
|
package com.yvan.logisticsModel; |
||||
|
|
||||
|
import com.yvan.logisticsEnv.EnvConfig; |
||||
|
import lombok.SneakyThrows; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.eclipse.paho.mqttv5.client.*; |
||||
|
import org.eclipse.paho.mqttv5.common.MqttException; |
||||
|
import org.eclipse.paho.mqttv5.common.MqttMessage; |
||||
|
import org.eclipse.paho.mqttv5.common.packet.MqttProperties; |
||||
|
|
||||
|
import java.nio.charset.StandardCharsets; |
||||
|
import java.util.concurrent.CountDownLatch; |
||||
|
|
||||
|
@Slf4j |
||||
|
public class FrontendPushService implements MqttCallback { |
||||
|
private final LogisticsRuntime runtime; |
||||
|
public final String rcsFrontendTopic; |
||||
|
private volatile MqttClient clientForSend; |
||||
|
private volatile MqttClient client; |
||||
|
private CountDownLatch connectLatch = new CountDownLatch(1); |
||||
|
|
||||
|
public FrontendPushService(LogisticsRuntime runtime) { |
||||
|
this.runtime = runtime; |
||||
|
this.rcsFrontendTopic = "/rcs/" + runtime.projectUUID + "/" + runtime.envId + "/frontend"; |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void disconnected(MqttDisconnectResponse disconnectResponse) { |
||||
|
log.info("mqtt disconnected"); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void mqttErrorOccurred(MqttException exception) { |
||||
|
log.error("mqtt error occurred"); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void messageArrived(String topic, MqttMessage message) throws Exception { |
||||
|
log.info("Message arrived on topic {}: {}", topic, new String(message.getPayload(), StandardCharsets.UTF_8)); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void deliveryComplete(IMqttToken token) { |
||||
|
log.info("Message delivery complete: {}", token); |
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void connectComplete(boolean reconnect, String serverURI) { |
||||
|
// BannerUtils.printConfig(log, "MQTT 开启监听", new String[]{serverURI + " topic: /agv_robot/status"});
|
||||
|
log.info("MQTT client connected to server: {}", serverURI); |
||||
|
connectLatch.countDown(); // 放行
|
||||
|
} |
||||
|
|
||||
|
@Override |
||||
|
public void authPacketArrived(int reasonCode, MqttProperties properties) { |
||||
|
} |
||||
|
|
||||
|
@SneakyThrows |
||||
|
public void stop() { |
||||
|
if (client != null && client.isConnected()) { |
||||
|
client.disconnect(); |
||||
|
client.close(); |
||||
|
log.info("MQTT client disconnected and closed."); |
||||
|
} else { |
||||
|
log.warn("MQTT client is not connected, no action taken."); |
||||
|
} |
||||
|
|
||||
|
if (clientForSend != null && clientForSend.isConnected()) { |
||||
|
clientForSend.disconnect(); |
||||
|
clientForSend.close(); |
||||
|
log.info("MQTT clientForSend disconnected and closed."); |
||||
|
} else { |
||||
|
log.warn("MQTT clientForSend is not connected, no action taken."); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 创建一个表示服务器离线的最后消息 |
||||
|
*/ |
||||
|
public MqttMessage createWillMessage() { |
||||
|
MqttMessage lastMessage = new MqttMessage(); |
||||
|
String msg = String.format(""" |
||||
|
{ |
||||
|
"id": "serverOffline", |
||||
|
"content": { |
||||
|
"project": "%s", |
||||
|
"env": %s, |
||||
|
"status": "offline", |
||||
|
"server": "%s", |
||||
|
} |
||||
|
} |
||||
|
""", runtime.projectUUID, runtime.envId, runtime.clientId); |
||||
|
lastMessage.setPayload(msg.getBytes(StandardCharsets.UTF_8)); |
||||
|
lastMessage.setQos(0); |
||||
|
lastMessage.setRetained(true); |
||||
|
return lastMessage; |
||||
|
} |
||||
|
|
||||
|
@SneakyThrows |
||||
|
public void start(EnvConfig.MqttConfig mqttConfig) { |
||||
|
String brokerUrl = mqttConfig.getBrokerUrl(); // "tcp://10.10.203.239:1885"
|
||||
|
String username = mqttConfig.getUsername(); // admin
|
||||
|
String password = mqttConfig.getPassword(); // admin
|
||||
|
|
||||
|
clientForSend = new MqttClient(brokerUrl, runtime.clientId + "_front_send"); // String clientId = "LUOYIFAN-PC_send"
|
||||
|
client = new MqttClient(brokerUrl, runtime.clientId + "_front"); // String clientId = "LUOYIFAN-PC"
|
||||
|
MqttConnectionOptions options = new MqttConnectionOptions(); |
||||
|
options.setServerURIs(new String[]{brokerUrl}); |
||||
|
options.setAutomaticReconnect(true); |
||||
|
options.setUserName(username); |
||||
|
options.setPassword(password.getBytes()); |
||||
|
options.setConnectionTimeout(1); |
||||
|
options.setKeepAliveInterval(20); |
||||
|
options.setExecutorServiceTimeout(1); |
||||
|
// 最后一条消息，表示服务离线
|
||||
|
options.setWill(this.rcsFrontendTopic, createWillMessage()); |
||||
|
|
||||
|
client.setCallback(this); |
||||
|
client.connect(options); |
||||
|
|
||||
|
clientForSend.connect(options); |
||||
|
connectLatch.await(); |
||||
|
|
||||
|
// 发送一条消息代表上线
|
||||
|
this.publish(String.format(""" |
||||
|
{ |
||||
|
"id": "serverOnline", |
||||
|
"content": { |
||||
|
"project": "%s", |
||||
|
"env": %s, |
||||
|
"status": "online", |
||||
|
"server": "%s" |
||||
|
} |
||||
|
} |
||||
|
""", runtime.projectUUID, runtime.envId, runtime.clientId)); |
||||
|
} |
||||
|
|
||||
|
@SneakyThrows |
||||
|
void publish(String payloadString) { |
||||
|
MqttMessage message = new MqttMessage(payloadString.getBytes(StandardCharsets.UTF_8)); |
||||
|
message.setQos(0); |
||||
|
message.setRetained(false); |
||||
|
|
||||
|
if (this.clientForSend.isConnected()) { |
||||
|
this.clientForSend.publish(this.rcsFrontendTopic, message); |
||||
|
} else { |
||||
|
throw new RuntimeException("MQTT client is not connected"); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue