db.sql.api.impl.cmd.struct.update.UpdateSet Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mybatis-mp-sql-api-impl Show documentation

There is a newer version: 1.7.6-RC2

Show newest version

package db.sql.api.impl.cmd.struct.update;


import db.sql.api.Cmd;
import db.sql.api.DbType;
import db.sql.api.SqlBuilderContext;
import db.sql.api.cmd.struct.update.IUpdateSet;
import db.sql.api.impl.cmd.basic.TableField;
import db.sql.api.impl.cmd.executor.AbstractUpdate;
import db.sql.api.impl.tookit.SqlConst;
import db.sql.api.tookit.CmdUtils;

import java.util.Objects;

public class UpdateSet implements IUpdateSet {

    private final TableField field;

    private final Cmd value;

    public UpdateSet(TableField field, Cmd value) {
        this.field = field;
        this.value = value;
    }

    @Override
    public TableField getField() {
        return this.field;
    }

    @Override
    public Cmd getValue() {
        return this.value;
    }

    @Override
    public StringBuilder sql(Cmd module, Cmd parent, SqlBuilderContext context, StringBuilder sqlBuilder) {
        if (context.getDbType() == DbType.PGSQL) {
            //PG update set 列 不支持别名 直接拼接列名
            sqlBuilder.append(this.field.getName(context.getDbType()));
        } else if (context.getDbType() == DbType.SQL_SERVER) {
            //SQL_SERVER 别名支持的话 需要 带有from ；否则 只能是表面
            AbstractUpdate abstractUpdate = (AbstractUpdate) module;
            if (Objects.nonNull(abstractUpdate.getFrom())) {
                this.field.sql(module, this, context, sqlBuilder);
            } else {
                sqlBuilder.append(this.field.getName(context.getDbType()));
            }
        } else {
            this.field.sql(module, this, context, sqlBuilder);
        }

        sqlBuilder.append(SqlConst.EQ);
        this.value.sql(module, this, context, sqlBuilder);
        return sqlBuilder;
    }

    @Override
    public boolean contain(Cmd cmd) {
        return CmdUtils.contain(cmd, this.field, this.value);
    }
}