如何修改Mongo Repository使其随机返回单个匹配条件元素

两种落地方式，可根据项目数据量选择：

方案1：数据库层聚合随机（推荐，生产环境优先选） #

利用MongoDB原生$sample聚合算子直接在数据库侧完成随机抽取，不需要把所有符合条件的数据加载到应用内存，大数据量下性能优势明显。
各层代码修改如下：

• Repository层：新增带聚合注解的查询方法，原有全量查询方法可按需保留

import org.springframework.data.mongodb.repository.Aggregation;
import org.springframework.data.mongodb.repository.MongoRepository;
import java.util.Optional;

public interface PackageHolidayRepository extends MongoRepository<PackageHoliday, String> {
    // 原有全量查询方法
    List<PackageHoliday> findPackageHolidayByTypeOfPackageHolidayContains(String escortedTours);

    // 新增随机取1条的方法：先匹配条件，再随机抽1条
    @Aggregation(pipeline = {
        "{$match: {typeOfPackageHoliday: {$regex: ?0}}}",
        "{$sample: {size: 1}}"
    })
    Optional<PackageHoliday> findRandomOneByTypeContains(String keyword);
}

• Service层：调用新增的Repository方法，处理空结果场景

import java.util.Optional;

@Override
public PackageHoliday getRandomEscortedTours() {
    Optional<PackageHoliday> result = packageHolidayRepository.findRandomOneByTypeContains("escortedTours");
    // 可替换为项目自定义的业务异常，比如资源未找到异常
    return result.orElseThrow(() -> new IllegalArgumentException("无匹配的跟团游数据"));
}

• Controller层：调整返回值类型为单个实体，不再返回列表

@GetMapping("/escortedTours")
public PackageHoliday getRandomEscortTours() {
    return packageHolidayService.getRandomEscortedTours();
}

性能提示：如果项目使用的Spring Data版本过低不支持@Aggregation注解，直接注入MongoTemplate手写聚合管道即可，逻辑和注解中定义的pipeline完全一致。

方案2：内存随机（仅适合匹配数据量极小的场景） #

如果符合条件的数据总量常年在几十条以内，可以直接复用现有全量查询方法，在应用内存中随机取1条，实现逻辑最简单，但数据量大时会产生不必要的网络传输和内存占用。
仅需修改Service层和Controller层即可：

• Service层修改

import java.util.List;
import java.util.Random;

@Override
public PackageHoliday getRandomEscortedTours() {
    List<PackageHoliday> allMatched = packageHolidayRepository.findPackageHolidayByTypeOfPackageHolidayContains("escortedTours");
    if (allMatched.isEmpty()) {
        throw new IllegalArgumentException("无匹配的跟团游数据");
    }
    int randomIdx = new Random().nextInt(allMatched.size());
    return allMatched.get(randomIdx);
}

• Controller层修改：和方案1一致，调整返回值为单个PackageHoliday实体即可。

注意：如果接口需要严格随机，$sample在数据量大于100条时的随机性满足绝大多数业务场景，不需要额外处理。

内容的提问来源于stack exchange，提问作者devnoob