From 9feee6ff46daf602d6ce3917aefc12bb6efb42a6 Mon Sep 17 00:00:00 2001
From: Srikanth Chekuri <srikanth.chekuri92@gmail.com>
Date: Fri, 3 Jan 2025 09:36:52 +0530
Subject: [PATCH] chore: add option skip web (#6736)

---
 ee/query-service/app/server.go | 9 ++++++---
 ee/query-service/main.go       | 6 ++++--
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/ee/query-service/app/server.go b/ee/query-service/app/server.go
index 5f36aaaf66..afd9dad4c5 100644
--- a/ee/query-service/app/server.go
+++ b/ee/query-service/app/server.go
@@ -79,6 +79,7 @@ type ServerOptions struct {
 	GatewayUrl        string
 	UseLogsNewSchema  bool
 	UseTraceNewSchema bool
+	SkipWebFrontend   bool
 }
 
 // Server runs HTTP api service
@@ -383,9 +384,11 @@ func (s *Server) createPublicServer(apiHandler *api.APIHandler, web *web.Web) (*
 
 	handler = handlers.CompressHandler(handler)
 
-	err := web.AddToRouter(r)
-	if err != nil {
-		return nil, err
+	if !s.serverOptions.SkipWebFrontend {
+		err := web.AddToRouter(r)
+		if err != nil {
+			return nil, err
+		}
 	}
 
 	return &http.Server{
diff --git a/ee/query-service/main.go b/ee/query-service/main.go
index 5fbbca1e18..3514376213 100644
--- a/ee/query-service/main.go
+++ b/ee/query-service/main.go
@@ -108,6 +108,7 @@ func main() {
 	var dialTimeout time.Duration
 	var gatewayUrl string
 	var useLicensesV3 bool
+	var skipWebFrontend bool
 
 	flag.BoolVar(&useLogsNewSchema, "use-logs-new-schema", false, "use logs_v2 schema for logs")
 	flag.BoolVar(&useTraceNewSchema, "use-trace-new-schema", false, "use new schema for traces")
@@ -125,7 +126,7 @@ func main() {
 	flag.StringVar(&cluster, "cluster", "cluster", "(cluster name - defaults to 'cluster')")
 	flag.StringVar(&gatewayUrl, "gateway-url", "", "(url to the gateway)")
 	flag.BoolVar(&useLicensesV3, "use-licenses-v3", false, "use licenses_v3 schema for licenses")
-
+	flag.BoolVar(&skipWebFrontend, "skip-web-frontend", false, "skip web frontend")
 	flag.Parse()
 
 	loggerMgr := initZapLog(enableQueryServiceLogOTLPExport)
@@ -148,7 +149,7 @@ func main() {
 	}
 
 	web, err := signozweb.New(zap.L(), config.Web)
-	if err != nil {
+	if err != nil && !skipWebFrontend {
 		zap.L().Fatal("Failed to create web", zap.Error(err))
 	}
 
@@ -169,6 +170,7 @@ func main() {
 		GatewayUrl:        gatewayUrl,
 		UseLogsNewSchema:  useLogsNewSchema,
 		UseTraceNewSchema: useTraceNewSchema,
+		SkipWebFrontend:   skipWebFrontend,
 	}
 
 	// Read the jwt secret key