nbcp.myoql.code.generator.db.mysql.EntityDbItemData.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of ktmyoql-code-generator Show documentation

kotlin orm -- mysql,mongo , just like ktorm

The newest version!

package nbcp.myoql.code.generator.db.mysql

import nbcp.base.comm.config
import nbcp.base.extend.HasValue
import nbcp.base.utils.StringUtil
import nbcp.myoql.code.generator.removeQuoteContent

/**
 * 以后使用MySqlTableCodeTemplateData替换它。
 */
class EntityDbItemData {
    var name = ""

    //原始的表注释
    var comment = ""

    //去除 （）
    val simpleComment: String
        get() {
            return comment.removeQuoteContent();
        }

    val group: String
        get() {
            // [组名]，先对表注释内容进行组名提取。
            var groups_all_value = Regex("""\[\s*([\w-_]+)\s*\]""")
                    .find(
                            comment
                    )
                    ?.groupValues ?: listOf()

            var group_value = ""
            if (groups_all_value.size > 0) {
                group_value = groups_all_value[1];
            }

            var groups_value = group_value.split(",").map { it.trim() }.filter { it.HasValue }

            if (groups_value.any()) {
                return groups_value.first();
            }
            groups_value = name.split("_", "-");
            if (groups_value.size > 0) {
                return groups_value.first()
            }

            return "";
        }

    /**
     * 除主键，单唯一键之外的联合唯一键。
     */
    var auks = arrayOf()

    var columns = mutableListOf()

    val className: String
        get() {
            var name2 = this.name
            if (name2[0].isDigit()) {
                name2 = "table_" + name2;
            }

            if (config.myoqlKeepDbName) {
                return StringUtil.splitWordParts(name2).joinToString("_")
            }

            return StringUtil.getBigCamelCase(name2)
        }
}