今天你的leader興致沖沖地找到你,希望你可以幫他一個(gè)小忙,他現(xiàn)在急著要去開會(huì)。要幫什么忙呢?你很好奇。
他對你說,當(dāng)前你們項(xiàng)目的數(shù)據(jù)庫中有一張用戶信息表,里面存放了很用戶的數(shù)據(jù),現(xiàn)在需要完成一個(gè)選擇性查詢用戶信息的功能。他說會(huì)傳遞給你一個(gè)包含許多用戶名的數(shù)組,你需要根據(jù)這些用戶名把他們相應(yīng)的數(shù)據(jù)都給查出來。
這個(gè)功能很簡單的嘛,你爽快地答應(yīng)了。由于你們項(xiàng)目使用的是MySQL數(shù)據(jù)庫,你很快地寫出了如下代碼:
require 'mysql' class QueryUtil def find_user_info usernames @db = Mysql.real_connect("localhost","root","123456","test",3306); sql = "select * from user_info where " usernames.each do |user| sql << "username = '" sql << user sql << "' or " end puts sql result = @db.query(sql); result.each_hash do |row| #處理從數(shù)據(jù)庫讀出來的數(shù)據(jù) end #后面應(yīng)將讀到的數(shù)據(jù)組裝成對象返回,這里略去 ensure @db.close end end
這里根據(jù)傳入的用戶名數(shù)組拼裝了SQL語句,然后去數(shù)據(jù)庫中查找相應(yīng)的行。為了方面調(diào)試,你還將拼裝好的SQL語句打印了出來。
然后,你寫了如下代碼來測試這個(gè)方法:
qUtil = QueryUtil.new qUtil.find_user_info ["Tom", "Jim", "Anna"]
現(xiàn)在運(yùn)行一下測試代碼,你發(fā)現(xiàn)程序出錯(cuò)了。于是你立刻去檢查了一下打印的SQL語句,果然發(fā)現(xiàn)了問題。
select * from user_info where username = 'Tom' or username = 'Jim' or username = 'Anna' or
拼裝出來的SQL語句在最后多加了一個(gè) or 關(guān)鍵字!因?yàn)閒or循環(huán)執(zhí)行到最后一條數(shù)據(jù)時(shí)不應(yīng)該再加上or,可是代碼很笨地給最后一條數(shù)據(jù)也加了or關(guān)鍵字,導(dǎo)致SQL語句語法出錯(cuò)了。
這可怎么辦呢?
有了!你靈光一閃,想出了一個(gè)解決辦法。等SQL語句拼裝完成后,截取到最后一個(gè)or之前的位置不就好了么。于是你將代碼改成如下所示:
require 'mysql' class QueryUtil def find_user_info usernames @db = Mysql.real_connect("localhost","root","123456","test",3306); sql = "select * from user_info where " usernames.each do |user| sql << "username = '" sql << user sql << "' or " end sql = sql[0 .. -" or ".length] puts sql result = @db.query(sql); result.each_hash do |row| #處理從數(shù)據(jù)庫讀出來的數(shù)據(jù) end #后面應(yīng)將讀到的數(shù)據(jù)組裝成對象返回,這里略去 ensure @db.close end end
使用String的截取子字符串方法,只取到最后一個(gè)or之前的部分,這樣再運(yùn)行測試代碼,一切就正常了,打印的SQL語句如下所示:
select * from user_info where username = 'Tom' or username = 'Jim' or username = 'Anna'
新聞熱點(diǎn)
疑難解答
圖片精選