Skip to content
zig 版本:0.13.0

函数

函数是编程语言中最为基本的语句。

基本使用

zig 的函数明显,你可以一眼就看出来它的组成,我们来用一个简单的函数作为说明:

zig
pub fn add(a: u8, b: u8) u8 {
    return a + b;
}

如果你有 C 的使用经验,一眼就可以看出来各自的作用。

下面来进行说明:

  1. pub 是访问修饰符,有且只有一个选择,那就是 pub,这代表着函数是公共可访问的(其他的文件import该文件后,可以直接使用这个函数)。
  2. fn 是关键字,代表着我们接下来定义了一个函数。
  3. add 是标识符,作为函数的名字。
  4. a: u8 是参数的标识符和类型,这里有两个参数,分别是 ab,它们的类型均是 u8
  5. u8 是函数的返回类型,在 zig 中,一个函数只能返回一个值。

如果没有返回值,请使用 voidzig 原则上不允许忽略函数的返回值,如果需要忽略可将返回值分配给 _,编译器将自动忽略该返回值。

🅿️ 提示

你可能注意到了有的函数定义是这样子的:

zig
fn max(comptime T: type, a: T, b: T) T {
    return if (a > b) a else b;
}

其中的 comptime T: type 你可能很陌生,这是编译期参数,它是用来实现鸭子类型(泛型)的关键语法!

关于函数命名

这里命名规则没什么强制性的要求,你只需要保证符合变量声明的 标识符规范 即可。

如果你需要一个命名的推荐规则的话,可以参照 zig 源码的命名方式,它使用的是小驼峰命名法

WARNING

函数体和函数指针之间是有区别的,函数体是仅限编译时的类型,而函数指针可能是运行时已知的。

关于函数指针,*const fn (a: i8, b: i8) i8 就是一个函数指针类型。

参数传递

参数传递是一个非常古老的问题,在高级语言的角度来看,存在着“值传递”和“引用传递”这两种情况,这无疑大大增加了程序员在编程时的心智负担。

zig 在这方面的处理则是,原始类型(整型、布尔这种)传递完全使用值传递,针对原始类型的这种策略开销非常小,通常只需要设置对应的寄存器即可。

像复合类型(结构体、联合、数组等),这些传递均是由编译器来决定究竟是使用“值传递”还是“引用传递”。

但作为开发者,只需要记住,函数的参数是不可变的 就行了。

🅿️ 提示

对于外部函数(使用 extern 修饰),Zig 遵循 C ABI 按值传递结构和联合类型。

内建函数

内建函数由编译器提供,并以 @ 为前缀。参数上的 comptime 关键字意味着该参数必须在编译期已知。

介于内建函数的数目过多,故不进行系统讲解,仅在对应章节说明涉及到的内建函数。

更多的内建函数文档请看 这里

高阶使用

以下是一些更加高级地使用,可以之后在学习!

闭包

先来看看维基百科的定义:

在计算机科学中,闭包(英语:Closure),又称词法闭包(Lexical Closure)或函数闭包(function closures),是在支持头等函数的编程语言中实现词法绑定的一种技术。

闭包在实现上是一个结构体,它存储了一个函数(通常是其入口地址)和一个关联的环境(相当于一个符号查找表)。

环境里是若干对符号和值的对应关系,它既要包括约束变量(该函数内部绑定的符号),也要包括自由变量(在函数外部定义但在函数内被引用),有些函数也可能没有自由变量。闭包跟函数最大的不同在于,当捕捉闭包的时候,它的自由变量会在捕捉时被确定,这样即便脱离了捕捉时的上下文,它也能照常运行。

捕捉时对于值的处理可以是值拷贝,也可以是名称引用,这通常由语言设计者决定,也可能由用户自行指定(如C++)。

在 zig 中,由于语言本身的限制(这是出于某种角度考虑而做出的限制),我们无法自由地使用闭包特性。

先来说一下广义的闭包:闭包是指一个函数与其引用的词法作用域(lexical scope)形成的组合。简言之,闭包允许一个函数访问其外部作用域中的变量,即使这个函数在其外部作用域之外被调用,其的主要特性是能够“记住”其创建时的环境。

在很多支持内存垃圾回收(GC)的语言中,它们的使用大概是这样的:

python
def outer_function():
    message = "Hello, World!"  # 外部函数的局部变量

    def inner_function():
        print(message)  # 内部函数访问外部函数的变量

    return inner_function  # 返回内部函数

# 创建闭包
closure = outer_function()

# 调用闭包
closure()  # 输出: Hello, World!

以上是一段 python 代码,其中 outer_function 函数最后返回一个函数类型(实际上它返回了函数 inner_function,但这不严谨,因为在解释运行的时候它不会叫做 inner_function)。

Zig 语言不允许在函数内声明函数,也不允许直接创建匿名函数。这两个功能构成了在其他编程语言(例如 JavaScript、PowerQuery-M 等)中实现闭包模式的通用模式。

这就导致我们的闭包不得不在编译期是已知的:

zig
fn bar(comptime x: i32) fn (i32) i32 {
    const res = struct {
        pub fn foo(y: i32) i32 {
            var counter = 0;
            for (x..y) |i| {
                if ((i % 2) == 0) {
                    counter += i * i;
                }
            }
            return counter;
        }
    }.foo;
    return res;
}

函数 bar 返回一个函数类型fn (i32) i32,接收一个编译期的参数,这使得该函数可以访问编译期传入的数据。

与此同时我们使用了匿名函数(匿名结构体方法):

zig
const res = struct {
    pub fn foo(y: i32) i32 {
        var counter = 0;
        for (x..y) |i| {
            if ((i % 2) == 0) {
                counter += i * i;
            }
        }
        return counter;
    }
}.foo;

关于闭包的更多内容可以参考以下文章或者帖子:

关于安德鲁拒绝匿名函数提案的解释:

anytype

函数参数可以用 anytype 代替类型来声明。在这种情况下,调用函数时将推断参数类型。使用 @TypeOf@typeInfo 获取有关推断类型的信息。

zig
fn addFortyTwo(x: anytype) @TypeOf(x) {
    return x + 42;
}

noreturn

noreturn 是一个特殊的类型,它代表以下内容:

  • break
  • continue
  • return
  • unreachable
  • while (true) {}

当一个函数不会返回时,你可以使用它来代替 void

该类型一般用在内核开发中,因为内核本身应当是一个不会退出的程序,还有一种使用场景是 exit 函数。

zig
const WINAPI = @import("std").os.windows.WINAPI;
extern "kernel32" fn ExitProcess(exit_code: c_uint) callconv(WINAPI) noreturn;

export

export 关键字保证函数可以在生成的 object 文件中可见,并且使用 C ABI。

zig
export fn sub(a: i8, b: i8) i8 {
    return a - b;
}

extern

extern 说明符用于声明一个将在链接时解析的函数(也就是该函数并非由 zig 定义,而是由外部库定义,一般遵循 C ABI,但 C ABI 本身有多种规范),链接可以是静态链接,也可以是动态链接。extern 关键字后面的引号中的标识符指定了包含该函数的库(例如 c -> libc.so )。callconv 说明符用于更改函数的调用约定。

zig
extern "c" fn atan2(a: f64, b: f64) f64;

TIP

externexport 的区别就是 extern 来引用非 Zig 实现的库,而 export 是 zig 将函数对外暴露供其他语言使用!

@setCold

@setCold(comptime is_cold: bool) void

告诉优化器当前函数很少被调用(或不被调用),该函数仅在函数作用域内有效。

zig
fn abort() noreturn {
    @setCold(true);
    while (true) {}
}

callconv

callconv 关键字告诉函数的调用约定,这在对外暴露函数或者裸汇编时会很有用。

zig
// 强制该函数在所有被调用位置内联,否则失败。
inline fn shiftLeftOne(a: u32) u32 {
    return a << 1;
}

关于可以使用的调用约定格式,可以参考这里std.builtin.CallingConvention