Add feature flag for graph exploration.

Author: apognu

api/routes.go

@@ -118,11 +118,13 @@ func addRoutes(r *gin.Engine, conf Configuration, uc usecases.Usecases, auth uti
 	router.POST("/ingestion/:object_type/batch", timeoutMiddleware(conf.BatchTimeout), handlePostCsvIngestion(uc))
 	router.GET("/ingestion/:object_type/upload-logs", tom, handleListUploadLogs(uc))
 
-	router.GET("/graph/relations", tom, handleListGraphRelations(uc))
-	router.POST("/graph/relations", tom, handleCreateGraphRelation(uc))
-	router.DELETE("/graph/relations/:relation_id", tom, handleDeleteGraphRelation(uc))
+	infra.RouteWithFeatureFlag(router, infra.GRAPH_EXPLORATION_FEATURE_FLAG, func(sub gin.IRoutes) {
+		sub.GET("/graph/relations", tom, handleListGraphRelations(uc))
+		sub.POST("/graph/relations", tom, handleCreateGraphRelation(uc))
+		sub.DELETE("/graph/relations/:relation_id", tom, handleDeleteGraphRelation(uc))
 
-	router.GET("/graph/:node_type/:node_id", tom, handleGraphWalk(uc))
+		sub.GET("/graph/:node_type/:node_id", tom, handleGraphWalk(uc))
+	})
 
 	router.GET("/client_data/:object_type/:object_id", tom, handleGetIngestedObject(uc))
 	router.GET("/client_data/:object_type/:object_id/annotations", tom, handleListEntityAnnotations(uc))

infra/feature_flags.go

@@ -16,6 +16,7 @@ type featureFlag string
 const (
 	TEST_UNUSED_FEATURE_FLAG        featureFlag = "TEST_UNUSED_FEATURE_FLAG"
 	BATCH_EXECUTION_V2_FEATURE_FLAG featureFlag = "BATCH_EXECUTION_V2"
+	GRAPH_EXPLORATION_FEATURE_FLAG  featureFlag = "GRAPH_EXPLORATION"
 )
 
 func HasGlobalFeatureFlag(flag featureFlag) bool {

usecases/worker_jobs/graph_build_job.go

@@ -11,6 +11,7 @@ import (
 	"github.com/google/uuid"
 	"github.com/riverqueue/river"
 
+	"github.com/checkmarble/marble-backend/infra"
 	"github.com/checkmarble/marble-backend/models"
 	"github.com/checkmarble/marble-backend/repositories"
 	"github.com/checkmarble/marble-backend/usecases/executor_factory"
@@ -180,6 +181,10 @@ func (w *GraphBuildWorker) Timeout(job *river.Job[models.GraphBuildArgs]) time.D
 func (w *GraphBuildWorker) Work(ctx context.Context, job *river.Job[models.GraphBuildArgs]) error {
 	logger := utils.LoggerFromContext(ctx)
 
+	if !infra.HasFeatureFlag(infra.GRAPH_EXPLORATION_FEATURE_FLAG, job.Args.OrgId) {
+		return nil
+	}
+
 	// Prevent herd effect
 	if err := AddStrideDelay(job, w.interval); err != nil {
 		return err