模版方法模式是设计模式中比较好理解的一种模式,它指的是在父类中定义一个public方法,包含此类的行为,而具体的行为是在子类中实现的。
案例:公司要求设计一个消息推送功能,包括微信推送、发送短信等,目前在用户支付订单和发货的时候,给用户通知。按照这个思路,我们实现抽象类如下:
# 抽象推送消息类
class AbstractPushMessage:
# 推送支付成功消息
def _sendPayMsg(self, msg):
pass
# 推送发货消息
def _sendDeliveryMsg(self, msg):
pass
# 模版方法,控制推送逻辑,对外只暴露这一个接口即可
def sendMsg(self, msg, type):
if type == 'PAY':
self._sendPayMsg(msg)
if type == 'DELIVERY':
self._sendDeliveryMsg(msg)
子类的实现也简单(微信推送、短信推送),这里就不写出来了。
模版方法模式有一个很重要的扩展功能:实现钩子方法(Hook Method)。钩子方法返回值是boolean类型的,用于控制模版方法的流程。回到刚才的案例,公司发现短信发送成本太高,要求紧急停止。此时怎么做呢?在各个调用方入口屏蔽,或是在实现子类将各个方法屏蔽?都不是很好的做法,此时我们加个“钩子”:
# 抽象推送消息类
class AbstractPushMessage:
# 推送支付成功消息
def _sendPayMsg(self, msg):
pass
# 推送发货消息
def _sendDeliveryMsg(self, msg):
pass
# 钩子方法,该推送渠道是否下线
def _isOff(self):
return False
# 模版方法,控制推送逻辑,对外只暴露这一个接口即可
def sendMsg(self, msg, type):
if self._isOff():
return
if type == 'PAY':
self._sendPayMsg(msg)
if type == 'DELIVERY':
self._sendDeliveryMsg(msg)
短息推送子类如下:
class SmsPushMessage(AbstractPushMessage):
def _sendPayMsg(self, msg):
print("推送短信用户支付消息:" + msg)
def _sendDeliveryMsg(self, msg):
print("推送短信发货消息:" + msg)
def _isOff(self):
return True
改动如此之少,是不是觉得重构的很优雅?
总结:
模版方法模式应用的非常广泛,在各种框架、中间件随处可见,亲爱的你或许也在经常实践它呢。