[INLONG-5285][Manager] Support custom query for group, stream, source and sink

ciscozhou · ciscozhou · commit 91250ad49625 · 2022-07-30T15:51:00.000+08:00
diff --git a/inlong-manager/manager-common/src/main/java/org/apache/inlong/manager/common/beans/PageRequest.java b/inlong-manager/manager-common/src/main/java/org/apache/inlong/manager/common/beans/PageRequest.java
@@ -32,6 +32,28 @@ public class PageRequest {
     @ApiModelProperty(value = "page size, default 10", required = true, example = "10")
     private int pageSize = 10;
 
+    @ApiModelProperty(value = "order field, default create_time", example = "create_time")
+    private String orderField = "create_time";
+
+    @ApiModelProperty(value = "order type, default desc", example = "desc")
+    private String orderType = "desc";
+
+    public String getOrderField() {
+        return orderField;
+    }
+
+    public void setOrderField(String orderField) {
+        this.orderField = orderField;
+    }
+
+    public String getOrderType() {
+        return orderType;
+    }
+
+    public void setOrderType(String orderType) {
+        this.orderType = orderType;
+    }
+
     public int getPageNum() {
         return pageNum;
     }
diff --git a/inlong-manager/manager-common/src/main/java/org/apache/inlong/manager/common/enums/OrderFieldEnum.java b/inlong-manager/manager-common/src/main/java/org/apache/inlong/manager/common/enums/OrderFieldEnum.java
@@ -0,0 +1,41 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.inlong.manager.common.enums;
+
+import org.apache.inlong.manager.common.beans.PageRequest;
+
+/**
+ * The order field enumeration.
+ */
+public enum OrderFieldEnum {
+
+    CREATE_TIME,
+
+    MODIFY_TIME;
+
+    public static void checkOrderField(PageRequest pageRequest) {
+        for (OrderFieldEnum value : values()) {
+            if (value.name().equalsIgnoreCase(pageRequest.getOrderField())) {
+                pageRequest.setOrderField(value.name().toLowerCase());
+                return;
+            }
+        }
+        pageRequest.setOrderField(CREATE_TIME.name().toLowerCase());
+    }
+
+}
diff --git a/inlong-manager/manager-dao/src/main/resources/mappers/InlongGroupEntityMapper.xml b/inlong-manager/manager-dao/src/main/resources/mappers/InlongGroupEntityMapper.xml
@@ -154,7 +154,17 @@
                 </foreach>
             </if>
         </where>
-        order by modify_time desc
+        <choose>
+            <when test="orderField != null and orderField != '' and orderType = 'desc'">
+                order by #{orderField} desc
+            </when>
+            <when test="orderField != null and orderField != '' and orderType = 'asc'">
+                order by #{orderField} asc
+            </when>
+            <otherwise>
+                order by create_time desc
+            </otherwise>
+        </choose>
     </select>
     <select id="selectBriefList" parameterType="org.apache.inlong.manager.common.pojo.group.InlongGroupPageRequest"
             resultType="org.apache.inlong.manager.common.pojo.group.InlongGroupBriefInfo">
diff --git a/inlong-manager/manager-dao/src/main/resources/mappers/InlongStreamEntityMapper.xml b/inlong-manager/manager-dao/src/main/resources/mappers/InlongStreamEntityMapper.xml
@@ -262,7 +262,17 @@
                 and stream.status = #{request.status, jdbcType=INTEGER}
             </if>
         </where>
-        order by stream.modify_time desc
+        <choose>
+            <when test="request.orderField != null and request.orderField != '' and request.orderType = 'desc'">
+                order by stream.#{request.orderField} desc
+            </when>
+            <when test="request.orderField != null and request.orderField != '' and request.orderType = 'asc'">
+                order by stream.#{request.orderField} asc
+            </when>
+            <otherwise>
+                order by stream.create_time desc
+            </otherwise>
+        </choose>
     </select>
     <select id="selectBriefList" resultType="org.apache.inlong.manager.common.pojo.stream.InlongStreamBriefInfo">
         select id, inlong_group_id, inlong_stream_id, name, mq_resource, modify_time
diff --git a/inlong-manager/manager-dao/src/main/resources/mappers/StreamSinkEntityMapper.xml b/inlong-manager/manager-dao/src/main/resources/mappers/StreamSinkEntityMapper.xml
@@ -107,8 +107,18 @@
             <if test="request.sortConsumerGroup != null and request.sortConsumerGroup != ''">
                 and sort_consumer_group = #{request.sortConsumerGroup, jdbcType=VARCHAR}
             </if>
-            order by modify_time desc
         </where>
+        <choose>
+            <when test="orderField != null and orderField != '' and orderType = 'desc'">
+                order by #{request.orderField} desc
+            </when>
+            <when test="orderField != null and orderField != '' and orderType = 'asc'">
+                order by #{request.orderField} asc
+            </when>
+            <otherwise>
+                order by create_time desc
+            </otherwise>
+        </choose>
     </select>
     <select id="selectSummary"
             resultType="org.apache.inlong.manager.common.pojo.sink.SinkBriefInfo">
diff --git a/inlong-manager/manager-dao/src/main/resources/mappers/StreamSourceEntityMapper.xml b/inlong-manager/manager-dao/src/main/resources/mappers/StreamSourceEntityMapper.xml
@@ -125,8 +125,18 @@
             <if test="request.status != null and request.status != ''">
                 and status = #{request.status, jdbcType=INTEGER}
             </if>
-            order by modify_time desc
         </where>
+        <choose>
+            <when test="orderField != null and orderField != '' and orderType = 'desc'">
+                order by #{request.orderField} desc
+            </when>
+            <when test="orderField != null and orderField != '' and orderType = 'asc'">
+                order by #{request.orderField} asc
+            </when>
+            <otherwise>
+                order by create_time desc
+            </otherwise>
+        </choose>
     </select>
     <select id="selectByAgentIp" resultType="org.apache.inlong.manager.dao.entity.StreamSourceEntity">
         select
diff --git a/inlong-manager/manager-service/src/main/java/org/apache/inlong/manager/service/core/impl/InlongStreamServiceImpl.java b/inlong-manager/manager-service/src/main/java/org/apache/inlong/manager/service/core/impl/InlongStreamServiceImpl.java
@@ -25,6 +25,7 @@
 import org.apache.inlong.manager.common.consts.InlongConstants;
 import org.apache.inlong.manager.common.enums.ErrorCodeEnum;
 import org.apache.inlong.manager.common.enums.GroupStatus;
+import org.apache.inlong.manager.common.enums.OrderFieldEnum;
 import org.apache.inlong.manager.common.enums.StreamStatus;
 import org.apache.inlong.manager.common.exceptions.BusinessException;
 import org.apache.inlong.manager.common.pojo.sink.SinkBriefInfo;
@@ -202,6 +203,7 @@ public PageInfo<InlongStreamBriefInfo> listBrief(InlongStreamPageRequest request
         LOGGER.debug("begin to list inlong stream page by {}", request);
 
         PageHelper.startPage(request.getPageNum(), request.getPageSize());
+        OrderFieldEnum.checkOrderField(request);
         Page<InlongStreamEntity> entityPage = (Page<InlongStreamEntity>) streamMapper.selectByCondition(request);
         List<InlongStreamBriefInfo> streamList = CommonBeanUtils.copyListProperties(entityPage,
                 InlongStreamBriefInfo::new);
diff --git a/inlong-manager/manager-service/src/main/java/org/apache/inlong/manager/service/group/InlongGroupServiceImpl.java b/inlong-manager/manager-service/src/main/java/org/apache/inlong/manager/service/group/InlongGroupServiceImpl.java
@@ -27,6 +27,7 @@
 import org.apache.inlong.manager.common.consts.InlongConstants;
 import org.apache.inlong.manager.common.enums.ErrorCodeEnum;
 import org.apache.inlong.manager.common.enums.GroupStatus;
+import org.apache.inlong.manager.common.enums.OrderFieldEnum;
 import org.apache.inlong.manager.common.enums.SourceType;
 import org.apache.inlong.manager.common.exceptions.BusinessException;
 import org.apache.inlong.manager.common.exceptions.WorkflowListenerException;
@@ -176,6 +177,7 @@ public PageInfo<InlongGroupBriefInfo> listBrief(InlongGroupPageRequest request)
             request.setPageSize(MAX_PAGE_SIZE);
         }
         PageHelper.startPage(request.getPageNum(), request.getPageSize());
+        OrderFieldEnum.checkOrderField(request);
         Page<InlongGroupEntity> entityPage = (Page<InlongGroupEntity>) groupMapper.selectByCondition(request);
 
         List<InlongGroupBriefInfo> briefInfos = CommonBeanUtils.copyListProperties(entityPage,
diff --git a/inlong-manager/manager-service/src/main/java/org/apache/inlong/manager/service/sink/StreamSinkServiceImpl.java b/inlong-manager/manager-service/src/main/java/org/apache/inlong/manager/service/sink/StreamSinkServiceImpl.java
@@ -27,6 +27,7 @@
 import org.apache.inlong.manager.common.consts.InlongConstants;
 import org.apache.inlong.manager.common.enums.ErrorCodeEnum;
 import org.apache.inlong.manager.common.enums.GroupStatus;
+import org.apache.inlong.manager.common.enums.OrderFieldEnum;
 import org.apache.inlong.manager.common.enums.SinkStatus;
 import org.apache.inlong.manager.common.enums.SinkType;
 import org.apache.inlong.manager.common.exceptions.BusinessException;
@@ -183,6 +184,7 @@ public PageInfo<? extends StreamSink> listByCondition(SinkPageRequest request) {
         Preconditions.checkNotNull(request.getInlongGroupId(), ErrorCodeEnum.GROUP_ID_IS_EMPTY.getMessage());
 
         PageHelper.startPage(request.getPageNum(), request.getPageSize());
+        OrderFieldEnum.checkOrderField(request);
         List<StreamSinkEntity> entityPage = sinkMapper.selectByCondition(request);
         Map<SinkType, Page<StreamSinkEntity>> sinkMap = Maps.newHashMap();
         for (StreamSinkEntity streamSink : entityPage) {
diff --git a/inlong-manager/manager-service/src/main/java/org/apache/inlong/manager/service/source/StreamSourceServiceImpl.java b/inlong-manager/manager-service/src/main/java/org/apache/inlong/manager/service/source/StreamSourceServiceImpl.java
@@ -27,6 +27,7 @@
 import org.apache.inlong.manager.common.consts.InlongConstants;
 import org.apache.inlong.manager.common.enums.ErrorCodeEnum;
 import org.apache.inlong.manager.common.enums.GroupStatus;
+import org.apache.inlong.manager.common.enums.OrderFieldEnum;
 import org.apache.inlong.manager.common.enums.SourceStatus;
 import org.apache.inlong.manager.common.enums.SourceType;
 import org.apache.inlong.manager.common.exceptions.BusinessException;
@@ -171,6 +172,7 @@ public PageInfo<? extends StreamSource> listByCondition(SourcePageRequest reques
         Preconditions.checkNotNull(request.getInlongGroupId(), ErrorCodeEnum.GROUP_ID_IS_EMPTY.getMessage());
 
         PageHelper.startPage(request.getPageNum(), request.getPageSize());
+        OrderFieldEnum.checkOrderField(request);
         List<StreamSourceEntity> entityList = sourceMapper.selectByCondition(request);
 
         // Encapsulate the paging query results into the PageInfo object to obtain related paging information
