快捷導(dǎo)航

go函數(shù)的參數(shù)設(shè)置默認(rèn)值的方法

更新時間：2025年01月10日 09:56:35 作者：半桶水專家

Go語言不直接支持函數(shù)參數(shù)默認(rèn)值,但可以通過指針、結(jié)構(gòu)體、變長參數(shù)和選項模式等方法模擬,下面給大家分享幾種方式模擬函數(shù)參數(shù)的默認(rèn)值功能,感興趣的朋友一起看看吧

1. 通過指針傳遞可選值

如果函數(shù)接收一個指針作為參數(shù)，你可以通過傳遞 nil 來模擬未傳遞的情況，并在函數(shù)內(nèi)部提供默認(rèn)值。

package main
import "fmt"
func greet(name *string) {
    if name == nil {
        defaultName := "Guest"
        name = &defaultName
    }
    fmt.Println("Hello,", *name)
}
func main() {
    greet(nil)         // 使用默認(rèn)值
    greet(&"Alice")    // 使用傳入值
}

輸出：

Hello, Guest
Hello, Alice

2. 使用結(jié)構(gòu)體模擬默認(rèn)參數(shù)值

你可以定義一個結(jié)構(gòu)體，作為函數(shù)的參數(shù)類型，然后根據(jù)情況為結(jié)構(gòu)體字段設(shè)置默認(rèn)值。

package main
import "fmt"
type Config struct {
    Name  string
    Age   int
    City  string
}
// 默認(rèn)值
func NewConfig() *Config {
    return &Config{
        Name: "John Doe",  // 默認(rèn)值
        Age:  30,          // 默認(rèn)值
        City: "New York",   // 默認(rèn)值
    }
}
func Greet(cfg *Config) {
    if cfg == nil {
        cfg = NewConfig() // 如果沒有傳入配置，使用默認(rèn)值
    }
    fmt.Printf("Name: %s, Age: %d, City: %s\n", cfg.Name, cfg.Age, cfg.City)
}
func main() {
    // 使用默認(rèn)配置
    Greet(nil)
    // 自定義配置
    customCfg := &Config{Name: "Alice", Age: 25}
    Greet(customCfg)
}

輸出：

Name: John Doe, Age: 30, City: New York
Name: Alice, Age: 25, City: New York

3. 使用變長參數(shù)（...）和自定義邏輯（推薦）

你也可以使用 Go 的變長參數(shù)來模擬默認(rèn)值的效果。當(dāng)調(diào)用函數(shù)時，如果沒有提供某個參數(shù)，你可以在函數(shù)內(nèi)檢查參數(shù)的數(shù)量并賦予默認(rèn)值。

package main
import "fmt"
func greet(args ...string) {
    if len(args) == 0 {
        fmt.Println("Hello, Guest!")
    } else {
        fmt.Println("Hello,", args[0])
    }
}
func main() {
    greet()             // 使用默認(rèn)值
    greet("Alice")      // 使用傳入值
}

輸出：

Hello, Guest!
Hello, Alice

4. 使用選項模式（構(gòu)造函數(shù)）

另一種常見的方式是使用構(gòu)造函數(shù)模式，這通常在需要多個可選參數(shù)時非常有用。你可以通過創(chuàng)建一個函數(shù)來設(shè)置所有可能的參數(shù)，然后只傳遞你想要的部分參數(shù)。

package main
import "fmt"
type Person struct {
    Name  string
    Age   int
    Email string
}
func NewPerson(name string, options ...func(*Person)) *Person {
    p := &Person{Name: name} // 必選參數(shù)
    // 通過 options 來設(shè)置可選參數(shù)
    for _, option := range options {
        option(p)
    }
    return p
}
// 可選設(shè)置 Age 和 Email
func WithAge(age int) func(*Person) {
    return func(p *Person) {
        p.Age = age
    }
}
func WithEmail(email string) func(*Person) {
    return func(p *Person) {
        p.Email = email
    }
}
func main() {
    p1 := NewPerson("John")  // 只有 Name
    fmt.Println(p1)           // 輸出: &{John 0 }
    p2 := NewPerson("Alice", WithAge(30), WithEmail("alice@example.com"))
    fmt.Println(p2)           // 輸出: &{Alice 30 alice@example.com}
}

輸出：