Multipledispatch2

本文讲述了笔者的multipledispatch2库的一些技术细节.

由于是从静态语言切换成的python使用者, 笔者对python的动态类型非常不适应. 因此几乎每个地方都会使用multipledispatch给保护一下.

然后用久了感觉有几个不方便:

  1. multipledispatch不支持函数参数的类型签名. python 3.5加入了typing库, 表示def foo(bar: int, baz: str) -> list这种写法是官方提倡的, 然而multipledispatch却不支持这种写法.
  2. multipledispatch不支持一个类型是多个类型的子类型的写法, 比如我想表达A <: B && A <: C时就无能为力了.

由于给作者提issue以后作者几个月没动静, 因此在multipledispatch的基础上, 笔者修改了一些代码, 发布了multipledispatch2.

主要改动是:

  1. 增加了对type annotation的支持, 即: 对于原来的

    @dispatch(int, str)
    def foo(a, b):
        pass
    

    现在可以写作:

    @dispatch
    def foo(a: int, b: str):
        pass
    

    相对更自然了一点.

  2. 增加了对subtype of multiple types的支持:

    class A: pass
    class B: pass
    class C(A, B): pass
    @dispatch
    def foo(a: [A, B]): 
        pass
    

    对于调用foo(x), 当且仅当入参x同时是AB的子类时, 对foo的调用才会成立. 在上例中只有foo(C())是成立的.

    这个新特性对于写库的人来说十分方便, 比如当你想要使用trait来做mixin的时候, users可能会拿你的trait混合出很多subtypes. 如果你想要提供一些函数来操作这些subtypes, 比如有一个函数当参数mixinclass A时产生behavior1, mixinclass B时产生behavior2, 同时mixinclass Aclass B时, 产生behavior3. 在这种情况下, 使用原有的multipledispatch是不可能的.

技术细节:

主要问题在于dispatch order上. 比如: 对于同一个函数名foo, 它有两种类型:

  1. foo(a: A, b: B): pass
  2. foo(a: C, b: B): pass

其中:

class A: pass
class B: pass
class C(A): pass
class D(C): pass

当你传入参数(C, B)的时候, 显然希望调用的是2而不是1, 当传入(D, B)的时候, 显然也希望调用2而不是1.只有当传入(A, B)时, 才希望调用的是1.

用形象的话来说: 对于入参X, 希望被调用的函数是拥有最具体的签名的那个函数.

于是这就要求我们有一个排序方法将foo的所有签名进行排序, 越具体的签名尽可能在前面. 进行签名搜索时, 应该从头开始搜, 并采纳第一个适合的签名.

还是上面那个例子, 如果我们能够产生一个搜索顺序: [(C, B), (A, B)], 那不就符合要求了?

对于排序, 显然需要一个操作符compare进行比较. 那么对于签名的compare该如何定义呢? 即如何确定签名与签名之间谁大谁小.

在这里我们定义compare如下:

令 A, B 为 tuple of types

if A.length == b.length then
    A compare B := A <: B (即A是B的子类型)
else
    return not comparable

而对于tuple of types<:定义如下:

for zip(all a in A, all b in B):
  a <: b

即A中每一个类型都是对应位置上B的子类型时, A <: B

这样我们定义了类型签名之间的subtyping关系. 于是我们使用这个compare关系对一个函数的所有签名进行拓扑排序, 结果将得到一个序列.

注意这里使用拓扑排序而不是别的排序方法, 是因为签名之间并不是良序关系. 两个签名之间可能其实是没有任何大小关系的(上面的return not comparable分支). 因此一个函数的所有签名其实构成了多个DAG(有向无环图).

按照拓扑排序得到的序列搜索, 一定能够得到最具体的那个签名.

加入multiple subtypes后的变化:

上面的定义很不错, 但是没有考虑一种情况: 在tuple of types<:定义中, 我们使用了a <: b这个比较. 然而当multiple subtypes存在的时候, ab可能是一个类型, 也可能是一个联合类型[type, type ...]. 那么如何求a <: b呢?

比如, 当

class A: pass
class B: pass
class C(A, B): pass

时, 如何证明C <: [A, B]呢?

甚至是当

class A: pass
class B: pass
class C(A): pass
class D(B): pass

时, 如何证明[C, D] <: [A, B]呢?

我们拓展一下subtyping关系即可:

对于types a and b

def a <: b as
  if a and b both are type then
      return a <: b
  else if a is type and b is tuple then
    return if a <: all types in b
  else if a is tuple and b is type then
    return if any types in a <: b
  else if a and b both are tuple then
    return for all types in b if any types in a <: types in b

经过这样一个补充定义以后, multipledispatch2就完美支持multiple subtypes啦.

PhD Student in Computing Science

PhD Student @ SFU

Related