Ruby使用Mysql2連接操作MySQL

更新時間：2022年04月18日 15:44:14 作者：駿馬金龍

這篇文章主要介紹了Ruby使用Mysql2連接操作MySQL,需要的朋友可以參考下

Ruby操作MySQL

使用mysql2連接mysql并操作mysql。相關文檔參見：

https://www.rubydoc.info/gems/mysql2/0.5.2
https://github.com/brianmario/mysql2

gem install mysql2

連接mysql

建立連接：

require 'mysql2'

conn = Mysql2::Client.new({ 
  host: '192.168.200.73',
  username: 'root',
  password: 'P@ssword1!'
})

接受的連接選項包括：

Mysql2::Client.new(
  :host,
  :username,
  :password,
  :port,
  :database,
  :socket = '/path/to/mysql.sock',
  :flags = REMEMBER_OPTIONS | LONG_PASSWORD | LONG_FLAG | TRANSACTIONS | PROTOCOL_41 | SECURE_CONNECTION | MULTI_STATEMENTS,
  :encoding = 'utf8',
  :read_timeout = seconds,
  :write_timeout = seconds,
  :connect_timeout = seconds,
  :connect_attrs = {:program_name => $PROGRAM_NAME, ...},
  :reconnect = true/false,
  :local_infile = true/false,
  :secure_auth = true/false,
  :ssl_mode = :disabled / :preferred / :required / :verify_ca / :verify_identity,
  :default_file = '$HOME/.my.cnf',   #=> 從文件讀取連接信息
  :default_group = 'my.cfg section', #=> 選擇.my.cnf中的section
  :default_auth = 'authentication_windows_client',
  :init_command => SQL_Statement  #=>主要用于設置本次連接時的某些變量
)

連接建立后就可以操作數據庫了，比如執(zhí)行SQL語句：

conn. query('create databse mytest')
conn.select_ db( 'mytest' )
# conn.query('use mytest' )
conn. query('create table tb(name varchar(1024),age int)')
conn. query <<-SQL
    insert into tb values 
    ('junmajinlong', 23), 
    ( 'woniu', 25),
    ('fairy', 26)
SQL

如果測試和mysql的連接是否斷開，可執(zhí)行ping()：

conn.ping

如果連接未斷開，ping()返回true，如果連接已斷開但已啟用auto-reconnect，則ping()會嘗試依次reconnect，連接成功則返回true，否則報錯。如果連接已斷開，且未啟用auto-reconnect，則報錯。

query()查詢和結果處理

query()用于執(zhí)行任何允許的SQL語句，比如執(zhí)行查詢語句。

查詢結果可使用each進行迭代，迭代時傳遞查詢到的每一行記錄，可使用hash索引的方式(默認以hash類型保存每一行)查詢某個字段的內容：

conn.query("show databases").each do |row| pp row end
=begin
{"Database"=>"information_schema"}
{"Database"=>"mysql"}
{"Database"=>"mytest"}
{"Database"=>"performance_schema"}
{"Database"=>"sys"}
=end

conn.query("select * from mytest.tb").each do |row|
  pp row
  pp row["name"]
end
=begin
{"name"=>"junmajinlong", "age"=>23}
"junmajinlong"
{"name"=>"woniu", "age"=>25}
"woniu"
{"name"=>"fairy", "age"=>26}
"fairy"
=end

可見，查詢結果中，每一行數據默認以hash格式保存。

實際上，對于增刪改的SQL語句，query()的返回值為nil，對于查詢類的語句，其返回值以Mysql2::Result對象返回

conn.query("create table mytest.t1(id int)")
#=> nil
res = conn.query("select * from mytest.tb")
p res

結果：

#<Mysql2::Result:0x00007fffe833a230
 @query_options=
  {:as=>:hash,
   :async=>false,
   :cast_booleans=>false,
   :symbolize_keys=>false,
   :database_timezone=>:local,
   :application_timezone=>nil,
   :cache_rows=>true,
   :connect_flags=>2148540933,
   :cast=>true,
   :default_file=>nil,
   :default_group=>nil,
   :host=>"192.168.200.73",
   :username=>"root",
   :password=>"P@ssword1!"},
 @server_flags=
   {:no_good_index_used=>false, 
     :no_index_used=>true, 
     :query_was_slow=>false}>

query()各查詢選項的含義以及默認的查詢選項參見下文。先了解兩個：

:as項表示以數組方式(:as=>:array)還是hash方式(:as=>:hash)存儲查詢結果
:symbolize_keys表示返回hash結果時，其key是否設置為符號類型，默認為false，即key是字符串類型

可在查詢時指定這些參數，也可在each迭代時指定這些參數，例如：

conn.query("select * from mytest.tb", symbolize_keys: true).each do |row|
  pp row
end

conn.query("select * from mytest.tb").each(symbolize_keys: true) do |row|
  pp row
end

結果：

{:name=>"junmajinlong", :age=>23}
{:name=>"woniu", :age=>25}
{:name=>"fairy", :age=>26}

雖然大多數時候使用hash保存每一行數據更方便，但有時候也會想要以數組方式操作每一行數據，例如連接每一個字段的值：

sql = 'select * from mytest.tb'
res = conn.query(sql)
res.each(as: :array) do |row|
  p row.join(",")
end

結果：

"junmajinlong,23"
"woniu,25"
"fairy,26"

Mysql2::Result自身具備的方法不多：

#count ? Object (also: #size)
#each(*args) ? Object
#fields ? Object
#free ? Object

外，Mysql2::Result已經mix-in了Enumerable模塊，所以可直接使用該模塊中的方法，例如first表示取第一行記錄。

sql = 'select * from mytest.tb'
res = conn.query(sql)
res.first
#=>{"name"=>"junmajinlong", "age"=>23}

需注意，所有SQL語句中涉及到的值都是未經轉義的，有時候需要也建議在執(zhí)行它們之前先對它們進行轉義。

escaped = conn.escape("gi'thu\"bbe\0r's")
results = conn.query("SELECT * FROM users WHERE group='#{escaped}'")

查詢選項含義以及默認查詢選項

query()默認的查詢選項可以通過Mysql2::Client.default_query_options獲取，它是一個hash結果：

Mysql2::Client.default_query_options
=begin
{:as=>:hash,
 :async=>false,
 :cast_booleans=>false,
 :symbolize_keys=>false,
 :database_timezone=>:local,
 :application_timezone=>nil,
 :cache_rows=>true,
 :connect_flags=>2148540933,
 :cast=>true,
 :default_file=>nil,
 :default_group=>nil}
=end

其中(重要)：

:as項表示以數組方式(:as=>:array)還是hash方式(:as=>:hash)存儲查詢結果
:symbolize_keys表示返回hash結果時，其key是否設置為符號類型，默認為false，即key是字符串類型
:async表示查詢是否異步模式，即是否非阻塞的查詢，參考https://github.com/brianmario/mysql2#async
:cast指示MySQL的查詢結果轉換為Ruby數據時是否進行類型轉換，如果確定本次查詢的字段類型和Ruby的類型完全對應，可禁用casting功能提升效率
:database_timezone指示Ruby接收MySQL返回的日期時間數據時的時區(qū)，mysql2將先以該時區(qū)創(chuàng)建日期時間對象來保存對應字段的值。僅支持:local和:utc兩個值
:application_timezone指示最終Mysql2::Result中的日期時間的時區(qū)，即程序端的時區(qū)。因此，mysql2先以”無損”的時區(qū)從MySQL獲取日期時間數據，并根據application_timezone將其轉換成程序端時區(qū)的日期時間對象
:cache_rows指示是否緩存構建出來的hash行或array行
Mysql2處理查詢結果的流程：
- Mysql2的MySQL C api從MySQL服務端查詢數據，并保存在Ruby的查詢結果集(結果集屬于C)
- Mysql2::Result和C端結果集是關聯的，當釋放Mysql2::Result，也會對C結果集進行GC
- Mysql2在需要取得結果集中的數據時(比如each迭代)，才從結果集中根據查詢選項構建所需行并返回，比如構建hash結構的行，構建數組結構的行，構建hash結構時將key轉換為Symbol類型等
- 默認情況下，從結果集中查詢并構建出來的hash行或array行會緩存在Ruby中，使得下次再次請求這一行時(比如再次迭代)，可用直接從緩存中取得hash行或array行
- 比如從MySQL服務端查詢了100行數據保存在C的結果集中，第一次以hash方式請求其中4行，這4行hash數據會緩存起來，如果下次再從頭開始以hash方式請求15行，則前4行來自于緩存，后11行來自于結果集的臨時構建
- 如果:cache_rows未禁用，當結果集中的所有行都被緩存，Mysql2::Result將會去釋放C端的結果集
- 如果能確保查詢的結果集只使用一次，可禁用:cache_rows，這會提升效率
:stream: true表示以Stream的方式處理查詢結果。有時候查詢結果數據量非常大，Ruby端不方便存放所有結果，可采用stream的方式去處理本次查詢：一邊從MySQL服務端取數據放進結果集，一邊從結果集中取數據進行處理(比如迭代)。使用stream時，會自動關閉cache_rows，因為它們是互相沖突的概念。此外，使用stream模式要求必須迭代完所有數據集才會執(zhí)行下一條查詢，因為一個mysql連接在某一時刻只能執(zhí)行一個操作，在迭代完之前，本次查詢操作還尚未完成。

修改Mysql2::Client.default_query_options可以設置默認query()的查詢選項。如果想要設置其中某選項，可以通過hash合并的方式來設置該選項。

Mysql2::Client.default_query_options
#=> {:as=>:hash, ...}

Mysql2::Client.default_query_options.merge!(:as => :array)
#=> {:as=>:array, ...}

prepare()+execute()

除了直接使用query()執(zhí)行SQL語句查詢數據庫，也可以使用prepare()方法將字符串準備成一個待執(zhí)行的SQL語句，其中可以使用?充當占位符。

prepare后的語句是一個Mysql2::Statement對象，該對象有一個execute()方法，可以用來執(zhí)行這個準備好的語句，它可指定查詢選項，且其返回值同query()一樣：對于增刪改操作，返回值為nil，對于查詢類操作，返回Mysql2::Result結果對象。

res_sql = conn.prepare('select * from mytest.tb where age >= ? and name like ?')
res = res_sql.execute(20, '%junma%', as: :array)
res.first

處理多結果集

有些存儲過程中可能包含多個查詢結果集，或者有時候會在一行SQL中包含多個select語句而同時返回多個結果集，Mysql2能很好地處理多結果集問題。

要處理多結果集，連接mysql時必須指定一個flag：

conn = Mysql2::Client.new({
  host: "192.168.200.73",
  username: "root",
  password: "P@ssword1!",
  flags: Mysql2::Client::MULTI_STATEMENTS
})

然后執(zhí)行多結果集的多個查詢語句：

res = conn.query('select 1;select 2;select 3')

雖然本次query()涉及了多個select語句，Mysql2也已經保存了這三個select的查詢結果集(保存在結果集隊列中)，但本次query()方法的返回值僅是第一個結果集，所以res中保存的是第一個結果集的內容。

res.first   #=> {"1"=>1}

要獲取剩余的結果集，可通過conn.next_result將結果集偏移指針移到下一個結果集，然后通過conn.store_result獲取下一個結果集，依次類推，直到沒有剩余結果集后，conn.next_result返回false。可通過more_results?()方法判斷是否還有剩余的結果集。

conn.next_result
res = conn.store_result
res.first #=> {"2"=>2}

conn.next_result
res = conn.store_result
res.first #=> {"3"=>3}

conn.next_result  #=> false

所以，可遍歷多個結果集：

res = conn.query('select 1;select 2;select 3')
loop do
  p res.first
  break unless conn.next_result
  res = conn.store_result
end

# 或者
p res.first
while conn.next_result
  res = conn.store_result
  p res.first
end

輸出結果：

{"1"=>1}
{"2"=>2}
{"3"=>3}

需注意，開啟多行語句(即多結果集)功能后，所查詢得到的所有結果集必須已經處理完成(嚴格來說，是存放結果集的隊列已經為空)，才能繼續(xù)執(zhí)行后續(xù)的SQL語句(事實上，經測試，結果集隊列未空的情況下執(zhí)行其它SQL語句會導致直接斷開mysql連接)?？墒褂?code>abandon_results!()方法強行丟棄所有剩余結果集，使得Mysql2馬上回歸正常狀態(tài)：可向MySQL服務端發(fā)送SQL語句。

# res和res1都只保存第一個結果集
# 但結果集隊列中保留的是select 5和select 6的結果集
res = conn.query('select 1;select 2;select 3')
conn.abandon_results!  # 丟棄所有剩余結果集
res1 = conn.query('select 4;select 5;select 6')

另外，如果多個查詢語句中間的某個查詢語句報錯，它將影響其后面的語句不會執(zhí)行，所以無法獲取后面的結果集。

res = conn.query('select 1;select 2;select A;select 3')
loop do
  p res.first
  break unless conn.next_result
  res = conn.store_result
end

結果：

{"1"=>1}
{"2"=>2}
Mysql2::Error: Unknown column 'A' in 'field list'

Mysql2的EventMachine

Mysql2支持EM，可以執(zhí)行異步的query()，此外，可以指定當query()查詢成功或失敗時的回調語句塊：

require 'mysql2/em'

EM.run do
  client1 = Mysql2::EM::Client.new
  defer1 = client1.query "SELECT sleep(3) as first_query"
  defer1.callback do |result|
    puts "Result: #{result.to_a.inspect}"
  end

  client2 = Mysql2::EM::Client.new
  defer2 = client2.query "SELECT sleep(1) second_query"
  defer2.callback do |result|
    puts "Result: #{result.to_a.inspect}"
  end
end

ORM之：Sequel

Active:Record應該是最為人熟知的orm，其功能極其豐富。

另一個輕量級的ORM是Sequel，它支持ADO, JDBC, MySQL, Mysql2, ODBC, Oracle, PostgreSQL, SQLite3等等。

Sequel官方手冊：

https://github.com/jeremyevans/sequel
http://sequel.jeremyevans.net/rdoc/

例如：

require 'sequel'

# 創(chuàng)建數據庫實例
DB = Sequel.connect(
  adapter: :mysql2,
  user: 'root',
  password: 'P@ssword1!',
  host: '192.168.200.73',
  port: 3306,
  database: 'mytest'
)

# 創(chuàng)建數據集，數據集表示的是一張表或表部分數據
# 此時不會去查詢數據，會推遲到需要數據時才查詢
dataset = DB[:tb]

# 迭代表數據
dataset.each do |row|
  pp row
end

# 條件查詢
pp dataset.where(name: 'junmajinlong', age: 23).first
pp dataset.where { name =~ "junmajinlong" and age =~ 23 }.first

您可能感興趣的文章:

Ruby
MySQL

Rails命令行常用操作命令簡明總結
這篇文章主要介紹了Rails命令行常用操作命令簡明總結,只列出了一些最常用的,總結的比較簡潔,需要的朋友參考下吧
2014-06-06
Ruby on Rails網站項目構建簡單指南
Rails項目通過Ruby世界中的gem和rake工具來構建起來真的相當方便,這里就給大家整理了一份Ruby on Rails網站項目構建簡單指南,需要的朋友可以參考下
2016-06-06
Ruby元編程的一些值得注意的地方
這篇文章主要介紹了Ruby元編程的一些值得注意的地方,作者自己用實際代碼給出了相關示例,需要的朋友可以參考下
2015-08-08
詳解Ruby當中的算數運算
這篇文章主要介紹了詳解Ruby當中的算數運算,是Ruby學習當中的基礎知識,需要的朋友可以參考下
2015-05-05
淺析Ruby中的Profiling工具的用法
這篇文章主要介紹了淺析Ruby中的Profiling工具的用法,作者也給出了工具相關開源包的一些下載來源和使用方法,需要的朋友可以參考下
2015-05-05
rudy 方法分析
rudy 方法分析...
2007-11-11
Ruby的語法和語言特性總結
這篇文章主要介紹了Ruby的語法和語言特性總結,包括代碼塊與元編程等Ruby編程中的核心特性,需要的朋友可以參考下
2016-06-06
ruby 實變量
ruby 實變量...
2007-11-11
使用RVM實現控制切換Ruby/Rails版本
RVM 是Ruby Version Manager的縮寫，是一個命令行工具，它可以讓你輕松地安裝，管理和使用多個版本的Ruby.不同的rails項目使用等ruby和rails版本不一樣的時候，可以使用RVM自由切換。
2017-06-06
ruby元編程之創(chuàng)建自己的動態(tài)方法
這篇文章主要介紹了ruby元編程之創(chuàng)建自己的動態(tài)方法,本文講解使用method_missing和respond_to?創(chuàng)建自己的動態(tài)方法,需要的朋友可以參考下
2015-05-05