diff --git a/ee/query-service/main.go b/ee/query-service/main.go index be651dfc63..edda0b171a 100644 --- a/ee/query-service/main.go +++ b/ee/query-service/main.go @@ -7,12 +7,14 @@ import ( "time" "github.com/SigNoz/signoz/ee/query-service/app" + "github.com/SigNoz/signoz/ee/sqlstore/postgressqlstore" "github.com/SigNoz/signoz/pkg/config" "github.com/SigNoz/signoz/pkg/config/envprovider" "github.com/SigNoz/signoz/pkg/config/fileprovider" "github.com/SigNoz/signoz/pkg/query-service/auth" baseconst "github.com/SigNoz/signoz/pkg/query-service/constants" "github.com/SigNoz/signoz/pkg/signoz" + "github.com/SigNoz/signoz/pkg/sqlstore/sqlstorehook" "github.com/SigNoz/signoz/pkg/types/authtypes" "github.com/SigNoz/signoz/pkg/version" @@ -94,12 +96,17 @@ func main() { version.Info.PrettyPrint(config.Version) + sqlStoreFactories := signoz.NewSQLStoreProviderFactories() + if err := sqlStoreFactories.Add(postgressqlstore.NewFactory(sqlstorehook.NewLoggingFactory())); err != nil { + zap.L().Fatal("Failed to add postgressqlstore factory", zap.Error(err)) + } + signoz, err := signoz.New( context.Background(), config, signoz.NewCacheProviderFactories(), signoz.NewWebProviderFactories(), - signoz.NewSQLStoreProviderFactories(), + sqlStoreFactories, signoz.NewTelemetryStoreProviderFactories(), ) if err != nil { diff --git a/pkg/sqlstore/postgressqlstore/dialect.go b/ee/sqlstore/postgressqlstore/dialect.go similarity index 100% rename from pkg/sqlstore/postgressqlstore/dialect.go rename to ee/sqlstore/postgressqlstore/dialect.go diff --git a/pkg/sqlstore/postgressqlstore/provider.go b/ee/sqlstore/postgressqlstore/provider.go similarity index 100% rename from pkg/sqlstore/postgressqlstore/provider.go rename to ee/sqlstore/postgressqlstore/provider.go diff --git a/pkg/signoz/provider.go b/pkg/signoz/provider.go index 4dc9c71228..fdb8aa9046 100644 --- a/pkg/signoz/provider.go +++ b/pkg/signoz/provider.go @@ -10,7 +10,6 @@ import ( "github.com/SigNoz/signoz/pkg/factory" "github.com/SigNoz/signoz/pkg/sqlmigration" "github.com/SigNoz/signoz/pkg/sqlstore" - "github.com/SigNoz/signoz/pkg/sqlstore/postgressqlstore" "github.com/SigNoz/signoz/pkg/sqlstore/sqlitesqlstore" "github.com/SigNoz/signoz/pkg/sqlstore/sqlstorehook" "github.com/SigNoz/signoz/pkg/telemetrystore" @@ -39,7 +38,6 @@ func NewSQLStoreProviderFactories() factory.NamedMap[factory.ProviderFactory[sql hook := sqlstorehook.NewLoggingFactory() return factory.MustNewNamedMap( sqlitesqlstore.NewFactory(hook), - postgressqlstore.NewFactory(hook), ) }