/
KMongo.kt
79 lines (72 loc) · 2.72 KB
/
KMongo.kt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
/*
* Copyright (C) 2016/2022 Litote
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.litote.kmongo.reactivestreams
import com.mongodb.ConnectionString
import com.mongodb.MongoClientSettings
import com.mongodb.MongoDriverInformation
import com.mongodb.reactivestreams.client.MongoClient
import com.mongodb.reactivestreams.client.MongoClients
import org.bson.codecs.configuration.CodecRegistry
import org.litote.kmongo.service.ClassMappingType
/**
* Main object used to create a [MongoClient](https://api.mongodb.com/java/current/com/mongodb/reactivestreams/client/MongoClient.html) instance.
*/
object KMongo {
/**
* Creates a new client with the default connection string "mongodb://localhost".
*
* @return the client
*/
fun createClient(): MongoClient = createClient(ConnectionString("mongodb://localhost"))
/**
* Create a new client with the given client settings.
*
* @param settings the settings
* @return the client
*/
fun createClient(settings: MongoClientSettings): MongoClient {
val codecRegistry = ClassMappingType.codecRegistry(settings.codecRegistry)
return MongoClients.create(
MongoClientSettings.builder(settings)
.codecRegistry(codecRegistry).build(),
MongoDriverInformation.builder().driverName("kmongo").driverPlatform(String.format("Kotlin/%s", KotlinVersion.CURRENT)).build()
)
}
/**
* Create a new client with the given connection string.
*
* @param connectionString the connection
* @return the client
*/
fun createClient(connectionString: String): MongoClient = createClient(ConnectionString(connectionString))
/**
* Create a new client with the given connection string.
*
* @param connectionString the settings
* @return the client
*/
fun createClient(connectionString: ConnectionString): MongoClient {
return createClient(
MongoClientSettings
.builder()
.applyConnectionString(connectionString)
.build()
)
}
internal fun configureRegistry(codecRegistry: CodecRegistry): CodecRegistry {
return ClassMappingType.codecRegistry(codecRegistry)
}
}