update task logic fix

qingshu-ui · Sep 26, 2024 · dba692e · dba692e
1 parent 15f6a6c
commit dba692e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/main/kotlin/io/github/qingshu/ayaka/example/plugin/RandomVideo.kt b/src/main/kotlin/io/github/qingshu/ayaka/example/plugin/RandomVideo.kt
@@ -74,6 +74,7 @@ class RandomVideo(
 
     @Scheduled(cron = "0 0/15 * * * ?")
     fun updateVideoInfo() {
+        if (service.allUnUpdatedCount() == 0) return
         log.info("开始更新视频信息")
         val baseConfig = EAConfig.base
         val botSession = sessionFactory.createSession("localhost")

diff --git a/src/main/kotlin/io/github/qingshu/ayaka/example/repository/DouYinVideoRepository.kt b/src/main/kotlin/io/github/qingshu/ayaka/example/repository/DouYinVideoRepository.kt
@@ -37,6 +37,10 @@ interface DouYinVideoRepository : JpaRepository<DouYinVideoEntity, Int> {
 
     fun findByUsedTodayIsTrue(): List<DouYinVideoEntity>
 
+    @Query(
+        value = "select v from DouYinVideoEntity v " +
+                "where v.updateStatus = :status order by function('random') "
+    )
     fun findByUpdateStatus(status: String, pageable: Pageable): List<DouYinVideoEntity>
 
     fun countByUpdateStatus(status: String = "pending"): Int