Class: FluidDb2::TinyTds

Inherits:
Base
  • Object
show all
Defined in:
lib/fluiddb2/tiny_tds.rb

Instance Attribute Summary

Attributes inherited from Base

#connection, #verbose

Instance Method Summary collapse

Methods inherited from Base

#initialize, #reconnect, #verbose_log

Constructor Details

This class inherits a constructor from FluidDb2::Base

Instance Method Details

#begin ⇒ Object

Transaction Semantics



140
141
142
 
# File 'lib/fluiddb2/tiny_tds.rb', line 140

def begin
  @connection.execute('BEGIN TRANSACTION')
end

#close ⇒ Object 



70
71
72
 
# File 'lib/fluiddb2/tiny_tds.rb', line 70

def close
  @connection.close
end

#commit ⇒ Object

Transaction Semantics



145
146
147
 
# File 'lib/fluiddb2/tiny_tds.rb', line 145

def commit
  @connection.execute('COMMIT')
end

#connect ⇒ Object

Connect to Db.

Parameters:

  • uri (String) —

    a location for the resource to which we will attach, eg tinytds://<user>:<pass>@<dataserver>/<database>



12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
 
# File 'lib/fluiddb2/tiny_tds.rb', line 12

def connect
  uri = @uri

  dataserver = uri.host
  database = uri.path.sub('/', '')
  username = URI.unescape(uri.user)
  password = uri.password

  if dataserver == '' || database == ''
    fail '*** You need to specify both a dataserver and a database for ' \
          'the tinytds driver. Expected format: ' \
          'tinytds://<user>:<pass>@<dataserver>/<database>\n' \
          '*** The specified dataserver should have an entry in ' \
          '/etc/freetds/freetds.conf'
  end

  if username == '' || password == ''
    puts '*** Warning - you will normally need to specify both a username ' \
         'and password for the tinytds driver to work correctly.'
  end

  hash = Hash[:username, username,
              :password, password,
              :database, database,
              :dataserver, dataserver]
  fail_over_data_server = nil
  unless uri.query.nil?
    cgi = CGI.parse(uri.query)
    hash[:timeout] = cgi['timeout'][0].to_i if cgi.key?('timeout')
    if cgi.key?('host')
      hash.delete(:dataserver)
      hash[:host] = dataserver
    end

    fail_over_data_server =
      cgi['failoverdataserver'][0] if cgi.key?('failoverdataserver')
  end

  begin
    @connection = ::TinyTds::Client.new(hash)
  rescue ::TinyTds::Error => e
    # Message for an incorrect password,
    # Login failed. The login is from an untrusted domain and cannot be used
    #  with Windows authentication.
    # Message for unavailable db
    # Cannot open user default database. Login failed.
    if e.message == 'Cannot open user default database. Login failed.' &&
       !fail_over_data_server.nil?
      hash[:dataserver] = fail_over_data_server
      @connection = ::TinyTds::Client.new(hash)
    else
      raise e
    end
  end

  fail 'Unable to connect to the database' unless @connection.active?
end

#escape_string(input) ⇒ Object 



74
75
76
 
# File 'lib/fluiddb2/tiny_tds.rb', line 74

def escape_string(input)
  @connection.escape(input)
end

#execute(sql, params = [], expected_affected_rows = nil) ⇒ Object 



122
123
124
125
126
127
128
129
130
131
132
133
 
# File 'lib/fluiddb2/tiny_tds.rb', line 122

def execute(sql, params = [], expected_affected_rows = nil)
  sql = FluidDb2.format_to_sql(sql, params)
  r = @connection.execute(sql)
  r.each

  if !expected_affected_rows.nil? &&
     r.affected_rows != expected_affected_rows
    msg = "Expected affected rows, #{expected_affected_rows}, " \
          "Actual affected rows, #{r.affected_rows}"
    fail ExpectedAffectedRowsError, msg
  end
end

#insert(_sql, _params) ⇒ Object 



135
136
137
 
# File 'lib/fluiddb2/tiny_tds.rb', line 135

def insert(_sql, _params)
  fail 'Not implemented'
end

#query_for_array(sql, params = []) ⇒ Object 



78
79
80
81
82
83
84
85
86
87
88
89
90
91
 
# File 'lib/fluiddb2/tiny_tds.rb', line 78

def query_for_array(sql, params = [])
  sql = FluidDb2.format_to_sql(sql, params)
  results = @connection.execute(sql)

  count = 0
  tuple = ''
  results.each do |row|
    count += 1
    fail FluidDb2::TooManyRowsError if count > 1
    tuple = row
  end
  fail FluidDb2::NoDataFoundError if count == 0
  tuple
end

#query_for_resultset(sql, params = []) ⇒ Object 



110
111
112
113
114
115
116
117
118
119
120
 
# File 'lib/fluiddb2/tiny_tds.rb', line 110

def query_for_resultset(sql, params = [])
  sql = FluidDb2.format_to_sql(sql, params)
  results = @connection.execute(sql)

  list = []
  results.each do |row|
    list << row
  end

  list
end

#query_for_value(sql, params = []) ⇒ Object 



93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
 
# File 'lib/fluiddb2/tiny_tds.rb', line 93

def query_for_value(sql, params = [])
  sql = FluidDb2.format_to_sql(sql, params)
  results = @connection.execute(sql)

  count = 0
  value = ''
  results.each do |row|
    count += 1
    fail FluidDb2::TooManyRowsError if count > 1
    value = row[results.fields[0]]
  end

  fail FluidDb2::NoDataFoundError if count == 0

  value
end

#rollback ⇒ Object

Transaction Semantics



150
151
152
 
# File 'lib/fluiddb2/tiny_tds.rb', line 150

def rollback
  @connection.execute('ROLLBACK')
end