From 704d6c73b456789e697fb26a19a385aad36cc167 Mon Sep 17 00:00:00 2001
From: diantu <diantu188@163.com>
Date: Mon, 6 Feb 2023 17:26:30 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D:=E8=BE=BE=E6=A2=A6=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E5=BA=93=E6=9F=A5=E8=AF=A2=E6=95=B0=E6=8D=AE=E6=97=B6?=
 =?UTF-8?q?,=E5=85=B3=E9=94=AE=E8=AF=8D=E5=AF=BC=E8=87=B4=E7=9A=84?=
 =?UTF-8?q?=E6=8A=A5=E9=94=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../db/service/database/DmServiceImpl.java    | 41 +++++++++++++++++++
 1 file changed, 41 insertions(+)

diff --git a/zyplayer-doc-db/src/main/java/com/zyplayer/doc/db/service/database/DmServiceImpl.java b/zyplayer-doc-db/src/main/java/com/zyplayer/doc/db/service/database/DmServiceImpl.java
index 69844970..2fc2ec4d 100644
--- a/zyplayer-doc-db/src/main/java/com/zyplayer/doc/db/service/database/DmServiceImpl.java
+++ b/zyplayer-doc-db/src/main/java/com/zyplayer/doc/db/service/database/DmServiceImpl.java
@@ -1,8 +1,12 @@
 package com.zyplayer.doc.db.service.database;
 
+import com.zyplayer.doc.db.controller.param.DataViewParam;
 import com.zyplayer.doc.db.framework.db.enums.DatabaseProductEnum;
+import org.apache.commons.lang3.StringUtils;
 import org.springframework.stereotype.Service;
 
+import java.util.Objects;
+
 /**
  * 达梦数据查询服务实现类
  *
@@ -26,4 +30,41 @@ public class DmServiceImpl extends DbBaseService {
     public String getUseDbSql(String dbName) {
         return null;
     }
+
+    /**
+     * 获取分页查询的SQL
+     *
+     * @return 分页查询的SQL
+     * @author diantu
+     * @since 2023年2月6日
+     */
+    @Override
+    public String getQueryPageSql(DataViewParam dataViewParam) {
+        String queryColumns = StringUtils.defaultIfBlank(dataViewParam.getRetainColumn(), "*");
+        if(!Objects.equals(queryColumns, "*")){
+            String[] queryColumnsArray = queryColumns.split(",");
+            String resultString = "";
+            for(int i=0;i<queryColumnsArray.length;i++){
+                if(queryColumnsArray[i].equalsIgnoreCase("IDENTITY")){
+                    queryColumnsArray[i] = "\"IDENTITY\"";
+                }
+                if(i < queryColumnsArray.length-1){
+                    resultString +=queryColumnsArray[i] + ",";
+                }else{
+                    resultString +=queryColumnsArray[i];
+                }
+            }
+            queryColumns = resultString;
+        }
+        StringBuilder sqlSb = new StringBuilder();
+        sqlSb.append(String.format("select %s from %s.%s", queryColumns, dataViewParam.getDbName(), dataViewParam.getTableName()));
+        if (StringUtils.isNotBlank(dataViewParam.getCondition())) {
+            sqlSb.append(String.format(" where %s", dataViewParam.getCondition()));
+        }
+        if (StringUtils.isNotBlank(dataViewParam.getOrderColumn()) && StringUtils.isNotBlank(dataViewParam.getOrderType())) {
+            sqlSb.append(String.format(" order by %s %s", dataViewParam.getOrderColumn(), dataViewParam.getOrderType()));
+        }
+        sqlSb.append(String.format(" limit %s offset %s", dataViewParam.getPageSize(), dataViewParam.getOffset()));
+        return sqlSb.toString();
+    }
 }