diff --git a/ee/query-service/app/server.go b/ee/query-service/app/server.go index 9a2c96734f..ac941cec37 100644 --- a/ee/query-service/app/server.go +++ b/ee/query-service/app/server.go @@ -8,7 +8,6 @@ import ( "io" "net" "net/http" - "net/http/httputil" _ "net/http/pprof" // http profiler "os" "regexp" @@ -118,33 +117,13 @@ func NewServer(serverOptions *ServerOptions) (*Server, error) { localDB.SetMaxOpenConns(10) - gatewayFeature := basemodel.Feature{ - Name: "GATEWAY", - Active: false, - Usage: 0, - UsageLimit: -1, - Route: "", - } - - //Activate this feature if the url is not empty - var gatewayProxy *httputil.ReverseProxy - if serverOptions.GatewayUrl == "" { - gatewayFeature.Active = false - gatewayProxy, err = gateway.NewNoopProxy() - if err != nil { - return nil, err - } - } else { - zap.L().Info("Enabling gateway feature flag ...") - gatewayFeature.Active = true - gatewayProxy, err = gateway.NewProxy(serverOptions.GatewayUrl, gateway.RoutePrefix) - if err != nil { - return nil, err - } + gatewayProxy, err := gateway.NewProxy(serverOptions.GatewayUrl, gateway.RoutePrefix) + if err != nil { + return nil, err } // initiate license manager - lm, err := licensepkg.StartManager("sqlite", localDB, gatewayFeature) + lm, err := licensepkg.StartManager("sqlite", localDB) if err != nil { return nil, err } diff --git a/ee/query-service/integrations/gateway/noop.go b/ee/query-service/integrations/gateway/noop.go index bbe930e2f9..ccb5d3269a 100644 --- a/ee/query-service/integrations/gateway/noop.go +++ b/ee/query-service/integrations/gateway/noop.go @@ -5,5 +5,5 @@ import ( ) func NewNoopProxy() (*httputil.ReverseProxy, error) { - return nil, nil + return &httputil.ReverseProxy{}, nil } diff --git a/ee/query-service/model/plans.go b/ee/query-service/model/plans.go index 41bca047d5..135d276cad 100644 --- a/ee/query-service/model/plans.go +++ b/ee/query-service/model/plans.go @@ -11,6 +11,7 @@ const Enterprise = "ENTERPRISE_PLAN" const DisableUpsell = "DISABLE_UPSELL" const Onboarding = "ONBOARDING" const ChatSupport = "CHAT_SUPPORT" +const Gateway = "GATEWAY" var BasicPlan = basemodel.FeatureSet{ basemodel.Feature{ @@ -111,6 +112,13 @@ var BasicPlan = basemodel.FeatureSet{ UsageLimit: -1, Route: "", }, + basemodel.Feature{ + Name: Gateway, + Active: false, + Usage: 0, + UsageLimit: -1, + Route: "", + }, } var ProPlan = basemodel.FeatureSet{ @@ -205,6 +213,13 @@ var ProPlan = basemodel.FeatureSet{ UsageLimit: -1, Route: "", }, + basemodel.Feature{ + Name: Gateway, + Active: true, + Usage: 0, + UsageLimit: -1, + Route: "", + }, } var EnterprisePlan = basemodel.FeatureSet{ @@ -313,4 +328,11 @@ var EnterprisePlan = basemodel.FeatureSet{ UsageLimit: -1, Route: "", }, + basemodel.Feature{ + Name: Gateway, + Active: true, + Usage: 0, + UsageLimit: -1, + Route: "", + }, }