使用CrudRepository时不调用自定义DynamoDb TableNameResolver

Question

我正在测试DynamoDB表，希望使用关键字“dev”为prod和dev环境设置不同的表名，用于开发，用于生产。

我有一个POJO

@DynamoDBTable(tableName = "abc_xy_dev_MyProjectName_Employee")
public class Employee implements Cloneable {

}

在Prod上，我希望它的名字是abc_xy_Prod_MyProjectName_Employee。

所以，我写了一个TableNameResolver

public static class MyTableNameResolver implements TableNameResolver {
    public static final MyTableNameResolver INSTANCE = new MyTableNameResolver();

    @Override
    public String getTableName(Class<?> clazz, DynamoDBMapperConfig config) {
        final TableNameOverride override = config.getTableNameOverride();
        String tableNameToReturn = null;
        if (override != null) {
            final String tableName = override.getTableName();
            if (tableName != null) {
                System.out.println("MyTableNameResolver ==================================");
                return tableName;
            }
        }
        String env = System.getenv("DEPLOYMENT_ENV");
        for(Annotation annotation : clazz.getAnnotations()){
                if(annotation instanceof DynamoDBTable){
                    DynamoDBTable myAnnotation = (DynamoDBTable) annotation;
                    if ("production".equals(env)){
                        tableNameToReturn = myAnnotation.tableName().replace("_dev_", "_prod_");
                    }
                    else {
                        tableNameToReturn = myAnnotation.tableName();
                    }
                }
            }

        return tableNameToReturn;
    }
}

这是通过创建一个名为abc_xy_prod_MyProjectName_Employee in production的表来实现的。

但是，我有一个包含以下代码的存储库

@EnableScan
public interface EmployeeRepository extends CrudRepository<Employee, String> 
{

    @Override
    <S extends Employee> S save(S employee);

    Optional<Employee> findById(String id);

    @Override
    List<Employee> findAll();

    Optional<Employee> findByEmployeeNumber(String EmployeeNumber);

}

因此，当我试图通过endpoint/测试用例调用findAll方法时，我得到了异常

出现意外错误（type=内部服务器错误，状态=500）。user： arn： aws： iam::87668976786： user/svc_nac_ps_MyProjectName_prod未被授权执行： Dynamodb： Scan on resource::/abc_xy_dev_MyProjectName_Employee（服务： Amazon DynamoDBv2；状态代码： 400；错误代码： AccessDeniedExcture；请求ID： aksdnhLDFL）

i、 执行respository方法时，不会在内部调用MyTableNameResolver。它仍然指向在注释@DynamoDBTable（tableName=“abc_xy_dev_MyProjectName_Employee”）中给出的名为abc_xy_dev_MyProjectName_Employee”的表名

Answer 1

您正在使用DynamoDBMapper（JavaSDK）。下面是我如何使用它。假设我有一个名为用户的表，其中有一个关联的用户POJO。在DynamoDB我有DEV_Users和LIVE_Users。

我有一个环境变量“ApplicationEnvironmentName”，它是DEV或LIVE。

我创建一个自定义DynamoDBMapper，如下所示：

public class ApplicationDynamoMapper {

    private static Map<String, DynamoDBMapper> mappers = new HashMap<>();

    private static AmazonDynamoDB client = AmazonDynamoDBClientBuilder.standard()
            .withRegion(System.getProperty("DynamoDbRegion")).build();

    protected ApplicationDynamoMapper() {
        // Exists only to defeat instantiation.
    }

    public static DynamoDBMapper getInstance(final String tableName) {
        final ApplicationLogContext LOG = new ApplicationLogContext();
        DynamoDBMapper mapper = mappers.get(tableName);
        if (mapper == null) {
            final String tableNameOverride = System.getProperty("ApplicationEnvironmentName") + "_" + tableName;
            LOG.debug("Creating DynamoDBMapper with overridden tablename {}.", tableNameOverride);
            final DynamoDBMapperConfig mapperConfig = new DynamoDBMapperConfig.Builder().withTableNameOverride(TableNameOverride.withTableNameReplacement(tableNameOverride)).build();
            mapper = new DynamoDBMapper(client, mapperConfig);
            mappers.put(tableName, mapper);
        }
        return mapper;
    }
}

我的用户POJO如下所示：

@DynamoDBTable(tableName = "Users")
public class User {
   ...
}

当我想使用数据库时，我会创建一个应用程序映射器，如下所示：

DynamoDBMapper userMapper = ApplicationDynamoMapper.getInstance(User.DB_TABLE_NAME);

如果我想加载一个用户，我会这样做：

User user = userMapper.load(User.class, userId);

希望有帮助。

Answer 2

下面的配置可以用来设置表名覆盖，作为Spring引导配置的一部分。

示例见https://github.com/ganesara/SpringExamples/tree/master/spring-dynamo

使用用户定义的映射器配置参考映射Dynamo db存储库

@EnableDynamoDBRepositories(basePackages = "home.poc.spring", dynamoDBMapperConfigRef="dynamoDBMapperConfig")

表覆盖的映射器配置如下所示

@Bean
public DynamoDBMapperConfig dynamoDBMapperConfig() {
    DynamoDBMapperConfig mapperConfig = new DynamoDBMapperConfig
            .Builder()
            .withTableNameOverride(DynamoDBMapperConfig.TableNameOverride.withTableNamePrefix("PROD_"))
            .build();
    return mapperConfig;
}

完整配置供参考

@Configuration
@EnableDynamoDBRepositories(basePackages = "home.poc.spring", dynamoDBMapperConfigRef="dynamoDBMapperConfig")
public class DynamoDBConfig {

    @Value("${amazon.dynamodb.endpoint}")
    private String amazonDynamoDBEndpoint;

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

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

    @Bean
    public AmazonDynamoDB amazonDynamoDB() {
        AmazonDynamoDB amazonDynamoDB
                = new AmazonDynamoDBClient(amazonAWSCredentials());

        if (!StringUtils.isEmpty(amazonDynamoDBEndpoint)) {
            amazonDynamoDB.setEndpoint(amazonDynamoDBEndpoint);
        }

        return amazonDynamoDB;
    }

    @Bean
    public AWSCredentials amazonAWSCredentials() {
        return new BasicAWSCredentials(
                amazonAWSAccessKey, amazonAWSSecretKey);
    }

    @Bean
    public DynamoDBMapperConfig dynamoDBMapperConfig() {
        DynamoDBMapperConfig mapperConfig = new DynamoDBMapperConfig
                .Builder()
                .withTableNameOverride(DynamoDBMapperConfig.TableNameOverride.withTableNamePrefix("PROD_"))
                .build();
        return mapperConfig;
    }

    @Bean
    public DynamoDBMapper dynamoDBMapper() {
        return new DynamoDBMapper(amazonDynamoDB(), dynamoDBMapperConfig());
    }
}