Skip to content

Commit

Permalink
Move common PromQL engine configuration to its own package
Browse files Browse the repository at this point in the history
  • Loading branch information
charleskorn committed Mar 25, 2024
1 parent 333c8df commit e3933a2
Show file tree
Hide file tree
Showing 6 changed files with 9 additions and 9 deletions.
2 changes: 1 addition & 1 deletion Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -329,7 +329,7 @@ lint: check-makefiles
faillint -paths "github.com/grafana/mimir/pkg/..." ./pkg/alertmanager/alertspb/...
faillint -paths "github.com/grafana/mimir/pkg/..." ./pkg/ruler/rulespb/...
faillint -paths "github.com/grafana/mimir/pkg/..." ./pkg/storage/sharding/...
faillint -paths "github.com/grafana/mimir/pkg/..." ./pkg/querier/engine/...
faillint -paths "github.com/grafana/mimir/pkg/..." ./pkg/querier/engine/common/...
faillint -paths "github.com/grafana/mimir/pkg/..." ./pkg/querier/api/...
faillint -paths "github.com/grafana/mimir/pkg/..." ./pkg/util/globalerror

Expand Down
4 changes: 2 additions & 2 deletions pkg/mimir/modules.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ import (
"context"
"flag"
"fmt"
"github.com/grafana/mimir/pkg/querier/engine/common"
"net/http"
"strconv"
"time"
Expand Down Expand Up @@ -47,7 +48,6 @@ import (
"github.com/grafana/mimir/pkg/frontend/transport"
"github.com/grafana/mimir/pkg/ingester"
"github.com/grafana/mimir/pkg/querier"
"github.com/grafana/mimir/pkg/querier/engine"
"github.com/grafana/mimir/pkg/querier/tenantfederation"
querier_worker "github.com/grafana/mimir/pkg/querier/worker"
"github.com/grafana/mimir/pkg/ruler"
Expand Down Expand Up @@ -712,7 +712,7 @@ func (t *Mimir) initQueryFrontendTripperware() (serv services.Service, err error
t.QueryFrontendCodec = querymiddleware.NewPrometheusCodec(t.Registerer, t.Cfg.Frontend.QueryMiddleware.QueryResultResponseFormat)
promqlEngineRegisterer := prometheus.WrapRegistererWith(prometheus.Labels{"engine": "query-frontend"}, t.Registerer)

engineOpts, engineExperimentalFunctionsEnabled := engine.NewPromQLEngineOptions(t.Cfg.Querier.EngineConfig, t.ActivityTracker, util_log.Logger, promqlEngineRegisterer)
engineOpts, engineExperimentalFunctionsEnabled := common.NewPromQLEngineOptions(t.Cfg.Querier.EngineConfig, t.ActivityTracker, util_log.Logger, promqlEngineRegisterer)

tripperware, err := querymiddleware.NewTripperware(
t.Cfg.Frontend.QueryMiddleware,
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
// SPDX-License-Identifier: AGPL-3.0-only

package engine
package common

import (
"flag"
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
// SPDX-License-Identifier: AGPL-3.0-only

package engine
package common

import (
"bytes"
Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
// SPDX-License-Identifier: AGPL-3.0-only

package engine
package common

import (
"context"
Expand Down
6 changes: 3 additions & 3 deletions pkg/querier/querier.go
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,7 @@ import (
"errors"
"flag"
"fmt"
"github.com/grafana/mimir/pkg/querier/engine/common"
"sync"
"time"

Expand All @@ -25,7 +26,6 @@ import (
"github.com/prometheus/prometheus/util/annotations"
"golang.org/x/sync/errgroup"

"github.com/grafana/mimir/pkg/querier/engine"
"github.com/grafana/mimir/pkg/querier/engine/streaming"
"github.com/grafana/mimir/pkg/querier/stats"
"github.com/grafana/mimir/pkg/storage/chunk"
Expand Down Expand Up @@ -59,7 +59,7 @@ type Config struct {
PromQLEngine string `yaml:"promql_engine" category:"experimental"`

// PromQL engine config.
EngineConfig engine.Config `yaml:",inline"`
EngineConfig common.Config `yaml:",inline"`
}

const (
Expand Down Expand Up @@ -150,7 +150,7 @@ func New(cfg Config, limits *validation.Overrides, distributor Distributor, stor
return lazyquery.NewLazyQuerier(querier), nil
})

opts, engineExperimentalFunctionsEnabled := engine.NewPromQLEngineOptions(cfg.EngineConfig, tracker, logger, reg)
opts, engineExperimentalFunctionsEnabled := common.NewPromQLEngineOptions(cfg.EngineConfig, tracker, logger, reg)

// Experimental functions can only be enabled globally, and not on a per-engine basis.
parser.EnableExperimentalFunctions = engineExperimentalFunctionsEnabled
Expand Down

0 comments on commit e3933a2

Please sign in to comment.