Class: Baza::Driver::Mysql2

Inherits:
BaseSqlDriver show all
Defined in:
lib/baza/drivers/mysql2.rb

Defined Under Namespace

Classes: Column, Columns, Index, Indexes, Result, Table, Tables

Instance Attribute Summary collapse

Attributes inherited from BaseSqlDriver

#baza, #cols, #indexes, #sep_col, #sep_table, #sep_val, #tables

Class Method Summary collapse

Instance Method Summary collapse

Methods inherited from BaseSqlDriver

#esc_col

Constructor Details

#initialize(baza) ⇒ Mysql2

Returns a new instance of Mysql2.



34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
 
# File 'lib/baza/drivers/mysql2.rb', line 34

def initialize(baza)
  super

  @opts = @baza.opts

  require "monitor"
  @mutex = Monitor.new

  if @opts[:encoding]
    @encoding = @opts[:encoding]
  else
    @encoding = "utf8"
  end

  if @baza.opts.key?(:port)
    @port = @baza.opts[:port].to_i
  else
    @port = 3306
  end

  reconnect
end

Instance Attribute Details

#conn ⇒ Object (readonly)

Returns the value of attribute conn.



14
15
16
 
# File 'lib/baza/drivers/mysql2.rb', line 14

def conn
  @conn
end

#conns ⇒ Object (readonly)

Returns the value of attribute conns.



14
15
16
 
# File 'lib/baza/drivers/mysql2.rb', line 14

def conns
  @conns
end

Class Method Details

.from_object(args) ⇒ Object

Helper to enable automatic registering of database using Baza::Db.from_object



17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
 
# File 'lib/baza/drivers/mysql2.rb', line 17

def self.from_object(args)
  if args[:object].class.name == "Mysql2::Client"
    return {
      type: :success,
      args: {
        type: :mysql2,
        conn: args[:object],
        query_args: {
          symbolize_keys: true
        }
      }
    }
  end

  return nil
end

Instance Method Details

#clean ⇒ Object

Cleans the wref-map holding the tables.



58
59
60
 
# File 'lib/baza/drivers/mysql2.rb', line 58

def clean
  tables.clean if tables
end

#close ⇒ Object

Closes the connection threadsafe.



157
158
159
 
# File 'lib/baza/drivers/mysql2.rb', line 157

def close
  @mutex.synchronize { @conn.close }
end

#destroy ⇒ Object

Destroyes the connection.



162
163
164
165
166
167
168
169
 
# File 'lib/baza/drivers/mysql2.rb', line 162

def destroy
  @conn = nil
  @baza = nil
  @mutex = nil
  @encoding = nil
  @query_args = nil
  @port = nil
end

#escape(string) ⇒ Object

Escapes a string to be safe to use in a query.



147
148
149
 
# File 'lib/baza/drivers/mysql2.rb', line 147

def escape(string)
  return @conn.escape(string.to_s)
end

#insert_multi(tablename, arr_hashes, args = nil) ⇒ Object

Inserts multiple rows in a table. Can return the inserted IDs if asked to in arguments.



172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
 
# File 'lib/baza/drivers/mysql2.rb', line 172

def insert_multi(tablename, arr_hashes, args = nil)
  sql = "INSERT INTO `#{tablename}` ("

  first = true
  if args && args[:keys]
    keys = args[:keys]
  elsif arr_hashes.first.is_a?(Hash)
    keys = arr_hashes.first.keys
  else
    raise "Could not figure out keys."
  end

  keys.each do |col_name|
    sql << "," unless first
    first = false if first
    sql << "`#{self.esc_col(col_name)}`"
  end

  sql << ") VALUES ("

  first = true
  arr_hashes.each do |hash|
    if first
      first = false
    else
      sql << "),("
    end

    first_key = true
    if hash.is_a?(Array)
      hash.each do |val|
        if first_key
          first_key = false
        else
          sql << ","
        end

        sql << @baza.sqlval(val)
      end
    else
      hash.each do |key, val|
        if first_key
          first_key = false
        else
          sql << ","
        end

        sql << @baza.sqlval(val)
      end
    end
  end

  sql << ")"

  return sql if args && args[:return_sql]

  query(sql)

  if args && args[:return_id]
    first_id = self.last_id
    raise "Invalid ID: #{first_id}" if first_id.to_i <= 0
    ids = [first_id]
    1.upto(arr_hashes.length - 1) do |count|
      ids << first_id + count
    end

    ids_length = ids.length
    arr_hashes_length = arr_hashes.length
    raise "Invalid length (#{ids_length}, #{arr_hashes_length})." if ids_length != arr_hashes_length

    return ids
  else
    return nil
  end
end

#last_id ⇒ Object

Returns the last inserted ID for the connection.



152
153
154
 
# File 'lib/baza/drivers/mysql2.rb', line 152

def last_id
  @mutex.synchronize { return @conn.last_id.to_i }
end

#query(str) ⇒ Object

Executes a query and returns the result.



113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
 
# File 'lib/baza/drivers/mysql2.rb', line 113

def query(str)
  str = str.to_s
  str = str.force_encoding("UTF-8") if @encoding == "utf8" && str.respond_to?(:force_encoding)
  tries = 0

  begin
    tries += 1
    @mutex.synchronize do
      return Baza::Driver::Mysql2::Result.new(self, @conn.query(str, @query_args))
    end
  rescue => e
    if tries <= 3
      if e.message == "MySQL server has gone away" || e.message == "closed MySQL connection" or e.message == "Can't connect to local MySQL server through socket"
        sleep 0.5
        reconnect
        retry
      elsif e.message.include?("No operations allowed after connection closed") or e.message == "This connection is still waiting for a result, try again once you have the result" or e.message == "Lock wait timeout exceeded; try restarting transaction"
        reconnect
        retry
      end
    end

    raise e
  end
end

#query_ubuf(str) ⇒ Object

Executes an unbuffered query and returns the result that can be used to access the data.



140
141
142
143
144
 
# File 'lib/baza/drivers/mysql2.rb', line 140

def query_ubuf(str)
  @mutex.synchronize do
    return Baza::Driver::Mysql2::Result.new(self, @conn.query(str, @query_args.merge(stream: true)))
  end
end

#reconnect ⇒ Object

Respawns the connection to the MySQL-database.



63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
 
# File 'lib/baza/drivers/mysql2.rb', line 63

def reconnect
  @mutex.synchronize do
    args = {
      host: @baza.opts[:host],
      username: @baza.opts[:user],
      password: @baza.opts[:pass],
      database: @baza.opts[:db],
      port: @port,
      symbolize_keys: true,
      cache_rows: false
    }

    #Symbolize keys should also be given here, else table-data wont be symbolized for some reason - knj.
    @query_args = {symbolize_keys: true,}
    @query_args[:cast] = false unless @baza.opts[:type_translation]
    @query_args.merge!(@baza.opts[:query_args]) if @baza.opts[:query_args]

    pos_args = [:as, :async, :cast_booleans, :database_timezone, :application_timezone, :cache_rows, :connect_flags, :cast]
    pos_args.each do |key|
      args[key] = @baza.opts[key] if @baza.opts.key?(key)
    end

    args[:as] = :array

    tries = 0
    begin
      tries += 1
      if @baza.opts[:conn]
        @conn = @baza.opts[:conn]
      else
        require "mysql2"
        @conn = Mysql2::Client.new(args)
      end
    rescue => e
      if tries <= 3
        if e.message == "Can't connect to local MySQL server through socket '/var/run/mysqld/mysqld.sock' (111)"
          sleep 1
          tries += 1
          retry
        end
      end

      raise e
    end

    query("SET NAMES '#{esc(@encoding)}'") if @encoding
  end
end

#transaction ⇒ Object 



248
249
250
251
252
253
254
255
256
257
258
 
# File 'lib/baza/drivers/mysql2.rb', line 248

def transaction
  @baza.q("START TRANSACTION")

  begin
    yield @baza
    @baza.q("COMMIT")
  rescue
    @baza.q("ROLLBACK")
    raise
  end
end