Refactor device-profile class-c fields.
Some checks are pending
CI / tests (postgres) (push) Waiting to run
CI / tests (sqlite) (push) Waiting to run
CI / dist (postgres) (push) Blocked by required conditions
CI / dist (sqlite) (push) Blocked by required conditions

This commit is contained in:
Orne Brocaar 2025-01-14 11:32:09 +00:00
parent 479856ce50
commit d521bb4790
10 changed files with 100 additions and 15 deletions

View File

@ -1,6 +1,7 @@
alter table device_profile
add column abp_params jsonb null,
add column class_b_params jsonb null;
add column class_b_params jsonb null,
add column class_c_params jsonb null;
update device_profile
set abp_params = json_build_object(
@ -16,7 +17,13 @@ update device_profile
'ping_slot_nb_k', class_b_ping_slot_nb_k,
'ping_slot_dr', class_b_ping_slot_dr,
'ping_slot_freq', class_b_ping_slot_freq)
where supports_class_b = false;
where supports_class_b = true;
update device_profile
set class_c_params = json_build_object(
'timeout', class_c_timeout)
where
supports_class_c = true;
alter table device_profile
drop column abp_rx1_delay,
@ -26,5 +33,6 @@ alter table device_profile
drop column class_b_timeout,
drop column class_b_ping_slot_nb_k,
drop column class_b_ping_slot_dr,
drop column class_b_ping_slot_freq;
drop column class_b_ping_slot_freq,
drop column class_c_timeout;

View File

@ -6,6 +6,7 @@ alter table device_profile add column class_b_timeout integer not null default 0
alter table device_profile add column class_b_ping_slot_nb_k integer not null default 0;
alter table device_profile add column class_b_ping_slot_dr smallint not null default 0;
alter table device_profile add column class_b_ping_slot_freq bigint not null default 0;
alter table device_profile add column class_c_timeout integer not null default 0;
update device_profile
set
@ -25,6 +26,13 @@ update device_profile
where
class_b_params is not null;
update device_profile
set
class_c_timeout = class_c_params->'timeout'
where
class_c_params is not null;
alter table device_profile drop column abp_params;
alter table device_profile drop column class_b_params;
alter table device_profile drop column class_c_params;

View File

@ -1,5 +1,6 @@
alter table device_profile add column abp_params text null;
alter table device_profile add column class_b_params text null;
alter table device_profile add column class_c_params text null;
update device_profile
set abp_params = json_object(
@ -15,7 +16,12 @@ update device_profile
'ping_slot_nb_k', class_b_ping_slot_nb_k,
'ping_slot_dr', class_b_ping_slot_dr,
'ping_slot_freq', class_b_ping_slot_freq)
where supports_class_b = false;
where supports_class_b = true;
update device_profile
set class_c_params = json_object(
'timeout', class_c_timeout)
where supports_class_c = true;
alter table device_profile drop column abp_rx1_delay;
alter table device_profile drop column abp_rx1_dr_offset;
@ -27,3 +33,4 @@ alter table device_profile drop column class_b_ping_slot_nb_k;
alter table device_profile drop column class_b_ping_slot_dr;
alter table device_profile drop column class_b_ping_slot_freq;
alter table device_profile drop column class_c_timeout;

View File

@ -60,7 +60,6 @@ impl DeviceProfileService for DeviceProfile {
supports_otaa: req_dp.supports_otaa,
supports_class_b: req_dp.supports_class_b,
supports_class_c: req_dp.supports_class_c,
class_c_timeout: req_dp.class_c_timeout as i32,
tags: fields::KeyValue::new(req_dp.tags.clone()),
measurements: fields::Measurements::new(
req_dp
@ -126,6 +125,13 @@ impl DeviceProfileService for DeviceProfile {
} else {
None
},
class_c_params: if req_dp.supports_class_c {
Some(fields::ClassCParams {
timeout: req_dp.class_c_timeout as u16,
})
} else {
None
},
..Default::default()
};
@ -159,6 +165,7 @@ impl DeviceProfileService for DeviceProfile {
let dp = device_profile::get(&dp_id).await.map_err(|e| e.status())?;
let abp_params = dp.abp_params.clone().unwrap_or_default();
let class_b_params = dp.class_b_params.clone().unwrap_or_default();
let class_c_params = dp.class_c_params.clone().unwrap_or_default();
let mut resp = Response::new(api::GetDeviceProfileResponse {
device_profile: Some(api::DeviceProfile {
@ -182,7 +189,7 @@ impl DeviceProfileService for DeviceProfile {
class_b_ping_slot_nb_k: class_b_params.ping_slot_nb_k as u32,
class_b_ping_slot_dr: class_b_params.ping_slot_dr as u32,
class_b_ping_slot_freq: class_b_params.ping_slot_freq as u32,
class_c_timeout: dp.class_c_timeout as u32,
class_c_timeout: class_c_params.timeout as u32,
abp_rx1_delay: abp_params.rx1_delay as u32,
abp_rx1_dr_offset: abp_params.rx1_dr_offset as u32,
abp_rx2_dr: abp_params.rx2_dr as u32,
@ -276,7 +283,6 @@ impl DeviceProfileService for DeviceProfile {
supports_otaa: req_dp.supports_otaa,
supports_class_b: req_dp.supports_class_b,
supports_class_c: req_dp.supports_class_c,
class_c_timeout: req_dp.class_c_timeout as i32,
tags: fields::KeyValue::new(req_dp.tags.clone()),
measurements: fields::Measurements::new(
req_dp
@ -342,6 +348,13 @@ impl DeviceProfileService for DeviceProfile {
} else {
None
},
class_c_params: if req_dp.supports_class_c {
Some(fields::ClassCParams {
timeout: req_dp.class_c_timeout as u16,
})
} else {
None
},
..Default::default()
})
.await

View File

@ -282,8 +282,12 @@ impl TxAck {
qi.is_pending = true;
if dev.enabled_class == DeviceClass::C {
let timeout =
Utc::now() + Duration::try_seconds(dp.class_c_timeout as i64).unwrap_or_default();
let timeout_sec = dp
.class_c_params
.as_ref()
.map(|v| v.timeout)
.unwrap_or_default() as i64;
let timeout = Utc::now() + Duration::try_seconds(timeout_sec).unwrap_or_default();
qi.timeout_after = Some(timeout);
}

View File

@ -34,7 +34,6 @@ pub struct DeviceProfile {
pub supports_otaa: bool,
pub supports_class_b: bool,
pub supports_class_c: bool,
pub class_c_timeout: i32,
pub tags: fields::KeyValue,
pub payload_codec_script: String,
pub flush_queue_on_activate: bool,
@ -68,6 +67,7 @@ pub struct DeviceProfile {
pub rx1_delay: i16,
pub abp_params: Option<fields::AbpParams>,
pub class_b_params: Option<fields::ClassBParams>,
pub class_c_params: Option<fields::ClassCParams>,
}
impl DeviceProfile {
@ -107,7 +107,6 @@ impl Default for DeviceProfile {
supports_otaa: false,
supports_class_b: false,
supports_class_c: false,
class_c_timeout: 0,
tags: fields::KeyValue::new(HashMap::new()),
measurements: fields::Measurements::new(HashMap::new()),
auto_detect_measurements: false,
@ -138,6 +137,7 @@ impl Default for DeviceProfile {
rx1_delay: 0,
abp_params: None,
class_b_params: None,
class_c_params: None,
}
}
}
@ -238,7 +238,6 @@ pub async fn update(dp: DeviceProfile) -> Result<DeviceProfile, Error> {
device_profile::supports_otaa.eq(&dp.supports_otaa),
device_profile::supports_class_b.eq(&dp.supports_class_b),
device_profile::supports_class_c.eq(&dp.supports_class_c),
device_profile::class_c_timeout.eq(&dp.class_c_timeout),
device_profile::tags.eq(&dp.tags),
device_profile::measurements.eq(&dp.measurements),
device_profile::auto_detect_measurements.eq(&dp.auto_detect_measurements),
@ -275,6 +274,7 @@ pub async fn update(dp: DeviceProfile) -> Result<DeviceProfile, Error> {
device_profile::rx1_delay.eq(&dp.rx1_delay),
device_profile::abp_params.eq(&dp.abp_params),
device_profile::class_b_params.eq(&dp.class_b_params),
device_profile::class_c_params.eq(&dp.class_c_params),
))
.get_result(&mut get_async_db_conn().await?)
.await

View File

@ -101,3 +101,48 @@ impl serialize::ToSql<Text, Sqlite> for ClassBParams {
Ok(serialize::IsNull::No)
}
}
#[derive(
Default, Debug, Clone, PartialEq, Eq, Deserialize, Serialize, AsExpression, FromSqlRow,
)]
#[cfg_attr(feature = "postgres", diesel(sql_type = Jsonb))]
#[cfg_attr(feature = "sqlite", diesel(sql_type = Text))]
pub struct ClassCParams {
pub timeout: u16,
}
#[cfg(feature = "postgres")]
impl deserialize::FromSql<Jsonb, Pg> for ClassCParams {
fn from_sql(value: <Pg as Backend>::RawValue<'_>) -> deserialize::Result<Self> {
let value = <serde_json::Value as deserialize::FromSql<Jsonb, Pg>>::from_sql(value)?;
Ok(serde_json::from_value(value)?)
}
}
#[cfg(feature = "postgres")]
impl serialize::ToSql<Jsonb, Pg> for ClassCParams {
fn to_sql<'b>(&'b self, out: &mut serialize::Output<'b, '_, Pg>) -> serialize::Result {
let value = serde_json::to_value(&self)?;
<serde_json::Value as serialize::ToSql<Jsonb, Pg>>::to_sql(&value, &mut out.reborrow())
}
}
#[cfg(feature = "sqlite")]
impl deserialize::FromSql<Text, Sqlite> for ClassCParams
where
*const str: deserialize::FromSql<Text, Sqlite>,
{
fn from_sql(value: <Sqlite as Backend>::RawValue<'_>) -> deserialize::Result<Self> {
let s =
<*const str as deserialize::FromSql<diesel::sql_types::Text, Sqlite>>::from_sql(value)?;
Ok(serde_json::from_str(unsafe { &*s })?)
}
}
#[cfg(feature = "sqlite")]
impl serialize::ToSql<Text, Sqlite> for ClassCParams {
fn to_sql<'b>(&'b self, out: &mut serialize::Output<'b, '_, Sqlite>) -> serialize::Result {
out.set_value(serde_json::to_string(&self)?);
Ok(serialize::IsNull::No)
}
}

View File

@ -9,7 +9,7 @@ mod uuid;
pub use big_decimal::BigDecimal;
pub use dev_nonces::DevNonces;
pub use device_profile::{AbpParams, ClassBParams};
pub use device_profile::{AbpParams, ClassBParams, ClassCParams};
pub use device_session::DeviceSession;
pub use key_value::KeyValue;
pub use measurements::*;

View File

@ -103,7 +103,6 @@ diesel::table! {
supports_otaa -> Bool,
supports_class_b -> Bool,
supports_class_c -> Bool,
class_c_timeout -> Int4,
tags -> Jsonb,
payload_codec_script -> Text,
flush_queue_on_activate -> Bool,
@ -138,6 +137,7 @@ diesel::table! {
rx1_delay -> Int2,
abp_params -> Nullable<Jsonb>,
class_b_params -> Nullable<Jsonb>,
class_c_params -> Nullable<Jsonb>,
}
}

View File

@ -92,7 +92,6 @@ diesel::table! {
supports_otaa -> Bool,
supports_class_b -> Bool,
supports_class_c -> Bool,
class_c_timeout -> Integer,
tags -> Text,
payload_codec_script -> Text,
flush_queue_on_activate -> Bool,
@ -126,6 +125,7 @@ diesel::table! {
rx1_delay -> SmallInt,
abp_params -> Nullable<Text>,
class_b_params -> Nullable<Text>,
class_c_params -> Nullable<Text>,
}
}