@@ -2,15 +2,15 @@ use std::collections::HashMap;
22use std:: sync:: Arc ;
33use std:: time:: { Duration , SystemTime } ;
44
5- use anyhow:: { anyhow , Context , Result } ;
5+ use anyhow:: { Context , Result , anyhow } ;
66use tokio:: net:: UdpSocket ;
77use tokio:: sync:: mpsc:: { UnboundedReceiver , UnboundedSender } ;
8- use tokio:: sync:: { oneshot , OnceCell , RwLock } ;
9- use tracing:: { debug, error, info, trace, warn, Instrument } ;
8+ use tokio:: sync:: { OnceCell , RwLock , oneshot } ;
9+ use tracing:: { Instrument , debug, error, info, trace, warn} ;
1010
1111use crate :: config;
1212use crate :: monitoring:: { inc_server_udp_received_count, inc_server_udp_sent_count} ;
13- use crate :: packets:: { get_random_token , GatewayId , PacketType } ;
13+ use crate :: packets:: { GatewayId , PacketType , get_random_token } ;
1414use crate :: traits:: PrintFullError ;
1515
1616static SERVERS : OnceCell < RwLock < Vec < Server > > > = OnceCell :: const_new ( ) ;
@@ -159,13 +159,13 @@ async fn handle_uplink_packet(gateway_id: GatewayId, data: &[u8]) -> Result<()>
159159 inc_server_udp_sent_count ( & server. server , packet_type) . await ;
160160 }
161161 PacketType :: TxAck => {
162- if let Some ( pull_resp_token) = socket. pull_resp_token {
163- if pull_resp_token == random_token {
164- info ! ( packet_type = %packet_type , "Sending UDP packet" ) ;
165- socket . pull_resp_token = None ;
166- socket. socket . send ( data ) . await . context ( "Send UDP packet" ) ? ;
167- inc_server_udp_sent_count ( & server . server , packet_type ) . await ;
168- }
162+ if let Some ( pull_resp_token) = socket. pull_resp_token
163+ && pull_resp_token == random_token
164+ {
165+ info ! ( packet_type = %packet_type , "Sending UDP packet" ) ;
166+ socket. pull_resp_token = None ;
167+ socket . socket . send ( data ) . await . context ( "Send UDP packet" ) ? ;
168+ inc_server_udp_sent_count ( & server . server , packet_type ) . await ;
169169 }
170170 }
171171 _ => { }
@@ -339,10 +339,10 @@ async fn set_pull_resp_token(srv: &str, gateway_id: GatewayId, token: u16) -> Re
339339 let mut servers = servers. write ( ) . await ;
340340
341341 for server in servers. iter_mut ( ) {
342- if server. server . eq ( srv) {
343- if let Some ( v) = server. sockets . get_mut ( & gateway_id) {
344- v . pull_resp_token = Some ( token ) ;
345- }
342+ if server. server . eq ( srv)
343+ && let Some ( v) = server. sockets . get_mut ( & gateway_id)
344+ {
345+ v . pull_resp_token = Some ( token ) ;
346346 }
347347 }
348348
0 commit comments