mart-w/kanidm
1
0
Fork 0
mirror of https://github.com/kanidm/kanidm.git synced 2025-02-23 04:27:02 +01:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Resolve upgrade in place error with cbor to json (#1028)

Browse source
This commit is contained in:
Firstyear 2022-09-09 12:17:01 +10:00 committed by William Brown
parent 985462590b
commit b20d531295
1 changed files with 57 additions and 40 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

75
kanidm_unix_int/src/db.rs
View file

@ -277,20 +277,25 @@ impl<'a> DbTxn<'a> {
return Err(());
}
let r: Result<Option<(_, _)>, ()> = data
.first()
.map(|(token, expiry)| {
if let Some((token, expiry)) = data.first() {
// token convert with json.
let t = serde_json::from_slice(token.as_slice()).map_err(|e| {
error!("json error -> {:?}", e);
})?;
// If this errors, we specifically return Ok(None) because that triggers
// the cache to refetch the token.
match serde_json::from_slice(token.as_slice()) {
Ok(t) => {
let e = u64::try_from(*expiry).map_err(|e| {
error!("u64 convert error -> {:?}", e);
})?;
Ok((t, e))
})
.transpose();
r
Ok(Some((t, e)))
}
Err(e) => {
warn!("recoverable - json error -> {:?}", e);
Ok(None)
}
}
} else {
Ok(None)
}
}
pub fn get_accounts(&self) -> Result<Vec<UnixUserToken>, ()> {
@ -314,14 +319,18 @@ impl<'a> DbTxn<'a> {
let data = data?;
data.iter()
.map(|token| {
Ok(data
.iter()
// We filter map here so that anything invalid is skipped.
.filter_map(|token| {
// token convert with json.
serde_json::from_slice(token.as_slice()).map_err(|e| {
error!("get_accounts json error -> {:?}", e);
serde_json::from_slice(token.as_slice())
.map_err(|e| {
warn!("get_accounts json error -> {:?}", e);
})
.ok()
})
.collect()
.collect())
}
pub fn update_account(&self, account: &UnixUserToken, expire: u64) -> Result<(), ()> {
@ -572,20 +581,25 @@ impl<'a> DbTxn<'a> {
return Err(());
}
let r: Result<Option<(_, _)>, ()> = data
.first()
.map(|(token, expiry)| {
if let Some((token, expiry)) = data.first() {
// token convert with json.
let t = serde_json::from_slice(token.as_slice()).map_err(|e| {
error!("json error -> {:?}", e);
})?;
// If this errors, we specifically return Ok(None) because that triggers
// the cache to refetch the token.
match serde_json::from_slice(token.as_slice()) {
Ok(t) => {
let e = u64::try_from(*expiry).map_err(|e| {
error!("u64 convert error -> {:?}", e);
})?;
Ok((t, e))
})
.transpose();
r
Ok(Some((t, e)))
}
Err(e) => {
warn!("recoverable - json error -> {:?}", e);
Ok(None)
}
}
} else {
Ok(None)
}
}
pub fn get_group_members(&self, g_uuid: &str) -> Result<Vec<UnixUserToken>, ()> {
@ -643,15 +657,18 @@ impl<'a> DbTxn<'a> {
let data = data?;
data.iter()
.map(|token| {
Ok(data
.iter()
.filter_map(|token| {
// token convert with json.
// debug!("{:?}", token);
serde_json::from_slice(token.as_slice()).map_err(|e| {
serde_json::from_slice(token.as_slice())
.map_err(|e| {
error!("json error -> {:?}", e);
})
.ok()
})
.collect()
.collect())
}
pub fn update_group(&self, grp: &UnixGroupToken, expire: u64) -> Result<(), ()> {