feat: 重构代码.

orion-ops-module-infra/orion-ops-module-infra-provider/src/main/java/com/orion/ops/module/infra/api/DataGroupApi.java

@@ -58,6 +58,14 @@ public interface DataGroupApi {
      */
     List<DataGroupDTO> getDataGroupTree(DataGroupTypeEnum type);
 
+    /**
+     * 通过 id 查询
+     *
+     * @param idList idList
+     * @return rows
+     */
+    List<DataGroupDTO> getByIdList(List<Long> idList);
+
     /**
      * 删除数据分组
      *

orion-ops-module-infra/orion-ops-module-infra-provider/src/main/java/com/orion/ops/module/infra/enums/DataPermissionTypeEnum.java

@@ -19,6 +19,16 @@ public enum DataPermissionTypeEnum {
      */
     HOST_GROUP(true),
 
+    /**
+     * 主机秘钥
+     */
+    HOST_KEY(true),
+
+    /**
+     * 主机身份
+     */
+    HOST_IDENTITY(true),
+
     ;
 
     /**

orion-ops-module-infra/orion-ops-module-infra-service/src/main/java/com/orion/ops/module/infra/api/impl/DataGroupApiImpl.java

@@ -3,6 +3,8 @@ package com.orion.ops.module.infra.api.impl;
 import com.orion.ops.framework.common.utils.Valid;
 import com.orion.ops.module.infra.api.DataGroupApi;
 import com.orion.ops.module.infra.convert.DataGroupProviderConvert;
+import com.orion.ops.module.infra.dao.DataGroupDAO;
+import com.orion.ops.module.infra.entity.domain.DataGroupDO;
 import com.orion.ops.module.infra.entity.dto.DataGroupCacheDTO;
 import com.orion.ops.module.infra.entity.dto.data.DataGroupCreateDTO;
 import com.orion.ops.module.infra.entity.dto.data.DataGroupDTO;
@@ -18,6 +20,7 @@ import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * 数据分组 对外服务实现类
@@ -33,6 +36,9 @@ public class DataGroupApiImpl implements DataGroupApi {
     @Resource
     private DataGroupService dataGroupService;
 
+    @Resource
+    private DataGroupDAO dataGroupDAO;
+
     @Override
     public Long createDataGroup(DataGroupTypeEnum type, DataGroupCreateDTO dto) {
         Valid.valid(dto);
@@ -67,6 +73,14 @@ public class DataGroupApiImpl implements DataGroupApi {
         return DataGroupProviderConvert.MAPPER.toList(rows);
     }
 
+    @Override
+    public List<DataGroupDTO> getByIdList(List<Long> idList) {
+        List<DataGroupDO> rows = dataGroupDAO.selectBatchIds(idList);
+        return rows.stream()
+                .map(DataGroupProviderConvert.MAPPER::to)
+                .collect(Collectors.toList());
+    }
+
     @Override
     public Integer deleteDataGroupById(Long id) {
         return dataGroupService.deleteDataGroupById(id);