读代码之golang装饰器模式

在v站看到一篇文章,说到了golang实现python装饰器的方法,有两个实现方式
其中一种是用类方式实现的,当时泛览了下代码,觉得 emmm...实现方式挺花哨的,貌似看懂了就没怎么在意,当时特意把代码复制到goland中作为参考了
某天上班摸鱼,仔细看了下这段代码,梳理细节后忽然发现没get到其中逻辑
郁闷很久未参透
有空就特意调试打印下这段代码
好几天才整明白,这大概就是菜比の理解能力了吧

代码如下:

// with interface. which we need create a `point` first. and use it later

package main

import "fmt"

type FibI interface {
    Fib(n int) int
    Wrap(fib FibI) FibI
}

type Fib struct {
    Wrapper FibI
}

func (this *Fib) Fib(n int) int {
    //wrapper := this.Wrapper
    if this.Wrapper == nil {
        this.Wrapper = this
    }
    fmt.Printf("我执行了...%T...%v\n", this.Wrapper, n)
    if n == 0 {
        return 0
    }
    if n == 1 {
        return 1
    }
    // call son
    fmt.Printf("leixing9...%T...\n", this.Wrapper)
    return this.Wrapper.Fib(n-1) + this.Wrapper.Fib(n-2)
}

func (this *Fib) Wrap(fib FibI) FibI {
    fmt.Printf("leixing1...%T...\n", this.Wrapper)
    this.Wrapper = fib
    fmt.Printf("leixing2...%T...\n", this.Wrapper)
    fmt.Println("fib 调用", this)
    return this
}

type CacheFib struct {
    Wrapper FibI
    cache   map[int]int
}

func (this *CacheFib) Wrap(fib FibI) FibI {
    this.Wrapper = fib
    fmt.Printf("leixing6...%T...\n", this.Wrapper)
    return this
}

func (this *CacheFib) Fib(n int) int {
    if this.cache == nil {
        this.cache = make(map[int]int)
    }
    if ans, ok := this.cache[n]; ok {
        return ans
    }
    ans := this.Wrapper.Fib(n)
    this.cache[n] = ans
    return ans
}

type CounterFib struct {
    Wrapper FibI
    Counter int
}

func (this *CounterFib) Wrap(fib FibI) FibI {
    this.Wrapper = fib
    fmt.Printf("leixing...%T...\n", this.Wrapper)
    return this
}

func (this *CounterFib) Fib(n int) int {
    this.Counter++
    return this.Wrapper.Fib(n)
}

func main() {
    fib := new(Fib)
    //fmt.Println("result fib", fib.Fib(10))

    cacheFib := new(CacheFib)
    //f := cacheFib.Wrap(fib.Wrap(cacheFib))
    //fmt.Println(f.Fib(10), "heihei")
    counterFib := new(CounterFib)
    counterCacheFib := cacheFib.Wrap(counterFib.Wrap(fib.Wrap(cacheFib)))
    fmt.Println("result cache:counter:fib", counterCacheFib.Fib(10))
    fmt.Printf("count: %d, cache: %v", counterFib.Counter, cacheFib.cache)
}

本文的目的没有那么高大上,就是简单的理解一下,这种装饰器实现及其后的核心原理
讲真,这个代码实现的挺花哨的>_<
所以,我把自己的那些调试打印信息也一并给出了

首先第一步:
背景:用装饰器的原理实现斐波那契数列的计算
如果有golang语法基础,相信这些main函数之前的代码应该理解起来不难

其中Fib结构是一个实现了FibI接口的对象
乍看之下好像仅此而已
CacheFib结构也同样实现了此接口,不过提供了一个map数据结构,会把之前计算过的数值缓存起来,后面调用直接取缓存
CounterFib结构就简单啦,它计算了CounterFib实例调用Fib方法的次数
三个Fib类的Wrap方法实现方式都一样:就是把自身的Wrapper属性赋值为参数中的(实现了)FibI值

一直到main函数之前,这些代码还都不难理解
最后注意下主函数内的调用方法

cacheFib.Wrap(counterFib.Wrap(fib.Wrap(cacheFib)))

瞬间懵逼有木有
乍看之下,为什么cacheFib会出现两次,看起来好像是自己引用了自己(递归也是自己调用自己,好好参悟一下)
嗯,那些能明白这行代码的童鞋可以先行告退了,没懂得同学咱门继续

不明白的话就运行一遍代码:
斐波那契数列计算完美,有计算次数,也优化了计算逻辑(缓存了)
但这并不妨碍咱们的懵逼状态ing~
整个方法的调用链是怎样的呢

我在各个位置加了打印信息,其核心就是Fib方法
通过打印可以发现,每次计算n数值的时候都会调用Fib 对象的Fib 方法(有点绕)
打印Fib 对象的Wrapper类型, 显示为main.CacheFib类型

所以,我们重新回顾下Fib对象,其实它没那么简单,首先对象的Wrapper 属性本身类型就是实现了FibI 接口的类型,其次对象本身也实现FibI 接口
Fib 对象的Fib 方法中调用了 Fib 对象的Wrapper属性的Fib方法(还是很绕,注意区分下Fib对象和Fib方法,这是两个概念)

那么回头来看main函数中的实现,cacheFib 是CacheFib类的实例,cacheFib使用Wrap方法用counterFib对象来装饰自身,但是counterFib对象的Wrapper被fib对象装饰了,同时fib对象又被cacheFib自身装饰了,那么当我们执行counterCacheFib.Fib(10)是时候,经过层层装饰,其实调用的方法是fib对象的Fib方法,fib对象的Fib方法在每次执行的时候又会调用自身Wrapper的Fib方法(不过这是一个类似递归的操作,只有当参数n大于1的时候才会调用自身的Wrapper的Fib方法),不过Fib.Wrapper又被cacheFib装饰了,所以绕了一圈又回来了,只要入参n大于1,就会一直递归往复(这是一个有退出条件的闭环), 所以小伙伴们都绕明白了吗

如果用python实现起来其实就相对简单多了,这里就不再给出代码了。换个思路,为了加强理解,如果用python实现一个上面golang风格的代码呢?感兴趣可以实现一下,会更好的帮助你理解golang以及python的风格用法特点,这里给出参考:

class CacheFib(object):

    def __init__(self):
        self.cache = dict()
        self.f = None

    def wrap(self, obj):
        self.f = obj.fib
        # return self

    def fib(self, n):
        if not self.cache.get(n):
            self.cache[n] = self.f(n)

        return self.cache[n]


class CountFib(object):

    def __init__(self):
        self.count = 0
        self.f = None

    def wrap(self, obj):
        self.f = obj.fib
        # return self

    def fib(self, n):
        self.count += 1
        return  self.f(n)


class Fib(object):

    def __init__(self):
        self.f = self.fib

    def wrap(self, obj):
        self.f = obj.fib
        # return self

    def fib(self, n)->int:
        if n == 0:
            return 0
        elif n == 1:
            return 1
        else:
            return self.f(n-1) + self.f(n-2)


if __name__ == '__main__':
    f = Fib()
    cache = CacheFib()
    count = CountFib()

    f.wrap(cache)
    count.wrap(f)
    cache.wrap(count)

    # cache = cache.wrap(count.wrap(f.wrap(cache))
    cache.fib(20)
    print("count: {}, cache: {}".format(count.count, cache.cache) )

你可能感兴趣的:(读代码之golang装饰器模式)