代码改变世界

Go的可变参数函数

2024-03-26 11:07  abce  阅读(79)  评论(0编辑  收藏  举报

可变函数是指可以接收可变数量的参数的函数。在 Golang 中,可以传递与函数签名中引用的类型相同的不同数量的参数。

在声明可变函数时,最后一个参数的类型前会有一个省略号"...",这表明该函数可以用任意数量的该类型参数来调用,可以是0个、1个或者多个。

这种类型的函数在不知道传递给函数的参数个数时非常有用,最好的例子就是 fmt 软件包中的内置 Println 函数,它就是一个可变函数。

func Println(a ...interface{}) (n int, err error)

先来看看 fmt.Println 函数示例

package main

import "fmt"

func main() {
    fmt.Println()
    fmt.Println("one")
    fmt.Println("one", "two")
    fmt.Println("one", "two", "three")
}

结果:


one one two one two three

第一条结果是空白,因为没有输入任何参数。

 

下面看看可变函数

可变函数的语法就是:

function function_name(para1, para2...type)type {// code...}

可变函数中:

·函数中...type 的行为和切片一样。假如我们定义的一个函数 add(b...int),那么这里的参数就是 []int 类型的

·可以向可变函数传递已经存在的切片

·如果没有向可变函数传递任何参数,则函数中的切片是空切片

·可变函数常用于字符串的格式化

·可以向可变函数传递多个切片

·不可以将可变参数作为返回值,但是可以以切片的方式返回

 

示例一

// Go program to illustrate the concept of variadic function
package main

import (
    "fmt"
    "strings"
)

// Variadic function to join strings
func joinStr(elements ...string) string {
    return strings.Join(elements, "-")
}

func main() {

    // zero argument
    fmt.Println(joinStr())

    // multiple arguments
    fmt.Println(joinStr("Variadic", "Function"))
    fmt.Println(joinStr("Variadic", "Function", "In", "Golang"))
    fmt.Println(joinStr("G", "o", "l", "a", "n", "g"))

}

结果:


Variadic-Function Variadic-Function-In-Golang G-o-l-a-n-g

 

示例二

// Go program to illustrate the concept of variadic function
package main

import (
    "fmt"
    "strings"
)

// Variadic function to join strings
func joinStr(elements ...string) string {
    return strings.Join(elements, "-")
}

func main() {
    // pass a slice in variadic function
    elements := []string{"Variadic", "Function", "In", "Golang"}
    fmt.Println(joinStr(elements...))
}

结果:

Variadic-Function-In-Golang

可变参数的顺序

函数中,只能包含一个可变参数,而且必须是函数的最后一个参数,不然编译就会报错。

 

何时使用可变函数

·替代需要向函数传递切片的场景

·不知道函数参数数量的场景

·提高代码的可读性

·传递的参数不属于同一种数据结构,比如切片、数组等,可变函数可以避免创建新的数据结构来传递这些参数