添加处理批量工单功能

common/templates/config.html

@@ -183,6 +183,28 @@ <h5 style="color: darkgrey"><b>SQL上线</b></h5>
                                            placeholder="自动驳回的等级，1表示警告驳回，2和空表示错误才驳回，其他表示不驳回" />
                                 </div>
                             </div>
+			    <div class="form-group">
+                                <label for="batch_passed_remark"
+                                       class="col-sm-4 control-label">BATCH_PASSED_REMARK</label>
+                                <div class="col-sm-5">
+                                    <input type="text" class="form-control"
+                                           id="batch_passed_remark"
+                                           key="batch_passed_remark"
+                                           value="{{ config.batch_passed_remark }}"
+                                           placeholder="批量审核通过原因" />
+                                </div>
+                            </div>
+                            <div class="form-group">
+                                <label for="batch_cancel_remark"
+                                       class="col-sm-4 control-label">BATCH_CANCEL_REMARK</label>
+                                <div class="col-sm-5">
+                                    <input type="text" class="form-control"
+                                           id="batch_cancel_remark"
+                                           key="batch_cancel_remark"
+                                           value="{{ config.batch_cancel_remark }}"
+                                           placeholder="批量审核终止原因" />
+                                </div>
+                            </div>
                             <div class="form-group">
                                 <label for="enable_backup_switch"
                                        class="col-sm-4 control-label">ENABLE_BACKUP_SWITCH</label>

sql/models.py

@@ -938,9 +938,12 @@ class Meta:
             ("menu_document", "菜单 相关文档"),
             ("menu_openapi", "菜单 OpenAPI"),
             ("sql_submit", "提交SQL上线工单"),
+	    ('sql_submitbatch', '提交批量SQL上线工单'),
             ("sql_review", "审核SQL上线工单"),
+	    ('sql_reviewbatch', '审核批量SQL上线工单'),
             ("sql_execute_for_resource_group", "执行SQL上线工单(资源组粒度)"),
             ("sql_execute", "执行SQL上线工单(仅自己提交的)"),
+	    ('sql_executebatch', '执行批量SQL上线工单'),
             ("sql_analyze", "执行SQL分析"),
             ("optimize_sqladvisor", "执行SQLAdvisor"),
             ("optimize_sqltuning", "执行SQLTuning"),

sql/resource_group.py

@@ -148,7 +148,7 @@ def instances(request):
     ins = (
         ins.filter(**filter_dict)
         .order_by(Convert("instance_name", "gbk").asc())
-        .values("id", "type", "db_type", "instance_name")
+        .values("id", "type", "db_type", "instance_name", "db_name")
     )
     rows = [row for row in ins]
     result = {"status": 0, "msg": "ok", "data": rows}

sql/sql_workflow.py

@@ -525,3 +525,128 @@ def osc_control(request):
         json.dumps(result, cls=ExtendJSONEncoder, bigint_as_string=True),
         content_type="application/json",
     )
+
+@permission_required('sql.sql_reviewbatch', raise_exception=True)
+def passedbatch(request):
+    """
+    批量审核通过，不执行
+    :param request:
+    :return:
+    """
+    workflow_id_list = request.POST['workflowid_array']
+    workflow_id_list = json.loads(workflow_id_list)
+    sys_config = SysConfig()
+    audit_remark = sys_config.get('batch_passed_remark')
+
+    for workflow_id in workflow_id_list:
+        passed(request, workflow_id, audit_remark)
+
+    return HttpResponseRedirect('/sqlworkflow')
+
+@permission_required('sql.sql_executebatch', raise_exception=True)
+def executebatch(request):
+    """
+    执行SQL
+    :param request:
+    :return:
+    """
+    workflow_id_list = request.POST['workflowid_array']
+    workflow_id_list = json.loads(workflow_id_list)
+    op_mode = 'auto'
+
+    for workflow_id in workflow_id_list:
+        execute(request, workflow_id, op_mode)
+    return HttpResponseRedirect('/sqlworkflow')
+
+@permission_required('sql.sql_reviewbatch', raise_exception=True)
+def cancelbatch(request):
+    """
+    终止流程
+    :param request:
+    :return:
+    """
+    workflow_id_list = request.POST['workflowid_array']
+    workflow_id_list = json.loads(workflow_id_list)
+    sys_config = SysConfig()
+    audit_remark = sys_config.get('batch_cancel_remark')
+
+    for workflow_id in workflow_id_list:
+        cancel(request,workflow_id,audit_remark)
+
+    return HttpResponseRedirect('/sqlworkflow')
+
+@permission_required('sql.sql_submitbatch', raise_exception=True)
+def checkbatch(request):
+    """SQL检测按钮, 此处没有产生工单"""
+    sql_content = request.POST.get('sql_content')
+    instance_names = request.POST.get('instance_name')
+    instance_names= json.loads(instance_names)
+    instances=[]
+    for instance_name in instance_names:
+        instance = Instance.objects.get(instance_name=instance_name)
+        instances.append(instance)
+
+    result = {'status': 0, 'msg': 'ok', 'data': {}}
+    # 服务器端参数验证
+    if sql_content is None or len(instance_names) ==0 :
+        result['status'] = 1
+        result['msg'] = '页面提交参数可能为空'
+        return HttpResponse(json.dumps(result), content_type='application/json')
+    warning_count_totle=0
+    error_count_totle=0
+    # 交给engine进行检测
+    check_result_arr=[]
+    for instance in instances:
+        try:
+            check_engine = get_engine(instance=instance)
+            check_result = check_engine.execute_check(db_name=instance.db_name, sql=sql_content.strip())
+            for i in range(len(check_result.to_dict())):
+                check_result.to_dict()[i]['instance'] = instance.instance_name
+            warning_count_totle= warning_count_totle+check_result.warning_count
+            error_count_totle=error_count_totle+check_result.error_count
+            check_result_arr=check_result_arr+check_result.to_dict()
+        except Exception as e:
+            result['status'] = 1
+            result['msg'] = str(e)
+            return HttpResponse(json.dumps(result), content_type='application/json')
+
+    # 处理检测结果
+    result['data']['rows'] =  check_result_arr
+    result['data']['CheckWarningCount'] = warning_count_totle
+    result['data']['CheckErrorCount'] = error_count_totle
+    return HttpResponse(json.dumps(result), content_type='application/json')
+
+
+@permission_required('sql.sql_submitbatch', raise_exception=True)
+def rollbackbatch(request):
+    """
+    回滚流程
+    :param request:
+    :return:
+    """
+    workflow_id_list = request.POST['workflowid_array']
+    workflow_id_list = json.loads(workflow_id_list)
+    workflow_list = list()
+    for workflow_id in workflow_id_list:
+        workflow = SqlWorkflow.objects.get(id=int(workflow_id))
+        rollback_workflow_name = f"【回滚工单】原工单Id:{workflow_id} ,{workflow.workflow_name}"
+        query_engine = get_engine(instance=workflow.instance)
+        list_backup_sql = query_engine.get_rollback(workflow=workflow)
+        ## 获取完整的回滚SQL
+        sql_content = '\n'.join(item[1] for item in list_backup_sql)
+        param = {
+            'workflow': {
+                'workflow_name': rollback_workflow_name,
+                'daemon_url': '',
+                'group_id': ResourceGroup.objects.get(group_name=workflow.group_name).group_id,
+                'instance': workflow.instance.id,
+                'db_name': workflow.db_name,
+                'is_backup': True,
+                'run_date_start': None,
+                'run_date_end': None,
+             },
+            'sql_content': sql_content
+        }
+        workflow_list.append(param)
+    result = {"status": 0, "msg": "ok", "data": workflow_list}
+    return HttpResponse(json.dumps(result), content_type="application/json")