nbcp.myoql.db.mongo.component.MongoBaseMetaCollection.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ktmyoql Show documentation
kotlin orm -- mysql,mongo , just like ktorm
The newest version!
package nbcp.myoql.db.mongo.component


import nbcp.base.extend.AllFields
import nbcp.base.extend.AsString
import nbcp.base.extend.FindField
import nbcp.base.extend.HasValue
import nbcp.myoql.db.BaseEntity
import nbcp.myoql.db.comm.BaseMetaData
import nbcp.myoql.db.db
import nbcp.myoql.db.mongo.MongoBaseInsertClip
import nbcp.myoql.db.mongo.base.MongoColumnName
import nbcp.myoql.db.mongo.query
import org.slf4j.LoggerFactory
import org.springframework.data.mongodb.core.MongoTemplate

typealias mongoQuery = org.springframework.data.mongodb.core.query.Query

/**
 * mongo 元数据实体的基类
 */
abstract class MongoBaseMetaCollection @JvmOverloads constructor(
        /**
         * 实体的类型
         */
        entityClass: Class,
        tableName: String = "",
        databaseId: String = ""
) : BaseMetaData(entityClass, tableName, databaseId) {
    //    abstract fun getColumns(): Array;
    companion object {
        private val logger = LoggerFactory.getLogger(this::class.java)
    }

    @Transient
    private var _columns = arrayOf()

    fun getColumns(): Array {
        if (_columns.isNotEmpty()) {
            return _columns;
        }

        _columns = this::class.java.AllFields.filter { MongoColumnName::class.java.isAssignableFrom(it.type) }
                .map { it.get(this) as MongoColumnName }
                .toTypedArray()
        return _columns;
    }


    fun getColumn(column: String): MongoColumnName? {
        return getColumns().firstOrNull { if (column == "id") it.toString() == "_id" else it.toString() == column }
    }

    fun getMongoTemplate(): MongoTemplate {
        val query = this.query();
        val settingResult = db.mongo.mongoEvents.onQuering(query);
        return query.getMongoTemplate(settingResult.lastOrNull { it.result.dataSource.HasValue }?.result?.dataSource)
    }


    /**
     * 插入单条实体
     */
    fun doInsert(entity: T): String {
        return insertAny(entity);
    }

    fun doInsert(entity: Map<*, Any?>): String {
        return insertAny(entity);
    }

    private fun insertAny(entity: Any): String {
        var batchInsert = MongoBaseInsertClip(this.tableName)
        batchInsert.addEntity(entity)
        batchInsert.exec();

        if (entity is BaseEntity) {
            return entity.id
        } else if (entity is Map<*, *>) {
            var idValue = entity.get("_id");
            if (idValue != null) {
                return idValue.AsString()
            }
        }

        var idField = entity.javaClass.FindField("id")
        if (idField != null) {
            return idField.get(entity).AsString();
        }

        throw RuntimeException("找不到 id")
    }
}