前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Ruby设计模式透析之 —— 策略(Strategy)

Ruby设计模式透析之 —— 策略(Strategy)

作者头像
用户1158055
发布2018-01-05 16:45:32
7750
发布2018-01-05 16:45:32
举报
文章被收录于专栏:郭霖郭霖

Java设计模式透析之 —— 策略(Strategy)

今天你的leader兴致冲冲地找到你,希望你可以帮他一个小忙,他现在急着要去开会。要帮什么忙呢?你很好奇。

他对你说,当前你们项目的数据库中有一张用户信息表,里面存放了很用户的数据,现在需要完成一个选择性查询用户信息的功能。他说会传递给你一个包含许多用户名的数组,你需要根据这些用户名把他们相应的数据都给查出来。

这个功能很简单的嘛,你爽快地答应了。由于你们项目使用的是MySQL数据库,你很快地写出了如下代码:

代码语言:javascript
复制
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语句打印了出来。

然后,你写了如下代码来测试这个方法:

代码语言:javascript
复制
qUtil = QueryUtil.new
qUtil.find_user_info ["Tom", "Jim", "Anna"]

现在运行一下测试代码,你发现程序出错了。于是你立刻去检查了一下打印的SQL语句,果然发现了问题。

代码语言:javascript
复制
select * from user_info where username = 'Tom' or username = 'Jim' or username = 'Anna' or 

拼装出来的SQL语句在最后多加了一个 or 关键字!因为for循环执行到最后一条数据时不应该再加上or,可是代码很笨地给最后一条数据也加了or关键字,导致SQL语句语法出错了。

这可怎么办呢?

有了!你灵光一闪,想出了一个解决办法。等SQL语句拼装完成后,截取到最后一个or之前的位置不就好了么。于是你将代码改成如下所示:

代码语言:javascript
复制
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语句如下所示:

代码语言:javascript
复制
select * from user_info where username = 'Tom' or username = 'Jim' or username = 'Anna'

好了,完工!你自信满满。

你的leader开完会后,过来看了下你的成果。总体来说,他还挺满意,但对于你使用的SQL语句拼装算法,他总是感觉有些不对劲,可是又说不上哪里不好。于是他告诉了你另一种拼装SQL语句的算法,让你加入到代码中,但是之前的那种算法也不要删除,先保留着再说,然后他又很忙似的跑开了。于是,你把他刚刚教你的算法加了进去,代码如下所示:

代码语言:javascript
复制
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|
			#处理从数据库读出来的数据
		end
		#后面应将读到的数据组装成对象返回,这里略去
	ensure
		@db.close
	end
end

可以看到,你leader教你的拼装算法,使用了一个布尔变量来控制是否需要加个or这个关键字,第一次执行for循环的时候因为该布尔值为false,所以不会加上or,在循环的最后将布尔值赋值为true,这样以后循环每次都会在头部加上一个or关键字,由于使用了头部添加or的方法,所以不用再担心SQL语句的尾部会多出一个or来。然后你为了将两个算法都保留,在find_user_info方法上加了一个参数,strategy值为1表示使用第一种算法,strategy值为2表示使用第二种算法。

这样测试代码也需要改成如下方式:

代码语言:javascript
复制
qUtil = QueryUtil.new
qUtil.find_user_info(["Tom", "Jim", "Anna"], 2)

这里你通过参数指明了使用第二种算法来拼装SQL语句,打印的结果和使用第一种算法是完全相同的。

你立刻把你的leader从百忙之中拖了过来,让他检验一下你当前的成果,可是他还是一如既往的挑剔。

“你这样写的话,find_user_info这个方法的逻辑就太复杂了,非常不利于阅读,也不利于将来的扩展,如果我还有第三第四种算法想加进去,这个方法还能看吗?”  你的leader指点你,遇到这种情况,就要使用策略模式来解决,策略模式的核心思想就是把算法提取出来放到一个独立的对象中

为了指点你,他不顾自己的百忙,开始教你如何使用策略模式进行优化。

首先定义一个父类,父类中包含了一个get_sql方法,这个方法就是简单的抛出了一个异常:

代码语言:javascript
复制
class Strategy
	def get_sql usernames
		raise "You should override this method in subclass."
	end
end

然后定义两个子类都继承上述父类,并将两种拼装SQL语句的算法分别加入两个子类中:

代码语言:javascript
复制
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
代码语言:javascript
复制
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方法中调用Strategy的get_sql方法就可以获得拼装好的SQL语句,代码如下所示:

代码语言:javascript
复制
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|
			#处理从数据库读出来的数据
		end
		#后面应将读到的数据组装成对象返回,这里略去
	ensure
		@db.close
	end
end

最后,测试代码在调用find_user_info方法时,只需要显示地指明需要使用哪一个策略对象就可以了:

代码语言:javascript
复制
qUtil = QueryUtil.new
qUtil.find_user_info(["Tom", "Jim", "Anna"], Strategy1.new)
qUtil.find_user_info(["Jac", "Joe", "Rose"], Strategy2.new)

打印出的SQL语句丝毫不出预料,如下所示:

代码语言:javascript
复制
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'

使用策略模式修改之后,代码的可读性和扩展性都有了很大的提高,即使以后还需要添加新的算法,你也是手到擒来了!

策略:它定义了算法家庭,分别封装起来。让它们之间可以互相替换,此模式让算法的变化,不会影响到使用算法的客户。 

本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2013-05-31 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
相关产品与服务
云数据库 SQL Server
腾讯云数据库 SQL Server (TencentDB for SQL Server)是业界最常用的商用数据库之一,对基于 Windows 架构的应用程序具有完美的支持。TencentDB for SQL Server 拥有微软正版授权,可持续为用户提供最新的功能,避免未授权使用软件的风险。具有即开即用、稳定可靠、安全运行、弹性扩缩等特点。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档