diff --git a/src/main/java/cn/fw/freya/dao/LivePoolDao.java b/src/main/java/cn/fw/freya/dao/LivePoolDao.java index 845f96c..0e3db5f 100644 --- a/src/main/java/cn/fw/freya/dao/LivePoolDao.java +++ b/src/main/java/cn/fw/freya/dao/LivePoolDao.java @@ -34,4 +34,13 @@ public interface LivePoolDao extends JpaRepository { */ @Query("select live from LivePool live where live.playbackUrl is null and live.roomId is not null and live.type = ?1 and live.duration >= ?2") List getWithoutPlaybackLive(Integer type, Double durationThreshold); + + /** + * 获取没有拿到回播的直播 + * + * @param accountNo 账户号 + * @return + */ + @Query("select live from LivePool live where live.phoneNo = ?1") + List getLiveByAccountNo(String accountNo); } diff --git a/src/main/java/cn/fw/freya/task/DataCaptureTask.java b/src/main/java/cn/fw/freya/task/DataCaptureTask.java index b72b5f3..543773d 100644 --- a/src/main/java/cn/fw/freya/task/DataCaptureTask.java +++ b/src/main/java/cn/fw/freya/task/DataCaptureTask.java @@ -133,7 +133,7 @@ public class DataCaptureTask { } if (flag) { final ArrayList livePools = new ArrayList<>(); - for (LivePool item : list) {// 该人有一次成功就证明已经处理过该人的所有直播了, 将他所有直播进行标记 + for (LivePool item : livePoolDao.getLiveByAccountNo(accountNo)) {// 该人有一次成功就证明已经处理过该人的所有直播了, 将他所有直播进行标记 final LivePool livePool = PublicUtil.copy(item, LivePool.class);// 对象拷贝 livePool.setGetPlaybackFailTimes(-1); livePools.add(livePool);