nbcp.myoql.db.sql.component.SqlSetEntityUpdateClip.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.sql.component


import nbcp.base.comm.JsonMap
import nbcp.base.extend.AllFields
import nbcp.base.utils.MyUtil
import nbcp.base.utils.ReflectUtil
import nbcp.myoql.db.db
import nbcp.myoql.db.sql.base.SqlBaseMetaTable
import nbcp.myoql.db.sql.base.SqlColumnName
import nbcp.myoql.db.sql.base.SqlParameterData
import nbcp.myoql.db.sql.extend.proc_value
import java.io.Serializable

/**
 * Created by udi on 17-4-7.
 */

//根据Id，更新Mongo的一个键。

/**
 * MongoUpdate
 * 不会更新 id
 */
class SqlSetEntityUpdateClip>(var mainEntity: M, var entity: Serializable) :
    SqlBaseExecuteClip(mainEntity.tableName) {
    companion object {
    }

    private var whereColumns = mutableSetOf()
    private var setColumns = mutableSetOf()
    private var unsetColumns = mutableSetOf()
    private var sets = mutableMapOf()

    fun withColumn(setFunc: (M) -> SqlColumnName): SqlSetEntityUpdateClip {
        this.setColumns.add(setFunc(this.mainEntity))
        return this;
    }

    fun withoutColumn(unsetFunc: (M) -> SqlColumnName): SqlSetEntityUpdateClip {
        this.unsetColumns.add(unsetFunc(this.mainEntity))
        return this;
    }

    fun whereColumn(whereFunc: (M) -> SqlColumnName): SqlSetEntityUpdateClip {
        this.whereColumns.add(whereFunc(this.mainEntity))
        return this;
    }

//    /**
//     * 不应该依赖客户端，不应该使用这个方法
//     */
//    fun withRequestParams(keys: Set): SqlSetEntityUpdateClip {
//
//    }

    //额外设置
    fun set(setItemAction: (M) -> Pair): SqlSetEntityUpdateClip {
        var setItem = setItemAction(this.mainEntity);
        this.sets.put(setItem.first, setItem.second);
        return this;
    }

    /**
     * 更新，默认按 id 更新
     */
    override fun exec(): Int {
        //调用这个方法很重要。
        this.toSql();
        return sqlUpdate.exec();
    }

    private var sqlUpdate = SqlUpdateClip(mainEntity);

    /**
     * 设置 sqlUpdate 对象
     */
    override fun toSql(): SqlParameterData {
        sqlUpdate = SqlUpdateClip(mainEntity);

        var columns = this.mainEntity.getColumns()

        var setValues = mutableMapOf()

        var field_names = mutableListOf();
        entity::class.java.AllFields.forEach { field ->

//            var ann = field.getAnnotation(SqlSpreadColumn::class.java);
//            if( ann!= null){
//                var ent_field_value = field.get(entity);
//                field.type.AllFields.forEach { subField ->
//                    var value = subField.get(ent_field_value);
//                    setValues.put(columns.first { it.name == field.name +"_" + subField.name }, value)
//                }
//            }

            var spread = this.mainEntity.getSpreadColumns().firstOrNull { it.column == field.name }
            if (spread != null) {
                var ent_field_value = field.get(entity);
                field.type.AllFields.forEach { subField ->
                    var value = subField.get(ent_field_value);
                    setValues.put(columns.first { it.name == spread.getPrefixName() + subField.name }, value)
                }
            }

            field_names.add(field.name);
        };


        //自增 id 不能更新。
        var auKey = columns.firstOrNull { it.name == this.mainEntity.getAutoIncrementKey() };
        var where = WhereData();

        var whereColumns2 = whereColumns.map { it }.toMutableList();

        if (whereColumns2.any() == false) {
            if (auKey != null) {
                whereColumns2.add(auKey)
            } else {
                var uks = this.mainEntity.getUks().toList();
                var pks = uks.groupBy { it.size }.minByOrNull { it.key }?.value?.firstOrNull()
                if (pks != null) {
                    whereColumns2.addAll(columns.filter { pks.contains(it.name) })
                }
            }
        }


        whereColumns2.forEach { column ->
            var value = ReflectUtil.getValueByWbsPath(entity, column.name)

            where.and(
                WhereData(
                    "${column.fullName} = :${column.paramVarKeyName}",
                    JsonMap(column.paramVarKeyName to value)
                )
            )
        }


        var unsetColumn_names = unsetColumns.map { it.name }
        var setColumn_names = listOf()

        if (setColumns.any() == false) {
            setColumn_names = columns.map { it.name }
        } else {
            setColumn_names = setColumns.map { it.name }
        }

        columns.minus(whereColumns2)
            .filter { column ->
                column.name != auKey?.name
                        && field_names.contains(column.name)
                        && !unsetColumn_names.contains(column.name)
                        && setColumn_names.contains(column.name)
            }
            .forEach { key ->
                var value = ReflectUtil.getValueByWbsPath(entity, key.name)
                if (value == null) {
                    setValues.put(key, null);
                    return@forEach
                }

                setValues.put(key, proc_value(value));
            }

        this.sets.forEach {
            setValues.put(it.key, it.value)
        }
        setValues.forEach {
            this.sqlUpdate.setData.put(it.key, it.value);
        }

        this.sqlUpdate.whereDatas.and(where);

        return sqlUpdate.toSql()
    }

    /**
     * 执行更新, == exec ,语义清晰
     */
    fun execUpdate(): Int {
        return this.exec();
    }

    /**
     * 执行插入
     */
    fun execInsert(): Int {
        val batchInsert = SqlInsertClip(mainEntity)
        batchInsert.addEntity(entity);

        return batchInsert.exec();
    }

    /**
     * 先更新，如果不存在，则插入。
     * @return: 返回插入的Id，如果是更新则返回空字串
     */
    fun updateOrAdd(): Int {
        //有一个问题，可能是阻止更新了。所以导致是0。
        if (this.execUpdate() == 0) {
            return this.execInsert()
        }
        return db.affectRowCount;
    }
}