Merge pull request #2770 from actiontech/audit_plan_add_model_for_del…

…_project

chore: add audit plan model for delete project

sqle/api/controller/v1/instance_audit_plan.go

@@ -109,8 +109,13 @@ func CreateInstanceAuditPlan(c echo.Context) error {
 		return controller.JSONBaseErrorReq(c, err)
 	}
 
+	instID, err := strconv.Atoi(req.InstanceId)
+	if err != nil {
+		return controller.JSONBaseErrorReq(c, err)
+	}
+
 	// check instance
-	inst, exist, err := dms.GetInstancesById(c.Request().Context(), req.InstanceId)
+	inst, exist, err := dms.GetInstanceInProjectById(c.Request().Context(), projectUid, uint64(instID))
 	if !exist {
 		return controller.JSONBaseErrorReq(c, ErrInstanceNotExist)
 	} else if err != nil {

sqle/model/instance_audit_plan.go

@@ -531,3 +531,9 @@ func (s *Storage) UpdateAuditPlanLastCollectionTime(auditPlanID uint, collection
 	}
 	return nil
 }
+
+func (s *Storage) GetAuditPlansByProjectId(projectID string) ([]*InstanceAuditPlan, error) {
+	instanceAuditPlan := []*InstanceAuditPlan{}
+	err := s.db.Model(InstanceAuditPlan{}).Where("project_id = ?", projectID).Find(&instanceAuditPlan).Error
+	return instanceAuditPlan, err
+}