Go设计模式:装饰器模式

今天我们来介绍decorator这个经典的设计模式。如果使用过python的话,我想decorator都不用我介绍了,我们来看个例子:

import functools


def foo():
    print("=== foo ===")


if __name__ == "__main__":
    foo()

如果我们想在执行foo函数之前或者之后做点什么事情,就可以用上decorator(比如在web开发中,我们就经常有这种需求,例如在一个 请求的开始,我们初始化一个事务,在请求结束之后,我们尝试提交或者回滚事务):

import functools


def with_tx(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        print("=== start tx ===")
        result = func(*args, **kwargs)
        print("=== commit tx ===")
        return result

    return wrapper


@with_tx
def foo():
    print("=== foo ===")


if __name__ == "__main__":
    foo()

执行一下:

$ python main.py 
=== start tx ===
=== foo ===
=== commit tx ===

我们先来介绍Python中decorator的语法糖,为什么在使用 with_tx 之后,我们仍然能以 foo 的名义来调用被装饰过的函数呢? 我们来把 with_tx 这个函数拆城几部分来看:

def with_tx(func):  # 定义 with_tx 函数,这个函数接收一个函数作为参数
    @functools.wraps(func)  # functools.wraps 的作用,是把传入的func的文档等资料,放到wrapper函数里,它也是一个decorator
    def wrapper(*args, **kwargs):  # 定义一个闭包函数,闭包函数可以使用外层的变量,因此也就可以使用func。
        print("=== start tx ===")  # 闭包函数决定何时调用被包装的函数,比如我们这里先print,再调用
        result = func(*args, **kwargs)
        print("=== commit tx ===")
        return result

    return wrapper  # 把闭包函数返回

所以,我们可以得出这么几个结论:

  • with_tx 接受一个函数作为参数,同时它返回一个函数
  • with_tx 内的闭包函数最后是被返回的,它的实现决定了何时调用被包装的函数
  • 使用 @with_tx 之后里面返回的函数,最后却是以 foo 的函数名调用,其实是因为这相当于,把返回的wrapper函数直接 重新赋值给 foo 函数,相当于 foo = with_tx(foo)

这就是装饰器模式,一种不改变原有代码,却能增加点功能的设计模式。

Go语言的装饰器模式

在了解Python中的decorator模式之后,我们再来看Go语言如何实现装饰器模式,就很简单了:

package main

import (
	"fmt"
)

type Decoer func(i int, s string) bool

func foo(i int, s string) bool {
	fmt.Printf("=== foo ===\n")
	return true
}

func withTx(fn Decoer) Decoer {
	return func(i int, s string) bool {
		fmt.Printf("=== start tx ===\n")
		result := fn(i, s)
		fmt.Printf("=== commit tx ===\n")

		return result
	}
}

func main() {
	foo := withTx(foo)
	foo(1, "hello")
}

由于Go没有Python中那样的语法糖,因此只能手动重新赋值给同名的变量。

这就是装饰器模式。


更多文章
  • Golang expvar库源码阅读
  • Golang SQL生成库 Squirrel 教程及源码阅读
  • Golang validator使用教程
  • 使用Redis的Stream模块实现群聊功能
  • 价值编程与职业发展
  • 解决k8s cron无法读取环境变量的问题
  • 应用内购的坑
  • 两种常见的访问控制模型
  • gunicorn max-requests 选项的作用
  • Redis使用中的几点注意事项
  • 给你的代码跑个分?pylint使用教程
  • 一个Gunicorn worker数量引发的血案
  • MySQL Boolean类型的坑
  • pip freeze是魔鬼
  • 一个feed流系统的演进