Skip to content

Commit 8b44853

Browse files
committed
code review related fixes
1 parent d4c0f9d commit 8b44853

File tree

3 files changed

+6
-16
lines changed
  • integration-kinesis-to-lambda
  • integration-kinesis-to-lambda-with-batch-item-handling
  • integration-sqs-to-lambda

3 files changed

+6
-16
lines changed

integration-kinesis-to-lambda-with-batch-item-handling/main.rs

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,6 @@ use aws_lambda_events::{
55
};
66
use lambda_runtime::{run, service_fn, Error, LambdaEvent};
77

8-
static UNDEFINED: &str = "undefined";
9-
108
async fn function_handler(event: LambdaEvent<KinesisEvent>) -> Result<KinesisEventResponse, Error> {
119
let mut response = KinesisEventResponse {
1210
batch_item_failures: vec![],
@@ -20,7 +18,7 @@ async fn function_handler(event: LambdaEvent<KinesisEvent>) -> Result<KinesisEve
2018
for record in &event.payload.records {
2119
tracing::info!(
2220
"EventId: {}",
23-
record.event_id.as_deref().unwrap_or(UNDEFINED)
21+
record.event_id.as_deref().unwrap_or_default()
2422
);
2523

2624
let record_processing_result = process_record(record);
@@ -46,13 +44,12 @@ async fn function_handler(event: LambdaEvent<KinesisEvent>) -> Result<KinesisEve
4644
fn process_record(record: &KinesisEventRecord) -> Result<(), Error> {
4745
let record_data = std::str::from_utf8(record.kinesis.data.as_slice());
4846

49-
if record_data.is_err() {
50-
let err = record_data.err().unwrap();
47+
if let Some(err) = record_data.err() {
5148
tracing::error!("Error: {}", err);
5249
return Err(Error::from(err));
5350
}
5451

55-
let record_data = record_data.unwrap();
52+
let record_data = record_data.unwrap_or_default();
5653

5754
// do something interesting with the data
5855
tracing::info!("Data: {}", record_data);
@@ -71,4 +68,4 @@ async fn main() -> Result<(), Error> {
7168
.init();
7269

7370
run(service_fn(function_handler)).await
74-
}
71+
}

integration-kinesis-to-lambda/main.rs

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,14 @@
11
use aws_lambda_events::event::kinesis::KinesisEvent;
22
use lambda_runtime::{run, service_fn, Error, LambdaEvent};
33

4-
static UNDEFINED: &str = "undefined";
5-
64
async fn function_handler(event: LambdaEvent<KinesisEvent>) -> Result<(), Error> {
7-
85
if event.payload.records.is_empty() {
96
tracing::info!("No records found. Exiting.");
107
return Ok(());
118
}
129

1310
event.payload.records.iter().for_each(|record| {
14-
15-
tracing::info!("EventId: {}", record.event_id.as_deref().unwrap_or(UNDEFINED));
11+
tracing::info!("EventId: {}",record.event_id.as_deref().unwrap_or_default());
1612

1713
let record_data = std::str::from_utf8(&record.kinesis.data);
1814

integration-sqs-to-lambda/main.rs

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,10 @@
11
use aws_lambda_events::event::sqs::SqsEvent;
22
use lambda_runtime::{run, service_fn, Error, LambdaEvent};
33

4-
static UNDEFINED: &str = "undefined";
5-
64
async fn function_handler(event: LambdaEvent<SqsEvent>) -> Result<(), Error> {
7-
85
event.payload.records.iter().for_each(|record| {
96
// process the record
10-
tracing::info!("Message body: {}", record.body.as_deref().unwrap_or(UNDEFINED))
7+
tracing::info!("Message body: {}", record.body.as_deref().unwrap_or_default())
118
});
129

1310
Ok(())

0 commit comments

Comments
 (0)