Hàm variadic trong Golang: tham số ...T và cách truyền s...

Tham số ...T

Đặt ... trước kiểu của tham số cuối cùng. Khi đó nơi gọi hàm có thể truyền bao nhiêu đối số kiểu đó cũng được, kể cả không truyền gì. Bên trong hàm, tham số này là một slice.

package main

import "fmt"

func sum(nums ...int) int {
	total := 0
	for _, n := range nums {
		total += n
	}
	return total
}

func main() {
	fmt.Println(sum())
	fmt.Println(sum(5))
	fmt.Println(sum(1, 2, 3, 4))
}

Chương trình in ra 0, 5 và 10. Khi không có đối số nào, nums là một slice nil có độ dài 0, nên vòng lặp đơn giản là không chạy.

Các tham số thường có thể đứng trước. Chỉ tham số cuối cùng được phép là variadic:

func logf(level string, format string, args ...any)

Truyền slice bằng s...

Nếu đã có sẵn một slice, bạn không thể truyền thẳng nó vào: sum(nums) báo lỗi cannot use nums (variable of type []int) as int value in argument to sum. Hãy trải slice ra bằng ... đặt sau đối số:

package main

import "fmt"

func sum(nums ...int) int {
	total := 0
	for _, n := range nums {
		total += n
	}
	return total
}

func main() {
	nums := []int{10, 20, 30}
	fmt.Println(sum(nums...))

	more := append([]int{1, 2}, nums...)
	fmt.Println(more)
}

append cũng là hàm variadic, vì thế nối hai slice được viết là append(a, b...).

Slice được trải ra phải là thứ duy nhất ở vị trí variadic. sum(1, nums...) báo lỗi too many arguments in call to sum. Hãy dựng slice gộp trước bằng append.

Slice được dùng chung, không sao chép

Khi bạn gọi hàm với các đối số riêng lẻ, Go tạo một slice mới cho chúng. Khi bạn trải một slice có sẵn bằng s..., hàm nhận đúng slice đó. Ghi vào phần tử của nó sẽ làm thay đổi dữ liệu của nơi gọi.

package main

import "fmt"

func zeroFirst(xs ...int) {
	if len(xs) > 0 {
		xs[0] = 0
	}
}

func main() {
	nums := []int{7, 8, 9}

	zeroFirst(nums[0], nums[1], nums[2])
	fmt.Println(nums) // untouched: a new slice was built

	zeroFirst(nums...)
	fmt.Println(nums) // changed: the same backing array
}

Đầu ra:

[7 8 9]
[0 8 9]

Một hàm variadic có sửa tham số của mình thì nên ghi rõ điều đó trong tài liệu, hoặc sao chép trước bằng slices.Clone. Phần lớn hàm variadic chỉ đọc đối số nên hiếm khi gặp vấn đề này, nhưng đây là điểm khác biệt thật so với các ngôn ngữ mà tham số rest luôn là một mảng mới.

...any và cách fmt.Println hoạt động

fmt.Println được khai báo là func Println(a ...any) (n int, err error). any (bí danh của interface{} từ Go 1.18) nhận mọi kiểu, nên bạn truyền chuỗi, số và struct trong cùng một lời gọi được.

package main

import (
	"fmt"
	"strings"
)

func logf(level, format string, args ...any) {
	prefix := "[" + strings.ToUpper(level) + "] "
	fmt.Printf(prefix+format+"\n", args...)
}

func main() {
	logf("info", "server started on port %d", 8080)
	logf("warn", "%d of %d workers idle", 3, 8)
	logf("error", "no arguments here")
}

Chi tiết quan trọng là việc chuyển tiếp: logf truyền args... cho Printf. Thiếu dấu ba chấm, Printf chỉ nhận một đối số là một []any, và dòng cảnh báo in ra [WARN] [3 8] of %!d(MISSING) workers idle. go vet chỉ báo missing ... in args forwarded to printf-like function khi tham số format được truyền nguyên vẹn, như fmt.Printf(format, args). Ở đây format được ghép bằng +, nên vet im lặng và việc nhớ dấu ba chấm là của bạn.

Không thể trải một []string vào ...any. Kiểu phần tử phải khớp chính xác, nên hãy chuyển đổi trước:

names := []string{"a", "b"}
args := make([]any, len(names))
for i, n := range names {
	args[i] = n
}
fmt.Println(args...)

Bắt buộc có ít nhất một đối số

Tham số variadic chấp nhận không có đối số nào. Khi hàm cần ít nhất một đối số, hãy biến đối số đầu tiên thành tham số thường:

package main

import "fmt"

func maxOf(first int, rest ...int) int {
	m := first
	for _, n := range rest {
		m = max(m, n)
	}
	return m
}

func main() {
	fmt.Println(maxOf(3))
	fmt.Println(maxOf(3, 9, 4))
}

Giờ maxOf() là lỗi biên dịch thay vì phải kiểm tra lúc chạy. Bản thân hàm có sẵn max cũng hoạt động như vậy: nó đòi ít nhất một đối số.

Tham số variadic làm đối số tùy chọn

Go không có giá trị mặc định cho tham số. Đôi khi người ta dùng tham số variadic để giả lập một đối số tùy chọn:

func connect(addr string, timeout ...time.Duration)

Cách này chạy được nhưng khó đọc, vì nơi gọi có thể truyền ba timeout và chữ ký hàm không cho biết cái nào được dùng. Khi có nhiều hơn một tùy chọn, nên dùng một struct cấu hình, hoặc mẫu functional options (...Option, trong đó mỗi Option là một hàm gán giá trị cho một trường). Mẫu này cũng là variadic, nhưng mỗi đối số đều có ý nghĩa rõ ràng.

Tóm tắt quy tắc