Integrate Spring Boot With Amazon DynamoDB

Published by Vignesh M on June 11, 2018

In this post, let’s see how to integrate Spring Boot with Amazon DynamoDB. I have been exploring the option of creating the REST API using Spring Boot backed by DynamoDB. I will share the information that I have learned during the development process with a simple project named Tasklet API.

Let’s build our project on top of the example we have used in the spring boot starter post. The complete project we are going to develop in this post can be downloaded from here or from the downloads section available at the end of this post.

Prerequisite tools

1. Java 1.8 or higher
2. Eclipse IDE
3. Maven

DynamoDB library for Java

To make the integration between Spring based applications and DynamoDB easier, we can make use of the community module Spring Data DynamoDB. I assume that you have imported the project used in the spring boot starter post.

Add the below dependency for spring-data-dynamodb in pom.xml

<dependency>
	<groupId>com.github.derjust</groupId>
	<artifactId>spring-data-dynamodb</artifactId>
	<version>5.0.2</version>
</dependency>

<groupId>com.github.derjust</groupId>

<artifactId>spring-data-dynamodb</artifactId>

</dependency>

Using the Tasklet API, users should be able to add, read, edit and delete the Tasks. As next step we need to create the data model class which represent the Task itself.

package com.rayfocus.api.tasklet.model;

import java.util.List;

import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBAttribute;
import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBHashKey;
import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBTable;
import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonProperty;

@DynamoDBTable(tableName = "Tasks")
public class Task {

	private String taskId;
	private String taskName;
	private String taskStatus;
	private List<SubTask> taskContent;
	private String userId;
	private String createdOn;

	@JsonCreator
	public Task(@JsonProperty("task_id") String taskId, @JsonProperty("task_name") String taskName,
			@JsonProperty("task_status") String taskStatus, @JsonProperty("task_content") List<SubTask> taskContent,
			@JsonProperty("user_id") String userId, @JsonProperty("created_on") String createdOn) {
		this.taskId = taskId;
		this.taskName = taskName;
		this.taskStatus = taskStatus;
		this.taskContent = taskContent;
		this.userId = userId;
		this.createdOn = createdOn;
	}

	public Task() 
	{
		// default constructor
	}

	/** Getters */
	@DynamoDBHashKey(attributeName = "task_id")
	public String getTaskId() {
		return taskId;
	}

	@DynamoDBAttribute(attributeName = "task_name")
	public String getTaskName() {
		return taskName;
	}
	
	@DynamoDBAttribute(attributeName = "task_status")
	public String getTaskStatus() {
		return taskStatus;
	}

	@DynamoDBAttribute(attributeName = "task_content")
	public List<SubTask> getTaskContent() {
		return taskContent;
	}

	@DynamoDBAttribute(attributeName = "user_id")
	public String getUserId() {
		return userId;
	}

	@DynamoDBAttribute(attributeName = "created_on")
	public String getCreatedOn() {
		return createdOn;
	}

	/** Setters */
	public void setTaskId(String taskId) {
		this.taskId = taskId;
	}

	public void setTaskName(String taskName) {
		this.taskName = taskName;
	}

	public void setTaskStatus(String taskStatus) {
		this.taskStatus = taskStatus;
	}

	public void setTaskContent(List<SubTask> taskContent) {
		this.taskContent = taskContent;
	}

	public void setUserId(String userId) {
		this.userId = userId;
	}

	public void setCreatedOn(String createdOn) {
		this.createdOn = createdOn;
	}

	
	@Override
	public String toString() {
		return "Task [taskId=" + taskId + ", taskName=" + taskName + ", taskStatus=" + taskStatus + ", taskContent="
				+ taskContent + ", userId=" + userId + ", createdOn=" + createdOn + "]";
	}

}

100

101

package com.rayfocus.api.tasklet.model;

import java.util.List;

import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBAttribute;

import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBHashKey;

import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBTable;

import com.fasterxml.jackson.annotation.JsonCreator;

import com.fasterxml.jackson.annotation.JsonProperty;

@DynamoDBTable(tableName = "Tasks")

public class Task {

private String taskId;

private String taskName;

private String taskStatus;

private List<SubTask> taskContent;

private String userId;

private String createdOn;

@JsonCreator

public Task(@JsonProperty("task_id") String taskId, @JsonProperty("task_name") String taskName,

@JsonProperty("task_status") String taskStatus, @JsonProperty("task_content") List<SubTask> taskContent,

@JsonProperty("user_id") String userId, @JsonProperty("created_on") String createdOn) {

this.taskId = taskId;

this.taskName = taskName;

this.taskStatus = taskStatus;

this.taskContent = taskContent;

this.userId = userId;

this.createdOn = createdOn;

}

public Task()

{

// default constructor

}

/** Getters */

@DynamoDBHashKey(attributeName = "task_id")

public String getTaskId() {

return taskId;

}

@DynamoDBAttribute(attributeName = "task_name")

public String getTaskName() {

return taskName;

}

@DynamoDBAttribute(attributeName = "task_status")

public String getTaskStatus() {

return taskStatus;

}

@DynamoDBAttribute(attributeName = "task_content")

public List<SubTask> getTaskContent() {

return taskContent;

}

@DynamoDBAttribute(attributeName = "user_id")

public String getUserId() {

return userId;

}

@DynamoDBAttribute(attributeName = "created_on")

public String getCreatedOn() {

return createdOn;

}

/** Setters */

public void setTaskId(String taskId) {

this.taskId = taskId;

}

public void setTaskName(String taskName) {

this.taskName = taskName;

}

public void setTaskStatus(String taskStatus) {

this.taskStatus = taskStatus;

}

public void setTaskContent(List<SubTask> taskContent) {

this.taskContent = taskContent;

}

public void setUserId(String userId) {

this.userId = userId;

}

public void setCreatedOn(String createdOn) {

this.createdOn = createdOn;

}

@Override

public String toString() {

return "Task [taskId=" + taskId + ", taskName=" + taskName + ", taskStatus=" + taskStatus + ", taskContent="

+ taskContent + ", userId=" + userId + ", createdOn=" + createdOn + "]";

}

We have to represent the sub tasks as well for which another Model class SubTask to be defined.

package com.rayfocus.api.tasklet.model;

import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBAttribute;
import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBDocument;
import com.fasterxml.jackson.annotation.JsonCreator;
import com.fasterxml.jackson.annotation.JsonProperty;

@DynamoDBDocument
public class SubTask {
	
	private String taskItem;
	private String status;

	@JsonCreator
	public SubTask(@JsonProperty("task_item") String taskItem, 
			       @JsonProperty("status") String status) {
		this.taskItem = taskItem;
		this.status = status;
	}

	public SubTask() 
	{
		// default constructor
	}
	
	/** Getters */
	@DynamoDBAttribute(attributeName = "task_item")
	public String getTaskItem() {
		return taskItem;
	}

	@DynamoDBAttribute(attributeName = "status")
	public String getStatus() {
		return status;
	}

	/** Setters */
	public void setTaskItem(String taskItem) {
		this.taskItem = taskItem;
	}

	public void setStatus(String status) {
		this.status = status;
	}

}

package com.rayfocus.api.tasklet.model;

import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBAttribute;

import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBDocument;

import com.fasterxml.jackson.annotation.JsonCreator;

import com.fasterxml.jackson.annotation.JsonProperty;

@DynamoDBDocument

public class SubTask {

private String taskItem;

private String status;

@JsonCreator

public SubTask(@JsonProperty("task_item") String taskItem,

@JsonProperty("status") String status) {

this.taskItem = taskItem;

this.status = status;

}

public SubTask()

{

// default constructor

}

/** Getters */

@DynamoDBAttribute(attributeName = "task_item")

public String getTaskItem() {

return taskItem;

}

@DynamoDBAttribute(attributeName = "status")

public String getStatus() {

return status;

}

/** Setters */

public void setTaskItem(String taskItem) {

this.taskItem = taskItem;

}

public void setStatus(String status) {

this.status = status;

}

DynamoDB Config

To access the DynamoDB from Spring Boot application, we need to create and register Spring beans for DynamoDB client.

package com.rayfocus.api.tasklet.db.config;

import org.socialsignin.spring.data.dynamodb.repository.config.EnableDynamoDBRepositories;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import com.amazonaws.auth.AWSCredentials;
import com.amazonaws.auth.BasicAWSCredentials;
import com.amazonaws.services.dynamodbv2.AmazonDynamoDB;
import com.amazonaws.services.dynamodbv2.AmazonDynamoDBClient;

@Configuration
@EnableDynamoDBRepositories(basePackages = "com.rayfocus.api.tasklet.db.repository")
public class DynamoDBConfig {

	@Value("${amazon.aws.accesskey}")
	private String awsAccessKey;

	@Value("${amazon.aws.secretkey}")
	private String awsSecretKey;

	@Bean
	public AmazonDynamoDB amazonDynamoDB(AWSCredentials awsCredentials) {
		@SuppressWarnings("deprecation")
		AmazonDynamoDB amazonDynamoDB = new AmazonDynamoDBClient(awsCredentials);
		return amazonDynamoDB;
	}

	@Bean
	public AWSCredentials awsCredentials() {
		return new BasicAWSCredentials(awsAccessKey, awsSecretKey);
	}

}

package com.rayfocus.api.tasklet.db.config;

import org.socialsignin.spring.data.dynamodb.repository.config.EnableDynamoDBRepositories;

import org.springframework.beans.factory.annotation.Value;

import org.springframework.context.annotation.Bean;

import org.springframework.context.annotation.Configuration;

import com.amazonaws.auth.AWSCredentials;

import com.amazonaws.auth.BasicAWSCredentials;

import com.amazonaws.services.dynamodbv2.AmazonDynamoDB;

import com.amazonaws.services.dynamodbv2.AmazonDynamoDBClient;

@Configuration

@EnableDynamoDBRepositories(basePackages = "com.rayfocus.api.tasklet.db.repository")

public class DynamoDBConfig {

@Value("${amazon.aws.accesskey}")

private String awsAccessKey;

@Value("${amazon.aws.secretkey}")

private String awsSecretKey;

@Bean

public AmazonDynamoDB amazonDynamoDB(AWSCredentials awsCredentials) {

@SuppressWarnings("deprecation")

AmazonDynamoDB amazonDynamoDB = new AmazonDynamoDBClient(awsCredentials);

return amazonDynamoDB;

}

@Bean

public AWSCredentials awsCredentials() {

return new BasicAWSCredentials(awsAccessKey, awsSecretKey);

}

To complete the configuration, add the entries for AWS access key and secret key in the application.properties.

amazon.aws.accesskey=[AWS_ACCESS_KEY]
amazon.aws.secretkey=[AWS_SECRET_KEY]

1 2	amazon.aws.accesskey=[AWS_ACCESS_KEY] amazon.aws.secretkey=[AWS_SECRET_KEY]

Note: When deploying the application into Amazon EC2 instance or Amazon ECS, we can make use of the IAM roles to access the other AWS services like DynamoDB or S3, instead of storing credentials in properties file. At this point of time, to develop this project let’s make use of credentials.

IAM Policy for accessing DynamoDB

For this project, I have created a IAM user named TaskletAPI and attached the below custom IAM policy. I have restricted the IAM policy to have access only for DynamoDB operations on a specific table named Tasks.

{
    "Version": "2012-10-17",
    "Statement": [
        {
            "Sid": "VisualEditor0",
            "Effect": "Allow",
            "Action": [
                "dynamodb:BatchGetItem",
                "dynamodb:BatchWriteItem",
                "dynamodb:PutItem",
                "dynamodb:DeleteItem",
                "dynamodb:Scan",
                "dynamodb:Query",
                "dynamodb:UpdateGlobalTable",
                "dynamodb:UpdateItem",
                "dynamodb:CreateTable",
                "dynamodb:DescribeTable",
                "dynamodb:DescribeGlobalTable",
                "dynamodb:GetItem",
                "dynamodb:CreateGlobalTable",
                "dynamodb:UpdateTable"
            ],
            "Resource": "arn:aws:dynamodb:us-east-1:<aws_account_id>:table/Tasks"
        },
        {
            "Sid": "VisualEditor1",
            "Effect": "Allow",
            "Action": "dynamodb:ListTables",
            "Resource": "*"
        }
    ]
}

{

"Version": "2012-10-17",

"Statement": [

{

"Sid": "VisualEditor0",

"Effect": "Allow",

"Action": [

"dynamodb:BatchGetItem",

"dynamodb:BatchWriteItem",

"dynamodb:PutItem",

"dynamodb:DeleteItem",

"dynamodb:Scan",

"dynamodb:Query",

"dynamodb:UpdateGlobalTable",

"dynamodb:UpdateItem",

"dynamodb:CreateTable",

"dynamodb:DescribeTable",

"dynamodb:DescribeGlobalTable",

"dynamodb:GetItem",

"dynamodb:CreateGlobalTable",

"dynamodb:UpdateTable"

"Resource": "arn:aws:dynamodb:us-east-1:<aws_account_id>:table/Tasks"

{

"Sid": "VisualEditor1",

"Effect": "Allow",

"Action": "dynamodb:ListTables",

"Resource": "*"

}

]

}

Repository interface for CRUD operations

Since we have spring-data-dynamodb in our classpath, we can make use of CRUD repository for our database operations. With @EnableScan annotation, the repository implementation will be registered as a component in the Spring container.

package com.rayfocus.api.tasklet.db.repository;

import java.util.List;

import org.socialsignin.spring.data.dynamodb.repository.EnableScan;
import org.springframework.data.repository.CrudRepository;

import com.rayfocus.api.tasklet.model.Task;

@EnableScan
public interface TaskRepository extends CrudRepository<Task, String> {
	List<Task> findByUserId(String userId);
}

package com.rayfocus.api.tasklet.db.repository;

import java.util.List;

import org.socialsignin.spring.data.dynamodb.repository.EnableScan;

import org.springframework.data.repository.CrudRepository;

import com.rayfocus.api.tasklet.model.Task;

@EnableScan

public interface TaskRepository extends CrudRepository<Task, String> {

List<Task> findByUserId(String userId);

}

REST controller

Okay. Now we are all set to write our controller class for handling the incoming requests. Let’s write handler methods for writing/fetching user task to/from DynamoDB. Our controller class looks as shown in below listing.

package com.rayfocus.api.tasklet.controller;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import com.rayfocus.api.tasklet.http.HttpResponse;
import com.rayfocus.api.tasklet.model.Task;
import com.rayfocus.api.tasklet.service.TaskletService;

@RestController
@RequestMapping("/tasklet")
public class TaskletController {

	@Autowired
	TaskletService taskletService;

	@PostMapping("/task")
	public ResponseEntity<HttpResponse> saveTask(@RequestBody Task newTask,
			@RequestParam(value = "version", required = false) String apiVersion) {
		return taskletService.saveTask(newTask);
	}

	@GetMapping("/task/{userid}")
	public ResponseEntity<HttpResponse> getTasksByUserId(@PathVariable("userid") String userId,
			@RequestParam(value = "version", required = false) String apiVersion) {
		return taskletService.getTaskByUserId(userId);
	}
}

package com.rayfocus.api.tasklet.controller;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.http.ResponseEntity;

import org.springframework.web.bind.annotation.GetMapping;

import org.springframework.web.bind.annotation.PathVariable;

import org.springframework.web.bind.annotation.PostMapping;

import org.springframework.web.bind.annotation.RequestBody;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RequestParam;

import org.springframework.web.bind.annotation.RestController;

import com.rayfocus.api.tasklet.http.HttpResponse;

import com.rayfocus.api.tasklet.model.Task;

import com.rayfocus.api.tasklet.service.TaskletService;

@RestController

@RequestMapping("/tasklet")

public class TaskletController {

@Autowired

TaskletService taskletService;

@PostMapping("/task")

public ResponseEntity<HttpResponse> saveTask(@RequestBody Task newTask,

@RequestParam(value = "version", required = false) String apiVersion) {

return taskletService.saveTask(newTask);

}

@GetMapping("/task/{userid}")

public ResponseEntity<HttpResponse> getTasksByUserId(@PathVariable("userid") String userId,

@RequestParam(value = "version", required = false) String apiVersion) {

return taskletService.getTaskByUserId(userId);

}

@PostMapping annotation indicates that the HTTP POST request with URI http://<domain>:<port>/tasklet/task will be handled and the data will be saved to database. The call is then delegated to service class from the controller.

Similarly, @GetMapping annotation on method getTaskByUserId indicates that HTTP GET request with URI http://<domain>:<port>/tasklet/task/{userid}will be handled and the data will be returned after fetching from database.

package com.rayfocus.api.tasklet.service;

import java.util.List;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Component;

import com.rayfocus.api.tasklet.db.repository.TaskRepository;
import com.rayfocus.api.tasklet.http.HttpResponse;
import com.rayfocus.api.tasklet.model.Task;
import com.rayfocus.api.tasklet.model.TaskMetaData;

@Component
public class TaskletService {

	@Autowired
	TaskRepository taskRepository;

	public ResponseEntity<HttpResponse> saveTask(Task newTask) {
		Task savedTask = taskRepository.save(newTask);

		TaskMetaData taskMetaData = null;
		if (savedTask == null) {
			taskMetaData = new TaskMetaData(newTask.getTaskId(), newTask.getTaskName());
			return new ResponseEntity<HttpResponse>(
					new HttpResponse("500", "ERROR", "Task creation failed", taskMetaData),
					HttpStatus.INTERNAL_SERVER_ERROR);
		}

		taskMetaData = new TaskMetaData(savedTask.getTaskId(), savedTask.getTaskName());
		return new ResponseEntity<HttpResponse>(
				new HttpResponse("200", "SUCCESS", "Task created successfully", taskMetaData), HttpStatus.OK);
	}
	
	public ResponseEntity<HttpResponse> getTaskByUserId(String userId){
		List<Task> tasks = taskRepository.findByUserId(userId);
		if (tasks == null) {
			return new ResponseEntity<HttpResponse>(
					new HttpResponse("500", "ERROR", "Error fetching task details", tasks),
					HttpStatus.INTERNAL_SERVER_ERROR);
		}
		return new ResponseEntity<HttpResponse>(new HttpResponse("200", "SUCCESS", "Task Details", tasks),
				HttpStatus.OK);
	}
}

package com.rayfocus.api.tasklet.service;

import java.util.List;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.http.HttpStatus;

import org.springframework.http.ResponseEntity;

import org.springframework.stereotype.Component;

import com.rayfocus.api.tasklet.db.repository.TaskRepository;

import com.rayfocus.api.tasklet.http.HttpResponse;

import com.rayfocus.api.tasklet.model.Task;

import com.rayfocus.api.tasklet.model.TaskMetaData;

@Component

public class TaskletService {

@Autowired

TaskRepository taskRepository;

public ResponseEntity<HttpResponse> saveTask(Task newTask) {

Task savedTask = taskRepository.save(newTask);

TaskMetaData taskMetaData = null;

if (savedTask == null) {

taskMetaData = new TaskMetaData(newTask.getTaskId(), newTask.getTaskName());

return new ResponseEntity<HttpResponse>(

new HttpResponse("500", "ERROR", "Task creation failed", taskMetaData),

HttpStatus.INTERNAL_SERVER_ERROR);

}

taskMetaData = new TaskMetaData(savedTask.getTaskId(), savedTask.getTaskName());

return new ResponseEntity<HttpResponse>(

new HttpResponse("200", "SUCCESS", "Task created successfully", taskMetaData), HttpStatus.OK);

}

public ResponseEntity<HttpResponse> getTaskByUserId(String userId){

List<Task> tasks = taskRepository.findByUserId(userId);

if (tasks == null) {

return new ResponseEntity<HttpResponse>(

new HttpResponse("500", "ERROR", "Error fetching task details", tasks),

HttpStatus.INTERNAL_SERVER_ERROR);

}

return new ResponseEntity<HttpResponse>(new HttpResponse("200", "SUCCESS", "Task Details", tasks),

HttpStatus.OK);

}

HttpResponse and TaskMetaData are helper classes to form the ResponseEntity.

HttpResponse.java

package com.rayfocus.api.tasklet.http;

public class HttpResponse {
	private final String statusCode;
	private final String statusMessage;
	private final String responseDesc;

	private final Object responseData;

	public HttpResponse(String statusCode, String statusMessage, String responseDesc, Object responseData) {
		this.statusCode = statusCode;
		this.statusMessage = statusMessage;
		this.responseDesc = responseDesc;
		this.responseData = responseData;
	}

	public String getStatusCode() {
		return statusCode;
	}

	public String getStatusMessage() {
		return statusMessage;
	}

	public String getResponseDesc() {
		return responseDesc;
	}

	public Object getResponseData() {
		return responseData;
	}

}

package com.rayfocus.api.tasklet.http;

public class HttpResponse {

private final String statusCode;

private final String statusMessage;

private final String responseDesc;

private final Object responseData;

public HttpResponse(String statusCode, String statusMessage, String responseDesc, Object responseData) {

this.statusCode = statusCode;

this.statusMessage = statusMessage;

this.responseDesc = responseDesc;

this.responseData = responseData;

}

public String getStatusCode() {

return statusCode;

}

public String getStatusMessage() {

return statusMessage;

}

public String getResponseDesc() {

return responseDesc;

}

public Object getResponseData() {

return responseData;

}

TaskMetaData.java

package com.rayfocus.api.tasklet.model;

public class TaskMetaData {
	private final String taskId;
	private final String taskName;

	public TaskMetaData(String taskId, String taskName) {
		this.taskId = taskId;
		this.taskName = taskName;
	}

	public String getTaskId() {
		return taskId;
	}

	public String getTaskName() {
		return taskName;
	}

}

package com.rayfocus.api.tasklet.model;

public class TaskMetaData {

private final String taskId;

private final String taskName;

public TaskMetaData(String taskId, String taskName) {

this.taskId = taskId;

this.taskName = taskName;

}

public String getTaskId() {

return taskId;

}

public String getTaskName() {

return taskName;

}

Managing API versions

If we need to manage the API versions going further, we can do that using Query params in the request URI. For example, if we have this application running on port 8080, then issue the request as below:

with query param

http://localhost:8080/tasklet/task?version=2

In this case, we can route to the appropriate service method for processing the request.

without query param

http://localhost:8080/tasklet/task

If we define the query param as optional, then by default we can route to the latest api version service method if the version param is not present in request.

Unit testing

We have completed developing our Tasklet service. Let’s write some unit tests to ensure that the application is working as expected.

package com.rayfocus.api.tasklet;

import static org.junit.Assert.assertTrue;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.http.ResponseEntity;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import com.amazonaws.services.dynamodbv2.AmazonDynamoDB;
import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBMapper;
import com.amazonaws.services.dynamodbv2.model.ProvisionedThroughput;
import com.amazonaws.services.dynamodbv2.util.TableUtils;
import com.amazonaws.services.dynamodbv2.util.TableUtils.TableNeverTransitionedToStateException;
import com.rayfocus.api.tasklet.http.HttpResponse;
import com.rayfocus.api.tasklet.model.SubTask;
import com.rayfocus.api.tasklet.model.Task;
import com.rayfocus.api.tasklet.model.TaskMetaData;
import com.rayfocus.api.tasklet.service.TaskletService;

@RunWith(SpringJUnit4ClassRunner.class)
@SpringBootTest
public final class TaskletServiceTests {

	private final Logger logger = LoggerFactory.getLogger(getClass());

	@Autowired
	private AmazonDynamoDB amazonDynamoDB;

	@Autowired
	private TaskletService taskletService;
	
	private static boolean taskTableExist = false;

	@Before
	public void setUp() throws TableNeverTransitionedToStateException, InterruptedException {
		if( ! taskTableExist ) {
			if(TableUtils.createTableIfNotExists(amazonDynamoDB, new DynamoDBMapper(amazonDynamoDB).generateCreateTableRequest(Task.class)
					.withProvisionedThroughput(new ProvisionedThroughput(2L, 2L)))) {
				logger.info("~~~ Waiting for Tasks table to be active ~~~");
				TableUtils.waitUntilActive(amazonDynamoDB, "Tasks");
			}
			else {
				logger.info("~~~ Tasks table already exists ~~~");
			}
			taskTableExist = true;
		}
	}

	@Test
	public void testSaveTask() {
		logger.info("Saving the taskable items");

		List<SubTask> subTasks = new ArrayList<SubTask>(
				Arrays.asList(new SubTask("Vegetables", "INCOMPLETE"), new SubTask("Fruits", "COMPLETE")));
		Task newTask = new Task("456", "Buy Groceries", "Active", subTasks, "users@taskletAPI.com", "11-JUN-2018");

		ResponseEntity<HttpResponse> response = taskletService.saveTask(newTask);

		assertTrue("Task saved successfully.", response.getBody().getStatusCode() == "200");
		assertTrue("Task Id matches", ((TaskMetaData) response.getBody().getResponseData()).getTaskId() == "456");
		logger.info("Save task done");
	}

	@SuppressWarnings("unchecked")
	@Test
	public void testGetTaskByUserId() {
		logger.info("Fetching the task items");

		ResponseEntity<HttpResponse> response = taskletService.getTaskByUserId("users@taskletAPI.com");

		assertTrue("Task fetched successfully.", response.getBody().getStatusCode() == "200");
		assertTrue("User Id matches", ((List<Task>) response.getBody().getResponseData()).get(0).getUserId()
				.equals("users@taskletAPI.com"));
		logger.info("Get task done");
	}
}

package com.rayfocus.api.tasklet;

import static org.junit.Assert.assertTrue;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.List;

import org.junit.Before;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.boot.test.context.SpringBootTest;

import org.springframework.http.ResponseEntity;

import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import com.amazonaws.services.dynamodbv2.AmazonDynamoDB;

import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBMapper;

import com.amazonaws.services.dynamodbv2.model.ProvisionedThroughput;

import com.amazonaws.services.dynamodbv2.util.TableUtils;

import com.amazonaws.services.dynamodbv2.util.TableUtils.TableNeverTransitionedToStateException;

import com.rayfocus.api.tasklet.http.HttpResponse;

import com.rayfocus.api.tasklet.model.SubTask;

import com.rayfocus.api.tasklet.model.Task;

import com.rayfocus.api.tasklet.model.TaskMetaData;

import com.rayfocus.api.tasklet.service.TaskletService;

@RunWith(SpringJUnit4ClassRunner.class)

@SpringBootTest

public final class TaskletServiceTests {

private final Logger logger = LoggerFactory.getLogger(getClass());

@Autowired

private AmazonDynamoDB amazonDynamoDB;

@Autowired

private TaskletService taskletService;

private static boolean taskTableExist = false;

@Before

public void setUp() throws TableNeverTransitionedToStateException, InterruptedException {

if( ! taskTableExist ) {

if(TableUtils.createTableIfNotExists(amazonDynamoDB, new DynamoDBMapper(amazonDynamoDB).generateCreateTableRequest(Task.class)

.withProvisionedThroughput(new ProvisionedThroughput(2L, 2L)))) {

logger.info("~~~ Waiting for Tasks table to be active ~~~");

TableUtils.waitUntilActive(amazonDynamoDB, "Tasks");

}

else {

logger.info("~~~ Tasks table already exists ~~~");

}

taskTableExist = true;

}

@Test

public void testSaveTask() {

logger.info("Saving the taskable items");

List<SubTask> subTasks = new ArrayList<SubTask>(

Arrays.asList(new SubTask("Vegetables", "INCOMPLETE"), new SubTask("Fruits", "COMPLETE")));

Task newTask = new Task("456", "Buy Groceries", "Active", subTasks, "users@taskletAPI.com", "11-JUN-2018");

ResponseEntity<HttpResponse> response = taskletService.saveTask(newTask);

assertTrue("Task saved successfully.", response.getBody().getStatusCode() == "200");

assertTrue("Task Id matches", ((TaskMetaData) response.getBody().getResponseData()).getTaskId() == "456");

logger.info("Save task done");

}

@SuppressWarnings("unchecked")

@Test

public void testGetTaskByUserId() {

logger.info("Fetching the task items");

ResponseEntity<HttpResponse> response = taskletService.getTaskByUserId("users@taskletAPI.com");

assertTrue("Task fetched successfully.", response.getBody().getStatusCode() == "200");

assertTrue("User Id matches", ((List<Task>) response.getBody().getResponseData()).get(0).getUserId()

.equals("users@taskletAPI.com"));

logger.info("Get task done");

}

Testing with Postman API client

Before testing with Postman client, make sure the application is up and running. It should run on the default port 8080 since we haven’t explicitly configured the server’s port.

Launch the Postman client and make a HTTP POST request to http://localhost:8080/tasklet/task?version=1 . Input the below sample request.

{
  "created_on": "11-JUN-2018",
  "task_content": [
    {
      "status": "INCOMPLETE",
      "task_item": "Vegetables"
    },
    {
      "status": "COMPLETE",
      "task_item": "Cakes & Fruits"
    },
    {
      "status": "COMPLETE",
      "task_item": "Get Rice"
    }
  ],
  "task_id": "123",
  "task_name": "Buy Groceries",
  "task_status": "Active",
  "user_id": "users@taskletAPI.com"
}

{

"created_on": "11-JUN-2018",

"task_content": [

{

"status": "INCOMPLETE",

"task_item": "Vegetables"

{

"status": "COMPLETE",

"task_item": "Cakes & Fruits"

{

"status": "COMPLETE",

"task_item": "Get Rice"

}

"task_id": "123",

"task_name": "Buy Groceries",

"task_status": "Active",

"user_id": "users@taskletAPI.com"

}

You should see the response similar to the one shown in the below listing.

Conclusion

We have seen how to integrate Spring Boot with Amazon DynamoDB . Finally, we have written unit tests to test our Tasklet API service. We will add more features of Spring Boot and Spring Cloud to our application in the upcoming posts. Stay tuned!

The source code we have worked on so far is available in the downloads section. If you have any questions, feel free to post it in the comments section. Thank you.

Download source code : spring-boot-with-dynamodb.zip

Categories: Amazon Web ServicesSpring framework

Tags: Amazon DynamoDBSpring Boot

Vignesh M

Java developer , AWS Certified Solutions Architect Associate and Cloud technology enthusiast. Currently working for Hexaware technologies. He believes that knowledge increases by sharing not by saving.

Integrate Spring Boot With Amazon DynamoDB

Prerequisite tools

DynamoDB library for Java

IAM Policy for accessing DynamoDB

Repository interface for CRUD operations

REST controller

Managing API versions

Unit testing

Conclusion

Vignesh M

Leave a Reply Cancel reply

Spring framework

Secure your Spring REST API using OAuth2

Spring framework

Spring Boot – Hello World Example

Amazon Web Services

Serverless framework with AWS – Example

Integrate Spring Boot With Amazon DynamoDB

Prerequisite tools

DynamoDB library for Java

IAM Policy for accessing DynamoDB

Repository interface for CRUD operations

REST controller

Managing API versions

Unit testing

Conclusion

Vignesh M

Leave a Reply Cancel reply

Related Posts

Spring framework

Secure your Spring REST API using OAuth2

Spring framework

Spring Boot – Hello World Example

Amazon Web Services

Serverless framework with AWS – Example