Cassandra：每个密钥空间的客户数据

Question

问题：我们的一个新客户希望数据存储在他自己的国家（法律法规）。然而，我们使用分布在不同国家的几个数据中心的现有客户数据。

问：我们如何在不太改变现有Cassandra架构的情况下，将新客户的数据分离到其自己的国家？

潜在的解决方案#1:为该客户使用单独的密钥空间。关键字空间之间的模式是相同的，这增加了数据迁移的复杂性，等等。DataStax支持确认可以为每个区域配置keyspace。然而，我们使用的Spring Data Cassandra不允许动态选择keyspace。唯一的方法是使用CqlTemplate并在每次调用之前运行use keyspace blabla，或者在表之前添加keyspaceselect*from blabla.mytable，但这听起来像是一个黑客。

Answer 1

下面的示例和解释与GitHub中的相同

GitHub中的示例现在正在工作。未来最有效的解决方案似乎是使用存储库扩展。将很快更新下面的示例。

请注意，我最初发布的解决方案有一些缺陷，这些缺陷是我在JMeter测试中发现的。Datastax Java驱动html" target="_blank">程序参考建议避免通过session对象设置keyspace。您必须在每个查询中显式设置keyspace。

我将首先定义如何在每个请求上设置租户ID。一个很好的例子是使用特定的HTTP头来定义REST API：

Tenant-Id: ACME

类似地，在每个远程协议上，您可以在每个消息上转发租户ID。假设您使用的是AMQP或JMS，您可以在消息头或属性中转发这个消息。

接下来，您应该在控制器中存储每个请求的传入头。您可以使用ThreadLocal，也可以尝试使用请求范围的bean。

@Component
@Scope(scopeName = "request", proxyMode= ScopedProxyMode.TARGET_CLASS)
public class TenantId {

    private String tenantId;

    public void set(String id) {
        this.tenantId = id;
    }

    public String get() {
        return tenantId;
    }
}

@RestController
public class UserController {

    @Autowired
    private UserRepository userRepo;
    @Autowired
    private TenantId tenantId;

    @RequestMapping(value = "/userByName")
    public ResponseEntity<String> getUserByUsername(
            @RequestHeader("Tenant-ID") String tenantId,
            @RequestParam String username) {
        // Setting the tenant ID
        this.tenantId.set(tenantId);
        // Finding user
        User user = userRepo.findOne(username);
        return new ResponseEntity<>(user.getUsername(), HttpStatus.OK);
    }
}

public class KeyspaceAwareCassandraRepository<T, ID extends Serializable>
        extends SimpleCassandraRepository<T, ID>  {

    private final CassandraEntityInformation<T, ID> metadata;
    private final CassandraOperations operations;

    @Autowired
    private TenantId tenantId;

    public KeyspaceAwareCassandraRepository(
            CassandraEntityInformation<T, ID> metadata,
            CassandraOperations operations) {
        super(metadata, operations);
        this.metadata = metadata;
        this.operations = operations;
    }

    private void injectDependencies() {
        SpringBeanAutowiringSupport
                .processInjectionBasedOnServletContext(this,
                getServletContext());
    }

    private ServletContext getServletContext() {
        return ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes())
                .getRequest().getServletContext();
    }

    @Override
    public T findOne(ID id) {
        injectDependencies();
        CqlIdentifier primaryKey = operations.getConverter()
                .getMappingContext()
                .getPersistentEntity(metadata.getJavaType())
                .getIdProperty().getColumnName();

        Select select = QueryBuilder.select().all()
                .from(tenantId.get(),
                        metadata.getTableName().toCql())
                .where(QueryBuilder.eq(primaryKey.toString(), id))
                .limit(1);

        return operations.selectOne(select, metadata.getJavaType());
    }

    // All other overrides should be similar
}

@SpringBootApplication
@EnableCassandraRepositories(repositoryBaseClass = KeyspaceAwareCassandraRepository.class)
public class DemoApplication {
...
}

null