首页 > 编程笔记 > Go语言笔记

Go语言为任意类型添加方法

Go语言可以对任何类型添加方法，给一种类型添加方法就像给结构体添加方法一样，因为结构体也是一种类型。

为基本类型添加方法

在Go语言中，使用 type 关键字可以定义出新的自定义类型，之后就可以为自定义类型添加各种方法了。我们习惯于使用面向过程的方式判断一个值是否为 0，例如： 
if  v == 0 {
    // v等于0
}
如果将 v 当做整型对象，那么判断 v 值就可以增加一个 IsZero() 方法，通过这个方法就可以判断 v 值是否为 0，例如： 
if  v.IsZero() {
    // v等于0
}
为基本类型添加方法的详细实现流程如下： 
package main

import (
    "fmt"
)

// 将int定义为MyInt类型
type MyInt int

// 为MyInt添加IsZero()方法
func (m MyInt) IsZero() bool {
    return m == 0
}

// 为MyInt添加Add()方法
func (m MyInt) Add(other int) int {
    return other + int(m)
}

func main() {

    var b MyInt

    fmt.Println(b.IsZero())

    b = 1

    fmt.Println(b.Add(2))
}
代码输出如下：

true
3

代码说明如下：

http包中的类型方法

Go语言提供的 http 包里也大量使用了类型方法，Go语言使用 http 包进行 HTTP 的请求，使用 http 包的 NewRequest() 方法可以创建一个 HTTP 请求，填充请求中的 http 头（req.Header），再调用 http.Client 的 Do 方法，将传入的 HTTP 请求发送出去。

下面代码演示创建一个 HTTP 请求，并且设定 HTTP 头。 
package main

import (
    "fmt"
    "io/ioutil"
    "net/http"
    "os"
    "strings"
)

func main() {
    client := &http.Client{}

    // 创建一个http请求
    req, err := http.NewRequest("POST", "http://www.163.com/", strings.NewReader("key=value"))

    // 发现错误就打印并退出
    if err != nil {
        fmt.Println(err)
        os.Exit(1)
        return
    }

    // 为标头添加信息
    req.Header.Add("User-Agent", "myClient")

    // 开始请求
    resp, err := client.Do(req)

    // 处理请求的错误
    if err != nil {
        fmt.Println(err)
        os.Exit(1)
        return
    }

    data, err := ioutil.ReadAll(resp.Body)
    fmt.Println(string(data))

    defer resp.Body.Close()

}
代码执行结果如下：

<html>
<head><title>405 Not Allowed</title></head>
<body bgcolor="white">
<center><h1>405 Not Allowed</h1></center>
<hr><center>nginx</center>
</body>
</html>

代码说明如下：
由于我们构造的请求不是网易服务器所支持的类型，所以服务器返回操作不被运行的 405 错误。

在本例子第 24 行中使用的 req.Header 的类型为 http.Header，就是典型的自定义类型，并且拥有自己的方法，http.Header 的部分定义如下： 
type Header map[string][]string

func (h Header) Add(key, value string) {
    textproto.MIMEHeader(h).Add(key, value)
}

func (h Header) Set(key, value string) {
    textproto.MIMEHeader(h).Set(key, value)
}


func (h Header) Get(key string) string {
    return textproto.MIMEHeader(h).Get(key)
}
代码说明如下：
为类型添加方法的过程是一个语言层特性，使用类型方法的代码经过编译器编译后的代码运行效率与传统的面向过程或面向对象的代码没有任何区别，因此，为了代码便于理解，可以在编码时使用Go语言的类型方法特性。

time 包中的类型方法

Go语言提供的 time 包主要用于时间的获取和计算等，在这个包中，也使用了类型方法，例如： 
package main

import (
    "fmt"
    "time"
)

func main() {
    fmt.Println(time.Second.String())
}
第 9 行的 time.Second 是一个常量，下面代码的加粗部分就是 time.Second 的定义： 
const (
    Nanosecond  Duration = 1
    Microsecond  = 1000 * Nanosecond
    Millisecond  = 1000 * Microsecond
    Second       = 1000 * Millisecond
    Minute       = 60 * Second
    Hour         = 60 * Minute
)
Second 的类型为 Duration，而 Duration 实际是一个 int64 的类型，定义如下：

type Duration int64

它拥有一个 String 的方法，部分定义如下： 
func (d Duration) String() string {

    // 一系列生成buf的代码
    …
   
    return string(buf[w:])
}
Duration.String 可以将 Duration 的值转为字符串。

所有教程

优秀文章