From e0b39a4cc8af78a6fe82af0b4b0c703956e52d62 Mon Sep 17 00:00:00 2001 From: Vladislav Sukhin Date: Tue, 23 Jul 2024 16:47:01 +0300 Subject: [PATCH] fix: rename option method Signed-off-by: Vladislav Sukhin --- cmd/api-server/main.go | 4 ++-- pkg/repository/result/mongo.go | 2 +- pkg/repository/testresult/mongo.go | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/cmd/api-server/main.go b/cmd/api-server/main.go index 00b8fc1cb75..21434447ac6 100644 --- a/cmd/api-server/main.go +++ b/cmd/api-server/main.go @@ -292,9 +292,9 @@ func main() { isDocDb := cfg.APIMongoDBType == storage.TypeDocDB sequenceRepository := sequence.NewMongoRepository(db) mongoResultsRepository := result.NewMongoRepository(db, cfg.APIMongoAllowDiskUse, isDocDb, result.WithFeatureFlags(features), - result.WithLogsClient(logGrpcClient), result.WithMongoRepositorySequenceCollection(sequenceRepository)) + result.WithLogsClient(logGrpcClient), result.WithMongoRepositorySequence(sequenceRepository)) resultsRepository = mongoResultsRepository - testResultsRepository = testresult.NewMongoRepository(db, cfg.APIMongoAllowDiskUse, isDocDb) + testResultsRepository = testresult.NewMongoRepository(db, cfg.APIMongoAllowDiskUse, isDocDb, testresult.WithMongoRepositorySequence(sequenceRepository)) testWorkflowResultsRepository = testworkflow2.NewMongoRepository(db, cfg.APIMongoAllowDiskUse) configRepository = configrepository.NewMongoRepository(db) triggerLeaseBackend = triggers.NewMongoLeaseBackend(db) diff --git a/pkg/repository/result/mongo.go b/pkg/repository/result/mongo.go index 36292a2380d..8d814acffcf 100644 --- a/pkg/repository/result/mongo.go +++ b/pkg/repository/result/mongo.go @@ -121,7 +121,7 @@ func WithMongoRepositoryResultCollection(collection *mongo.Collection) MongoRepo } } -func WithMongoRepositorySequenceCollection(sequenceRepository sequence.Repository) MongoRepositoryOpt { +func WithMongoRepositorySequence(sequenceRepository sequence.Repository) MongoRepositoryOpt { return func(r *MongoRepository) { r.sequenceRepository = sequenceRepository } diff --git a/pkg/repository/testresult/mongo.go b/pkg/repository/testresult/mongo.go index 13022171b64..3cf5c0111be 100644 --- a/pkg/repository/testresult/mongo.go +++ b/pkg/repository/testresult/mongo.go @@ -51,7 +51,7 @@ func WithMongoRepositoryCollection(collection *mongo.Collection) MongoRepository } } -func WithMongoRepositorySequenceCollection(sequenceRepository sequence.Repository) MongoRepositoryOpt { +func WithMongoRepositorySequence(sequenceRepository sequence.Repository) MongoRepositoryOpt { return func(r *MongoRepository) { r.sequenceRepository = sequenceRepository }