今天你的leader興致沖沖地找到你,希望你可以幫他一個(gè)小忙,他現(xiàn)在急著要去開(kāi)會(huì)。要幫什么忙呢?你很好奇。
他對(duì)你說(shuō),當(dāng)前你們項(xiàng)目的數(shù)據(jù)庫(kù)中有一張用戶信息表,里面存放了很用戶的數(shù)據(jù),現(xiàn)在需要完成一個(gè)選擇性查詢用戶信息的功能。他說(shuō)會(huì)傳遞給你一個(gè)包含許多用戶名的數(shù)組,你需要根據(jù)這些用戶名把他們相應(yīng)的數(shù)據(jù)都給查出來(lái)。
這個(gè)功能很簡(jiǎn)單的嘛,你爽快地答應(yīng)了。由于你們項(xiàng)目使用的是MySQL數(shù)據(jù)庫(kù),你很快地寫出了如下代碼:
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ù)庫(kù)讀出來(lái)的數(shù)據(jù) end #后面應(yīng)將讀到的數(shù)據(jù)組裝成對(duì)象返回,這里略去 ensure @db.close end end
這里根據(jù)傳入的用戶名數(shù)組拼裝了SQL語(yǔ)句,然后去數(shù)據(jù)庫(kù)中查找相應(yīng)的行。為了方面調(diào)試,你還將拼裝好的SQL語(yǔ)句打印了出來(lái)。
然后,你寫了如下代碼來(lái)測(cè)試這個(gè)方法:
qUtil = QueryUtil.new qUtil.find_user_info ["Tom", "Jim", "Anna"]
現(xiàn)在運(yùn)行一下測(cè)試代碼,你發(fā)現(xiàn)程序出錯(cuò)了。于是你立刻去檢查了一下打印的SQL語(yǔ)句,果然發(fā)現(xiàn)了問(wèn)題。
select * from user_info where username = 'Tom' or username = 'Jim' or username = 'Anna' or
拼裝出來(lái)的SQL語(yǔ)句在最后多加了一個(gè) or 關(guān)鍵字!因?yàn)閒or循環(huán)執(zhí)行到最后一條數(shù)據(jù)時(shí)不應(yīng)該再加上or,可是代碼很笨地給最后一條數(shù)據(jù)也加了or關(guān)鍵字,導(dǎo)致SQL語(yǔ)句語(yǔ)法出錯(cuò)了。
這可怎么辦呢?
有了!你靈光一閃,想出了一個(gè)解決辦法。等SQL語(yǔ)句拼裝完成后,截取到最后一個(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ù)庫(kù)讀出來(lái)的數(shù)據(jù) end #后面應(yīng)將讀到的數(shù)據(jù)組裝成對(duì)象返回,這里略去 ensure @db.close end end
使用String的截取子字符串方法,只取到最后一個(gè)or之前的部分,這樣再運(yùn)行測(cè)試代碼,一切就正常了,打印的SQL語(yǔ)句如下所示:
select * from user_info where username = 'Tom' or username = 'Jim' or username = 'Anna'
好了,完工!你自信滿滿。
你的leader開(kāi)完會(huì)后,過(guò)來(lái)看了下你的成果。總體來(lái)說(shuō),他還挺滿意,但對(duì)于你使用的SQL語(yǔ)句拼裝算法,他總是感覺(jué)有些不對(duì)勁,可是又說(shuō)不上哪里不好。于是他告訴了你另一種拼裝SQL語(yǔ)句的算法,讓你加入到代碼中,但是之前的那種算法也不要?jiǎng)h除,先保留著再說(shuō),然后他又很忙似的跑開(kāi)了。于是,你把他剛剛教你的算法加了進(jìn)去,代碼如下所示:
require 'mysql' class QueryUtil def find_user_info(usernames, strategy) @db = Mysql.real_connect("localhost","root","123456","test",3306); sql = "select * from user_info where " if strategy == 1 usernames.each do |user| sql << "username = '" sql << user sql << "' or " end sql = sql[0 .. -" or ".length] elsif strategy == 2 need_or = false usernames.each do |user| sql << " or " if need_or sql << "username = '" sql << user sql << "'" need_or = true end end puts sql result = @db.query(sql); result.each_hash do |row| #處理從數(shù)據(jù)庫(kù)讀出來(lái)的數(shù)據(jù) end #后面應(yīng)將讀到的數(shù)據(jù)組裝成對(duì)象返回,這里略去 ensure @db.close end end
可以看到,你leader教你的拼裝算法,使用了一個(gè)布爾變量來(lái)控制是否需要加個(gè)or這個(gè)關(guān)鍵字,第一次執(zhí)行for循環(huán)的時(shí)候因?yàn)樵摬紶栔禐閒alse,所以不會(huì)加上or,在循環(huán)的最后將布爾值賦值為true,這樣以后循環(huán)每次都會(huì)在頭部加上一個(gè)or關(guān)鍵字,由于使用了頭部添加or的方法,所以不用再擔(dān)心SQL語(yǔ)句的尾部會(huì)多出一個(gè)or來(lái)。然后你為了將兩個(gè)算法都保留,在find_user_info方法上加了一個(gè)參數(shù),strategy值為1表示使用第一種算法,strategy值為2表示使用第二種算法。
這樣測(cè)試代碼也需要改成如下方式:
qUtil = QueryUtil.new qUtil.find_user_info(["Tom", "Jim", "Anna"], 2)
這里你通過(guò)參數(shù)指明了使用第二種算法來(lái)拼裝SQL語(yǔ)句,打印的結(jié)果和使用第一種算法是完全相同的。
你立刻把你的leader從百忙之中拖了過(guò)來(lái),讓他檢驗(yàn)一下你當(dāng)前的成果,可是他還是一如既往的挑剔。
“你這樣寫的話,find_user_info這個(gè)方法的邏輯就太復(fù)雜了,非常不利于閱讀,也不利于將來(lái)的擴(kuò)展,如果我還有第三第四種算法想加進(jìn)去,這個(gè)方法還能看嗎?” 你的leader指點(diǎn)你,遇到這種情況,就要使用策略模式來(lái)解決,策略模式的核心思想就是把算法提取出來(lái)放到一個(gè)獨(dú)立的對(duì)象中。
為了指點(diǎn)你,他不顧自己的百忙,開(kāi)始教你如何使用策略模式進(jìn)行優(yōu)化。
首先定義一個(gè)父類,父類中包含了一個(gè)get_sql方法,這個(gè)方法就是簡(jiǎn)單的拋出了一個(gè)異常:
class Strategy def get_sql usernames raise "You should override this method in subclass." end end
然后定義兩個(gè)子類都繼承上述父類,并將兩種拼裝SQL語(yǔ)句的算法分別加入兩個(gè)子類中:
class Strategy1 def get_sql usernames sql = "select * from user_info where " usernames.each do |user| sql << "username = '" sql << user sql << "' or " end sql = sql[0 .. -" or ".length] end end class Strategy2 def get_sql usernames sql = "select * from user_info where " need_or = false usernames.each do |user| sql << " or " if need_or sql << "username = '" sql << user sql << "'" need_or = true end end end
然后在QueryUtil的find_user_info方法中調(diào)用Strategy的get_sql方法就可以獲得拼裝好的SQL語(yǔ)句,代碼如下所示:
require 'mysql' class QueryUtil def find_user_info(usernames, strategy) @db = Mysql.real_connect("localhost","root","123456","test",3306); sql = strategy.get_sql(usernames) puts sql result = @db.query(sql); result.each_hash do |row| #處理從數(shù)據(jù)庫(kù)讀出來(lái)的數(shù)據(jù) end #后面應(yīng)將讀到的數(shù)據(jù)組裝成對(duì)象返回,這里略去 ensure @db.close end end
最后,測(cè)試代碼在調(diào)用find_user_info方法時(shí),只需要顯示地指明需要使用哪一個(gè)策略對(duì)象就可以了:
qUtil = QueryUtil.new qUtil.find_user_info(["Tom", "Jim", "Anna"], Strategy1.new) qUtil.find_user_info(["Jac", "Joe", "Rose"], Strategy2.new)
打印出的SQL語(yǔ)句絲毫不出預(yù)料,如下所示:
select * from user_info where username = 'Tom' or username = 'Jim' or username = 'Anna' select * from user_info where username = 'Jac' or username = 'Joe' or username = 'Rose'
使用策略模式修改之后,代碼的可讀性和擴(kuò)展性都有了很大的提高,即使以后還需要添加新的算法,你也是手到擒來(lái)了!
策略模式和簡(jiǎn)單工廠模式結(jié)合的實(shí)例
需求:
商場(chǎng)收銀軟件,根據(jù)客戶購(gòu)買物品的單價(jià)和數(shù)量,計(jì)算費(fèi)用,會(huì)有促銷活動(dòng),打八折,滿三百減一百之類的。
1.使用工廠模式
# -*- encoding: utf-8 -*-#現(xiàn)金收費(fèi)抽象類class CashSuper def accept_cash(money) endend#正常收費(fèi)子類class CashNormal < CashSuper def accept_cash(money) money endend#打折收費(fèi)子類class CashRebate < CashSuper attr_accessor :mony_rebate def initialize(mony_rebate) @mony_rebate = mony_rebate end def accept_cash(money) money * mony_rebate endend#返利收費(fèi)子類class CashReturn < CashSuper attr_accessor :mony_condition, :mony_return def initialize(mony_condition, mony_return) @mony_condition = mony_condition @mony_return = mony_return end def accept_cash(money) if money > mony_condition money - (money/mony_condition) * mony_return end endend#現(xiàn)金收費(fèi)工廠類class CashFactory def self.create_cash_accept(type) case type when '正常收費(fèi)' CashNormal.new() when '打8折' CashRebate.new(0.8) when '滿三百減100' CashReturn.new(300,100) end endendcash0 = CashFactory.create_cash_accept('正常收費(fèi)')p cash0.accept_cash(700)cash1 = CashFactory.create_cash_accept('打8折')p cash1.accept_cash(700)cash2 = CashFactory.create_cash_accept('滿三百減100')p cash2.accept_cash(700)
做到了自定義折扣比例和滿減的數(shù)量。
存在的問(wèn)題:
增加活動(dòng)的種類時(shí),打五折,滿五百減二百,需要在工廠類中添加分支結(jié)構(gòu)。
活動(dòng)是多種多樣的,也有可能增加積分活動(dòng),滿100加10積分,積分一定可以領(lǐng)取活動(dòng)獎(jiǎng)品,這時(shí)就要增加一個(gè)子類。
但是每次增加活動(dòng)的時(shí)候,都要去修改工廠類,是很糟糕的處理方式,面對(duì)算法有改動(dòng)時(shí),應(yīng)該有更好的辦法。
2.策略模式
CashSuper和子類都是不變的,增加以下內(nèi)容:
class CashContext attr_accessor :cs def initialize(c_super) @cs = c_super end def result(money) cs.accept_cash(money) endendtype = '打8折'cs=case type when '正常收費(fèi)' CashContext.new(CashNormal.new()) when '打8折' CashContext.new(CashRebate.new(0.8)) when '滿三百減100' CashContext.new(CashReturn.new(300,100)) endp cs.result(700)
CashContext類對(duì)不同的CashSuper子類進(jìn)行了封裝,會(huì)返回對(duì)應(yīng)的result。也就是對(duì)不同的算法進(jìn)行了封裝,無(wú)論算法如何變化。都可以使用result得到結(jié)果。
不過(guò),目前有一個(gè)問(wèn)題,使用者需要去做判斷,來(lái)選擇使用哪個(gè)算法。可以和簡(jiǎn)單工場(chǎng)類結(jié)合。
3.策略和簡(jiǎn)單工場(chǎng)結(jié)合
class CashContext attr_accessor :cs def initialize(type) case type when '正常收費(fèi)' @cs = CashNormal.new() when '打8折' @cs = CashRebate.new(0.8) when '滿三百減100' @cs = CashReturn.new(300,100) end end def result(money) cs.accept_cash(money) endendcs=CashContext.new('打8折')p cs.result(700)
CashContext中實(shí)例化了不同的子類。(簡(jiǎn)單工廠)
將子類選擇的過(guò)程轉(zhuǎn)移到了內(nèi)部,封裝了算法(策略模式)。
調(diào)用者使用更簡(jiǎn)單,傳入?yún)?shù)(活動(dòng)類型,原價(jià)),即可得到最終的結(jié)果。
這里使用者只需要知道一個(gè)類(CashContext)就可以了,而簡(jiǎn)單工場(chǎng)需要知道兩個(gè)類(CashFactory的accept_cash方法和CashFactory),也就是說(shuō)封裝的更徹底。
新聞熱點(diǎn)
疑難解答
圖片精選