nbcp.myoql.db.mongo.MongoDocument2EntityUtil.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 org.bson.Document
import org.bson.types.ObjectId
import org.slf4j.LoggerFactory

object MongoDocument2EntityUtil {
    private val logger = LoggerFactory.getLogger(this::class.java);

    /**
     * 把 _id 转换为 id
     */
    @JvmOverloads
    @JvmStatic
    fun procResultData_id2Id(value: Collection<*>, remove_id: Boolean = true) {
        value.forEach { v ->
            if (v == null) {
                return@forEach
            }

            if (v is MutableMap<*, *>) {
                procResultData_id2Id(v, remove_id);
            } else if (v is Collection<*>) {
                procResultData_id2Id(v, remove_id);
            } else if (v is Array<*>) {
                procResultData_id2Id(v, remove_id);
            }
        }
    }


    /**
     * 把 _id 转换为 id
     */
    @JvmOverloads
    @JvmStatic
    fun procResultData_id2Id(value: Array<*>, remove_id: Boolean = true) {
        value.forEach { v ->
            if (v == null) {
                return@forEach
            }

            if (v is MutableMap<*, *>) {
                procResultData_id2Id(v, remove_id);
            } else if (v is Collection<*>) {
                procResultData_id2Id(v, remove_id);
            } else if (v is Array<*>) {
                procResultData_id2Id(v, remove_id);
            }
        }
    }

    @JvmOverloads
    @JvmStatic
    fun procResultData_id2Id(value: MutableMap<*, *>, remove_id: Boolean = true) {
        var keys = value.keys.toTypedArray();
        var needReplace = keys.contains("_id")

//        if (needReplace && value.get("id").AsString().HasValue) {
//            needReplace = false;
//        }

        for (k in keys) {
            var v = value.get(k);
            if (needReplace && (k == "_id")) {
                if (v == null) {
                    v = "";
                } else if (v is ObjectId) {
                    v = v.toString()
                }

                (value as MutableMap).set("id", v);
                if (remove_id) {
                    value.remove("_id")
                }
                needReplace = false;
                continue;
            }
            if (v == null) {
                continue;
            }
            if (v is MutableMap<*, *>) {
                procResultData_id2Id(v, remove_id);
            } else if (v is Collection<*>) {
                procResultData_id2Id(v, remove_id);
            }
        }
    }


    /**
     *value 可能会是： Document{{answerRole=Patriarch}}
     */
    @JvmStatic
    fun procDocumentJson(value: Document) {
//        fun testDocumentString(item: Any?): Boolean {
//            if (item == null) return false;
//            var type = item::class.java;
//            if (type.IsStringType == false) return false;
//            var v_string_value = item.toString()
//            return v_string_value.contains("{{") && v_string_value.endsWith("}}")
//        }
//
//        fun procDocumentString(v_string_value: String): Any {
//            //Document{{answerRole=Patriarch}}
//            //目前只发现一个键值对形式的。
//            val startIndex = v_string_value.indexOf("{{");
//            if (startIndex < 0) {
//                return v_string_value;
//            }
//
//            logger.Important("发现mongo Document怪异值: " + v_string_value);
//            val json = StringMap();
//            v_string_value.Slice(startIndex + 2, -2).split(",").forEach { item ->
//                val sect = item.split("=");
//                json.put(sect[0], sect[1]);
//            }
//            return json;
//        }

        procResultData_id2Id(value);

//        RecursionUtil.recursionAny(value, { json ->
//            json.keys.toTypedArray().forEachIndexed { _, key ->
//                if (key == null) {
//                    return@forEachIndexed
//                }
//                var documentStringValue = json.get(key);
//                if (!testDocumentString(documentStringValue)) {
//                    return@forEachIndexed
//                }
//
//                logger.Important("发现怪异mongo Document: " + value.ToJson())
//                logger.Important("发现怪异mongo Document,当前: " + json.ToJson())
//                (json as MutableMap).set(key, procDocumentString(documentStringValue.toString()));
//
//                return@forEachIndexed
//            }
//            return@recursionAny true
//        }, { list ->
//            var arrayList = list as MutableList
//            arrayList.forEachIndexed { index, it ->
//                if (it == null || !testDocumentString(it)) {
//                    return@forEachIndexed
//                }
//
//                logger.Important("发现怪异mongo Document: " + value.ToJson())
//                logger.Important("发现怪异mongo Array[${index}]: " + it.ToJson())
//                arrayList[index] = procDocumentString(it.toString());
//            }
//
//            return@recursionAny true
//        })
    }
}