forked from aws-samples/amazon-textract-searchable-pdf
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathDemoLambda.java
executable file
·36 lines (29 loc) · 1.47 KB
/
DemoLambda.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import com.amazonaws.services.lambda.runtime.Context;
import com.amazonaws.services.lambda.runtime.RequestHandler;
import com.amazonaws.services.lambda.runtime.events.S3Event;
import com.amazonaws.services.lambda.runtime.events.models.s3.S3EventNotification;
public class DemoLambda implements RequestHandler<S3Event, String> {
@Override
public String handleRequest(S3Event event, Context ctx) {
S3EventNotification.S3EventNotificationRecord record = event.getRecords().get(0);
String bucketName = record.getS3().getBucket().getName();
String keyName = record.getS3().getObject().getKey();
String keyNameLower = record.getS3().getObject().getKey().toLowerCase();
String filename = keyName.substring(keyName.lastIndexOf("/") + 1);
System.out.println("Bucket Name is " + bucketName);
System.out.println("File Path is " + keyName);
try {
if (keyNameLower.endsWith("pdf")) {
DemoPdfFromS3Pdf s3Pdf = new DemoPdfFromS3Pdf();
s3Pdf.run(bucketName, keyName, filename);
} else if (keyNameLower.endsWith("jpg") || keyNameLower.endsWith("jpeg") || keyNameLower.endsWith("png")) {
DemoPdfFromS3Image s3Image = new DemoPdfFromS3Image();
s3Image.run(bucketName, keyName, filename);
}
} catch (Exception e) {
e.printStackTrace();
System.out.println(e.getMessage());
}
return null;
}
}