今天你的leader興致沖沖地找到你,希望你可以幫他一個小忙,他現在急著要去開會。要幫什么忙呢?你很好奇。
他對你說,當前你們項目的數據庫中有一張用戶信息表,里面存放了很用戶的數據,現在需要完成一個選擇性查詢用戶信息的功能。他說會傳遞給你一個包含許多用戶名的數組,你需要根據這些用戶名把他們相應的數據都給查出來。
這個功能很簡單的嘛,你爽快地答應了。由于你們項目使用的是MySQL數據庫,你很快地寫出了如下代碼:
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| #處理從數據庫讀出來的數據 end #后面應將讀到的數據組裝成對象返回,這里略去 ensure @db.close end end
這里根據傳入的用戶名數組拼裝了SQL語句,然后去數據庫中查找相應的行。為了方面調試,你還將拼裝好的SQL語句打印了出來。
然后,你寫了如下代碼來測試這個方法:
qUtil = QueryUtil.new qUtil.find_user_info ["Tom", "Jim", "Anna"]
現在運行一下測試代碼,你發現程序出錯了。于是你立刻去檢查了一下打印的SQL語句,果然發現了問題。
select * from user_info where username = 'Tom' or username = 'Jim' or username = 'Anna' or
拼裝出來的SQL語句在最后多加了一個 or 關鍵字!因為for循環執行到最后一條數據時不應該再加上or,可是代碼很笨地給最后一條數據也加了or關鍵字,導致SQL語句語法出錯了。
這可怎么辦呢?
有了!你靈光一閃,想出了一個解決辦法。等SQL語句拼裝完成后,截取到最后一個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| #處理從數據庫讀出來的數據 end #后面應將讀到的數據組裝成對象返回,這里略去 ensure @db.close end end
使用String的截取子字符串方法,只取到最后一個or之前的部分,這樣再運行測試代碼,一切就正常了,打印的SQL語句如下所示:
select * from user_info where username = 'Tom' or username = 'Jim' or username = 'Anna'
新聞熱點
疑難解答