optimize: count for episodes (#710)

* optimize: EpisodeRepository.countBySubjectId

* optimize: DefaultEpisodeService.countMatchingBySubjectId

* docs: update CHANGELOG.MD

CHANGELOG.MD

@@ -2,6 +2,11 @@
 
 更新日志文档，版本顺序从新到旧，最新版本在最前(上)面。
 
+# 0.17.6
+
+- 优化查询条目剧集数的逻辑
+- 优化查询条目匹配了资源的剧集数逻辑
+
 # 0.17.5
 
 ## 优化

gradle.properties

@@ -1 +1 @@
-version=0.17.5
+version=0.17.6

server/src/main/java/run/ikaros/server/core/episode/DefaultEpisodeService.java

@@ -5,6 +5,7 @@
 import java.util.List;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.context.ApplicationEventPublisher;
+import org.springframework.r2dbc.core.DatabaseClient;
 import org.springframework.stereotype.Service;
 import org.springframework.util.Assert;
 import reactor.core.publisher.Flux;
@@ -25,18 +26,21 @@ public class DefaultEpisodeService implements EpisodeService {
     private final AttachmentReferenceRepository attachmentReferenceRepository;
     private final AttachmentRepository attachmentRepository;
     private final ApplicationEventPublisher applicationEventPublisher;
+    private final DatabaseClient databaseClient;
 
     /**
      * Construct.
      */
     public DefaultEpisodeService(EpisodeRepository episodeRepository,
                                  AttachmentReferenceRepository attachmentReferenceRepository,
                                  AttachmentRepository attachmentRepository,
-                                 ApplicationEventPublisher applicationEventPublisher) {
+                                 ApplicationEventPublisher applicationEventPublisher,
+                                 DatabaseClient databaseClient) {
         this.episodeRepository = episodeRepository;
         this.attachmentReferenceRepository = attachmentReferenceRepository;
         this.attachmentRepository = attachmentRepository;
         this.applicationEventPublisher = applicationEventPublisher;
+        this.databaseClient = databaseClient;
     }
 
 
@@ -114,13 +118,12 @@ public Mono<Long> countBySubjectId(Long subjectId) {
     @Override
     public Mono<Long> countMatchingBySubjectId(Long subjectId) {
         Assert.isTrue(subjectId >= 0, "'subjectId' must >= 0.");
-        return episodeRepository.findAllBySubjectId(subjectId)
-            .flatMap(entity -> attachmentReferenceRepository.existsByTypeAndReferenceId(
-                AttachmentReferenceType.EPISODE, entity.getId()))
-            .filter(exists -> exists)
-            .collectList()
-            .map(List::size)
-            .map(Long::valueOf);
+        return databaseClient.sql("select count(e.ID) from EPISODE e, ATTACHMENT_REFERENCE ar "
+                + "where ar.TYPE = 'EPISODE' and e.ID = ar.REFERENCE_ID "
+                + "and e.SUBJECT_ID = :subjectId")
+            .bind("subjectId", subjectId)
+            .map(row -> row.get(0, Long.class))
+            .one();
     }
 
     @Override

server/src/main/java/run/ikaros/server/store/repository/EpisodeRepository.java

@@ -1,5 +1,6 @@
 package run.ikaros.server.store.repository;
 
+import org.springframework.data.r2dbc.repository.Query;
 import org.springframework.data.r2dbc.repository.R2dbcRepository;
 import reactor.core.publisher.Flux;
 import reactor.core.publisher.Mono;
@@ -9,6 +10,10 @@
 public interface EpisodeRepository extends R2dbcRepository<EpisodeEntity, Long> {
     Flux<EpisodeEntity> findAllBySubjectId(Long subjectId);
 
+    @Query(value = "select id from episode where subject_id = :subjectId")
+    Flux<Long> findAllIdBySubjectId(Long subjectId);
+
+    @Query(value = "select count(id) from episode where subject_id = :subjectId")
     Mono<Long> countBySubjectId(Long subjectId);
 
     Flux<EpisodeEntity> findAllBySubjectIdOrderByGroupDescSequenceAscAirTimeAscCreateTimeAsc(