Spring Boot + Oracle 实现 Web 服务实例唯一数据集分配

2024-09-11 18:13:01

在分布式系统中，处理大量数据时，我们常常需要多个服务实例协同工作。如何确保每个实例处理的数据不重复，成为了一个关键问题。本文将探讨在Java和Spring Boot环境下，如何利用Oracle数据库的特性，实现基于特定条件为每个Web服务实例分配唯一数据集。

我们假设一个场景：四个Web服务实例需要从名为INBOUND_REQUEST的数据库表中获取数据并进行处理。这张表存储了待处理请求的信息，包括请求ID、账户代码、请求消息、创建时间和状态等。

我们的目标是避免重复处理，即每个服务实例只能处理特定账户代码对应的请求。比如，如果实例1处理了账户代码为“APPLE”的所有请求，那么其他实例就不能再处理“APPLE”的任何请求。

为了实现这个目标，我们可以结合数据库特性和Java代码，设计一种分配机制。

1. 为数据库表添加处理状态字段:

首先，我们需要在INBOUND_REQUEST表中添加一个字段，用于记录请求的处理状态。比如，我们可以添加一个名为processing_instance的字段，存储正在处理该请求的服务实例ID。初始状态可以设置为NULL，表示该请求尚未被任何实例处理。

2. 利用乐观锁:

Oracle数据库的乐观锁机制可以帮助我们在更新processing_instance字段时，确保只有一个实例能够成功获取该请求的处理权。

3. Java代码实现:

在Spring Boot应用中，我们可以使用JdbcTemplate或JPA等技术访问数据库。下面是一个使用JdbcTemplate实现的示例代码：

@Service
public class RequestProcessor {

    @Autowired
    private JdbcTemplate jdbcTemplate;

    public List<Request> getRequestsForProcessing(String instanceId) {
        // 尝试获取一个状态为PENDING且尚未被处理的请求，并将其processing_instance字段设置为当前实例ID
        String sql = "UPDATE INBOUND_REQUEST SET processing_instance = ? WHERE status = 'PENDING' AND processing_instance IS NULL AND ROWNUM = 1 RETURNING requestID, Account_cd, Requestmsg, createdDateTime, status";
        List<Request> requests = jdbcTemplate.query(sql, new BeanPropertyRowMapper<>(Request.class), instanceId);

        if (!requests.isEmpty()) {
            // 如果成功获取到一个请求，则继续将该账户代码下的所有未处理请求的processing_instance字段设置为当前实例ID
            String accountCd = requests.get(0).getAccountCd();
            sql = "UPDATE INBOUND_REQUEST SET processing_instance = ? WHERE status = 'PENDING' AND processing_instance IS NULL AND Account_cd = ?";
            jdbcTemplate.update(sql, instanceId, accountCd);

            // 最后返回所有分配给当前实例的请求列表
            sql = "SELECT requestID, Account_cd, Requestmsg, createdDateTime, status FROM INBOUND_REQUEST WHERE processing_instance = ?";
            return jdbcTemplate.query(sql, new BeanPropertyRowMapper<>(Request.class), instanceId);
        }

        return requests;
    }
}

这段代码的逻辑是：首先尝试获取一个状态为PENDING且尚未被处理的请求，并将它的processing_instance字段设置为当前实例的ID。如果成功获取到一个请求，就继续将该账户代码下所有未处理请求的processing_instance字段设置为当前实例的ID，最后返回分配给当前实例的所有请求列表。

4. 服务实例部署:

我们需要将多个Web服务实例部署到不同的服务器或容器中，每个实例都使用相同的数据库连接配置。