Class: Sequel::DB2::Database

Inherits:
Sequel::Database show all
Includes:
DB2CLI
Defined in:
lib/sequel_core/adapters/db2.rb

Constant Summary

Constants inherited from Sequel::Database

Sequel::Database::ADAPTERS, Sequel::Database::SQL_BEGIN, Sequel::Database::SQL_COMMIT, Sequel::Database::SQL_ROLLBACK

Constants included from Schema::SQL

Schema::SQL::AUTOINCREMENT, Schema::SQL::CASCADE, Schema::SQL::COMMA_SEPARATOR, Schema::SQL::NOT_NULL, Schema::SQL::NO_ACTION, Schema::SQL::NULL, Schema::SQL::PRIMARY_KEY, Schema::SQL::RESTRICT, Schema::SQL::SET_DEFAULT, Schema::SQL::SET_NULL, Schema::SQL::TYPES, Schema::SQL::UNDERSCORE, Schema::SQL::UNIQUE, Schema::SQL::UNSIGNED

Instance Attribute Summary

Attributes inherited from Sequel::Database

#loggers, #opts, #pool, #quote_identifiers

Instance Method Summary collapse

Methods inherited from Sequel::Database

#<<, #[], adapter_class, adapter_scheme, #add_column, #add_index, #alter_table, connect, #create_or_replace_view, #create_table, #create_table!, #create_view, #drop_column, #drop_index, #drop_table, #drop_view, #fetch, #from, #get, #initialize, #inspect, #log_info, #logger, #logger=, #multi_threaded?, #query, quote_identifiers=, #quote_identifiers?, #rename_column, #rename_table, #select, #serial_primary_key_options, #set_column_default, #set_column_type, single_threaded=, #single_threaded?, #synchronize, #table_exists?, #transaction, #typecast_value, #uri, uri_to_options

Methods included from Schema::SQL

#alter_table_sql, #alter_table_sql_list, #auto_increment_sql, #column_definition_sql, #column_list_sql, #constraint_definition_sql, #create_table_sql_list, #default_index_name, #drop_table_sql, #filter_expr, #index_definition_sql, #index_list_sql_list, #literal, #on_delete_clause, #quote_identifier, #rename_table_sql, #schema, #schema_utility_dataset, #type_literal

Constructor Details

This class inherits a constructor from Sequel::Database

Instance Method Details

#check_error(rc, msg) ⇒ Object

AUTO_INCREMENT = ‘IDENTITY(1,1)’.freeze

def auto_increment_sql

AUTO_INCREMENT

end



15
16
17
18
19
20
21
22
# File 'lib/sequel_core/adapters/db2.rb', line 15

def check_error(rc, msg)
  case rc
  when SQL_SUCCESS, SQL_SUCCESS_WITH_INFO
    nil
  else
    raise Error, msg
  end
end

#connectObject



27
28
29
30
31
32
33
34
35
# File 'lib/sequel_core/adapters/db2.rb', line 27

def connect
  rc, dbc = SQLAllocHandle(SQL_HANDLE_DBC, @@env) 
  check_error(rc, "Could not allocate database connection")
  
  rc = SQLConnect(dbc, @opts[:database], @opts[:user], @opts[:password]) 
  check_error(rc, "Could not connect to database")
  
  dbc
end

#dataset(opts = nil) ⇒ Object



52
53
54
# File 'lib/sequel_core/adapters/db2.rb', line 52

def dataset(opts = nil)
  DB2::Dataset.new(self, opts)
end

#disconnectObject



37
38
39
40
41
42
43
44
45
# File 'lib/sequel_core/adapters/db2.rb', line 37

def disconnect
  @pool.disconnect do |conn|
    rc = SQLDisconnect(conn)
    check_error(rc, "Could not disconnect from database")

    rc = SQLFreeHandle(SQL_HANDLE_DBC, conn)
    check_error(rc, "Could not free Database handle")
  end
end

#execute(sql, &block) ⇒ Object Also known as: do



56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
# File 'lib/sequel_core/adapters/db2.rb', line 56

def execute(sql, &block)
  log_info(sql)
  @pool.hold do |conn|
    rc, sth = SQLAllocHandle(SQL_HANDLE_STMT, @handle) 
    check_error(rc, "Could not allocate statement")

    begin
      rc = SQLExecDirect(sth, sql) 
      check_error(rc, "Could not execute statement")
      
      block[sth] if block

      rc, rpc = SQLRowCount(sth)
      check_error(rc, "Could not get RPC") 
      rpc
    ensure
      rc = SQLFreeHandle(SQL_HANDLE_STMT, sth)
      check_error(rc, "Could not free statement")
    end
  end
end

#test_connectionObject



47
48
49
50
# File 'lib/sequel_core/adapters/db2.rb', line 47

def test_connection
  @pool.hold {|conn|}
  true
end