Enable cron (#4239)

* Added API for enable and disable cron Signed-off-by: Saranya-jena <[email protected]> * Added backend implementation to toggle enable/disable cron experiment Signed-off-by: Saranya-jena <[email protected]> * fixed imports Signed-off-by: Saranya-jena <[email protected]> * removed code for disable cron from stop experiment Signed-off-by: Saranya-jena <[email protected]> --------- Signed-off-by: Saranya-jena <[email protected]>
litmuschaos · Nov 7, 2023 · 8cf7902 · 8cf7902
1 parent 997d1ab
commit 8cf7902
Show file tree

Hide file tree

Showing 9 changed files with 360 additions and 48 deletions.
diff --git a/chaoscenter/graphql/definitions/shared/chaos_experiment.graphqls b/chaoscenter/graphql/definitions/shared/chaos_experiment.graphqls
@@ -768,4 +768,12 @@ extend type Mutation {
     experimentRunID: String
     projectID: ID!
   ): Boolean!
+
+  """
+  Enable/Disable cron experiment state
+  """
+  updateCronExperimentState(experimentID: String!
+    disable: Boolean!
+    projectID: ID!
+  ): Boolean! @authorized
 }
diff --git a/chaoscenter/graphql/server/graph/chaos_experiment.resolvers.go b/chaoscenter/graphql/server/graph/chaos_experiment.resolvers.go
@@ -135,6 +135,29 @@ func (r *mutationResolver) DeleteChaosExperiment(ctx context.Context, experiment
 	return uiResponse, err
 }
 
+func (r *mutationResolver) UpdateCronExperimentState(ctx context.Context, experimentID string, disable bool, projectID string) (bool, error) {
+	logFields := logrus.Fields{
+		"projectId":         projectID,
+		"chaosExperimentId": experimentID,
+	}
+
+	logrus.WithFields(logFields).Info("request received to update cron chaos experiment")
+
+	err := authorization.ValidateRole(ctx, projectID,
+		authorization.MutationRbacRules[authorization.UpdateChaosWorkflow],
+		model.InvitationAccepted.String())
+	if err != nil {
+		return false, err
+	}
+
+	uiResponse, err := r.chaosExperimentHandler.UpdateCronExperimentState(ctx, experimentID, disable, projectID, data_store.Store)
+	if err != nil {
+		logrus.WithFields(logFields).Error(err)
+		return false, err
+	}
+	return uiResponse, err
+}
+
 func (r *queryResolver) GetExperiment(ctx context.Context, projectID string, experimentID string) (*model.GetExperimentResponse, error) {
 	logFields := logrus.Fields{
 		"projectId":         projectID,

diff --git a/chaoscenter/graphql/server/graph/generated/generated.go b/chaoscenter/graphql/server/graph/generated/generated.go