Java AWS Lambda with Spring Cloud Function

If you are familiar with Spring Boot you might wonder if you can use this knowledge to write a serverless application using AWS Lambda. While plain Java is enough for simple use cases, it might be helpful to use Spring Framework features (e.g the WebClient, data access, etc.).  With Spring Cloud Function you can achieve this and use the AWS adapter to deploy it as an AWS Lambda. In combination with the Serverless Framework, you get a running function in under five minutes.

By demonstrating five different use cases, I'll demonstrate how to use Spring Cloud Function to deploy AWS Lambda functions.

Maven project setup

To make use of Spring Boot for writing our Lambda functions, we can include the Spring Boot Maven parent for our application. It's not mandatory to include Spring Boot, as Spring Cloud Function also works without it.

As I'll demonstrate different examples of using Spring Cloud Function with AWS, our project contains multiple AWS, Spring Boot and Spring Cloud dependencies:

Besides including all required dependencies, we have to configure the way the .jar is built. Once we want to deploy the application to AWS Lambda, we can't rely on the default way the Spring Boot Maven plugin bundles our project.

It's required to create a shaded .jar (you can find a good explanation of shading dependencies here and here).  For this we'll make use of the Maven Shade Plugin and configure the build section of our pom.xml as the following:

Furthermore, we use the Spring Boot Thin Launcher to optimize the size of the .jar file. You can read more about this experimental feature on GitHub. For this example, the thin layout reduces the size of the artifact by almost 50% (39 MB to 20MB).

While building the application with mvn package, Maven now outputs two .jar files. For deploying the application to AWS only the shaded build artifact is important. In our case that's: spring-cloud-function-aws-1.0.0-shaded.jar

Introduction to Spring Cloud Function

Before we write our first AWS Lambda with Spring Cloud Function, I quickly want to summarize what Spring Cloud Function is all about.

The documentation states the following goals about this Spring Cloud project:

• Promote the implementation of business logic via functions.
• Decouple the development lifecycle of business logic from any specific runtime target so that the same code can run as a web endpoint, a stream processor, or a task.
• Support a uniform programming model across serverless providers, as well as the ability to run standalone (locally or in a PaaS).
• Enable Spring Boot features (auto-configuration, dependency injection, metrics) on serverless providers.

It abstracts away all of the transport details and infrastructure, allowing the developer to keep all the familiar tools and processes, and focus firmly on business logic.

In a nutshell, you simply write either a Java 8 Function, Supplier or Consumer, make use of well-known Spring Boot features, select an adapter for your cloud provider (e.g. AWS, Azure, etc.), and get a running function without caring about provisioning hardware (aka. serverless). That's it.

What's left is to pick the right AWS Request Handler provided by the Spring Cloud Function AWS adapter. There are multiple to choose from and I'll cover most of them throughout the different examples:

• FunctionInvoker
• SpringBootRequestHandler
• SpringBootRequestStreamHandler
• SpringBootKinesisEventHandler
• SpringBootApiGatewayRequestHandler

All of them use either the RequestHandler or RequestStreamHandler interface of the underlying AWS Lambda Java dependency.

FYI: In a previous blog post I developed an AWS Lambda function while implementing one of these interfaces without Spring Cloud Function.

Short note deployment

Throughout the following examples, I'm using Serverless to deploy the functions to AWS Lambda. The basic setup looks like the following:

Besides the standard region and runtime configuration for AWS Lambda, I'm adding an S3 Bucket role to access S3 later on. It's important to use at least 1024 MB of memory and a timeout of 10 seconds, as the first invocation (cold start) of the AWS Lambda takes more time.

With the package attribute, we point to the shaded .jar file which will be uploaded to AWS for each deployment.

You'll find the specific function configuration within each section and take a look at this blog post for more information on using the Serverless Framework.

AWS Lambda to uppercase a String

Let's start with a simple use case, the Hello World of writing functions: uppercasing a String.

Within our Spring Boot application, we can define a Function<String, String> and make it available to the Spring context using @Bean:

For the handler, I'm choosing FunctionInvoker, which is one of the most generic ones and implements RequestStreamHandler under the hood. The handler will take care of serializing the payload, invoking the correct function, and deserializing the result.

As we can have multiple functions within the Spring context, we have to tell Spring Cloud Function which function we want to trigger. For the FunctionInvokerwe can achieve this with an environment variable SPRING_CLOUD_FUNCTION_DEFINITION:

Once we build the project and deploy it to AWS (using serverless deploy), we can invoke it with either Serverless or the AWS Console:

AWS Lambda to generate random UUIDs

Next, let's add an example of using the Supplier interface.

This is relevant for use cases where it's not necessary to consume a value but only return one:

For this example, we'll use the SpringBootRequestHandler, which we can extend to define the input and output types of our processing:

The input type is Void in this case, as we don't care about any incoming value and just produce a random String.

We can now use our EmptyInputHandler as the handlerand have to specify which function we want to invoke using  the FUNCTION_NAME environment variable in this case.

Invoking the AWS Lambda without any input returns the random UUID:

AWS Lambda to process S3 events

For our next example, I want to make use of the Consumer interface to only consume values but don't return anything.

As a more realistic use case, we'll trigger the function whenever a file is uploaded to an S3 bucket:

The handler for this processing looks like the following:

With Serverless we can now configure the event on which the Lambda is triggered:

… and create the following output on each upload:

AWS Lambda behind an API Gateway Part I

As AWS Lambda functions can also be behind an API Gateway and triggered by HTTP calls, let's add an example for this.

Let's say we want a REST API endpoint /persons to process and store Person entities somewhere:

Within the processing, we get access to the serialized Java object and can perform any operation, e.g. store it in a database:

Here I'm wrapping the input and output types within the Message interface from org.springeframework.messaging. This is not required but as we'll use the SpringBootApiGatewayRequestHandler we can use this wrapper to add metadata (e.g headers) to the response object.

The handler will take care of creating the correct response type (including the statuscode field, etc.) for the API Gateway:

Once you deploy the function with the Serverless framework, you'll receive an HTTP endpoint to trigger your Lambda:

AWS Lambda behind an API Gateway Part II

In the last example, I want to show you how to have more control over the serialization of the incoming payload. For this, we'll use the SpringBootStreamHandler and are going to parse incoming XML payload to our AWS Lambda (behind an API Gateway).

Like the example above, we'll accept entities via a REST API and process them:

The input and output types are coming from the aws-lambda-java-events dependency and represent the payload of the API Gateway. As we are not defining the actual payload type, we can parse the body of the APIGatewayProxyRequestEvent (containing XML as a string) for ourselves.

You can even go further and request the raw InputStream for your function and handle everything for yourself.

The deployment configuration is similar to the other Lambda behind the API Gateway:

Spring Cloud Function for AWS Lambda conclusion

After reading this blog post, you shouldn't jump into writing your whole application using this approach. Use it wisely where the use case makes sense to write an AWS Lambda function. Whether you use Spring Cloud Function or the Java AWS Lambda interfaces (RequestHandler/RequestStreamHandler) directly, depends on you.

There is definitely an overhead when using Spring Cloud Function (bigger .jar files, slow cold starts as we have to start the Spring context), but being able to stay in the well-known Spring ecosystem may also result in more productivity.

While Spring Cloud Function provides a helpful abstraction, the biggest pain point for me, in the beginning, was to find the correct handler for my function. The documentation on them is limited.  Inspecting the source code of each handler in the IDE helped me to quickly understand their differences.

You can find the source code for the examples above on GitHub.

Have fun writing AWS Lambda functions with Spring Cloud Function,

Philip