node-oracledb/test/sql.js

/* Copyright (c) 2017, Oracle and/or its affiliates. All rights reserved. */

/******************************************************************************
 *
 * You may not use the identified files except in compliance with the Apache
 * License, Version 2.0 (the "License.")
 *
 * You may obtain a copy of the License at
 * http://www.apache.org/licenses/LICENSE-2.0.
 *
 * 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.
 *
 * The node-oracledb test suite uses 'mocha', 'should' and 'async'.
 * See LICENSE.md for relevant licenses.
 *
 * NAME
 *   sql.js
 *
 * DESCRIPTION
 *   generate sql
 *****************************************************************************/
'use strict';
var should   = require('should');
var async    = require('async');

var sql = exports;
module.exports = sql;

sql.createTable = function(tableName, dataType) {
  var element = dataType;
  if(dataType === "CHAR") {
    element = element + "(1000)";
  }
  if(dataType === "NCHAR") {
    element = element + "(1000)";
  }
  if(dataType === "VARCHAR2") {
    element = element + "(1000)";
  }
  if(dataType === "RAW") {
    element = element + "(1000)";
  }

  var sql = "BEGIN \n" +
            "    DECLARE \n" +
            "        e_table_missing EXCEPTION; \n" +
            "        PRAGMA EXCEPTION_INIT(e_table_missing, -00942);\n" +
            "    BEGIN \n" +
            "        EXECUTE IMMEDIATE ('DROP TABLE " + tableName + " PURGE' ); \n" +
            "    EXCEPTION \n" +
            "        WHEN e_table_missing \n" +
            "        THEN NULL; \n" +
            "    END; \n" +
            "    EXECUTE IMMEDIATE ( ' \n" +
            "        CREATE TABLE " + tableName + " ( \n" +
            "            id        NUMBER, \n" +
            "            content   " + element + " \n" +
            "        ) \n" +
            "    '); \n" +
            "END;  ";
  return sql;
};

sql.createAllTable = function(tableName, dataTypeArray) {
  var sql = "BEGIN \n" +
            "    DECLARE \n" +
            "        e_table_missing EXCEPTION; \n" +
            "        PRAGMA EXCEPTION_INIT(e_table_missing, -00942);\n" +
            "    BEGIN \n" +
            "        EXECUTE IMMEDIATE ('DROP TABLE " + tableName + " PURGE' ); \n" +
            "    EXCEPTION \n" +
            "        WHEN e_table_missing \n" +
            "        THEN NULL; \n" +
            "    END; \n" +
            "    EXECUTE IMMEDIATE ( ' \n" +
            "        CREATE TABLE " + tableName + " ( \n" +
            "            id    NUMBER, \n" ;

  async.forEach(dataTypeArray, function(element, cb) {
    var index = dataTypeArray.indexOf(element);
    var length = dataTypeArray.length;
    var col_name = "col_" + ( index + 1 );
    var col_type = element;
    var isLast;

    if(col_type === "CHAR") {
      element = element + "(2000)";
    }
    if(col_type === "NCHAR") {
      element = element + "(1000)";
    }
    if(col_type === "VARCHAR2") {
      element = element + "(4000)";
    }
    if(col_type === "RAW") {
      element = element + "(2000)";
    }
    isLast = (index == (length - 1)) ? true : false;
    sql = appendSql(sql, col_name, element, isLast);
    // cb();
  }, function(err) {
    should.not.exist(err);
  });
  sql = sql +
        "        ) \n" +
        "    '); \n" +
        "END;  ";
  return sql;
};

sql.executeSql = function(connection, sql) {
  connection.execute(
    sql,
    function(err) {
      should.not.exist(err);
    }
  );
};
var appendSql = function(sql, col_name, col_type, isLast) {
  if(isLast === true) {
    sql = sql + "            " + col_name + " " + col_type + " \n";
  } else {
    sql = sql + "            " +col_name + " " + col_type + ", \n";
  }
  return sql;
};