zl程序教程

您现在的位置是:首页 >  Javascript

当前栏目

这需求快让我崩溃了,不过幸亏我懂装饰器模式

2023-03-09 22:08:30 时间

目的

装饰器模式(Decorator Pattern) 的目的非常简单,那就是:

在不修改原有代码的情况下增加逻辑。

这句话听起来可能有些矛盾,既然都要增加逻辑了,怎么可能不去修改原有的代码?但 SOLID (向对象设计5大重要原则)的开放封闭原则就是在试图解决这个问题,其内容是不去改动已经写好的核心逻辑,但又能够扩充新逻辑,也就是对扩展开放,对修改关闭。

举个例子,假如产品的需求是实现一个专门在浏览器的控制台中输出文本的功能,你可能会这样做:

  1. class Printer { 
  2.   print(text) { 
  3.     console.log(text); 
  4.   } 
  5.  
  6. const printer = new Printer(); 
  7. printer.print('something'); // something 

在你满意的看着自己的成果时,产品过来说了一句:“我觉得颜色不够突出,还是把它改成黄色的吧!”

小菜一碟!你自信的打开百度一通操作之后,把代码改成了下面这样子:

  1. class Printer { 
  2.   print(text) { 
  3.     console.log(`%c${text}`,'color: yellow;'); 
  4.   } 

但产品看了看又说:“这个字体有点太小了,再大一点,最好是高端大气上档次那种。

”好吧。。。“你强行控制着自己拿刀的冲动,一边琢磨多大的字体才是高端大气上档次,一边修改 print 的代码:

  1. class Printer { 
  2.   print(text) { 
  3.     console.log(`%c${text}`,'color: yellow;font-size: 36px;'); 
  4.   } 

这次改完你之后你心中已经满是 mmp 了,而且偷偷给产品贴了个标签:

你无法保证这次是最后的修改,而且也可能会不只一个产品来对你指手划脚。你呆呆的看着显示器,直到电脑进入休眠模式,屏幕中映出你那张苦大仇深的脸,想着不断变得乱七八糟的 print 方法,不知道该怎么去应付那些永无休止的需求。。。

在上面的例子中,最开始的 Printer 按照需求写出它应该要有的逻辑,那就是在控制台中输出一些文本。换句话说,当写完“在控制台中输出一些文本”这段逻辑后,就能将 Printer 结束了,因为它就是 Printer 的全部逻辑了。那在这个情况下该如何改变字体或是颜色的逻辑呢?

这时你该需要装饰器模式了。

Decorator Pattern(装饰器模式)

首先修改原来的 Printer,使它可以支持扩充样式:

  1. class Printer { 
  2.   print(text = ''style = '') { 
  3.     console.log(`%c${text}`, style); 
  4.   } 

之后分别创建改变字体和颜色的装饰器:

  1. const yellowStyle = (printer) => ({ 
  2.   ...printer, 
  3.   print: (text = ''style = '') => { 
  4.     printer.print(text, `${style}color: yellow;`); 
  5.   } 
  6. }); 
  7.  
  8. const boldStyle = (printer) => ({ 
  9.   ...printer, 
  10.   print: (text = ''style = '') => { 
  11.     printer.print(text, `${style}font-weight: bold;`); 
  12.   } 
  13. }); 
  14.  
  15. const bigSizeStyle = (printer) => ({ 
  16.   ...printer, 
  17.   print: (text = ''style = '') => { 
  18.     printer.print(text, `${style}font-size: 36px;`); 
  19.   } 
  20. }); 

代码中的 yellowStyle、boldStyle 和 bigSizeStyle 分别是给 print 方法的装饰器,它们都会接收 printer,并以 printer 为基础复制出一个一样的对象出来并返回,而返回的 printer 与原来的区别是,各自 Decorator 都会为 printer 的 print 方法加上各自装饰的逻辑(例如改变字体、颜色或字号)后再调用 printer 的 print。

使用方式如下:

只要把所有装饰的逻辑抽出来,就能够自由的搭配什么时候要输出什么样式,加入要再增加一个斜体样式,也只需要再新增一个装饰器就行了,不需要改动原来的 print 逻辑。

不过要注意的是上面的代码只是简单的把 Object 用解构复制,如果在 prototype 上存在方法就有可能会出错,所以要深拷贝一个新对象的话,还需要另外编写逻辑:

  1. const copyObj = (originObj) => { 
  2.   const originPrototype = Object.getPrototypeOf(originObj); 
  3.   let newObj = Object.create(originPrototype); 
  4.     
  5.   const originObjOwnProperties = Object.getOwnPropertyNames(originObj); 
  6.   originObjOwnProperties.forEach((property) => { 
  7.     const prototypeDesc = Object.getOwnPropertyDescriptor(originObj, property); 
  8.      Object.defineProperty(newObj, property, prototypeDesc); 
  9.   }); 
  10.    
  11.   return newObj; 

然后装饰器内改使上面代码中的 copyObj,就能正确复制相同的对象了:

  1. const yellowStyle = (printer) => { 
  2.   const decorator = copyObj(printer); 
  3.  
  4.   decorator.print = (text = ''style = '') => { 
  5.     printer.print(text, `${style}color: yellow;`); 
  6.   }; 
  7.  
  8.   return decorator; 
  9. }; 

其他案例

因为我们用的语言是 JavaScript,所以没有用到类,只是简单的装饰某个方个方法,比如下面这个用来发布文章的 publishArticle:

  1. const publishArticle = () => { 
  2.   console.log('发布文章'); 
  3. }; 

如果你想要再发布文章之后在 微博或QQ空间之类的平台上发个动态,那又该怎么处理呢?是像下面的代码这样吗?

  1. const publishArticle = () => { 
  2.   console.log('发布文章'); 
  3.  
  4.   console.log('发 微博 动态'); 
  5.   console.log('发 QQ空间 动态'); 
  6. }; 

这样显然不好!publishArticle 应该只需要发布文章的逻辑就够了!而且如果之后第三方服务平台越来越多,那 publishArticle 就会陷入一直加逻辑一直爽的情况,在明白了装饰器模式后就不能再这样做了!

所以把这个需求套上装饰器:

  1. const publishArticle = () => { 
  2.   console.log('发布文章'); 
  3. }; 
  4.  
  5. const publishWeibo = (publish) => (...args) => { 
  6.   publish(args); 
  7.   console.log('发 微博 动态'); 
  8. }; 
  9.  
  10. const publishQzone = (publish) => (...args) => { 
  11.   publish(args); 
  12.   console.log('发 QQ空间 动态'); 
  13. }; 
  14.  
  15.  
  16. const publishArticleAndWeiboAndQzone = publishWeibo(publishQzone(publishArticle)); 

前面 Printer 的例子是复制一个对象并返回,但如果是方法就不用复制了,只要确保每个装饰器都会返回一个新方法,然后会去执行被装饰的方法就行了。

总结

装饰器模式是一种非常有用的设计模式,在项目中也会经常用到,当需求变动时,觉得某个逻辑很多余,那么直接不装饰它就行了,也不需要去修改实现逻辑的代码。每一个装饰器都做他自己的事情,与其他装饰器互不影响。