Class DB2::Statement < Object

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 101
def initialize(connection)
  @conn = connection
  @handle_type = SQL_HANDLE_STMT
  @parms = []                           #yun
  @sql = ''                             #yun
  @numParms = 0                         #yun
  @prepared = false                     #yun
  @parmArray = []                       #yun. attributes of the parameter markers
  rc, @handle = SQLAllocHandle(@handle_type, @conn.handle)
  check_rc(rc)
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 340
def adjust_content(c)
  case c.class.to_s
  when 'DB2CLI::NullClass'
    return nil
  when 'DB2CLI::Time'
    "%02d:%02d:%02d" % [c.hour, c.minute, c.second]
  when 'DB2CLI::Date'
    "%04d-%02d-%02d" % [c.year, c.month, c.day]
  when 'DB2CLI::Timestamp'
    "%04d-%02d-%02d %02d:%02d:%02d" % [c.year, c.month, c.day, c.hour, c.minute, c.second]
  else
    return c
  end
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 219
def bind(sql, args)                #does not use SQLBindParams
  arg_index = 0
  result = ""
  tokens(sql).each do |part|
    case part
    when '?'
      result << "'" + (args[arg_index]) + "'"  #put it into quotes
      arg_index += 1
    when '??'
      result << "?"
    else
      result << part
    end
  end
  if arg_index < args.size
    raise "Too many SQL parameters"
  elsif arg_index > args.size
    raise "Not enough SQL parameters"
  end
  result
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 199
def bindParms(parms)        # This is the real thing. It uses SQLBindParms
  1.upto(@numParms) do |i|  # parameter number starts from 1
    rc, dataType, parmSize, decimalDigits = SQLDescribeParam(@handle, i)
    check_rc(rc)
    if parms[i - 1].class == String
      value = parms[i - 1]
    else
      value = parms[i - 1].to_s
    end
    rc = SQLBindParameter(@handle, i, dataType, parmSize, decimalDigits, value)
    check_rc(rc)
  end
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 113
def columns(table_name, schema_name = '%')
  check_rc(SQLColumns(@handle, '', schema_name.upcase, table_name.upcase, '%'))
  fetch_all
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 254
def exec_direct(sql)
  check_rc(SQLExecDirect(@handle, sql))
  self
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 148
def execute(*parms)
  raise "The statement was not prepared" if @prepared == false

  if parms.size == 1 and parms[0].class == Array
    parms = parms[0]
  end

  if @numParms != parms.size
    raise "Number of parameters supplied does not match with the SQL statement"
  end

  if @numParms > 0            #need to bind parameters
    #--------------------------------------------------------------------
    #calling bindParms may not be safe. Look comment below.
    #--------------------------------------------------------------------
    #bindParms(parms)

    valueArray = []
    1.upto(@numParms) do |i|  # parameter number starts from 1
      type = @parmArray[i - 1].class
      size = @parmArray[i - 1].size
      decimalDigits = @parmArray[i - 1].decimalDigits

      if parms[i - 1].class == String
        valueArray << parms[i - 1]
      else
        valueArray << parms[i - 1].to_s
      end

      rc = SQLBindParameter(@handle, i, type, size, decimalDigits, valueArray[i - 1])
      check_rc(rc)
    end
  end

  check_rc(SQLExecute(@handle))

  if @numParms != 0
    check_rc(SQLFreeStmt(@handle, SQL_RESET_PARAMS)) # Reset parameters
  end

  self
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 296
def fetch
  cols = get_col_desc
  rc = SQLFetch(@handle)
  if rc == SQL_NO_DATA_FOUND
    SQLFreeStmt(@handle, SQL_CLOSE)        # Close cursor
    SQLFreeStmt(@handle, SQL_RESET_PARAMS) # Reset parameters
    return nil
  end
  raise "ERROR" unless rc == SQL_SUCCESS

  retval = []
  cols.each_with_index do |c, i|
    rc, content = SQLGetData(@handle, i + 1, c[1], c[2] + 1) #yun added 1 to c[2]
    retval << adjust_content(content)
  end
  retval
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 282
def fetch_all
  if block_given?
    while row = fetch do
      yield row
    end
  else
    res = []
    while row = fetch do
      res << row
    end
    res
  end
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 314
def fetch_as_hash
  cols = get_col_desc
  rc = SQLFetch(@handle)
  if rc == SQL_NO_DATA_FOUND
    SQLFreeStmt(@handle, SQL_CLOSE)        # Close cursor
    SQLFreeStmt(@handle, SQL_RESET_PARAMS) # Reset parameters
    return nil
  end
  raise "ERROR" unless rc == SQL_SUCCESS

  retval = {}
  cols.each_with_index do |c, i|
    rc, content = SQLGetData(@handle, i + 1, c[1], c[2] + 1)   #yun added 1 to c[2]
    retval[c[0]] = adjust_content(content)
  end
  retval
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 332
def get_col_desc
  rc, nr_cols = SQLNumResultCols(@handle)
  cols = (1..nr_cols).collect do |c|
    rc, name, bl, type, col_sz = SQLDescribeCol(@handle, c, 1024)
    [name.downcase, type, col_sz]
  end
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 264
def get_cursor_name
  rc, name = SQLGetCursorName(@handle)
  check_rc(rc)
  name
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 123
def indexes(table_name, schema_name = '')
  check_rc(SQLStatistics(@handle, '', schema_name.upcase, table_name.upcase, SQL_INDEX_ALL, SQL_ENSURE))
  fetch_all
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 276
def num_result_cols
  rc, cols = SQLNumResultCols(@handle)
  check_rc(rc)
  cols
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 128
def prepare(sql)
  @sql = sql
  check_rc(SQLPrepare(@handle, sql))
  rc, @numParms = SQLNumParams(@handle) #number of question marks
  check_rc(rc)
  #--------------------------------------------------------------------------
  # parameter attributes are stored in instance variable @parmArray so that
  # they are available when execute method is called.
  #--------------------------------------------------------------------------
  if @numParms > 0           # get parameter marker attributes
    1.upto(@numParms) do |i| # parameter number starts from 1
      rc, type, size, decimalDigits = SQLDescribeParam(@handle, i)
      check_rc(rc)
      @parmArray << Parameter.new(type, size, decimalDigits)
    end
  end
  @prepared = true
  self
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 270
def row_count
  rc, rowcount = SQLRowCount(@handle)
  check_rc(rc)
  rowcount
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 259
def set_cursor_name(name)
  check_rc(SQLSetCursorName(@handle, name))
  self
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 118
def tables(schema_name = '%')
  check_rc(SQLTables(@handle, '', schema_name.upcase, '%', 'TABLE'))
  fetch_all
end

# File rails/activerecord/lib/active_record/vendor/db2.rb, line 249
def tokens(sql)
  toks = sql.scan(%r('([^'\\]|''|\\.)*'|"([^"\\]|""|\\.)*"|\?\??|[^'"?]+)/)
  toks.collect { |t| t[0] }
end