|
|
@@ -1,14 +1,16 @@ |
|
|
|
use anyhow::{Context, Result}; |
|
|
|
use rumqttc::{Client, MqttOptions, QoS, Transport}; |
|
|
|
use serde::{Serialize, Deserialize}; |
|
|
|
use serde_json; |
|
|
|
use std::io::{self, Read}; |
|
|
|
use std::time::Duration; |
|
|
|
use std::thread; |
|
|
|
use serialport; |
|
|
|
use actix_web::{web, App, HttpServer, HttpResponse}; |
|
|
|
use actix_cors::Cors; |
|
|
|
use std::fs; |
|
|
|
use futures_util::{SinkExt, StreamExt}; |
|
|
|
use tokio::net::TcpListener; |
|
|
|
use tokio_tungstenite::accept_async; |
|
|
|
use tokio::sync::broadcast; |
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize)] |
|
|
|
struct WeightData { |
|
|
@@ -26,23 +28,11 @@ struct ScaleTypeResponse { |
|
|
|
scale_type: String, |
|
|
|
} |
|
|
|
|
|
|
|
#[derive(Deserialize)] |
|
|
|
struct MqttConfig { |
|
|
|
client_id: String, |
|
|
|
host: String, |
|
|
|
port: u16, |
|
|
|
username: String, |
|
|
|
password: String, |
|
|
|
keep_alive_secs: u64, |
|
|
|
topic_prefix: String, |
|
|
|
} |
|
|
|
|
|
|
|
#[derive(Deserialize)] |
|
|
|
struct Config { |
|
|
|
scale_type: String, |
|
|
|
serial_port: String, |
|
|
|
baud_rate: u32, |
|
|
|
mqtt: MqttConfig, |
|
|
|
} |
|
|
|
|
|
|
|
fn read_config() -> Result<Config> { |
|
|
@@ -88,8 +78,111 @@ async fn get_scale() -> HttpResponse { |
|
|
|
HttpResponse::Ok().json(response) |
|
|
|
} |
|
|
|
|
|
|
|
// WebSocket测试客户端函数 |
|
|
|
async fn test_websocket_client() -> Result<()> { |
|
|
|
println!("开始测试WebSocket客户端..."); |
|
|
|
|
|
|
|
// 连接到WebSocket服务器 |
|
|
|
println!("正在连接到WebSocket服务器..."); |
|
|
|
let (ws_stream, _) = tokio_tungstenite::connect_async("ws://127.0.0.1:8081") |
|
|
|
.await |
|
|
|
.context("无法连接到WebSocket服务器")?; |
|
|
|
println!("已成功连接到WebSocket服务器"); |
|
|
|
|
|
|
|
let (_, mut read) = ws_stream.split(); |
|
|
|
|
|
|
|
// 持续接收消息 |
|
|
|
println!("等待接收数据..."); |
|
|
|
while let Some(message) = read.next().await { |
|
|
|
match message { |
|
|
|
Ok(msg) => { |
|
|
|
if let Ok(text) = msg.into_text() { |
|
|
|
println!("收到原始数据: {}", text); |
|
|
|
// 尝试解析JSON数据 |
|
|
|
match serde_json::from_str::<WeightData>(&text) { |
|
|
|
Ok(weight_data) => { |
|
|
|
println!("收到重量数据: {} (时间戳: {})", weight_data.weight, weight_data.timestamp); |
|
|
|
} |
|
|
|
Err(e) => { |
|
|
|
println!("JSON解析错误: {}", e); |
|
|
|
println!("原始数据: {}", text); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
Err(e) => { |
|
|
|
eprintln!("WebSocket接收错误: {}", e); |
|
|
|
break; |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
Ok(()) |
|
|
|
} |
|
|
|
|
|
|
|
// 模拟串口数据发送 |
|
|
|
async fn simulate_serial_data() -> Result<()> { |
|
|
|
println!("开始模拟串口数据发送..."); |
|
|
|
|
|
|
|
// 创建一个TCP连接到WebSocket服务器 |
|
|
|
println!("正在连接到WebSocket服务器..."); |
|
|
|
let (ws_stream, _) = tokio_tungstenite::connect_async("ws://127.0.0.1:8081") |
|
|
|
.await |
|
|
|
.context("无法连接到WebSocket服务器")?; |
|
|
|
println!("已成功连接到WebSocket服务器"); |
|
|
|
|
|
|
|
let (mut write, _) = ws_stream.split(); |
|
|
|
|
|
|
|
// 模拟每秒发送一个随机重量数据 |
|
|
|
let mut interval = tokio::time::interval(Duration::from_secs(1)); |
|
|
|
let mut counter = 0; |
|
|
|
|
|
|
|
println!("开始发送模拟数据 (按Ctrl+C退出)..."); |
|
|
|
loop { |
|
|
|
interval.tick().await; |
|
|
|
counter += 1; |
|
|
|
|
|
|
|
// 生成一个模拟的重量值(10.5 - 20.5之间的随机数) |
|
|
|
let weight = format!("{:.1}", 10.5 + (counter % 10) as f32); |
|
|
|
|
|
|
|
// 创建数据结构 |
|
|
|
let weight_data = WeightData { |
|
|
|
weight: weight.clone(), |
|
|
|
timestamp: std::time::SystemTime::now() |
|
|
|
.duration_since(std::time::UNIX_EPOCH) |
|
|
|
.unwrap() |
|
|
|
.as_secs(), |
|
|
|
}; |
|
|
|
|
|
|
|
// 序列化并发送数据 |
|
|
|
if let Ok(json_data) = serde_json::to_string(&weight_data) { |
|
|
|
println!("发送模拟数据: {} (JSON: {})", weight, json_data); |
|
|
|
if let Err(e) = write.send(tokio_tungstenite::tungstenite::Message::Text(json_data)).await { |
|
|
|
eprintln!("发送数据错误: {}", e); |
|
|
|
break; |
|
|
|
} |
|
|
|
println!("数据发送成功"); |
|
|
|
} |
|
|
|
|
|
|
|
// 等待一小段时间 |
|
|
|
tokio::time::sleep(Duration::from_secs(1)).await; |
|
|
|
} |
|
|
|
|
|
|
|
Ok(()) |
|
|
|
} |
|
|
|
|
|
|
|
#[actix_web::main] |
|
|
|
async fn main() -> Result<()> { |
|
|
|
// 检查命令行参数 |
|
|
|
let args: Vec<String> = std::env::args().collect(); |
|
|
|
if args.len() > 1 { |
|
|
|
match args[1].as_str() { |
|
|
|
"--test-client" => return test_websocket_client().await, |
|
|
|
"--simulate" => return simulate_serial_data().await, |
|
|
|
_ => {} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
println!("程序启动..."); |
|
|
|
|
|
|
|
// 启动HTTP服务器 |
|
|
@@ -114,10 +207,10 @@ async fn main() -> Result<()> { |
|
|
|
|
|
|
|
println!("HTTP服务器已启动,监听在 http://127.0.0.1:8080"); |
|
|
|
|
|
|
|
// 在新线程中运行MQTT和串口服务 |
|
|
|
let mqtt_handle = tokio::spawn(async { |
|
|
|
if let Err(e) = run_mqtt_and_serial().await { |
|
|
|
eprintln!("MQTT/串口服务错误: {}", e); |
|
|
|
// 在新线程中运行WebSocket和串口服务 |
|
|
|
let _websocket_handle = tokio::spawn(async { |
|
|
|
if let Err(e) = run_websocket_and_serial().await { |
|
|
|
eprintln!("WebSocket/串口服务错误: {}", e); |
|
|
|
} |
|
|
|
}); |
|
|
|
|
|
|
@@ -127,119 +220,154 @@ async fn main() -> Result<()> { |
|
|
|
Ok(()) |
|
|
|
} |
|
|
|
|
|
|
|
// MQTT和串口处理函数 |
|
|
|
async fn run_mqtt_and_serial() -> Result<()> { |
|
|
|
// WebSocket和串口处理函数 |
|
|
|
async fn run_websocket_and_serial() -> Result<()> { |
|
|
|
// 读取配置 |
|
|
|
let config = read_config()?; |
|
|
|
|
|
|
|
// 创建 MQTT 客户端 |
|
|
|
let mut mqttopts = MqttOptions::new( |
|
|
|
&config.mqtt.client_id, |
|
|
|
&config.mqtt.host, |
|
|
|
config.mqtt.port |
|
|
|
); |
|
|
|
mqttopts.set_keep_alive(Duration::from_secs(60)); // 增加keep-alive时间到60秒 |
|
|
|
mqttopts.set_credentials(&config.mqtt.username, &config.mqtt.password); |
|
|
|
mqttopts.set_clean_session(true); |
|
|
|
mqttopts.set_transport(Transport::Tcp); |
|
|
|
mqttopts.set_max_packet_size(100 * 1024, 100 * 1024); // 设置发送和接收的最大包大小 |
|
|
|
let (mut client, mut connection) = Client::new(mqttopts, 10); |
|
|
|
|
|
|
|
// 在单独的线程中处理 MQTT 连接 |
|
|
|
thread::spawn(move || { |
|
|
|
loop { // 添加无限循环来保持重连 |
|
|
|
for notification in connection.iter() { |
|
|
|
match notification { |
|
|
|
Ok(_) => {}, |
|
|
|
Err(e) => { |
|
|
|
eprintln!("MQTT连接错误: {:?}", e); |
|
|
|
// 短暂等待后继续尝试重连 |
|
|
|
thread::sleep(Duration::from_secs(5)); |
|
|
|
break; |
|
|
|
// 创建一个广播通道,用于向所有WebSocket客户端发送数据 |
|
|
|
let (tx, _rx) = broadcast::channel::<String>(100); |
|
|
|
|
|
|
|
// 启动WebSocket服务器 |
|
|
|
let listener = TcpListener::bind("127.0.0.1:8081").await?; |
|
|
|
println!("WebSocket服务器已启动,监听在 ws://127.0.0.1:8081"); |
|
|
|
|
|
|
|
// 在单独的任务中处理串口数据 |
|
|
|
let tx_serial = tx.clone(); |
|
|
|
let _serial_handle = tokio::spawn(async move { |
|
|
|
// 打开配置的串口 |
|
|
|
println!("正在尝试打开串口 {}...", config.serial_port); |
|
|
|
let mut port = serialport::new(&config.serial_port, config.baud_rate) |
|
|
|
.timeout(Duration::from_millis(1000)) |
|
|
|
.open() |
|
|
|
.with_context(|| format!("无法打开串口 {}", config.serial_port))?; |
|
|
|
|
|
|
|
println!("成功打开串口 {}!", config.serial_port); |
|
|
|
println!("正在读取数据... (按 Ctrl+C 退出)"); |
|
|
|
|
|
|
|
// 增加缓冲区大小并使用String来累积数据 |
|
|
|
let mut serial_buf: Vec<u8> = vec![0; 1024]; |
|
|
|
let mut accumulated_data = String::new(); |
|
|
|
|
|
|
|
loop { |
|
|
|
match port.read(serial_buf.as_mut_slice()) { |
|
|
|
Ok(t) => { |
|
|
|
if t > 0 { |
|
|
|
// 将新数据添加到累积的字符串中 |
|
|
|
if let Ok(data) = String::from_utf8(serial_buf[..t].to_vec()) { |
|
|
|
accumulated_data.push_str(&data); |
|
|
|
|
|
|
|
// 检查是否有完整的数据行(以换行符或回车符结束) |
|
|
|
if accumulated_data.contains('\n') || accumulated_data.contains('\r') { |
|
|
|
// 处理累积的数据 |
|
|
|
let lines: Vec<&str> = accumulated_data |
|
|
|
.split(|c| c == '\n' || c == '\r') |
|
|
|
.filter(|s| !s.is_empty()) |
|
|
|
.collect(); |
|
|
|
|
|
|
|
for line in lines { |
|
|
|
let trimmed_data = line.trim(); |
|
|
|
if !trimmed_data.is_empty() { |
|
|
|
println!("收到串口数据: {}", trimmed_data); |
|
|
|
|
|
|
|
// 创建带时间戳的数据结构 |
|
|
|
let weight_data = WeightData { |
|
|
|
weight: trimmed_data.to_string(), |
|
|
|
timestamp: std::time::SystemTime::now() |
|
|
|
.duration_since(std::time::UNIX_EPOCH) |
|
|
|
.unwrap() |
|
|
|
.as_secs(), |
|
|
|
}; |
|
|
|
|
|
|
|
// 序列化数据 |
|
|
|
if let Ok(json_data) = serde_json::to_string(&weight_data) { |
|
|
|
println!("发送串口数据到WebSocket: {}", json_data); |
|
|
|
// 发送数据到所有WebSocket客户端 |
|
|
|
if let Err(e) = tx_serial.send(json_data) { |
|
|
|
eprintln!("发送数据到广播通道失败: {}", e); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
// 清除已处理的数据 |
|
|
|
accumulated_data.clear(); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
Err(ref e) if e.kind() == io::ErrorKind::TimedOut => { |
|
|
|
// 超时是正常的,继续尝试读取 |
|
|
|
continue; |
|
|
|
} |
|
|
|
Err(e) => { |
|
|
|
eprintln!("串口读取错误: {}", e); |
|
|
|
break; |
|
|
|
} |
|
|
|
} |
|
|
|
eprintln!("MQTT连接断开,5秒后尝试重连..."); |
|
|
|
thread::sleep(Duration::from_secs(5)); |
|
|
|
} |
|
|
|
Ok::<_, anyhow::Error>(()) |
|
|
|
}); |
|
|
|
|
|
|
|
// 打开配置的串口 |
|
|
|
println!("正在尝试打开串口 {}...", config.serial_port); |
|
|
|
let mut port = serialport::new(&config.serial_port, config.baud_rate) |
|
|
|
.timeout(Duration::from_millis(1000)) |
|
|
|
.open() |
|
|
|
.with_context(|| format!("无法打开串口 {}", config.serial_port))?; |
|
|
|
// 处理WebSocket连接 |
|
|
|
while let Ok((stream, addr)) = listener.accept().await { |
|
|
|
println!("新的WebSocket连接: {}", addr); |
|
|
|
let ws_stream = accept_async(stream).await.expect("Failed to accept WebSocket"); |
|
|
|
let tx = tx.clone(); |
|
|
|
|
|
|
|
// 为每个WebSocket连接创建一个新任务 |
|
|
|
tokio::spawn(async move { |
|
|
|
println!("开始处理WebSocket连接: {}", addr); |
|
|
|
let (mut ws_sender, mut ws_receiver) = ws_stream.split(); |
|
|
|
let mut rx = tx.subscribe(); |
|
|
|
|
|
|
|
println!("成功打开串口 {}!", config.serial_port); |
|
|
|
println!("正在读取数据... (按 Ctrl+C 退出)"); |
|
|
|
// 创建两个任务:一个用于接收消息,一个用于发送消息 |
|
|
|
let mut send_task = tokio::spawn(async move { |
|
|
|
println!("[{}] 等待接收广播数据...", addr); |
|
|
|
while let Ok(msg) = rx.recv().await { |
|
|
|
println!("[{}] 准备发送数据: {}", addr, msg); |
|
|
|
if let Err(e) = ws_sender.send(tokio_tungstenite::tungstenite::Message::Text(msg.clone())).await { |
|
|
|
eprintln!("[{}] 发送WebSocket消息失败: {}", addr, e); |
|
|
|
break; |
|
|
|
} |
|
|
|
println!("[{}] 成功发送数据", addr); |
|
|
|
} |
|
|
|
}); |
|
|
|
|
|
|
|
// 增加缓冲区大小并使用String来累积数据 |
|
|
|
let mut serial_buf: Vec<u8> = vec![0; 1024]; |
|
|
|
let mut accumulated_data = String::new(); |
|
|
|
|
|
|
|
loop { |
|
|
|
match port.read(serial_buf.as_mut_slice()) { |
|
|
|
Ok(t) => { |
|
|
|
if t > 0 { |
|
|
|
// 将新数据添加到累积的字符串中 |
|
|
|
if let Ok(data) = String::from_utf8(serial_buf[..t].to_vec()) { |
|
|
|
accumulated_data.push_str(&data); |
|
|
|
|
|
|
|
// 检查是否有完整的数据行(以换行符或回车符结束) |
|
|
|
if accumulated_data.contains('\n') || accumulated_data.contains('\r') { |
|
|
|
// 处理累积的数据 |
|
|
|
let lines: Vec<&str> = accumulated_data |
|
|
|
.split(|c| c == '\n' || c == '\r') |
|
|
|
.filter(|s| !s.is_empty()) |
|
|
|
.collect(); |
|
|
|
|
|
|
|
for line in lines { |
|
|
|
let trimmed_data = line.trim(); |
|
|
|
if !trimmed_data.is_empty() { |
|
|
|
println!("收到完整数据: {}", trimmed_data); |
|
|
|
|
|
|
|
// 创建权重数据结构 |
|
|
|
let weight_data = WeightData { |
|
|
|
weight: trimmed_data.to_string(), |
|
|
|
timestamp: std::time::SystemTime::now() |
|
|
|
.duration_since(std::time::UNIX_EPOCH) |
|
|
|
.unwrap() |
|
|
|
.as_secs(), |
|
|
|
}; |
|
|
|
|
|
|
|
// 序列化数据 |
|
|
|
let json_data = serde_json::to_string(&weight_data)?; |
|
|
|
println!("当前JSON数据: {}", json_data); |
|
|
|
|
|
|
|
// 获取MAC地址 |
|
|
|
let mac_address = get_mac_address(); |
|
|
|
println!("MAC地址: {}", mac_address); |
|
|
|
// 发布到 MQTT,主题中包含MAC地址 |
|
|
|
let topic = format!("{}/{}", config.mqtt.topic_prefix, mac_address); |
|
|
|
if let Err(e) = client.publish(topic, QoS::AtLeastOnce, false, json_data) { |
|
|
|
eprintln!("MQTT发布错误: {:?}", e); |
|
|
|
} else { |
|
|
|
println!("成功发送数据到MQTT"); |
|
|
|
} |
|
|
|
let tx_clone = tx.clone(); |
|
|
|
let mut recv_task = tokio::spawn(async move { |
|
|
|
println!("[{}] 开始监听WebSocket消息...", addr); |
|
|
|
while let Some(result) = ws_receiver.next().await { |
|
|
|
match result { |
|
|
|
Ok(msg) => { |
|
|
|
if let Ok(text) = msg.into_text() { |
|
|
|
println!("[{}] 收到WebSocket消息: {}", addr, text); |
|
|
|
// 将收到的消息广播给所有客户端 |
|
|
|
if let Err(e) = tx_clone.send(text.clone()) { |
|
|
|
eprintln!("[{}] 广播消息失败: {}", addr, e); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
// 清空累积的数据 |
|
|
|
accumulated_data.clear(); |
|
|
|
} |
|
|
|
Err(e) => { |
|
|
|
eprintln!("[{}] WebSocket接收错误: {}", addr, e); |
|
|
|
break; |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
}); |
|
|
|
|
|
|
|
// 等待任意一个任务结束 |
|
|
|
tokio::select! { |
|
|
|
_ = (&mut send_task) => { |
|
|
|
println!("[{}] 发送任务结束", addr); |
|
|
|
} |
|
|
|
_ = (&mut recv_task) => { |
|
|
|
println!("[{}] 接收任务结束", addr); |
|
|
|
} |
|
|
|
} |
|
|
|
Err(ref e) if e.kind() == io::ErrorKind::TimedOut => { |
|
|
|
// Timeout is not an error, just continue |
|
|
|
continue; |
|
|
|
} |
|
|
|
Err(e) => { |
|
|
|
eprintln!("发生错误: {}", e); |
|
|
|
break; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
println!("[{}] WebSocket连接结束", addr); |
|
|
|
}); |
|
|
|
} |
|
|
|
|
|
|
|
Ok(()) |
|
|
|