每天一個設計模式之裝飾者模式

心譚發表於2019-01-23

作者按:《每天一個設計模式》旨在初步領會設計模式的精髓,目前採用javascriptpython兩種語言實現。誠然,每種設計模式都有多種實現方式,但此小冊只記錄最直截了當的實現方式 :)

原文地址是:《每天一個設計模式之裝飾者模式》

如果您也想進行知識整理 + 搭建功能完善/設計簡約/快速啟動的個人部落格,可以直接戳:theme-bmw

0. 專案地址

1. 什麼是“裝飾者模式”?

裝飾者模式:在不改變物件自身的基礎上,動態地新增功能程式碼。

根據描述,裝飾者顯然比繼承等方式更靈活,而且不汙染原來的程式碼,程式碼邏輯鬆耦合。

2. 應用場景

裝飾者模式由於鬆耦合,多用於一開始不確定物件的功能、或者物件功能經常變動的時候。 尤其是在引數檢查引數攔截等場景。

3. 程式碼實現

3.1 ES6 實現

ES6的裝飾器語法規範只是在“提案階段”,而且不能裝飾普通函式或者箭頭函式。

下面的程式碼,addDecorator可以為指定函式增加裝飾器。

其中,裝飾器的觸發可以在函式執行之前,也可以在函式執行之後。

注意:裝飾器需要儲存函式的執行結果,並且返回。

const addDecorator = (fn, before, after) => {
  let isFn = fn => typeof fn === "function";

  if (!isFn(fn)) {
    return () => {};
  }

  return (...args) => {
    let result;
    // 按照順序執行“裝飾函式”
    isFn(before) && before(...args);
    // 儲存返回函式結果
    isFn(fn) && (result = fn(...args));
    isFn(after) && after(...args);
    // 最後返回結果
    return result;
  };
};

/******************以下是測試程式碼******************/

const beforeHello = (...args) => {
  console.log(`Before Hello, args are ${args}`);
};

const hello = (name = "user") => {
  console.log(`Hello, ${name}`);
  return name;
};

const afterHello = (...args) => {
  console.log(`After Hello, args are ${args}`);
};

const wrappedHello = addDecorator(hello, beforeHello, afterHello);

let result = wrappedHello("godbmw.com");
console.log(result);
複製程式碼

3.2 Python3 實現

python直接提供裝飾器的語法支援。用法如下:

# 不帶引數
def log_without_args(func):
    def inner(*args, **kw):
        print("args are %s, %s" % (args, kw))
        return func(*args, **kw)
    return inner

# 帶引數
def log_with_args(text):
    def decorator(func):
        def wrapper(*args, **kw):
            print("decorator's arg is %s" % text)
            print("args are %s, %s" % (args, kw))
            return func(*args, **kw)
        return wrapper
    return decorator

@log_without_args
def now1():
    print('call function now without args')

@log_with_args('execute')
def now2():
    print('call function now2 with args')

if __name__ == '__main__':
    now1()
    now2()
複製程式碼

其實python中的裝飾器的實現,也是通過“閉包”實現的。

以上述程式碼中的now1函式為例,裝飾器與下列語法等價:

# ....
def now1():
    print('call function now without args')
# ... 
now_without_args = log_without_args(now1) # 返回被裝飾後的 now1 函式
now_without_args() # 輸出與前面程式碼相同
複製程式碼

4. 參考

相關文章