社区精选 | 手写编程语言-实现运算符重载

业界 作者：SegmentFault 2022-10-05 12:21:55 阅读：900

今天小编为大家带来的是社区作者 crossoverJie 的文章，让我们一起来学习手写编程语言-实现运算符重载。

前言

先带来日常的 GScript 更新：新增了可变参数的特性，语法如下：

int add(string s, int ...num){
    println(s);
    int sum = 0;
    for(int i=0;i<len(num);i++){
        int v = num[i];
        sum = sum+v;
    }
    return sum;
}
int x = add("abc", 1,2,3,4);
println(x);
assertEqual(x, 10);

//formats according to a format specifier and writes to standard output.
printf(string format, any ...a){}

//formats according to a format specifier and returns the resulting string.
string sprintf(string format, any ...a){}

使用

println(100+2*2);

class Person{
    int age;
    Person(int a){
        age = a;
    }
}
Person p1 = Person(10);
Person p2 = Person(20);
Person p3 = p1+p2;

Person operator + (Person p1, Person p2){
    Person pp = Person(p1.age+p2.age);
    return pp;
}
Person p3 = p1+p2;
println("p3.age="+p3.age);
assertEqual(p3.age, 30);

• 函数名必须是 operator
• 名称后跟上运算符即可。

实现

1. 编译期间：记录所有的重载函数和运算符的关系。
2. 运行期：根据当前的运算找到声明的函数，直接运行即可。

// OpOverload 重载符
type OpOverload struct {
    function  *Func
    tokenType int
}

// 运算符重载自定义函数
opOverloads []*symbol.OpOverload

// GetOpFunction 获取运算符重载函数
// 通过返回值以及运算符号(+-*/) 匹配重载函数
func (a *AnnotatedTree) GetOpFunction(returnType symbol.Type, tokenType int) *symbol.Func {
    for _, overload := range a.opOverloads {
        isType := overload.GetFunc().GetReturnType().IsType(returnType)
        if isType && overload.GetTokenType() == tokenType {
            return overload.GetFunc()
        }
    }
    return nil
}

• 编译期：
• https://github.com/crossoverJie/gscript/blob/ae729ce7d4cf39fe115121993fcd2222716755e5/resolver/type_scope_resolver.go#L127
• 运行期：
• https://github.com/crossoverJie/gscript/blob/499236af549be47ff827c6d55de1fc8e5600b9b3/visitor.go#L387

总结

a5 = (a1.Add(a2).Add(a3)).Mul(a4);

a5 = (a1+a2+a3)*a4;