chore: 初始化测试场景
This commit is contained in:
Del Wang
@@ -0,0 +1,305 @@
|
||||
use crate::audio::codec::OpusCodec;
|
||||
use crate::audio::config::AudioConfig;
|
||||
use crate::audio::wav::{WavReader, WavWriter};
|
||||
use crate::net::discovery::Discovery;
|
||||
use crate::net::network::{AudioSocket, ControlConnection, ServerNetwork};
|
||||
use crate::net::protocol::{AudioPacket, ControlPacket, DeviceInfo, RpcResult};
|
||||
use anyhow::{Context, Result};
|
||||
use parking_lot::Mutex;
|
||||
use std::collections::HashMap;
|
||||
use std::net::SocketAddr;
|
||||
use std::sync::Arc;
|
||||
use std::sync::atomic::{AtomicU32, Ordering};
|
||||
use tokio::sync::{broadcast, oneshot};
|
||||
|
||||
pub struct RpcManager {
|
||||
next_id: AtomicU32,
|
||||
pending: Mutex<HashMap<u32, oneshot::Sender<RpcResult>>>,
|
||||
}
|
||||
|
||||
impl RpcManager {
|
||||
pub fn new() -> Self {
|
||||
Self {
|
||||
next_id: AtomicU32::new(1),
|
||||
pending: Mutex::new(HashMap::new()),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn alloc_id(&self) -> (u32, oneshot::Receiver<RpcResult>) {
|
||||
let id = self.next_id.fetch_add(1, Ordering::SeqCst);
|
||||
let (tx, rx) = oneshot::channel();
|
||||
self.pending.lock().insert(id, tx);
|
||||
(id, rx)
|
||||
}
|
||||
|
||||
pub fn fulfill(&self, id: u32, result: RpcResult) {
|
||||
if let Some(tx) = self.pending.lock().remove(&id) {
|
||||
let _ = tx.send(result);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub struct Session {
|
||||
pub info: DeviceInfo,
|
||||
pub control: Arc<tokio::sync::Mutex<ControlConnection>>,
|
||||
pub addr: SocketAddr,
|
||||
pub rpc: Arc<RpcManager>,
|
||||
}
|
||||
|
||||
pub struct Server {
|
||||
sessions: Arc<Mutex<HashMap<SocketAddr, Arc<Session>>>>,
|
||||
audio_socket: Arc<AudioSocket>,
|
||||
}
|
||||
|
||||
impl Server {
|
||||
pub async fn new() -> Result<Self> {
|
||||
Ok(Self {
|
||||
sessions: Arc::new(Mutex::new(HashMap::new())),
|
||||
audio_socket: Arc::new(AudioSocket::bind().await?),
|
||||
})
|
||||
}
|
||||
|
||||
pub async fn run(self: Arc<Self>, port: u16) -> Result<()> {
|
||||
let network = ServerNetwork::setup(port).await?;
|
||||
println!("服务端启动在: {}", network.local_addr()?);
|
||||
|
||||
// 启动广播
|
||||
Discovery::start_broadcast(port).await?;
|
||||
|
||||
loop {
|
||||
let (control, addr) = network.accept().await?;
|
||||
let server = self.clone();
|
||||
tokio::spawn(async move {
|
||||
if let Err(e) = server.handle_connection(control, addr).await {
|
||||
eprintln!("连接 {} 出错: {}", addr, e);
|
||||
}
|
||||
server.sessions.lock().remove(&addr);
|
||||
println!("客户端 {} 断开连接", addr);
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
async fn handle_connection(
|
||||
&self,
|
||||
mut control: ControlConnection,
|
||||
addr: SocketAddr,
|
||||
) -> Result<()> {
|
||||
println!("新客户端连接: {}", addr);
|
||||
|
||||
// 握手认证
|
||||
let info = match control.recv_packet().await? {
|
||||
ControlPacket::ClientIdentify { info } => info,
|
||||
p => return Err(anyhow::anyhow!("预期的握手包，收到: {:?}", p)),
|
||||
};
|
||||
|
||||
println!(
|
||||
"客户端识别: {} ({}) v{}",
|
||||
info.model, info.mac, info.version
|
||||
);
|
||||
control.send_packet(&ControlPacket::IdentifyOk).await?;
|
||||
|
||||
let rpc = Arc::new(RpcManager::new());
|
||||
let control = Arc::new(tokio::sync::Mutex::new(control));
|
||||
let session = Arc::new(Session {
|
||||
info,
|
||||
control: control.clone(),
|
||||
addr,
|
||||
rpc: rpc.clone(),
|
||||
});
|
||||
|
||||
self.sessions.lock().insert(addr, session.clone());
|
||||
|
||||
// 处理控制消息循环
|
||||
loop {
|
||||
let mut ctrl = control.lock().await;
|
||||
let packet = ctrl.recv_packet().await?;
|
||||
drop(ctrl); // 释放锁以允许发送
|
||||
|
||||
match packet {
|
||||
ControlPacket::RpcResponse { id, result } => {
|
||||
rpc.fulfill(id, result);
|
||||
}
|
||||
ControlPacket::Ping => {
|
||||
control
|
||||
.lock()
|
||||
.await
|
||||
.send_packet(&ControlPacket::Pong)
|
||||
.await?;
|
||||
}
|
||||
_ => {}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 暴露给外部调用的方法
|
||||
pub async fn call_shell(&self, addr: SocketAddr, cmd: &str) -> Result<RpcResult> {
|
||||
let session = self
|
||||
.sessions
|
||||
.lock()
|
||||
.get(&addr)
|
||||
.cloned()
|
||||
.context("未找到 Session")?;
|
||||
let (id, rx) = session.rpc.alloc_id();
|
||||
session
|
||||
.control
|
||||
.lock()
|
||||
.await
|
||||
.send_packet(&ControlPacket::RpcRequest {
|
||||
id,
|
||||
method: "shell".to_string(),
|
||||
args: vec![cmd.to_string()],
|
||||
})
|
||||
.await?;
|
||||
Ok(rx.await?)
|
||||
}
|
||||
|
||||
pub async fn start_recording(&self, addr: SocketAddr, config: AudioConfig) -> Result<()> {
|
||||
let session = self
|
||||
.sessions
|
||||
.lock()
|
||||
.get(&addr)
|
||||
.cloned()
|
||||
.context("未找到 Session")?;
|
||||
|
||||
// 发送开始录音指令
|
||||
session
|
||||
.control
|
||||
.lock()
|
||||
.await
|
||||
.send_packet(&ControlPacket::StartRecording {
|
||||
config: config.clone(),
|
||||
})
|
||||
.await?;
|
||||
|
||||
let audio_socket = self.audio_socket.clone();
|
||||
tokio::spawn(async move {
|
||||
if let Err(e) = save_audio_to_wav(audio_socket, config, "temp/recorded.wav").await {
|
||||
eprintln!("保存录音失败: {}", e);
|
||||
}
|
||||
});
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub async fn stop_recording(&self, addr: SocketAddr) -> Result<()> {
|
||||
let session = self
|
||||
.sessions
|
||||
.lock()
|
||||
.get(&addr)
|
||||
.cloned()
|
||||
.context("未找到 Session")?;
|
||||
session
|
||||
.control
|
||||
.lock()
|
||||
.await
|
||||
.send_packet(&ControlPacket::StopRecording)
|
||||
.await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub async fn start_playback(&self, addr: SocketAddr, config: AudioConfig) -> Result<()> {
|
||||
let session = self
|
||||
.sessions
|
||||
.lock()
|
||||
.get(&addr)
|
||||
.cloned()
|
||||
.context("未找到 Session")?;
|
||||
|
||||
session
|
||||
.control
|
||||
.lock()
|
||||
.await
|
||||
.send_packet(&ControlPacket::StartPlayback {
|
||||
config: config.clone(),
|
||||
})
|
||||
.await?;
|
||||
|
||||
let audio_socket = self.audio_socket.clone();
|
||||
tokio::spawn(async move {
|
||||
if let Err(e) = stream_wav_to_client(audio_socket, addr, config, "temp/test.wav").await
|
||||
{
|
||||
eprintln!("推流失败: {}", e);
|
||||
}
|
||||
});
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub async fn stop_playback(&self, addr: SocketAddr) -> Result<()> {
|
||||
let session = self
|
||||
.sessions
|
||||
.lock()
|
||||
.get(&addr)
|
||||
.cloned()
|
||||
.context("未找到 Session")?;
|
||||
session
|
||||
.control
|
||||
.lock()
|
||||
.await
|
||||
.send_packet(&ControlPacket::StopPlayback)
|
||||
.await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub fn get_sessions(&self) -> Vec<SocketAddr> {
|
||||
self.sessions.lock().keys().cloned().collect()
|
||||
}
|
||||
}
|
||||
|
||||
async fn save_audio_to_wav(
|
||||
socket: Arc<AudioSocket>,
|
||||
config: AudioConfig,
|
||||
path: &str,
|
||||
) -> Result<()> {
|
||||
std::fs::create_dir_all("temp")?;
|
||||
let mut writer = WavWriter::create(path, config.sample_rate, config.channels)?;
|
||||
let mut codec = OpusCodec::new(&config)?;
|
||||
let mut pcm_buf = vec![0i16; config.frame_size];
|
||||
let mut udp_buf = vec![0u8; 4096];
|
||||
|
||||
println!("正在录制到 {}...", path);
|
||||
|
||||
// 这里需要一个停止机制，目前简单起见，如果一段时间没收到包就停止，或者通过全局状态
|
||||
// 为简单演示，我们录制 100 个包
|
||||
for _ in 0..100 {
|
||||
let (packet, _) = socket.recv_packet(&mut udp_buf).await?;
|
||||
let pcm_len = codec.decode(&packet.data, &mut pcm_buf)?;
|
||||
writer.write_samples(&pcm_buf[..pcm_len])?;
|
||||
}
|
||||
|
||||
writer.finalize()?;
|
||||
println!("录制完成: {}", path);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
async fn stream_wav_to_client(
|
||||
socket: Arc<AudioSocket>,
|
||||
target: SocketAddr,
|
||||
config: AudioConfig,
|
||||
path: &str,
|
||||
) -> Result<()> {
|
||||
let mut reader = WavReader::open(path)?;
|
||||
let mut codec = OpusCodec::new(&config)?;
|
||||
let mut pcm_buf = vec![0i16; config.frame_size];
|
||||
let mut opus_buf = vec![0u8; 4096];
|
||||
|
||||
println!("正在从 {} 推流...", path);
|
||||
|
||||
loop {
|
||||
let n = reader.read_samples(&mut pcm_buf)?;
|
||||
if n == 0 {
|
||||
break;
|
||||
}
|
||||
let opus_len = codec.encode(&pcm_buf[..n], &mut opus_buf)?;
|
||||
let packet = AudioPacket {
|
||||
data: opus_buf[..opus_len].to_vec(),
|
||||
};
|
||||
socket.send_packet(&packet, target).await?;
|
||||
|
||||
// 控制发送频率，约 20ms 一帧
|
||||
tokio::time::sleep(std::time::Duration::from_millis(20)).await;
|
||||
}
|
||||
|
||||
println!("推流结束");
|
||||
Ok(())
|
||||
}
|
||||
@@ -1,6 +1,57 @@
|
||||
use crate::app::server::core::Server;
|
||||
use crate::audio::config::AudioConfig;
|
||||
use anyhow::Result;
|
||||
use std::sync::Arc;
|
||||
|
||||
pub async fn run_server() -> Result<()> {
|
||||
println!("Hello, Server!");
|
||||
Ok(())
|
||||
let server = Arc::new(Server::new().await?);
|
||||
let s = server.clone();
|
||||
|
||||
// 运行服务器
|
||||
tokio::spawn(async move {
|
||||
if let Err(e) = s.run(53531).await {
|
||||
eprintln!("Server error: {}", e);
|
||||
}
|
||||
});
|
||||
|
||||
// 等待一个客户端连接并进行演示
|
||||
println!("等待客户端连接以进行功能演示...");
|
||||
loop {
|
||||
let sessions = server.get_sessions();
|
||||
if !sessions.is_empty() {
|
||||
let addr = sessions[0];
|
||||
println!("开始对 {} 进行功能测试...", addr);
|
||||
|
||||
// 1. 测试 RPC
|
||||
println!("测试 RPC: echo hello");
|
||||
let res = server.call_shell(addr, "echo hello").await?;
|
||||
println!("RPC 结果: {:?}", res);
|
||||
|
||||
// 2. 测试录音
|
||||
println!("测试录制 2 秒音频...");
|
||||
server.start_recording(addr, AudioConfig::voice()).await?;
|
||||
tokio::time::sleep(std::time::Duration::from_secs(2)).await;
|
||||
server.stop_recording(addr).await?;
|
||||
println!("录制结束，请检查 temp/recorded.wav");
|
||||
|
||||
// 3. 测试播放 (如果 temp/test.wav 存在)
|
||||
if std::path::Path::new("temp/test.wav").exists() {
|
||||
println!("测试播放 temp/test.wav...");
|
||||
server.start_playback(addr, AudioConfig::voice()).await?;
|
||||
tokio::time::sleep(std::time::Duration::from_secs(5)).await;
|
||||
server.stop_playback(addr).await?;
|
||||
println!("播放结束");
|
||||
} else {
|
||||
println!("跳过播放测试 (temp/test.wav 不存在)");
|
||||
}
|
||||
|
||||
break;
|
||||
}
|
||||
tokio::time::sleep(std::time::Duration::from_secs(1)).await;
|
||||
}
|
||||
|
||||
// 保持运行
|
||||
loop {
|
||||
tokio::time::sleep(std::time::Duration::from_secs(3600)).await;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1 +1,2 @@
|
||||
pub mod core;
|
||||
pub mod entry;
|
||||
|
||||
Reference in New Issue
Block a user