2022年5月26日17:21:35

王明元
1 parent e2be3e7b
Showing 2 changed files with 10 additions and 1 deletions
src/main/java/cn/fw/freya/dao/LivePoolDao.java
src/main/java/cn/fw/freya/task/DataCaptureTask.java
@@ -34,4 +34,13 @@ public interface LivePoolDao extends JpaRepository&lt;LivePool, Long&gt; {
      */
     @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<LivePool> getWithoutPlaybackLive(Integer type, Double durationThreshold);
+
+    /**
+     * 获取没有拿到回播的直播
+     *
+     * @param accountNo 账户号
+     * @return
+     */
+    @Query("select live from LivePool live where live.phoneNo = ?1")
+    List<LivePool> getLiveByAccountNo(String accountNo);
 }
@@ -133,7 +133,7 @@ public class DataCaptureTask {
             }
             if (flag) {
                 final ArrayList<LivePool> 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);