Skip to content

Commit 9b097cd

Browse files
author
Roman Janusz
committed
review fixes in TypedMongoCollection
1 parent c1838ad commit 9b097cd

File tree

1 file changed

+30
-17
lines changed

1 file changed

+30
-17
lines changed

commons-mongo/jvm/src/main/scala/com/avsystem/commons/mongo/typed/TypedMongoCollection.scala

Lines changed: 30 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,8 @@ class TypedMongoCollection[E <: BaseMongoEntity] private(
101101
setupOptions: RenameCollectionOptions => RenameCollectionOptions = identity
102102
): Task[Unit] =
103103
empty(optionalizeFirstArg(
104-
nativeCollection.renameCollection(sessionOrNull, namespace, setupOptions(new RenameCollectionOptions))))
104+
nativeCollection.renameCollection(sessionOrNull, namespace, setupOptions(new RenameCollectionOptions))
105+
))
105106

106107
def countDocuments(
107108
filter: MongoDocumentFilter[E] = MongoFilter.empty,
@@ -114,9 +115,8 @@ class TypedMongoCollection[E <: BaseMongoEntity] private(
114115
def estimatedDocumentCount(
115116
setupOptions: EstimatedDocumentCountOptions => EstimatedDocumentCountOptions = identity
116117
): Task[Long] =
117-
single(optionalizeFirstArg(
118-
nativeCollection.estimatedDocumentCount(setupOptions(new EstimatedDocumentCountOptions))
119-
)).asInstanceOf[Task[Long]]
118+
single(nativeCollection.estimatedDocumentCount(setupOptions(new EstimatedDocumentCountOptions)))
119+
.asInstanceOf[Task[Long]]
120120

121121
def exists(
122122
filter: MongoDocumentFilter[E],
@@ -182,7 +182,8 @@ class TypedMongoCollection[E <: BaseMongoEntity] private(
182182
projection match {
183183
case SelfRef =>
184184
singleOpt(optionalizeFirstArg(
185-
nativeCollection.findOneAndUpdate(sessionOrNull, filterBson, updateBson, options)).asInstanceOf[Publisher[T]])
185+
nativeCollection.findOneAndUpdate(sessionOrNull, filterBson, updateBson, options)
186+
).asInstanceOf[Publisher[T]])
186187
case proj =>
187188
val optionsWithProj = options.projection(proj.toProjectionBson)
188189
singleOpt(optionalizeFirstArg(
@@ -226,11 +227,13 @@ class TypedMongoCollection[E <: BaseMongoEntity] private(
226227
projection match {
227228
case SelfRef =>
228229
singleOpt(optionalizeFirstArg(
229-
nativeCollection.findOneAndDelete(sessionOrNull, filterBson, options)).asInstanceOf[Publisher[T]])
230+
nativeCollection.findOneAndDelete(sessionOrNull, filterBson, options)
231+
).asInstanceOf[Publisher[T]])
230232
case proj =>
231233
val optionsWithProj = options.projection(proj.toProjectionBson)
232234
singleOpt(optionalizeFirstArg(
233-
docCollection.findOneAndDelete(sessionOrNull, filterBson, optionsWithProj))).map(_.map(proj.decodeFrom))
235+
docCollection.findOneAndDelete(sessionOrNull, filterBson, optionsWithProj))
236+
).map(_.map(proj.decodeFrom))
234237
}
235238
}
236239

@@ -255,28 +258,32 @@ class TypedMongoCollection[E <: BaseMongoEntity] private(
255258
setupOptions: InsertOneOptions => InsertOneOptions = identity
256259
): Task[InsertOneResult] =
257260
single(optionalizeFirstArg(
258-
nativeCollection.insertOne(sessionOrNull, value, setupOptions(new InsertOneOptions))))
261+
nativeCollection.insertOne(sessionOrNull, value, setupOptions(new InsertOneOptions))
262+
))
259263

260264
def insertMany(
261265
values: Seq[E],
262266
setupOptions: InsertManyOptions => InsertManyOptions = identity
263267
): Task[InsertManyResult] =
264268
single(optionalizeFirstArg(
265-
nativeCollection.insertMany(sessionOrNull, values.asJava, setupOptions(new InsertManyOptions))))
269+
nativeCollection.insertMany(sessionOrNull, values.asJava, setupOptions(new InsertManyOptions))
270+
))
266271

267272
def deleteOne(
268273
filter: MongoDocumentFilter[E],
269274
setupOptions: DeleteOptions => DeleteOptions = identity
270275
): Task[DeleteResult] =
271276
single(optionalizeFirstArg(
272-
nativeCollection.deleteOne(sessionOrNull, filter.toBson, setupOptions(new DeleteOptions))))
277+
nativeCollection.deleteOne(sessionOrNull, filter.toBson, setupOptions(new DeleteOptions))
278+
))
273279

274280
def deleteMany(
275281
filter: MongoDocumentFilter[E],
276282
setupOptions: DeleteOptions => DeleteOptions = identity
277283
): Task[DeleteResult] =
278284
single(optionalizeFirstArg(
279-
nativeCollection.deleteMany(sessionOrNull, filter.toBson, setupOptions(new DeleteOptions))))
285+
nativeCollection.deleteMany(sessionOrNull, filter.toBson, setupOptions(new DeleteOptions))
286+
))
280287

281288
def updateOne(
282289
filter: MongoDocumentFilter[E],
@@ -290,7 +297,8 @@ class TypedMongoCollection[E <: BaseMongoEntity] private(
290297
options.arrayFilters(arrayFilters)
291298
}
292299
single(optionalizeFirstArg(
293-
nativeCollection.updateOne(sessionOrNull, filter.toBson, updateBson, options)))
300+
nativeCollection.updateOne(sessionOrNull, filter.toBson, updateBson, options)
301+
))
294302
}
295303

296304
def updateMany(
@@ -305,7 +313,8 @@ class TypedMongoCollection[E <: BaseMongoEntity] private(
305313
options.arrayFilters(arrayFilters)
306314
}
307315
single(optionalizeFirstArg(
308-
nativeCollection.updateMany(sessionOrNull, filter.toBson, updateBson, options)))
316+
nativeCollection.updateMany(sessionOrNull, filter.toBson, updateBson, options)
317+
))
309318
}
310319

311320
def replaceOne(
@@ -316,7 +325,8 @@ class TypedMongoCollection[E <: BaseMongoEntity] private(
316325
): Task[UpdateResult] = {
317326
val options = setupOptions(new ReplaceOptions).upsert(upsert)
318327
single(optionalizeFirstArg(
319-
nativeCollection.replaceOne(sessionOrNull, filter.toBson, replacement, options)))
328+
nativeCollection.replaceOne(sessionOrNull, filter.toBson, replacement, options)
329+
))
320330
}
321331

322332
def bulkWrite(
@@ -325,12 +335,14 @@ class TypedMongoCollection[E <: BaseMongoEntity] private(
325335
): Task[BulkWriteResult] = {
326336
val requests = writes.iterator.map(_.toWriteModel).to(JList)
327337
single(optionalizeFirstArg(
328-
nativeCollection.bulkWrite(sessionOrNull, requests, setupOptions(new BulkWriteOptions))))
338+
nativeCollection.bulkWrite(sessionOrNull, requests, setupOptions(new BulkWriteOptions))
339+
))
329340
}
330341

331342
def createIndex(index: MongoIndex[E]): Task[String] =
332343
single(optionalizeFirstArg(
333-
nativeCollection.createIndex(sessionOrNull, index.toBson, index.setupOptions(new IndexOptions))))
344+
nativeCollection.createIndex(sessionOrNull, index.toBson, index.setupOptions(new IndexOptions))
345+
))
334346

335347
def createIndexes(
336348
indexes: Seq[MongoIndex[E]],
@@ -340,7 +352,8 @@ class TypedMongoCollection[E <: BaseMongoEntity] private(
340352
.map(index => new IndexModel(index.toBson, index.setupOptions(new IndexOptions)))
341353
.to(JList)
342354
single(optionalizeFirstArg(
343-
nativeCollection.createIndexes(sessionOrNull, indexModels, setupOptions(new CreateIndexOptions))))
355+
nativeCollection.createIndexes(sessionOrNull, indexModels, setupOptions(new CreateIndexOptions))
356+
))
344357
}
345358

346359
@bincompat private[typed] def this(rawCollection: MongoCollection[_], format: MongoAdtFormat[E]) =

0 commit comments

Comments
 (0)