in experimental/aws-lambda-java-profiler/extension/src/main/java/com/amazonaws/services/lambda/extension/ExtensionMain.java [23:51]
public static void main(String[] args) {
final String extension = ExtensionClient.registerExtension();
Logger.debug("Extension registration complete, extensionID: " + extension);
s3Manager = new S3Manager();
while (true) {
try {
String response = ExtensionClient.getNext(extension);
if (response != null && !response.isEmpty()) {
final String eventType = extractInfo(EVENT_TYPE, response);
Logger.debug("eventType = " + eventType);
if (eventType != null) {
switch (eventType) {
case "INVOKE":
handleInvoke(response);
break;
case "SHUTDOWN":
handleShutDown();
break;
default:
Logger.error("invalid event type received " + eventType);
}
}
}
} catch (Exception e) {
Logger.error("error while processing extension -" + e.getMessage());
e.printStackTrace();
}
}
}