Class: Fluent::RedshiftOutput::RedshiftConnection

Inherits:
Object
  • Object
show all
Defined in:
lib/fluent/plugin/out_redshift.rb

Constant Summary collapse

REDSHIFT_CONNECT_TIMEOUT =

10sec

10.0

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(db_conf) ⇒ RedshiftConnection

Returns a new instance of RedshiftConnection.



373
374
375
376
 
# File 'lib/fluent/plugin/out_redshift.rb', line 373

def initialize(db_conf)
  @db_conf = db_conf
  @connection = nil
end

Instance Attribute Details

#db_conf ⇒ Object (readonly)

Returns the value of attribute db_conf.



378
379
380
 
# File 'lib/fluent/plugin/out_redshift.rb', line 378

def db_conf
  @db_conf
end

Instance Method Details

#close ⇒ Object 



406
407
408
409
 
# File 'lib/fluent/plugin/out_redshift.rb', line 406

def close
  @connection.close rescue nil if @connection
  @connection = nil
end

#connect_start ⇒ Object 



402
403
404
 
# File 'lib/fluent/plugin/out_redshift.rb', line 402

def connect_start
  @connection = create_redshift_connection
end

#exec(sql, &block) ⇒ Object 



388
389
390
391
392
393
394
395
396
397
398
399
400
 
# File 'lib/fluent/plugin/out_redshift.rb', line 388

def exec(sql, &block)
  conn = @connection
  conn = create_redshift_connection if conn.nil?
  if block
    conn.exec(sql) {|result| block.call(result)}
  else
    conn.exec(sql)
  end
rescue PG::Error => e
  raise RedshiftError.new(e)
ensure
  conn.close if conn && @connection.nil?
end

#fetch_table_columns(table_name, schema_name) ⇒ Object 



380
381
382
383
384
385
386
 
# File 'lib/fluent/plugin/out_redshift.rb', line 380

def fetch_table_columns(table_name, schema_name)
  columns = nil
  exec(fetch_columns_sql(table_name, schema_name)) do |result|
    columns = result.collect{|row| row['column_name']}
  end
  columns
end