Skip to content

Commit 7767150

Browse files
committed
feat: added some more logs
1 parent 36b4e15 commit 7767150

File tree

2 files changed

+22
-21
lines changed

2 files changed

+22
-21
lines changed

src/flow_queue/service.rs

Lines changed: 21 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -211,68 +211,69 @@ impl RabbitmqClient {
211211

212212
match consumer_res {
213213
Ok(consumer) => consumer,
214-
Err(err) => panic!("{}", err),
214+
Err(err) => panic!("Cannot consume messages: {}", err),
215215
}
216216
};
217217

218-
println!("Starting to consume from {}", queue_name);
218+
debug!("Starting to consume from {}", queue_name);
219219

220220
while let Some(delivery) = consumer.next().await {
221221
let delivery = match delivery {
222222
Ok(del) => del,
223223
Err(err) => {
224-
println!("Error receiving message: {}", err);
224+
log::error!("Error receiving message: {}", err);
225225
return Err(err);
226226
}
227227
};
228228

229229
let data = &delivery.data;
230230
let message_str = match std::str::from_utf8(&data) {
231231
Ok(str) => {
232-
println!("Received message: {}", str);
232+
log::info!("Received message: {}", str);
233233
str
234234
}
235235
Err(err) => {
236-
println!("Error decoding message: {}", err);
236+
log::error!("Error decoding message: {}", err);
237237
return Ok(());
238238
}
239239
};
240240
// Parse the message
241241
let inc_message = match serde_json::from_str::<Message>(message_str) {
242-
Ok(mess) => {
243-
println!("Parsed message with telegram_id: {}", mess.message_id);
244-
mess
245-
}
242+
Ok(mess) => mess,
246243
Err(err) => {
247-
println!("Error parsing message: {}", err);
244+
log::error!("Error parsing message: {}", err);
248245
return Ok(());
249246
}
250247
};
251248

252249
let message = match handle_message(inc_message) {
253-
Ok(mess) => {
254-
println!("Handled message with telegram_id: {}", mess.message_id);
255-
mess
256-
}
250+
Ok(mess) => mess,
257251
Err(err) => {
258-
println!("Error handling message: {}", err);
252+
log::error!("Error handling message: {}", err);
259253
return Ok(());
260254
}
261255
};
262256

263-
let message_json = serde_json::to_string(&message).unwrap();
264-
265-
println!("{}", message_json);
257+
let message_json = match serde_json::to_string(&message) {
258+
Ok(json) => json,
259+
Err(err) => {
260+
log::error!("Error serializing message: {}", err);
261+
return Ok(());
262+
}
263+
};
266264

267265
{
268266
let _ = self.send_message(message_json, "recieve_queue").await;
269267
}
270268

271269
// Acknowledge the message
272-
delivery
270+
match delivery
273271
.ack(lapin::options::BasicAckOptions::default())
274272
.await
275-
.expect("Failed to acknowledge message");
273+
{
274+
Ok(_) => Ok(()),
275+
Err(err) => log::error!("Failed to acknowledge message: {}", err),
276+
}
276277
}
277278

278279
Ok(())

src/flow_store/service.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ impl FlowStoreServiceBase for FlowStoreService {
109109
match connection.keys("*").await {
110110
Ok(res) => res,
111111
Err(error) => {
112-
print!("Can't retrieve keys from redis. Reason: {error}");
112+
error!("Can't retrieve keys from redis. Reason: {error}");
113113
return Err(error);
114114
}
115115
}

0 commit comments

Comments
 (0)