refactor: make all the services started in the same way with a start_ func and a handler task
This commit is contained in:
+41
-27
@@ -1,5 +1,9 @@
|
||||
use core::net::SocketAddr;
|
||||
use embassy_net::udp::{PacketMetadata, UdpSocket};
|
||||
use embassy_executor::Spawner;
|
||||
use embassy_net::{
|
||||
Stack,
|
||||
udp::{PacketMetadata, UdpSocket},
|
||||
};
|
||||
use embassy_time::{Duration, Timer};
|
||||
use portable_atomic::{AtomicU64, Ordering};
|
||||
use sntpc::{NtpContext, get_time};
|
||||
@@ -10,11 +14,38 @@ use sntpc_time_embassy::EmbassyTimestampGenerator;
|
||||
const NTP_SERVER: &str = "pool.ntp.org";
|
||||
const NTP_PORT: u16 = 123;
|
||||
|
||||
pub fn now() -> embassy_time::Instant {
|
||||
embassy_time::Instant::now() + Duration::from_secs(NTP_BOOT_OFFSET.load(Ordering::Relaxed))
|
||||
}
|
||||
|
||||
pub fn start_ntp_sync(spawner: Spawner, net_stack: Stack<'static>) {
|
||||
spawner.spawn(handler(net_stack).unwrap());
|
||||
}
|
||||
|
||||
#[embassy_executor::task]
|
||||
async fn handler(net_stack: Stack<'static>) {
|
||||
loop {
|
||||
match fetch_ntp_time(net_stack).await {
|
||||
Ok(unix_secs) => {
|
||||
NTP_BOOT_OFFSET.store(
|
||||
unix_secs - embassy_time::Instant::now().as_secs(),
|
||||
portable_atomic::Ordering::Relaxed,
|
||||
);
|
||||
Timer::after(Duration::from_secs(6 * 60 * 60)).await
|
||||
}
|
||||
Err(err) => {
|
||||
defmt::error!("Failed to sync the clock from NTP: {}", err);
|
||||
Timer::after(Duration::from_secs(10)).await
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
static NTP_BOOT_OFFSET: AtomicU64 = AtomicU64::new(0);
|
||||
|
||||
pub async fn fetch_ntp_time(stack: embassy_net::Stack<'static>) -> Result<u64, &'static str> {
|
||||
pub async fn fetch_ntp_time(net_stack: Stack<'static>) -> Result<u64, &'static str> {
|
||||
// Resolve the NTP server hostname
|
||||
let server_addr = stack
|
||||
let server_addr = net_stack
|
||||
.dns_query(NTP_SERVER, embassy_net::dns::DnsQueryType::A)
|
||||
.await
|
||||
.map_err(|_| "DNS resolution failed")?
|
||||
@@ -30,7 +61,13 @@ pub async fn fetch_ntp_time(stack: embassy_net::Stack<'static>) -> Result<u64, &
|
||||
let mut tx_meta = [PacketMetadata::EMPTY; 16];
|
||||
let mut tx_buf = [0u8; 48];
|
||||
|
||||
let mut socket = UdpSocket::new(stack, &mut rx_meta, &mut rx_buf, &mut tx_meta, &mut tx_buf);
|
||||
let mut socket = UdpSocket::new(
|
||||
net_stack,
|
||||
&mut rx_meta,
|
||||
&mut rx_buf,
|
||||
&mut tx_meta,
|
||||
&mut tx_buf,
|
||||
);
|
||||
socket.bind(0).map_err(|_| "bind failed")?;
|
||||
|
||||
let wrapped = UdpSocketWrapper::from(socket);
|
||||
@@ -47,26 +84,3 @@ pub async fn fetch_ntp_time(stack: embassy_net::Stack<'static>) -> Result<u64, &
|
||||
|
||||
Ok(unix_secs)
|
||||
}
|
||||
|
||||
pub fn now() -> embassy_time::Instant {
|
||||
embassy_time::Instant::now() + Duration::from_secs(NTP_BOOT_OFFSET.load(Ordering::Relaxed))
|
||||
}
|
||||
|
||||
#[embassy_executor::task]
|
||||
pub async fn ntp_upkeep_service(stack: embassy_net::Stack<'static>) {
|
||||
loop {
|
||||
match fetch_ntp_time(stack).await {
|
||||
Ok(unix_secs) => {
|
||||
NTP_BOOT_OFFSET.store(
|
||||
unix_secs - embassy_time::Instant::now().as_secs(),
|
||||
portable_atomic::Ordering::Relaxed,
|
||||
);
|
||||
Timer::after(Duration::from_secs(6 * 60 * 60)).await
|
||||
}
|
||||
Err(err) => {
|
||||
defmt::error!("Failed to sync the clock from NTP: {}", err);
|
||||
Timer::after(Duration::from_secs(10)).await
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user