[Optimize-4422][admin] CatalogueService add a new method findByTaskId (#4442)

Author: aiwenmo

dinky-admin/src/main/java/org/dinky/controller/CatalogueController.java

@@ -162,6 +162,28 @@ public Result<List<TreeVo>> getCatalogueSortType() {
         return Result.succeed(catalogueSortType);
     }
 
+    /**
+     * find catalogue by task id
+     * @param taskId task id
+     * @return {@link Result}< {@link Catalogue}>}
+     */
+    @GetMapping("/findByTaskId")
+    @ApiOperation("Find Catalogue By Task ID")
+    @ApiImplicitParam(
+            name = "taskId",
+            value = "taskId",
+            required = true,
+            dataType = "Integer",
+            dataTypeClass = Integer.class)
+    public Result<Catalogue> findByTaskId(@RequestParam Integer taskId) {
+        Catalogue catalogue = catalogueService.findByTaskId(taskId);
+        if (catalogue != null) {
+            return Result.succeed(catalogue);
+        } else {
+            return Result.failed(Status.CATALOGUE_NOT_EXIST);
+        }
+    }
+
     /**
      * create catalogue and task
      * @param catalogueTaskDTO {@link CatalogueTaskDTO}

dinky-admin/src/main/java/org/dinky/service/catalogue/CatalogueService.java

@@ -61,6 +61,14 @@ public interface CatalogueService extends ISuperService<Catalogue> {
      */
     Catalogue findByParentIdAndName(Integer parentId, String name);
 
+    /**
+     * Find a catalogue by its task ID.
+     *
+     * @param taskId The ID of the task to find the catalogue for.
+     * @return A {@link Catalogue} object representing the found catalogue, or null if not found.
+     */
+    Catalogue findByTaskId(Integer taskId);
+
     /**
      * Save or update a catalogue and its tasks.
      *

dinky-admin/src/main/java/org/dinky/service/catalogue/impl/CatalogueServiceImpl.java

@@ -248,6 +248,11 @@ public Catalogue findByParentIdAndName(Integer parentId, String name) {
                 .eq(Catalogue::getName, name));
     }
 
+    @Override
+    public Catalogue findByTaskId(Integer taskId) {
+        return baseMapper.selectOne(new LambdaQueryWrapper<Catalogue>().eq(Catalogue::getTaskId, taskId));
+    }
+
     /**
      * check catalogue task name is exist
      *

dinky-admin/src/test/java/org/dinky/service/catalogue/impl/CatalogueServiceImplTest.java

@@ -177,4 +177,47 @@ public void importCatalogueTest() {
         catalogueServiceImplTest.importCatalogue(importCatalogueDto);
         verify(taskService, times(2)).saveBatch(anyList());
     }
+
+    @Test
+    public void findByTaskIdTest() {
+        // Given
+        Integer taskId = 123;
+        Catalogue expectedCatalogue = new Catalogue();
+        expectedCatalogue.setId(1);
+        expectedCatalogue.setName("Test Catalogue");
+        expectedCatalogue.setTaskId(taskId);
+        expectedCatalogue.setType("FlinkSQL");
+        expectedCatalogue.setParentId(0);
+        expectedCatalogue.setIsLeaf(true);
+
+        // Mock
+        when(catalogueMapper.selectOne(any(LambdaQueryWrapper.class))).thenReturn(expectedCatalogue);
+
+        // When
+        Catalogue result = catalogueServiceImplTest.findByTaskId(taskId);
+
+        // Then
+        assertNotNull(result);
+        assertEquals(expectedCatalogue.getId(), result.getId());
+        assertEquals(expectedCatalogue.getName(), result.getName());
+        assertEquals(expectedCatalogue.getTaskId(), result.getTaskId());
+        assertEquals(expectedCatalogue.getType(), result.getType());
+        assertEquals(expectedCatalogue.getParentId(), result.getParentId());
+        assertEquals(expectedCatalogue.getIsLeaf(), result.getIsLeaf());
+    }
+
+    @Test
+    public void findByTaskIdNotFoundTest() {
+        // Given
+        Integer taskId = 999;
+
+        // Mock - return null when catalogue not found
+        when(catalogueMapper.selectOne(any(LambdaQueryWrapper.class))).thenReturn(null);
+
+        // When
+        Catalogue result = catalogueServiceImplTest.findByTaskId(taskId);
+
+        // Then
+        assertNull(result);
+    }
 }