#1154 make delete return entity

[MatejNedic]

📢 Type of change

• Bugfix
• New feature
• Enhancement
• Refactoring

📜 Description

#1154

💡 Motivation and Context

💚 How did you test it?

📝 Checklist

• I reviewed submitted code
• I added tests to verify changes
• I updated reference documentation to reflect the change
• All tests passing
• No breaking changes

🔮 Next steps

[alexanderankin]

thank you!

adding to my codebase for now until i can change the import and type declaration back to the official one is released:

import io.awspring.cloud.dynamodb.DefaultDynamoDbTableNameResolver;
import io.awspring.cloud.dynamodb.DefaultDynamoDbTableSchemaResolver;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;
import org.springframework.util.Assert;
import software.amazon.awssdk.enhanced.dynamodb.DynamoDbEnhancedClient;
import software.amazon.awssdk.enhanced.dynamodb.DynamoDbTable;
import software.amazon.awssdk.enhanced.dynamodb.Key;

/**
 * @see <a href="https://github.com/awspring/spring-cloud-aws/pull/1185">awspring/spring-cloud-aws#1185</a>
 */
@RequiredArgsConstructor
@Component
public class DynamoDbTemplateIssue1185 {
    private final DefaultDynamoDbTableNameResolver tableNameResolver;
    private final DefaultDynamoDbTableSchemaResolver dynamoDbTableSchemaResolver;
    private final DynamoDbEnhancedClient dynamoDbEnhancedClient;

    public <T> T delete(Key key, Class<T> clazz) {
        Assert.notNull(key, "key is required");
        Assert.notNull(clazz, "clazz is required");
        return prepareTable(clazz).deleteItem(key);
    }

    public <T> T delete(T entity) {
        Assert.notNull(entity, "entity is required");
        return prepareTable(entity).deleteItem(entity);
    }

    protected <T> DynamoDbTable<T> prepareTable(Class<T> clazz) {
        Assert.notNull(clazz, "clazz is required");
        String tableName = tableNameResolver.resolve(clazz);
        return dynamoDbEnhancedClient.table(tableName, dynamoDbTableSchemaResolver.resolve(clazz));
    }

    @SuppressWarnings("unchecked")
    protected <T> DynamoDbTable<T> prepareTable(T entity) {
        Assert.notNull(entity, "entity is required");
        String tableName = tableNameResolver.resolve(entity.getClass());
        return (DynamoDbTable<T>) dynamoDbEnhancedClient.table(tableName,
                dynamoDbTableSchemaResolver.resolve(entity.getClass()));
    }
}