-
Notifications
You must be signed in to change notification settings - Fork 13
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #43 from flytegg/dev/mongo
merge: dev/mongo into master - add MongoDB framework
- Loading branch information
Showing
6 changed files
with
404 additions
and
27 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,44 +1,190 @@ | ||
| package gg.flyte.twilight.data | ||
|
|
||
| import com.mongodb.ConnectionString | ||
| import com.mongodb.MongoClientSettings | ||
| import com.mongodb.MongoClientSettings.getDefaultCodecRegistry | ||
| import com.mongodb.client.MongoClient | ||
| import com.mongodb.client.MongoClients | ||
| import com.mongodb.client.MongoCollection | ||
| import com.mongodb.client.MongoDatabase | ||
| import com.mongodb.client.model.Filters.eq | ||
| import com.mongodb.client.model.ReplaceOptions | ||
| import com.mongodb.client.result.DeleteResult | ||
| import com.mongodb.client.result.UpdateResult | ||
| import com.mongodb.kotlin.client.MongoClient | ||
| import com.mongodb.kotlin.client.MongoCollection | ||
| import com.mongodb.kotlin.client.MongoDatabase | ||
| import com.mongodb.kotlin.client.MongoIterable | ||
| import gg.flyte.twilight.Twilight | ||
| import gg.flyte.twilight.data.MongoDB.executor | ||
| import gg.flyte.twilight.environment.Environment | ||
| import gg.flyte.twilight.gson.GSON | ||
| import gg.flyte.twilight.gson.toJson | ||
| import gg.flyte.twilight.string.Case | ||
| import gg.flyte.twilight.string.formatCase | ||
| import gg.flyte.twilight.string.pluralize | ||
| import org.bson.Document | ||
| import org.bson.UuidRepresentation | ||
| import org.bson.codecs.configuration.CodecRegistries.fromProviders | ||
| import org.bson.codecs.configuration.CodecRegistries.fromRegistries | ||
| import org.bson.codecs.pojo.PojoCodecProvider | ||
| import org.bson.conversions.Bson | ||
| import java.util.concurrent.CompletableFuture | ||
| import java.util.concurrent.Executor | ||
| import java.util.concurrent.Executors | ||
| import kotlin.reflect.KClass | ||
| import kotlin.reflect.KProperty | ||
| import kotlin.reflect.KProperty1 | ||
| import kotlin.reflect.KType | ||
| import kotlin.reflect.full.memberProperties | ||
| import kotlin.reflect.jvm.javaField | ||
| import kotlin.reflect.jvm.javaType | ||
|
|
||
| object MongoDB { | ||
|
|
||
| private val codecRegistry = fromRegistries( | ||
| getDefaultCodecRegistry(), | ||
| fromProviders(PojoCodecProvider.builder().automatic(true).build()) | ||
| ) | ||
|
|
||
| private lateinit var client: MongoClient | ||
| private lateinit var database: MongoDatabase | ||
| internal lateinit var database: MongoDatabase | ||
| internal val executor: Executor = Executors.newCachedThreadPool() | ||
|
|
||
| fun mongo(mongo: Settings) { | ||
| client = MongoClients.create(mongo.uri) | ||
| client = MongoClient.create( | ||
| MongoClientSettings.builder() | ||
| .uuidRepresentation(UuidRepresentation.STANDARD) | ||
| .codecRegistry(fromProviders(getDefaultCodecRegistry())) | ||
| .applyConnectionString(ConnectionString(mongo.uri)) | ||
| .build() | ||
| ) | ||
| database = client.getDatabase(mongo.database + if (Twilight.usingEnv && Environment.isDev()) "-dev" else "") | ||
| .withCodecRegistry(codecRegistry) | ||
| } | ||
|
|
||
| fun collection(name: String): MongoCollection<Document> { | ||
| return database.getCollection(name) | ||
| } | ||
| fun collection(name: String): MongoCollection<Document> = database.getCollection(name) | ||
|
|
||
| fun <T> collection(name: String, `class`: Class<T>): MongoCollection<T> { | ||
| return database.getCollection(name, `class`) | ||
| } | ||
| @Deprecated("Use collection(clazz: KClass<out MongoSerializable>)", ReplaceWith("collection(`class`)")) | ||
| fun <T : Any> collection(name: String, `class`: Class<T>): MongoCollection<T> = | ||
| database.getCollection(name, `class`) | ||
|
|
||
| class Settings { | ||
| var uri: String = if (Twilight.usingEnv) Environment.get("MONGO_URI") else "" | ||
| var database: String = if (Twilight.usingEnv) Environment.get("MONGO_DATABASE") else "" | ||
| } | ||
|
|
||
| } | ||
| val collections = mutableMapOf<KClass<out MongoSerializable>, TwilightMongoCollection<out MongoSerializable>>() | ||
|
|
||
| inline fun <reified T : MongoSerializable> collection( | ||
| name: String = T::class.simpleName!!.pluralize().formatCase(Case.CAMEL) | ||
| ): TwilightMongoCollection<T> { | ||
| @Suppress("unchecked_cast") | ||
| return collections.getOrPut(T::class) { TwilightMongoCollection(T::class, name) } as TwilightMongoCollection<T> | ||
| } | ||
|
|
||
| fun collection(clazz: KClass<out MongoSerializable>): TwilightMongoCollection<out MongoSerializable> { | ||
| return collections.getOrPut(clazz) { | ||
| TwilightMongoCollection( | ||
| clazz, | ||
| clazz.simpleName!!.pluralize().formatCase(Case.CAMEL) | ||
| ) | ||
| } | ||
| } | ||
|
|
||
| } | ||
|
|
||
| class TwilightMongoCollection<T : MongoSerializable>( | ||
| private val clazz: KClass<out MongoSerializable>, | ||
| val name: String | ||
| ) { | ||
|
|
||
| val idField = IdField(clazz) | ||
| val documents: MongoCollection<Document> = MongoDB.database.getCollection(name, Document::class.java) | ||
|
|
||
| fun saveSync(serializable: MongoSerializable): UpdateResult = with(serializable.toDocument()) { | ||
| documents.replaceOne( | ||
| eq(idField.name, this[idField.name]), | ||
| this, | ||
| ReplaceOptions().upsert(true) | ||
| ) | ||
| } | ||
|
|
||
| fun save(serializable: MongoSerializable): CompletableFuture<UpdateResult> = | ||
| CompletableFuture.supplyAsync({ saveSync(serializable) }, executor) | ||
|
|
||
| fun findSync(filter: Bson? = null): MongoIterable<T> = | ||
| (if (filter == null) documents.find() else documents.find(filter)).map { | ||
| @Suppress("unchecked_cast") | ||
| GSON.fromJson(it.toJson(), clazz.java) as T | ||
| } | ||
|
|
||
| fun find(filter: Bson? = null): CompletableFuture<MongoIterable<T>> = | ||
| CompletableFuture.supplyAsync({ findSync(filter) }, executor) | ||
|
|
||
| fun findByIdSync(id: Any): MongoIterable<T> { | ||
| require(id::class.javaObjectType == idField.type.javaType) { | ||
| "id must be of type ${idField.type} (Java: ${idField.type.javaType})" | ||
| } | ||
| return findSync(eq(idField.name, id)) | ||
| } | ||
|
|
||
| fun findById(id: Any): CompletableFuture<MongoIterable<T>> = | ||
| CompletableFuture.supplyAsync({ findByIdSync(id) }, executor) | ||
|
|
||
| fun deleteSync(filter: Bson): DeleteResult = documents.deleteMany(filter) | ||
|
|
||
| fun delete(filter: Bson): CompletableFuture<DeleteResult> = | ||
| CompletableFuture.supplyAsync({ deleteSync(filter) }, executor) | ||
|
|
||
| fun deleteByIdSync(id: Any): DeleteResult { | ||
| require(id::class.javaObjectType == idField.type.javaType) { | ||
| "id must be of type ${idField.type} (Java: ${idField.type.javaType})" | ||
| } | ||
| return deleteSync(eq(idField.name, id)) | ||
| } | ||
|
|
||
| fun deleteById(id: Any): CompletableFuture<DeleteResult> = | ||
| CompletableFuture.supplyAsync({ deleteByIdSync(id) }, executor) | ||
|
|
||
| } | ||
|
|
||
| interface MongoSerializable { | ||
| fun saveSync(): UpdateResult = MongoDB.collection(this::class).saveSync(this) | ||
|
|
||
| fun save(): CompletableFuture<UpdateResult> = MongoDB.collection(this::class).save(this) | ||
|
|
||
| fun deleteSync(): DeleteResult = with(MongoDB.collection(this::class)) { | ||
| deleteSync(eq(idField.name, idField.value(this@MongoSerializable))) | ||
| } | ||
|
|
||
| fun delete(): CompletableFuture<DeleteResult> = with(MongoDB.collection(this::class)) { | ||
| delete(eq(idField.name, idField.value(this@MongoSerializable))) | ||
| } | ||
|
|
||
| fun toDocument(): Document = Document.parse(toJson()) | ||
| } | ||
|
|
||
| @Target(AnnotationTarget.FIELD) | ||
| annotation class Id | ||
|
|
||
| data class IdField(val clazz: KClass<out MongoSerializable>) { | ||
|
|
||
| private val idField: KProperty1<out MongoSerializable, *> | ||
| val name: String | ||
| val type: KType | ||
|
|
||
| init { | ||
| val idFields = clazz.memberProperties.filter { it.javaField?.isAnnotationPresent(Id::class.java) == true } | ||
| println(idFields) | ||
|
|
||
| require(idFields.size == 1) { | ||
| when (idFields.size) { | ||
| 0 -> "Class does not have a field annotated with @Id" | ||
| else -> "Class must not have more than one field annotated with @Id" | ||
| } | ||
| } | ||
|
|
||
| idField = idFields.first() | ||
|
|
||
| name = idField.name | ||
| type = idField.returnType | ||
| } | ||
|
|
||
| @Suppress("unchecked_cast") | ||
| fun value(instance: MongoSerializable): Any = (idField as KProperty1<Any, *>).get(instance) | ||
| ?: throw IllegalStateException("Field annotated with @Id must not be null") | ||
|
|
||
| } | ||
|
|
||
| fun Any.toDocument(): Document = Document.parse(toJson()) | ||
|
|
||
| infix fun <V> KProperty<V>.eq(other: Any): Bson = eq(this.name, other) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.