dnutiu/social-media-news-bot
dnutiu/social-media-news-bot
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor read_stream

Browse source
This commit is contained in:
Denis-Cosmin Nutiu 2024-12-28 10:22:18 +02:00
parent 9a2304c8b0
commit 291cc55c4d
1 changed files with 13 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
infrastructure/src/redis.rs
View file

@ -3,7 +3,7 @@ use log::error;
use redis::aio::MultiplexedConnection;
use redis::streams::StreamReadReply;
use redis::Value::BulkString;
use redis::{AsyncCommands, RedisError, RedisResult};
use redis::{AsyncCommands, RedisError, RedisResult, Value};
use serde::{Deserialize, Serialize};
pub struct RedisService {
@ -111,23 +111,22 @@ impl RedisService {
match result {
Ok(data) => {
if data.keys.is_empty() {
return Err(anyhow!("read stream entry with empty keys"));
}
if data.keys[0].ids.is_empty() {
return Err(anyhow!("read stream entry with empty ids"));
}
let stream = data.keys[0].ids[0].map.get("data");
if let Some(BulkString(data)) = stream {
let stream_data: Option<&Value> = data
.keys
.first()
.and_then(|f| f.ids.first().and_then(|i| i.map.get("data")));
if let Some(BulkString(data)) = stream_data {
let string_data = std::str::from_utf8(data);
return match string_data {
match string_data {
Ok(string_data) => Ok(serde_json::from_str(string_data)?),
Err(err) => Err(anyhow!("can't convert data to string: {err}")),
};
}
} else {
Err(anyhow!(
"invalid type read from streams, expected BulkString"
))
}
Err(anyhow!(
"invalid type read from streams, expected BulkString"
))
}
Err(err) => Err(err.into()),
}