在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) )