nbcp.myoql.db.mongo.component.MongoBaseInsertClip.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

import nbcp.base.extend.HasValue
import nbcp.base.extend.minus
import nbcp.base.extend.usingScope
import nbcp.myoql.db.db
import nbcp.myoql.db.enums.MyOqlDbScopeEnum
import nbcp.myoql.db.mongo.component.MongoClipBase
import nbcp.myoql.db.mongo.logger.logInsert
import org.slf4j.LoggerFactory
import java.time.LocalDateTime


open class MongoBaseInsertClip(tableName: String) : MongoClipBase(tableName) {
    companion object {
        private val logger = LoggerFactory.getLogger(this::class.java.declaringClass)
    }

    var entities = mutableListOf()

    /**
     * 批量添加中的添加实体。
     * 支持两种类型：Serializable,Map。  Document,DbObject 算是Map
     */
    fun addEntity(entity: Any) {
        this.entities.add(entity);
    }


    fun addEntities(entities: List) {
        this.entities.addAll(entities)
    }

    fun exec(): Int {
        db.affectRowCount = -1;

        var settingResult = db.mongo.mongoEvents.onInserting(this)
        if (settingResult.any { it.result.result == false }) {
            return 0;
        }

        var startAt = LocalDateTime.now()
        var error: Exception? = null;
        try {
            getMongoTemplate(settingResult.lastOrNull { it.result.dataSource.HasValue }?.result?.dataSource)
                    .insert(entities, this.actualTableName)
            this.executeTime = LocalDateTime.now() - startAt

            usingScope(arrayOf(MyOqlDbScopeEnum.IGNORE_AFFECT_ROW, MyOqlDbScopeEnum.IGNORE_EXECUTE_TIME)) {
                settingResult.forEach {
                    it.event.insert(this, it.result)
                }
            }

            this.affectRowCount = entities.size
            return this.affectRowCount
        } catch (e: Exception) {
            error = e;
            throw e;
        } finally {
            logger.logInsert(error, actualTableName, entities)
//            logger.InfoError(ret < 0) {
//                """[insert] ${this.collectionName}
//${
//                    if (config.debug) "[entities] ${entities.ToJson()}" else "[enities.ids] ${
//                        entities.map {
//                            if (it is BaseEntity) {
//                                return@map it.id
//                            } else if (it is Map<*, *>) {
//                                return@map it.get("id").AsString()
//                            }
//                            //反射 id.
//                            var idField = it.javaClass.FindField("id")
//                            if (idField != null) {
//                                idField.isAccessible = true;
//                                return@map idField.get(it);
//                            }
//                            return@map ""
//                        }.joinToString(",")
//                    }"
//                }
//[result] ${ret}
//[耗时] ${db.executeTime}
//"""
//            };
        }

//        return ret;
    }
}