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

func (m MyInt) IsZero() bool {
	return m == 0
}

func (m MyInt) Add(other int) int {
	return other + int(m)
}

func main() {

	var b MyInt
	fmt.Print(b.IsZero())

	b = 1
	fmt.Println("\n", b.Add(2))
}

二、http 包中的类型方法

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

package main

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

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

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

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

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

	//开始请求
	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()
}　

三、time包中的类型方法

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

package main

import (
    "fmt"
    "time"
)

func main(){
    fmt.Println(time.Second.String())
}