Merge pull request #2492 from actiontech/fix_audit_plan_ce

Fix audit plan ce

Author: sjjian

sqle/api/controller/v1/instance_audit_plan.go

@@ -91,6 +91,14 @@ func CreateInstanceAuditPlan(c echo.Context) error {
 		}
 		instanceID = inst.ID
 		instanceType = inst.DbType
+		// check instance audit plan exist
+		_, exist, err = model.GetStorage().GetInstanceAuditPlanByInstanceID(int64(inst.ID))
+		if err != nil {
+			return controller.JSONBaseErrorReq(c, err)
+		}
+		if exist {
+			return controller.JSONBaseErrorReq(c, fmt.Errorf("current instance has audit plan"))
+		}
 		// check operation
 		user, err := controller.GetCurrentUser(c, dms.GetUser)
 		if err != nil {
@@ -103,6 +111,7 @@ func CreateInstanceAuditPlan(c echo.Context) error {
 		if !canCreateAuditPlan {
 			return controller.JSONBaseErrorReq(c, errors.NewUserNotPermissionError(model.GetOperationCodeDesc(uint(model.OP_AUDIT_PLAN_SAVE))))
 		}
+
 	} else {
 		instanceType = req.InstanceType
 	}

sqle/server/auditplan/task_type_mysql_schema_meta.go

@@ -84,7 +84,7 @@ func (at *BaseSchemaMetaTaskV2) extractSQL(logger *logrus.Entry, ap *AuditPlan,
 			return nil, fmt.Errorf("use schema fail, error: %v", err)
 		}
 		for _, table := range tables {
-			sql, err := db.ShowCreateTable(utils.SupplementalQuotationMarks(ap.InstanceDatabase), utils.SupplementalQuotationMarks(table))
+			sql, err := db.ShowCreateTable(utils.SupplementalQuotationMarks(schema), utils.SupplementalQuotationMarks(table))
 			if err != nil {
 				return nil, fmt.Errorf("show create table fail, error: %v", err)
 			}