Skip to content

Commit 97764fe

Browse files
author
Isabella Siu
committed
GODRIVER-563 Add latencySelector for writeSelector property in mongo.Database and mongo.Collection
Change-Id: I323d44542822c865e9016be38ca97c35738ca130
1 parent 8f47540 commit 97764fe

File tree

2 files changed

+10
-2
lines changed

2 files changed

+10
-2
lines changed

mongo/collection.go

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,11 @@ func newCollection(db *Database, name string, opts ...collectionopt.Option) *Col
7474
description.LatencySelector(db.client.localThreshold),
7575
})
7676

77+
writeSelector := description.CompositeSelector([]description.ServerSelector{
78+
description.WriteSelector(),
79+
description.LatencySelector(db.client.localThreshold),
80+
})
81+
7782
coll := &Collection{
7883
client: db.client,
7984
db: db,
@@ -82,7 +87,7 @@ func newCollection(db *Database, name string, opts ...collectionopt.Option) *Col
8287
readConcern: rc,
8388
writeConcern: wc,
8489
readSelector: readSelector,
85-
writeSelector: db.writeSelector,
90+
writeSelector: writeSelector,
8691
registry: db.registry,
8792
}
8893

mongo/database.go

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,10 @@ func newDatabase(client *Client, name string, opts ...dbopt.Option) *Database {
7070
description.LatencySelector(db.client.localThreshold),
7171
})
7272

73-
db.writeSelector = description.WriteSelector()
73+
db.writeSelector = description.CompositeSelector([]description.ServerSelector{
74+
description.WriteSelector(),
75+
description.LatencySelector(db.client.localThreshold),
76+
})
7477

7578
return db
7679
}

0 commit comments

Comments
 (0)