feat(1449): update method reference in DefaultDynamoDbTableSchemaResolver to use fromClass

Author: marcusvoltolim

spring-cloud-aws-dynamodb/src/main/java/io/awspring/cloud/dynamodb/DefaultDynamoDbTableSchemaResolver.java

@@ -19,15 +19,18 @@
 import java.util.List;
 import java.util.Map;
 import java.util.concurrent.ConcurrentHashMap;
+
 import software.amazon.awssdk.enhanced.dynamodb.TableSchema;
 
 /**
  * Default implementation with simple cache for {@link TableSchema}.
  *
  * @author Matej Nedic
  * @author Maciej Walkowiak
+ * @author Marcus Voltolim
  */
 public class DefaultDynamoDbTableSchemaResolver implements DynamoDbTableSchemaResolver {
+
 	private final Map<Class<?>, TableSchema> tableSchemaCache = new ConcurrentHashMap<>();
 
 	public DefaultDynamoDbTableSchemaResolver() {
@@ -42,10 +45,11 @@ public DefaultDynamoDbTableSchemaResolver(List<TableSchema<?>> tableSchemas) {
 
 	@Override
 	public <T> TableSchema<T> resolve(Class<T> clazz) {
-		return tableSchemaCache.computeIfAbsent(clazz, TableSchema::fromBean);
+		return tableSchemaCache.computeIfAbsent(clazz, TableSchema::fromClass);
 	}
 
 	Map<Class<?>, TableSchema> getTableSchemaCache() {
 		return tableSchemaCache;
 	}
+
 }

spring-cloud-aws-dynamodb/src/test/java/io/awspring/cloud/dynamodb/DynamoDbTableSchemaResolverTest.java

@@ -19,6 +19,7 @@
 import static org.assertj.core.api.Assertions.assertThatThrownBy;
 
 import java.util.List;
+
 import org.junit.jupiter.api.Test;
 import software.amazon.awssdk.enhanced.dynamodb.TableSchema;
 import software.amazon.awssdk.enhanced.dynamodb.mapper.StaticTableSchema;
@@ -28,6 +29,7 @@
  *
  * @author Matej Nedic
  * @author Maciej Walkowiak
+ * @author Marcus Voltolim
  */
 class DynamoDbTableSchemaResolverTest {
 
@@ -46,6 +48,21 @@ void tableSchemaResolved_successfully() {
 		assertThat(defaultTableSchemaResolver.getTableSchemaCache()).hasSize(1);
 	}
 
+	@Test
+	void tableSchemaForImmutableResolved_successfully() {
+		// given
+		DefaultDynamoDbTableSchemaResolver defaultTableSchemaResolver = new DefaultDynamoDbTableSchemaResolver();
+		// when
+		TableSchema<PersonImmutable> tableSchema = defaultTableSchemaResolver.resolve(PersonImmutable.class);
+
+		// Call one more time to see if cache is being filled properly.
+		defaultTableSchemaResolver.resolve(PersonImmutable.class);
+
+		// then
+		assertThat(tableSchema).isNotNull();
+		assertThat(defaultTableSchemaResolver.getTableSchemaCache()).hasSize(1);
+	}
+
 	@Test
 	void tableSchemaResolved_whenSchemaPassedThroughConstructor() {
 		StaticTableSchema<Library> librarySchema = StaticTableSchema.builder(Library.class).build();
@@ -86,4 +103,5 @@ void tableSchemaResolver_fail_entity_not_annotated() {
 
 	static class Library {
 	}
+
 }

spring-cloud-aws-dynamodb/src/test/java/io/awspring/cloud/dynamodb/PersonImmutable.java

@@ -0,0 +1,53 @@
+/*
+ * Copyright 2013-2022 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package io.awspring.cloud.dynamodb;
+
+import software.amazon.awssdk.enhanced.dynamodb.mapper.annotations.DynamoDbImmutable;
+
+/**
+ * Immutable version of {@link Person}.
+ *
+ * @author Marcus Voltolim
+ */
+@DynamoDbImmutable(builder = PersonImmutable.Builder.class)
+public class PersonImmutable extends Person {
+
+	private PersonImmutable(String firstName, String lastName) {
+		super(firstName, lastName);
+	}
+
+	public static class Builder {
+
+		private String firstName;
+		private String lastName;
+
+		public Builder firstName(String firstName) {
+			this.firstName = firstName;
+			return this;
+		}
+
+		public Builder lastName(String lastName) {
+			this.lastName = lastName;
+			return this;
+		}
+
+		public PersonImmutable build() {
+			return new PersonImmutable(firstName, lastName);
+		}
+
+	}
+
+}