feat(flyer): 添加任务类型匹配标识功能

linwei

2026-07-08 f5ea859946f70d9d3a50c52fa2bf0c3e9939ba09

 drone-service/drone-gd/src/main/java/org/sxkj/gd/flyer/service/impl/GdFlyerServiceImpl.java

@@ -47,9 +47,72 @@
@Service
public class GdFlyerServiceImpl extends BaseServiceImpl<GdFlyerMapper, GdFlyerEntity> implements IGdFlyerService {

   /**
    * 分页查询飞手信息列表
    * @param page 分页对象
    * @param gdFlyer 查询参数
    * @return 分页结果
    */
   @Override
   public IPage<GdFlyerVO> selectGdFlyerPage(IPage<GdFlyerVO> page, GdFlyerPageParam gdFlyer) {
      return page.setRecords(baseMapper.selectGdFlyerPage(page, gdFlyer));
      // 步骤1：执行数据库查询获取飞手列表
      List<GdFlyerVO> gdFlyerVOS = baseMapper.selectGdFlyerPage(page, gdFlyer);

      // 步骤2：判断是否有擅长任务类型的查询条件
      if (StringUtil.isNotBlank(gdFlyer.getSkilledTaskType())) {
         // 步骤3：获取查询的任务类型
         String searchTaskType = gdFlyer.getSkilledTaskType();

         // 步骤4：遍历结果，设置匹配标识并根据匹配数量排序
         gdFlyerVOS.forEach(vo -> {
            // 步骤5：计算匹配数量
            int matchCount = countMatchedTaskTypes(vo.getSkilledTaskType(), searchTaskType);
            // 步骤6：设置匹配标识（匹配数量大于0返回1，否则返回0）
            vo.setTaskTypeMatchFlag(matchCount > 0 ? 1 : 0);
         });

         // 步骤7：对结果进行排序，匹配数量多的排在前面
         gdFlyerVOS.sort((vo1, vo2) -> {
            // 步骤8：计算记录1匹配的任务类型数量
            int matchCount1 = countMatchedTaskTypes(vo1.getSkilledTaskType(), searchTaskType);
            // 步骤9：计算记录2匹配的任务类型数量
            int matchCount2 = countMatchedTaskTypes(vo2.getSkilledTaskType(), searchTaskType);
            // 步骤10：匹配数量多的排在前面（降序排列）
            return Integer.compare(matchCount2, matchCount1);
         });
      } else {
         // 步骤11：无查询条件时，将所有记录的匹配标识设置为0
         gdFlyerVOS.forEach(vo -> vo.setTaskTypeMatchFlag(0));
      }

      return page.setRecords(gdFlyerVOS);
   }

   /**
    * 计算任务类型列表中匹配的任务类型数量
    * @param skilledTaskTypes 擅长的任务类型列表（二维数组）
    * @param searchTaskType 要搜索的任务类型（多个用逗号分隔，如：road4,road5）
    * @return 匹配的任务类型数量
    */
   private int countMatchedTaskTypes(List<List<String>> skilledTaskTypes, String searchTaskType) {
      // 步骤1：校验参数有效性
      if (skilledTaskTypes == null || StringUtil.isBlank(searchTaskType)) {
         return 0;
      }
      // 步骤2：拆分多个任务类型（支持逗号分隔）
      String[] taskTypes = searchTaskType.split(",");
      // 步骤3：计算匹配数量
      int matchCount = 0;
      for (String taskType : taskTypes) {
         String trimmedTaskType = taskType.trim();
         // 步骤4：检查二维数组中是否包含该任务类型
         boolean matched = skilledTaskTypes.stream()
            .anyMatch(innerList -> innerList != null && innerList.contains(trimmedTaskType));
         if (matched) {
            matchCount++;
         }
      }
      return matchCount;
   }