refactor: 统一轮询任务实现,移除 scan_interval_s 字段

This commit is contained in:
caoqianming 2026-03-05 09:30:30 +08:00
parent d156108148
commit a6be0827d0
4 changed files with 153 additions and 150 deletions

View File

@ -0,0 +1,2 @@
-- 移除 scan_interval_s 字段,因为现在使用统一的轮询任务
ALTER TABLE point DROP COLUMN scan_interval_s;

View File

@ -59,8 +59,10 @@ struct PointWriteTarget {
external_id: String, external_id: String,
} }
#[derive(Debug, Clone)]
pub struct PollPointInfo { pub struct PollPointInfo {
handle: JoinHandle<()>, pub point_id: Uuid,
pub external_id: String,
} }
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)] #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
@ -93,7 +95,8 @@ pub struct ConnectionStatus {
pub next_client_handle: u32, pub next_client_handle: u32,
pub client_handle_map: HashMap<u32, Uuid>, // client_handle -> point_id pub client_handle_map: HashMap<u32, Uuid>, // client_handle -> point_id
pub monitored_item_map: HashMap<Uuid, u32>, // point_id -> monitored_item_id pub monitored_item_map: HashMap<Uuid, u32>, // point_id -> monitored_item_id
pub poll_points: HashMap<Uuid, PollPointInfo>, // 正在轮询的点集合 pub poll_points: Vec<PollPointInfo>, // 正在轮询的点集合
poll_handle: Option<JoinHandle<()>>, // 统一的轮询任务句柄
} }
#[derive(Clone)] #[derive(Clone)]
@ -216,70 +219,85 @@ impl ConnectionManager {
self.point_monitor_data.read().await self.point_monitor_data.read().await
} }
async fn start_polling_for_point( async fn start_unified_poll_task(&self, source_id: Uuid, session: Arc<Session>) {
&self, let event_manager = match self.event_manager.clone() {
source_id: Uuid, Some(em) => em,
point: PointSubscriptionInfo, None => {
session: Arc<Session>, tracing::warn!("Event manager is not initialized, cannot start unified poll task");
) -> Result<(), String> { return;
let interval_s = point.scan_interval_s;
if interval_s <= 0 {
return Err(format!(
"Point {} has invalid scan_interval_s {}",
point.point_id, point.scan_interval_s
));
} }
};
let node_id = NodeId::from_str(&point.external_id) // 停止旧的轮询任务
.map_err(|e| format!("Invalid node id {}: {}", point.external_id, e))?;
let event_manager = self
.event_manager
.clone()
.ok_or_else(|| "Event manager is not initialized".to_string())?;
{ {
let status = self.status.read().await; let mut status = self.status.write().await;
if let Some(conn_status) = status.get(&source_id) { if let Some(conn_status) = status.get_mut(&source_id) {
if conn_status.poll_points.contains_key(&point.point_id) { if let Some(handle) = conn_status.poll_handle.take() {
return Ok(()); handle.abort();
} }
} }
} }
let point_id = point.point_id; tracing::info!(
let external_id = point.external_id.clone(); "Starting unified poll task for source {}",
let interval_sec_u64 = u64::try_from(interval_s) source_id
.map_err(|_| format!("Invalid scan_interval_s {} for point {}", interval_s, point_id))?; );
// 克隆 status 引用,以便在异步任务中使用
let status_ref = self.status.clone();
// 启动新的轮询任务
let handle = tokio::spawn(async move { let handle = tokio::spawn(async move {
let mut ticker = tokio::time::interval(Duration::from_secs(interval_sec_u64)); let mut ticker = tokio::time::interval(Duration::from_secs(1));
ticker.set_missed_tick_behavior(tokio::time::MissedTickBehavior::Skip); ticker.set_missed_tick_behavior(tokio::time::MissedTickBehavior::Skip);
loop { loop {
ticker.tick().await; ticker.tick().await;
let read_request = ReadValueId { // 在任务内部获取轮询点列表
node_id: node_id.clone(), let poll_points = {
let status = status_ref.read().await;
status.get(&source_id)
.map(|conn_status| conn_status.poll_points.clone())
.unwrap_or_default()
};
if poll_points.is_empty() {
continue;
}
// 构建批量读取请求
let read_requests: Vec<ReadValueId> = poll_points
.iter()
.filter_map(|p| {
NodeId::from_str(&p.external_id).ok().map(|node_id| ReadValueId {
node_id,
attribute_id: AttributeId::Value as u32, attribute_id: AttributeId::Value as u32,
index_range: NumericRange::None, index_range: NumericRange::None,
data_encoding: Default::default(), data_encoding: Default::default(),
}; })
})
.collect();
match session if read_requests.is_empty() {
.read(&[read_request], TimestampsToReturn::Both, 0f64) continue;
.await }
{
Ok(result) if !result.is_empty() => { // 执行批量读取
let dv = &result[0]; match session.read(&read_requests, TimestampsToReturn::Both, 0f64).await {
Ok(results) => {
for (i, result) in results.iter().enumerate() {
if i >= poll_points.len() {
break;
}
let poll_point = &poll_points[i];
let dv = result;
let val = dv.value.clone(); let val = dv.value.clone();
let unified_value = let unified_value = val.as_ref().map(crate::telemetry::opcua_variant_to_data);
val.as_ref().map(crate::telemetry::opcua_variant_to_data); let unified_value_type = val.as_ref().map(crate::telemetry::opcua_variant_type);
let unified_value_type =
val.as_ref().map(crate::telemetry::opcua_variant_type);
let unified_value_text = val.as_ref().map(|v| v.to_string()); let unified_value_text = val.as_ref().map(|v| v.to_string());
let quality = dv let quality = dv.status
.status
.as_ref() .as_ref()
.map(crate::telemetry::PointQuality::from_status_code) .map(crate::telemetry::PointQuality::from_status_code)
.unwrap_or(crate::telemetry::PointQuality::Unknown); .unwrap_or(crate::telemetry::PointQuality::Unknown);
@ -287,7 +305,7 @@ impl ConnectionManager {
let _ = event_manager.send(crate::event::ReloadEvent::PointNewValue( let _ = event_manager.send(crate::event::ReloadEvent::PointNewValue(
crate::telemetry::PointNewValue { crate::telemetry::PointNewValue {
source_id, source_id,
point_id: Some(point_id), point_id: Some(poll_point.point_id),
client_handle: 0, client_handle: 0,
value: unified_value, value: unified_value,
value_type: unified_value_type, value_type: unified_value_type,
@ -299,18 +317,11 @@ impl ConnectionManager {
}, },
)); ));
} }
Ok(_) => {
tracing::warn!(
"Poll read returned empty result for point {} node {}",
point_id,
external_id
);
} }
Err(e) => { Err(e) => {
tracing::warn!( tracing::warn!(
"Poll read failed for point {} node {}: {:?}", "Unified poll read failed for source {}: {:?}",
point_id, source_id,
external_id,
e e
); );
} }
@ -318,43 +329,10 @@ impl ConnectionManager {
} }
}); });
{ // 保存轮询任务句柄
let mut status = self.status.write().await; let mut status = self.status.write().await;
if let Some(conn_status) = status.get_mut(&source_id) { if let Some(conn_status) = status.get_mut(&source_id) {
conn_status.poll_points.insert( conn_status.poll_handle = Some(handle);
point_id,
PollPointInfo { handle },
);
}
}
Ok(())
}
async fn stop_polling_for_point(&self, point_id: Uuid) {
let mut status = self.status.write().await;
for conn_status in status.values_mut() {
if let Some(poll_info) = conn_status.poll_points.remove(&point_id) {
poll_info.handle.abort();
}
}
}
async fn stop_polling_for_source(&self, source_id: Uuid) {
let poll_infos = {
let mut status = self.status.write().await;
status
.get_mut(&source_id)
.map(|conn_status| conn_status.poll_points.drain().collect::<Vec<_>>())
.unwrap_or_default()
};
if poll_infos.is_empty() {
return;
}
for (_, poll_info) in poll_infos {
poll_info.handle.abort();
} }
} }
@ -365,28 +343,33 @@ impl ConnectionManager {
session: Arc<Session>, session: Arc<Session>,
) -> usize { ) -> usize {
let mut started = 0usize; let mut started = 0usize;
for point in points.iter().cloned() {
match self // 添加新的轮询点
.start_polling_for_point(source_id, point.clone(), session.clone())
.await
{ {
Ok(()) => { let mut status = self.status.write().await;
if let Some(conn_status) = status.get_mut(&source_id) {
for point in points {
// 检查点是否已经在轮询列表中
if !conn_status.poll_points.iter().any(|p| p.point_id == point.point_id) {
conn_status.poll_points.push(PollPointInfo {
point_id: point.point_id,
external_id: point.external_id.clone(),
});
started += 1; started += 1;
tracing::info!( tracing::info!(
"Point {} switched to poll mode with scan_interval_s {}", "Point {} switched to poll mode",
point.point_id, point.point_id
point.scan_interval_s
);
}
Err(e) => {
tracing::warn!(
"Point {} cannot switch to poll mode: {}",
point.point_id,
e
); );
} }
} }
} }
}
// 如果有新的轮询点,启动或重启统一的轮询任务
if started > 0 {
self.start_unified_poll_task(source_id, session).await;
}
started started
} }
@ -501,7 +484,8 @@ impl ConnectionManager {
next_client_handle: 1000, next_client_handle: 1000,
client_handle_map: HashMap::new(), client_handle_map: HashMap::new(),
monitored_item_map: HashMap::new(), monitored_item_map: HashMap::new(),
poll_points: HashMap::new(), poll_points: Vec::new(),
poll_handle: None,
}, },
); );
@ -522,12 +506,22 @@ impl ConnectionManager {
client_handle_map: HashMap::new(), client_handle_map: HashMap::new(),
monitored_item_map: HashMap::new(), monitored_item_map: HashMap::new(),
next_client_handle: 1000, next_client_handle: 1000,
poll_points: HashMap::new(), poll_points: Vec::new(),
poll_handle: None,
}, },
); );
} }
pub async fn disconnect(&self, source_id: Uuid) -> Result<(), String> { pub async fn disconnect(&self, source_id: Uuid) -> Result<(), String> {
self.stop_polling_for_source(source_id).await; // 停止轮询任务并清空轮询点列表
{
let mut status = self.status.write().await;
if let Some(conn_status) = status.get_mut(&source_id) {
conn_status.poll_points.clear();
if let Some(handle) = conn_status.poll_handle.take() {
handle.abort();
}
}
}
let conn_status = self.status.write().await.remove(&source_id); let conn_status = self.status.write().await.remove(&source_id);
if let Some(conn_status) = conn_status { if let Some(conn_status) = conn_status {
@ -546,7 +540,16 @@ impl ConnectionManager {
let source_ids: Vec<Uuid> = self.status.read().await.keys().copied().collect(); let source_ids: Vec<Uuid> = self.status.read().await.keys().copied().collect();
for source_id in source_ids { for source_id in source_ids {
self.stop_polling_for_source(source_id).await; // 停止轮询任务并清空轮询点列表
{
let mut status = self.status.write().await;
if let Some(conn_status) = status.get_mut(&source_id) {
conn_status.poll_points.clear();
if let Some(handle) = conn_status.poll_handle.take() {
handle.abort();
}
}
}
let conn_status = self.status.write().await.remove(&source_id); let conn_status = self.status.write().await.remove(&source_id);
if let Some(conn_status) = conn_status { if let Some(conn_status) = conn_status {
@ -1020,8 +1023,9 @@ impl ConnectionManager {
conn_status conn_status
.monitored_item_map .monitored_item_map
.insert(point.point_id, monitored_item_result.result.monitored_item_id); .insert(point.point_id, monitored_item_result.result.monitored_item_id);
// 从轮询列表中移除该点
conn_status.poll_points.retain(|p| p.point_id != point.point_id);
} }
self.stop_polling_for_point(point.point_id).await;
} else { } else {
tracing::error!( tracing::error!(
"Failed to create monitored item for point {}: {:?}", "Failed to create monitored item for point {}: {:?}",
@ -1144,8 +1148,12 @@ impl ConnectionManager {
.retain(|_, point_id| !removed_set.contains(point_id)); .retain(|_, point_id| !removed_set.contains(point_id));
} }
for point_id in &removed_point_ids { // 从轮询列表中移除已取消订阅的点
self.stop_polling_for_point(*point_id).await; {
let mut status = self.status.write().await;
if let Some(conn_status) = status.get_mut(&source_id) {
conn_status.poll_points.retain(|p| !removed_set.contains(&p.point_id));
}
} }
let _ = self let _ = self
.remove_point_write_target_cache_by_point_ids(&removed_point_ids) .remove_point_write_target_cache_by_point_ids(&removed_point_ids)

View File

@ -92,7 +92,6 @@ pub struct Point {
pub name: String, pub name: String,
pub description: Option<String>, pub description: Option<String>,
pub unit: Option<String>, pub unit: Option<String>,
pub scan_interval_s: i32, // s
pub tag_id: Option<Uuid>, pub tag_id: Option<Uuid>,
#[serde(serialize_with = "utc_to_local_str")] #[serde(serialize_with = "utc_to_local_str")]
pub created_at: DateTime<Utc>, pub created_at: DateTime<Utc>,
@ -104,7 +103,6 @@ pub struct Point {
pub struct PointSubscriptionInfo { pub struct PointSubscriptionInfo {
pub point_id: Uuid, pub point_id: Uuid,
pub external_id: String, pub external_id: String,
pub scan_interval_s: i32,
} }
#[derive(Debug, Serialize, Deserialize, FromRow, Clone)] #[derive(Debug, Serialize, Deserialize, FromRow, Clone)]

View File

@ -30,8 +30,7 @@ pub async fn get_points_grouped_by_source(
SELECT SELECT
p.id as point_id, p.id as point_id,
n.source_id, n.source_id,
n.external_id, n.external_id
p.scan_interval_s
FROM point p FROM point p
INNER JOIN node n ON p.node_id = n.id INNER JOIN node n ON p.node_id = n.id
WHERE p.id = ANY($1) WHERE p.id = ANY($1)
@ -54,7 +53,6 @@ pub async fn get_points_grouped_by_source(
let info = PointSubscriptionInfo { let info = PointSubscriptionInfo {
point_id, point_id,
external_id: row.get("external_id"), external_id: row.get("external_id"),
scan_interval_s: row.get("scan_interval_s"),
}; };
result.entry(source_id).or_default().push(info); result.entry(source_id).or_default().push(info);
@ -73,8 +71,7 @@ pub async fn get_points_with_ids(
r#" r#"
SELECT SELECT
p.id as point_id, p.id as point_id,
n.external_id, n.external_id
p.scan_interval_s
FROM point p FROM point p
INNER JOIN node n ON p.node_id = n.id INNER JOIN node n ON p.node_id = n.id
WHERE n.source_id = $1 WHERE n.source_id = $1
@ -89,8 +86,7 @@ pub async fn get_points_with_ids(
r#" r#"
SELECT SELECT
p.id as point_id, p.id as point_id,
n.external_id, n.external_id
p.scan_interval_s
FROM point p FROM point p
INNER JOIN node n ON p.node_id = n.id INNER JOIN node n ON p.node_id = n.id
WHERE n.source_id = $1 WHERE n.source_id = $1
@ -110,7 +106,6 @@ pub async fn get_points_with_ids(
.map(|row| PointSubscriptionInfo { .map(|row| PointSubscriptionInfo {
point_id: row.get("point_id"), point_id: row.get("point_id"),
external_id: row.get("external_id"), external_id: row.get("external_id"),
scan_interval_s: row.get("scan_interval_s"),
}) })
.collect()) .collect())
} }