From c78bf80885e27aeea81495dce8e326dcf8cb0f87 Mon Sep 17 00:00:00 2001
From: yanta <hai.yang@sayweee.com>
Date: Wed, 17 Jul 2024 17:17:46 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E4=B8=9A=E5=8A=A1=E4=BD=8E?=
 =?UTF-8?q?=E5=B3=B0=E6=97=B6=E9=97=B4=E8=8C=83=E5=9B=B4=EF=BC=8Cddl?=
 =?UTF-8?q?=E5=8F=AA=E8=83=BD=E5=A4=9F=E5=9C=A8=E5=9C=A8=E4=B8=9A=E5=8A=A1?=
 =?UTF-8?q?=E4=BD=8E=E5=B3=B0=E6=89=A7=E8=A1=8C=E9=80=89=E9=A1=B9=EF=BC=8C?=
 =?UTF-8?q?=E9=81=BF=E5=85=8D=E9=9D=9Edba=E6=89=A7=E8=A1=8Csql=E9=94=81?=
 =?UTF-8?q?=E8=A1=A8=E5=BD=B1=E5=93=8D=E4=B8=9A=E5=8A=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 common/templates/config.html | 27 +++++++++++++++++++++++++++
 sql/sql_workflow.py          |  9 +++++++++
 sql/utils/sql_review.py      | 20 ++++++++++++++++++++
 sql_api/api_workflow.py      |  9 ++++++++-
 4 files changed, 64 insertions(+), 1 deletion(-)

diff --git a/common/templates/config.html b/common/templates/config.html
index 93af8eeb1f..1e3d619c0e 100755
--- a/common/templates/config.html
+++ b/common/templates/config.html
@@ -161,6 +161,26 @@ <h4 style="color: darkgrey"><b>功能模块配置</b></h4>
                         <h5 style="color: darkgrey"><b>SQL上线</b></h5>
                         <hr/>
                         <div class="form-horizontal">
+                                                        <div class="form-group">
+                                <label for="query_low_peak_start"
+                                       class="col-sm-4 control-label">QUERY_LOW_PEAK_START</label>
+                                <div class="col-sm-5">
+                                    <input type="number" class="form-control" id="query_low_peak_start"
+                                           key="query_low_peak_start"
+                                           value="{{ config.query_low_peak_start }}"
+                                           placeholder="每天业务低峰开始时间，单位小时，默认为空" />
+                                </div>
+                            </div>
+                            <div class="form-group">
+                                <label for="query_low_peak_end"
+                                       class="col-sm-4 control-label">QUERY_LOW_PEAK_END</label>
+                                <div class="col-sm-5">
+                                    <input type="number" class="form-control" id="query_low_peak_end"
+                                           key="query_low_peak_end"
+                                           value="{{ config.query_low_peak_end }}"
+                                           placeholder="每天业务低峰结束时间，单位小时，默认为空" />
+                                </div>
+                            </div>
                             <div class="form-group">
                                 <label for="critical_ddl_regex"
                                        class="col-sm-4 control-label">CRITICAL_DDL_REGEX</label>
@@ -1319,6 +1339,13 @@ <h5 class="control-label text-bold">当前审批流程：<b id="workflow_auditor
                     return;
                 }
             }
+            // 业务低峰期时间设置，开始时间和结束时间必须同时填写
+            var start = $("#query_low_peak_start").val();
+            var end = $("#query_low_peak_end").val();
+            if ((start === "" && end !== "") || (start !== "" && end === "")) {
+                alert('业务低峰开始时间和结束时间必须同时填写！');
+                return;
+            }
             var sys_config = $("#div-system-config");
             var configs = [];
             sys_config.find('[key]').each(
diff --git a/sql/sql_workflow.py b/sql/sql_workflow.py
index b812be7c51..830c8909f5 100644
--- a/sql/sql_workflow.py
+++ b/sql/sql_workflow.py
@@ -24,6 +24,7 @@
     can_timingtask,
     can_cancel,
     can_execute,
+    on_query_low_peak_time_ddl,
     on_correct_time_period,
     can_view,
     can_rollback,
@@ -306,6 +307,14 @@ def execute(request):
             "errMsg": "不在可执行时间范围内，如果需要修改执行时间请重新提交工单!"
         }
         return render(request, "error.html", context)
+    sys_config = SysConfig()
+    if not request.user.is_superuser and on_query_low_peak_time_ddl(workflow_id) is False:
+        start = sys_config.get("query_low_peak_start", 0)
+        end = sys_config.get("query_low_peak_end", 24)
+        context = {
+            "errMsg": "管理员设置了业务低峰期时间范围:每天%s:00至%s,你只能在业务低峰时间范围执行DDL工单操作!" % (start, end)
+        }
+        return render(request, "error.html", context)
     # 获取审核信息
     audit_id = Audit.detail_by_workflow_id(
         workflow_id=workflow_id, workflow_type=WorkflowType.SQL_REVIEW
diff --git a/sql/utils/sql_review.py b/sql/utils/sql_review.py
index 78296247f6..b44eb3649a 100644
--- a/sql/utils/sql_review.py
+++ b/sql/utils/sql_review.py
@@ -40,6 +40,26 @@ def can_execute(user, workflow_id):
         result = True
     return result
 
+def on_query_low_peak_time_ddl(workflow_id, run_date=None):
+    """
+    判断是否是ddl，ddl必须在业务低峰期执行，包括人工执行和定时执行
+    :param workflow_id:
+    :param run_date:
+    :return:
+    """
+    config = SysConfig()
+    workflow_detail = SqlWorkflow.objects.get(id=workflow_id)
+    start = int(config.get("query_low_peak_start", ""))
+    end = int(config.get("query_low_peak_end", ""))
+    result = True
+    ctime = run_date or datetime.datetime.now()
+    hour = ctime.hour
+    syntax_type = workflow_detail.syntax_type
+    if syntax_type == 1:
+        if (start and hour < start) or (end and hour > end):
+            result = False
+    return result
+
 
 def on_correct_time_period(workflow_id, run_date=None):
     """
diff --git a/sql_api/api_workflow.py b/sql_api/api_workflow.py
index d834a5af30..b2533ff2b0 100644
--- a/sql_api/api_workflow.py
+++ b/sql_api/api_workflow.py
@@ -26,7 +26,7 @@
 from sql.notify import notify_for_audit, notify_for_execute
 from sql.query_privileges import _query_apply_audit_call_back
 from sql.utils.resource_group import user_groups
-from sql.utils.sql_review import can_cancel, can_execute, on_correct_time_period
+from sql.utils.sql_review import can_cancel, can_execute, on_correct_time_period,on_query_low_peak_time_ddl
 from sql.utils.tasks import del_schedule
 from sql.utils.workflow_audit import Audit, get_auditor, AuditException
 from .filters import WorkflowFilter, WorkflowAuditFilter
@@ -340,6 +340,13 @@ def post(self, request):
                         "errors": "不在可执行时间范围内，如果需要修改执行时间请重新提交工单!"
                     }
                 )
+            sys_config = SysConfig()
+            if not request.user.is_superuser and on_query_low_peak_time_ddl(workflow_id) is False:
+                start = sys_config.get("query_low_peak_start", 0)
+                end = sys_config.get("query_low_peak_end", 24)
+                raise serializers.ValidationError(
+                    {"errMsg": "管理员设置了业务低峰期时间范围:每天%s:00至%s,你只能在业务低峰时间范围执行DDL工单操作!" % (start, end)}
+                )
 
             # 获取审核信息
             audit_id = Audit.detail_by_workflow_id(