如何通过Spring Data JPA按tenant_id查询复合主键关联的Account列表

首先需要调整服务层方法，使其接收tenant_id参数：

1. 修改服务层接口与实现 #

AccountService接口 #

public interface AccountService {
    List<Account> getByTenantId(Integer tenantId);
}

AccountServiceImpl实现类 #

@Service
public class AccountServiceImpl implements AccountService {
    private final AccountRepository accountRepository;

    // 优先使用构造器注入（Spring 4.3+推荐，替代@Autowired字段注入）
    public AccountServiceImpl(AccountRepository accountRepository) {
        this.accountRepository = accountRepository;
    }

    @Override
    public List<Account> getByTenantId(Integer tenantId) {
        return accountRepository.findByTenantId(tenantId);
    }
}

2. 实现AccountRepository的几种方式 #

假设你的AccountRepository继承自JpaRepository<Account, AccountId>，以下是三种Spring Data JPA原生实现方案：

方式一：利用方法命名规则自动生成查询 #

直接定义符合Spring Data JPA命名规范的方法，框架会自动生成对应SQL：

public interface AccountRepository extends JpaRepository<Account, AccountId> {
    // 通过复合主键中的tenantId字段查询
    List<Account> findByIdTenantId(Integer tenantId);

    // 或者通过关联的Tenant对象的id查询
    List<Account> findByTenantId(Integer tenantId);
}

方式二：使用@Query注解编写JPQL #

如果需要更灵活的查询逻辑，可通过JPQL明确指定查询条件：

public interface AccountRepository extends JpaRepository<Account, AccountId> {
    @Query("SELECT a FROM Account a WHERE a.id.tenantId = :tenantId")
    List<Account> findByTenantId(@Param("tenantId") Integer tenantId);

    // 或通过关联的Tenant对象查询
    @Query("SELECT a FROM Account a WHERE a.tenant.id = :tenantId")
    List<Account> findByTenantId(@Param("tenantId") Integer tenantId);
}

方式三：通过Specification实现动态查询（适配多条件场景） #

若后续需扩展多条件查询，可借助Specification：

public interface AccountRepository extends JpaRepository<Account, AccountId>, JpaSpecificationExecutor<Account> {
}

在Service层构建查询条件：

@Override
public List<Account> getByTenantId(Integer tenantId) {
    Specification<Account> spec = (root, query, cb) -> 
        cb.equal(root.get("id").get("tenantId"), tenantId);
    
    // 也可通过关联对象构建条件
    // Specification<Account> spec = (root, query, cb) -> 
    //     cb.equal(root.get("tenant").get("id"), tenantId);
    
    return accountRepository.findAll(spec);
}

3. 额外优化建议 #

• 若查询时需要返回Tenant关联数据，可在JPQL中使用FETCH JOIN避免N+1查询问题，例如：SELECT a FROM Account a JOIN FETCH a.tenant WHERE a.id.tenantId = :tenantId
• 坚持使用构造器注入，提升代码可测试性与依赖透明性

内容的提问来源于stack exchange，提问作者Đỗ Như Vỹ