golang 高阶函数使用 Functional Options

原创于 2021-07-13 14:59:20 发布 · 158 阅读

0 ·

CC 4.0 BY-SA版权

go 专栏收录该内容

92 篇文章

订阅专栏

该博客展示了如何使用Go语言定义一个`Server`结构体来存储服务器配置信息，包括地址、端口、协议、超时时间和最大连接数。通过高阶函数模式，可以方便地设置和初始化这些配置，例如将协议更改为UDP并调整最大连接数。示例中创建了一个服务器实例，并打印了配置的协议和最大连接数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package main

import (
    "crypto/tls"
    "fmt"
    "time"
)

// 业务实体

type Server struct {
    Addr     string
    Port     int
    Protocol string
    Timeout  time.Duration
    MaxConns int
    TLS      *tls.Config
}

// 高阶函数

type Option func(*Server)

func Protocol(p string) Option {
    return func(server *Server) {
        server.Protocol = p
    }
}

func Timeout(timeout time.Duration) Option {
    return func(server *Server) {
        server.Timeout = timeout
    }
}

func MaxConns(maxconns int) Option {
    return func(server *Server) {
        server.MaxConns = maxconns
    }
}

func TLS(tls *tls.Config) Option {
    return func(server *Server) {
        server.TLS = tls
    }
}

func NewServer(addr string, port int, options ...func(*Server)) (*Server, error) {
    serv := Server{
        Addr: addr,
        Port: port,
        Protocol: "tcp",
        Timeout: 30 * time.Second,
        MaxConns: 1000,
        TLS: nil,
    }

    for _, option := range options {
        option(&serv)
    }
    return &serv, nil
}

func main()  {
    server, _ := NewServer("127.0.0.1", 9999, Protocol("UDP"), MaxConns(100))
    fmt.Println(server.Protocol)
    fmt.Println(server.MaxConns)
}