All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • org.tinygroup.tinysqldsl.base.InsertContext Maven / Gradle / Ivy

    The newest version!
    /**
 * Copyright (c) 1997-2013, www.tinygroup.org ([email protected]).
 * 
    
 * Licensed under the GPL, Version 3.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
    
 * http://www.gnu.org/licenses/gpl.html
 * 
    
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.tinygroup.tinysqldsl.base;

import org.apache.commons.collections.map.CaseInsensitiveMap;
import org.tinygroup.commons.tools.ArrayUtil;
import org.tinygroup.tinysqldsl.expression.Expression;
import org.tinygroup.tinysqldsl.expression.relational.ExpressionList;
import org.tinygroup.tinysqldsl.insert.InsertBody;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * 插入操作的上下文
 *
 * @author renhui
 *
 */
public class InsertContext {

    private Map params = new CaseInsensitiveMap();

    private List paramValues = new ArrayList();

    private List allColumnNames = new ArrayList();

    private List noNullColumns = new ArrayList();

    private ExpressionList itemsList = new ExpressionList();

    private SelectBody selectBody;

    private Table table;

    private String schema;

    private String tableName;

    private boolean useValues = true;

    private List values = new ArrayList();

    public InsertContext() {
        super();
    }

    public Table getTable() {
        return table;
    }

    public void setTable(Table table) {
        this.table = table;
    }

    public void addValues(Value... values) {
        for (Value value : values) {
            Column column = value.getColumn();
            addColumnName(column.getColumnName());// 所有新增字段都加到列表中
            Object paramValue = value.getValue();
            Expression expression = value.getExpression();
            if (paramValue == null) {
                if (expression instanceof NamedCondition) {
                    paramValue = ((NamedCondition) expression).getValue();
                } else if (expression instanceof Condition) {
                    Condition condition = (Condition) expression;
                    Object[] valueArray = condition.getValues();
                    if (ArrayUtil.isEmptyArray(valueArray)) {
                        paramValue = null;
                    } else if (valueArray.length == 1) {
                        paramValue = valueArray[0];
                    } else {
                        paramValue = valueArray;
                    }
                }
            }
            if (paramValue != null) {
                noNullColumns.add(column);
                putParam(column.getColumnName(), paramValue);
                itemsList.addExpression(expression);
                this.values.add(value);
            }
        }
    }

    public Object[] getParamValues() {
        return paramValues.toArray();
    }

    public Map getParams() {
        return params;
    }

    public void setUseValues(boolean useValues) {
        this.useValues = useValues;
    }

    private void putParam(String columnName, Object value) {
        params.put(columnName, value);
        paramValues.add(value);
    }

    public List getColumnNames() {
        return allColumnNames;
    }

    public String[] getColumnNameArray() {
        return allColumnNames.toArray(new String[0]);
    }

    private void addColumnName(String columnName) {
        allColumnNames.add(columnName);
    }

    public String getSchema() {
        return schema;
    }

    public void setSchema(String schema) {
        this.schema = schema;
    }

    public String getTableName() {
        return tableName;
    }

    public void setTableName(String tableName) {
        this.tableName = tableName;
    }

    public void setSelectBody(SelectBody selectBody) {
        this.selectBody = selectBody;
    }

    public boolean existParam(String columnName) {
        return params.containsKey(columnName);
    }

    public Object getParamValue(String columnName) {
        return params.get(columnName);
    }

    public InsertContext copyContext() {
        InsertContext context = new InsertContext();
        context.setSchema(schema);
        context.setTable(table);
        context.setTableName(tableName);
        context.setUseValues(true);
        context.addValues(values.toArray(new Value[0]));
        return context;
    }

    public InsertBody createInsert() {
        InsertBody insertBody = new InsertBody();
        insertBody.setTable(table);
        insertBody.setColumns(noNullColumns);
        if (useValues) {
            insertBody.setItemsList(itemsList);
        } else {
            insertBody.setSelectBody(selectBody);
        }
        return insertBody;
    }

    public void setColumns(List columns) {
        this.noNullColumns = columns;
    }

    public void setItemsList(ExpressionList itemsList) {
        this.itemsList = itemsList;
    }
}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy