com.dream.flex.def.UpdateColumnDef Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dream-orm-flex Show documentation
The newest version!
package com.dream.flex.def;

import com.dream.antlr.smt.*;
import com.dream.struct.invoker.TakeMarkInvokerStatement;

import java.util.Map;
import java.util.Set;


public interface UpdateColumnDef extends UpdateDef {
    default UpdateColumn set(ColumnDef columnDef, ColumnDef columnDef2) {
        Statement where = statement().getConditionList();
        if (where == null) {
            where = new ListColumnStatement(",");
            statement().setConditionList(where);
        }
        ListColumnStatement listColumnStatement = (ListColumnStatement) where;
        ConditionStatement conditionStatement = new ConditionStatement();
        conditionStatement.setLeft(columnDef.getStatement());
        conditionStatement.setOper(new OperStatement.EQStatement());
        conditionStatement.setRight(columnDef2.getStatement());
        listColumnStatement.add(conditionStatement);
        return (UpdateColumn) this;
    }

    default UpdateColumn set(ColumnDef columnDef, Object value) {
        set(columnDef, new ColumnDef(new TakeMarkInvokerStatement(columnDef.column(), value)));
        return (UpdateColumn) this;
    }

    default UpdateColumn set(ColumnDef columnDef, QueryDef queryDef) {
        BraceStatement braceStatement = new BraceStatement(queryDef.statement());
        set(columnDef, new ColumnDef(braceStatement));
        return (UpdateColumn) this;
    }

    default UpdateColumn setMap(Map valueMap) {
        if (valueMap != null && !valueMap.isEmpty()) {
            Set> entries = valueMap.entrySet();
            for (Map.Entry entry : entries) {
                set(entry.getKey(), entry.getValue());
            }
        }
        return (UpdateColumn) this;
    }

    default Update where(ConditionDef conditionDef) {
        WhereStatement whereStatement = new WhereStatement();
        whereStatement.setStatement(conditionDef.getStatement());
        statement().setWhere(whereStatement);
        return (Update) creatorFactory().newUpdateDef(statement());
    }
}