nbcp.myoql.bean.MyOqlBsonBeanConfig.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.bean

import com.fasterxml.jackson.databind.ObjectMapper
import nbcp.base.component.DbJsonMapper
import nbcp.base.component.WebJsonMapper
import nbcp.base.extend.*
import nbcp.base.utils.SpringUtil
import nbcp.myoql.db.mongo.DocumentJsonDeserializer
import nbcp.myoql.db.mongo.DocumentJsonSerializer
import nbcp.myoql.db.mongo.ObjectIdJsonDeserializer
import nbcp.myoql.db.mongo.ObjectIdJsonSerializer
import org.bson.Document
import org.bson.types.ObjectId
import org.springframework.beans.factory.config.BeanPostProcessor
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass
import org.springframework.boot.context.event.ApplicationStartedEvent
import org.springframework.context.event.EventListener
import org.springframework.stereotype.Component

@Component
@ConditionalOnClass(ObjectId::class)
class MyOqlBsonBeanConfig : BeanPostProcessor {
    companion object {
        private var inited = false;
    }

    override fun postProcessBeforeInitialization(bean: Any, beanName: String): Any? {
        if (inited == false) {
            inited = true;


            clazzesIsSimpleDefine.add(ObjectId::class.java)
        }

        if (bean is ObjectMapper) {
            bindExtendObjectMappers();
        }

        return super.postProcessBeforeInitialization(bean, beanName)
    }

    @EventListener
    fun bsonBeanInitOnApplicationStarted(ev: ApplicationStartedEvent) {
        bindExtendObjectMappers();
    }

    private fun bindExtendObjectMappers() {

        DbJsonMapper.registeObjectMapperTypeModule(
            ObjectId::class.java,
            ObjectIdJsonSerializer(),
            ObjectIdJsonDeserializer()
        );

        WebJsonMapper.registeObjectMapperTypeModule(
            ObjectId::class.java,
            ObjectIdJsonSerializer(),
            ObjectIdJsonDeserializer()
        );

//        DbJsonMapper.registeObjectMapperTypeModule(
//            Document::class.java,
//            DocumentJsonSerializer(),
//            DocumentJsonDeserializer()
//        );
    }
}