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.fmap.dal.store.jdbc2.impl.expressionevaluator;

import org.apache.commons.lang3.Range;

import org.gvsig.expressionevaluator.Interpreter;

import org.gvsig.expressionevaluator.spi.AbstractFunction;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorerParameters;

import org.gvsig.fmap.dal.DataServerExplorerPoolEntry;

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

import org.gvsig.fmap.dal.store.jdbc2.JDBCServerExplorer;

import static org.gvsig.fmap.dal.store.jdbc2.impl.expressionevaluator.JDBCSymbolTableFactory.GROUP_JDBC;

/**

 *

 * @author jjdelcerro

 */

public class ExecuteSQLFunction extends AbstractFunction {

  public static final String FUNCTION_EXECUTESQL = "EXECUTESQL";

  public ExecuteSQLFunction() {

    super(  GROUP_JDBC,

            FUNCTION_EXECUTESQL,

            Range.between(1,2),

            "Receive a connection and SQL string as argument.",

            FUNCTION_EXECUTESQL + "({{connection}}, sql)",

            new String[]{

              "connection - conextion string (optional)",

              "sql - sql string"

            },

            "Object",

            false

    );

  }

  @Override

  public boolean allowConstantFolding() {

    return false;

  }

  @Override

  public boolean useArgumentsInsteadObjects() {

    return false;

  }

  @Override

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

      String connectionName;

      String sql;

      JDBCServerExplorer explorer;

      if (args.length==1) {

          sql = this.getStr(args, 0);

          FeatureStore store = (FeatureStore) interpreter.getSymbolTable().function(DataManager.FUNCTION_CURRENT_STORE).call(interpreter,(Object[]) null);

          explorer = (JDBCServerExplorer) store.getExplorer();

      } else {

        connectionName = this.getStr(args, 0);

        sql = this.getStr(args, 1);

        if (this.getObject(args, 0) instanceof JDBCServerExplorer) {

            explorer = (JDBCServerExplorer) this.getObject(args, 0);

        } else if (this.getObject(args, 0) instanceof FeatureStore) {

            explorer = (JDBCServerExplorer) ((FeatureStore) this.getObject(args, 0)).getExplorer();

        } else {

            DataManager dalManager = DALLocator.getManager();

            DataServerExplorerPoolEntry poolEntry = dalManager.getDataServerExplorerPool().get(connectionName);

            DataServerExplorerParameters explorerParameters = poolEntry.getExplorerParameters();

            explorer = (JDBCServerExplorer) dalManager.openServerExplorer(explorerParameters.getProviderName(), explorerParameters);

        }

      }

      Object result = explorer.execute(sql);

      return result;

  }

}