Application: gvSIG desktop

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2020 gvSIG Association.

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 3

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.expressionevaluator.impl.function.dataaccess;

import java.util.ArrayList;

import java.util.List;

import java.util.Objects;

import org.apache.commons.lang3.Range;

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.lang3.tuple.ImmutablePair;

import org.apache.commons.lang3.tuple.Pair;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.CodeBuilder;

import org.gvsig.expressionevaluator.Codes;

import static org.gvsig.expressionevaluator.ExpressionBuilder.FUNCTION_GETATTR;

import static org.gvsig.expressionevaluator.ExpressionBuilder.FUNCTION_TUPLE;

import org.gvsig.expressionevaluator.ExpressionRuntimeException;

import org.gvsig.expressionevaluator.ExpressionUtils;

import org.gvsig.expressionevaluator.Interpreter;

import org.gvsig.expressionevaluator.Optimizer;

import org.gvsig.expressionevaluator.SymbolTable;

import org.gvsig.expressionevaluator.impl.DALFunctions;

import org.gvsig.expressionevaluator.spi.AbstractFunction;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import static org.gvsig.fmap.dal.DataManager.FUNCTION_SELECT;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.expressionevaluator.ExpressionEvaluator;

import static org.gvsig.fmap.dal.expressionevaluator.FeatureSymbolTable.SYMBOL_CURRENT_TABLE;

import org.gvsig.fmap.dal.expressionevaluator.TableAttributeHandler;

import org.gvsig.fmap.dal.feature.EditableFeature;

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.dal.feature.FeatureQuery;

import org.gvsig.fmap.dal.feature.FeatureQueryOrder;

import org.gvsig.fmap.dal.feature.FeatureSet;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.impl.expressionevaluator.DefaultExpressionEvaluator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.exception.BaseException;

/**

 *

 * @author jjdelcerro

 */

public class InsertIntoTableFunction 

        extends AbstractFunction 

        implements Optimizer.FunctionOptimizer

  {

  public InsertIntoTableFunction() {

    super(  DALFunctions.GROUP_DATA_ACCESS,

            DataManager.FUNCTION_INSERT_INTO_TABLE,

            Range.is(7),

            "Inserts in the indicated table the records obtained from the specified SELECT statement.\n" +

              "Return the number of inserted records.\n" +

              "This statement must always end with a semicolon.\n" +

              "See the SELECT statement for more information on this.",

            "INSERT INTO {{table_name}} SELECT * FROM table WHERE boolean_expression ORDER BY order_column LIMIT limit;",

            new String[]{

              "table_name - Name of the table in which to insert the records.",

              "column_names/asterisk - Names of the columns table to retrieve.",

              "table_name - Name of the table",

              "filter - boolean expression to apply as filter",

              "order_column - the order used to retrieve the features. It is a list of column names separated by a comma. The column name can optionally be followed by ASC or DESC to indicate whether the order should be ascending or descending.",

              "limit - Maximum number of features to return"

            },

            "Number",

            false

    );

  }

  @Override

  public boolean allowConstantFolding() {

    return false;

  }

  @Override

  public boolean useArgumentsInsteadObjects() {

    return true;

  }

  @Override

  public Object call(Interpreter interpreter, Object[] args) throws Exception {

    throw new UnsupportedOperationException();

  }

  private static final int COLUMNS = 1;

  private static final int TABLE = 2;

  private static final int WHERE = 3;

  private static final int ORDER = 4;

  private static final int ORDER_MODE = 5;

  private static final int LIMIT = 6;

  private Code.Callable getTupleOrNull(Codes args, int argn) {

    Code code = args.get(argn);

    if( code.code()==Code.CONSTANT ) {

      if( ((Code.Constant)code).value()!=null ) {

        throw new ExpressionRuntimeException("Tupple or null expected in argument "+argn+ " of function '" + FUNCTION_SELECT + "'.");

      }

      return null;

    }

    if( code.code()!=Code.CALLABLE ) {

      throw new ExpressionRuntimeException("Tupple or null expected in argument "+argn+ " of function '" + FUNCTION_SELECT + "'.");

    }

    Code.Callable caller = (Code.Callable) code;

    if( !StringUtils.equalsIgnoreCase(FUNCTION_TUPLE, caller.name()) ) {

      throw new ExpressionRuntimeException("Tupple or null expected in argument "+argn+ " of function '" + FUNCTION_SELECT + "'.");

    }

    return caller;

  }

  @Override

  @SuppressWarnings("UseSpecificCatch")

  public Object call(Interpreter interpreter, Codes args) throws Exception {

    FeatureStore targetStore = null;

    FeatureStore sourceStore = null;

    try {

      String targetTableName = Objects.toString(getObject(interpreter, args, 0),null);

      if( targetTableName == null ) {

        throw new ExpressionRuntimeException("Target table name can't be null.");    

      }

      DataManager dataManager = DALLocator.getDataManager();  

      targetStore = (FeatureStore) dataManager.getStoresRepository().getStore(targetTableName);

      if( targetStore==null ) {

        throw new ExpressionRuntimeException("Can't locate target table '"+targetTableName+"'.");    

      }

      Code.Identifier storeName =  (Code.Identifier) args.get(TABLE);

      Code columns = getTupleOrNull(args, COLUMNS);

      Code where = args.get(WHERE);

      Number limit = (Number) getObject(interpreter, args, LIMIT);

      Code.Callable order = getTupleOrNull(args, ORDER);

      Code.Callable order_mode = getTupleOrNull(args, ORDER_MODE);

      FeatureQueryOrder queryOrder = null;

      if( order!=null || order_mode!=null ) {

        for( int n=0 ; n<order.parameters().size(); n++) {

          String member = (String) interpreter.run(order.parameters().get(n));

          Boolean mode = (Boolean) interpreter.run(order_mode.parameters().get(n));

          if( queryOrder == null ) {

            queryOrder = new FeatureQueryOrder();

          }

          queryOrder.add(member, mode);

        }

      }

      // FIXME: add columns to query.addAttributeName() 

      DataStore store = this.getStore(storeName.name());

      if (store == null ) {

        throw new ExpressionRuntimeException("Cant locate the store '" + storeName + "' in function '" + FUNCTION_SELECT + "'.");

      }

      if (!(store instanceof FeatureStore)) {

        throw new ExpressionRuntimeException("The store'" + storeName + "' is not valid for function '" + FUNCTION_SELECT + "', a FeatureStore is required.");

      }

      sourceStore = (FeatureStore) store;

      FeatureSet features;

      if (where == null && queryOrder == null && limit==null ) {

        features = sourceStore.getFeatureSet();

      } else {

        FeatureQuery query = sourceStore.createFeatureQuery();

        if (where != null) {

          removeOuterTablesReferences(interpreter, where);

          ExpressionEvaluator filter = new DefaultExpressionEvaluator(where.toString());

          filter.getSymbolTable().addSymbolTable(interpreter.getSymbolTable());

          query.addFilter(filter);

        }

        if (queryOrder != null) {

          query.getOrder().copyFrom(queryOrder);

        }

        if( limit!=null ) {

          query.setLimit(limit.longValue());

        }

        query.retrievesAllAttributes();

        features = sourceStore.getFeatureSet(query);

      }

      long count = 0;

      targetStore.edit(FeatureStore.MODE_APPEND);

      for (Feature feature : features) {

        EditableFeature f = targetStore.createNewFeature();

        f.copyFrom(feature);

        targetStore.insert(f);

        count++;

      }

      targetStore.finishEditing();

      return count;

    } catch (ExpressionRuntimeException ex) {

      throw ex;

    } catch (Exception ex) {

      if( targetStore!=null && targetStore.isAppending() ) {

        targetStore.cancelEditing();

      }

      DisposeUtils.disposeQuietly(targetStore);

      DisposeUtils.disposeQuietly(sourceStore);

      throw new ExpressionRuntimeException("Problems calling '" + FUNCTION_SELECT + "' function", ex);

    }

  }

  protected DataStore getStore(String storeName) {

    DataManager dataManager = DALLocator.getDataManager();

    DataStore store = dataManager.getStoresRepository().getStore(storeName);

    return store;

  }

  private void removeOuterTablesReferences(Interpreter interpreter, Code where) {

    try {

      SymbolTable symbolTable = interpreter.getSymbolTable();

      TableAttributeHandler table = (TableAttributeHandler) symbolTable.value(SYMBOL_CURRENT_TABLE);

      List<Pair<Code,Code>>replaces = new ArrayList<>();

      CodeBuilder codeBuilder = ExpressionUtils.createCodeBuilder();

      where.accept((Object o) -> {

        Code code = (Code) o;

        if( code!=null && code.code() == Code.CALLABLE ) {

          Code.Callable caller = (Code.Callable) code;

          if( StringUtils.equalsIgnoreCase(caller.name(),FUNCTION_GETATTR) ) {

            Codes args = caller.parameters();

            Code arg0 = args.get(0);

            Code arg1 = args.get(1);

            if( arg0 instanceof Code.Identifier && arg1 instanceof Code.Identifier ) {

              Object tt = symbolTable.value(((Code.Identifier)arg0).name());

              if( tt instanceof TableAttributeHandler && 

                  StringUtils.equalsIgnoreCase(((TableAttributeHandler)tt).getName(), table.getName()) ) {

                String columnName = Objects.toString(((Code.Identifier)arg1).name(), null);

                if( columnName!=null ) {

                  Object value = table.get(columnName);

                  replaces.add(

                          new ImmutablePair<>(

                                  caller,

                                  codeBuilder.constant(value)

                          )

                  );

                }

              }

            }

          }

        }

      });

      for (Pair<Code, Code> replace : replaces) {

        if( replace!=null ) {

          where.replace(replace.getLeft(), replace.getRight());

        }

      }

    } catch (BaseException ex) {

      throw new ExpressionRuntimeException("Can't remove references to outer tables.", ex);

    }

  }

  @Override

  public Code optimize(Optimizer optimizer, Code.Callable caller) {

    return caller; // Don't optimize SELECT

  }

}