老男孩教育专注IT教育10余年,只培养IT技术精英

全国免费咨询电话(渠道合作):400-609-2893

Go语言错误总结(六)!老男孩IT教育

老男孩IT教育

技术博客

2022年4月15日 16:48

Go语言常见错误总结第六弹,快跟着小编学习一下吧。

  36、关闭HTTP的连接

  一些HTTP服务器保持会保持一段时间的网络连接(根据HTTP 1.1的说明和服务器端的“keep-alive”配置)。默认情况下,标准http库只在目标HTTP服务器要求关闭时才会关闭网络连接。这意味着你的应用在某些条件下消耗完sockets/file的描述符。

  你可以通过设置请求变量中的Close域的值为true,来让http库在请求完成时关闭连接。

  另一个选项是添加一个Connection的请求头,并设置为close。目标HTTP服务器应该也会响应一个Connection: close的头。当http库看到这个响应头时,它也将会关闭连接。

package main

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

func main() {
    req, err := http.NewRequest("GET", "http://golang.org", nil)
    if err != nil {
        fmt.Println(err)
        return
    }
    req.Close = true
    //or do this:
    //req.Header.Add("Connection", "close")
    resp, err := http.DefaultClient.Do(req)
    if resp != nil {
        defer resp.Body.Close()
    }
    if err != nil {
        fmt.Println(err)
        return
    }
    body, err := ioutil.ReadAll(resp.Body)
    if err != nil {
        fmt.Println(err)
        return
    }
    fmt.Println(len(string(body)))
}

  你也可以取消http的全局连接复用。你将需要为此创建一个自定义的http传输配置。


package main

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

func main() {
    tr := &http.Transport{DisableKeepAlives: true}
    client := &http.Client{Transport: tr}
    resp, err := client.Get("http://golang.org")
    if resp != nil {
        defer resp.Body.Close()
    }
    if err != nil {
        fmt.Println(err)
        return
    }
    fmt.Println(resp.StatusCode)
    body, err := ioutil.ReadAll(resp.Body)
    if err != nil {
        fmt.Println(err)
        return
    }
    fmt.Println(len(string(body)))
}

  如果你向同一个HTTP服务器发送大量的请求,那么把保持网络连接的打开是没问题的。然而,如果你的应用在短时间内向大量不同的HTTP服务器发送一两个请求,那么在引用收到响应后立刻关闭网络连接是一个好主意。增加打开文件的限制数可能也是个好主意。当然,正确的选择源自于应用。

  37、比较Structs, Arrays, Slices, and Maps

  如果结构体中的各个元素都可以用你可以使用等号来比较的话,那就可以使用相号, == 来比较结构体变量。


package main

import "fmt"

type data struct {
    num     int
    fp      float32
    complex complex64
    str     string
    char    rune
    yes     bool
    events  <-chan string
    handler interface{}
    ref     *byte
    raw     [10]byte
}

func main() {
    v1 := data{}
    v2 := data{}
    fmt.Println("v1 == v2:", v1 == v2) //prints: v1 == v2: true
}

  运行结果:

v1 == v2: true

  如果结构体中的元素无法比较,那使用等号将导致编译错误。注意数组仅在它们的数据元素可比较的情况下才可以比较。

package main

import "fmt"

type data struct {
    num    int               //ok
    checks [10]func() bool   //not comparable
    doit   func() bool       //not comparable
    m      map[string]string //not comparable
    bytes  []byte            //not comparable
}

func main() {
    v1 := data{}
    v2 := data{}
    fmt.Println("v1 == v2:", v1 == v2)
}

  编译错误:


./main.go:16:30: invalid operation: v1 == v2 (struct containing [10]func() bool cannot be compared)

  Go确实提供了一些助手函数,用于比较那些无法使用等号比较的变量。

  最常用的方法是使用reflect包中的 DeepEqual()函数。

package main

import (
    "fmt"
    "reflect"
)

type data struct {
    num    int               //ok
    checks [10]func() bool   //not comparable
    doit   func() bool       //not comparable
    m      map[string]string //not comparable
    bytes  []byte            //not comparable
}

func main() {
    v1 := data{}
    v2 := data{}
    fmt.Println("v1 == v2:", reflect.DeepEqual(v1, v2)) //prints: v1 == v2: true
    m1 := map[string]string{"one": "a", "two": "b"}
    m2 := map[string]string{"two": "b", "one": "a"}
    fmt.Println("m1 == m2:", reflect.DeepEqual(m1, m2)) //prints: m1 == m2: true
    s1 := []int{1, 2, 3}
    s2 := []int{1, 2, 3}
    fmt.Println("s1 == s2:", reflect.DeepEqual(s1, s2)) //prints: s1 == s2: true
}

  运行结果:

v1 == v2: true
m1 == m2: true
s1 == s2: true

  除了很慢(这个可能会也可能不会影响你的应用),DeepEqual()也有其他自身的技巧。

package main

import (
    "fmt"
    "reflect"
)

func main() {
    var b1 []byte = nil
    b2 := []byte{}
    fmt.Println("b1 == b2:", reflect.DeepEqual(b1, b2)) //prints: b1 == b2: false
}

  运行结果:

b1 == b2: false

  DeepEqual()不会认为空的slice与“nil”的slice相等。这个行为与你使用bytes.Equal()函数的行为不同。bytes.Equal()认为“nil”和空的slice是相等的。

package main

import (
    "bytes"
    "fmt"
)

func main() {
    var b1 []byte = nil
    b2 := []byte{}
    fmt.Println("b1 == b2:", bytes.Equal(b1, b2)) //prints: b1 == b2: true
}

  运行结果:

b1 == b2: true

  DeepEqual()在比较slice时并不总是完美的。

package main

import (
    "encoding/json"
    "fmt"
    "reflect"
)

func main() {
    var str string = "one"
    var in interface{} = "one"
    fmt.Println("str == in:", str == in, reflect.DeepEqual(str, in))
    //prints: str == in: true true
    v1 := []string{"one", "two"}
    v2 := []interface{}{"one", "two"}
    fmt.Println("v1 == v2:", reflect.DeepEqual(v1, v2))
    //prints: v1 == v2: false (not ok)
    data := map[string]interface{}{
        "code":  200,
        "value": []string{"one", "two"},
    }
    encoded, _ := json.Marshal(data)
    var decoded map[string]interface{}
    json.Unmarshal(encoded, &decoded)
    fmt.Println("data == decoded:", reflect.DeepEqual(data, decoded))
    //prints: data == decoded: false (not ok)
}

  运行结果:

str == in: true true
v1 == v2: false
data == decoded: false

  如果你的byte slice(或者字符串)中包含文字数据,而当你要不区分大小写形式的值时(在使用==,bytes.Equal(),或者bytes.Compare()),你可能会尝试使用“bytes”和“string”包中的ToUpper()或者ToLower()函数。对于英语文本,这么做是没问题的,但对于许多其他的语言来说就不行了。这时应该使用strings.EqualFold()和bytes.EqualFold()。

  如果你的byte slice中包含需要验证用户数据的隐私信息(比如,加密哈希、tokens等),不要使用reflect.DeepEqual()、bytes.Equal(),或者bytes.Compare(),因为这些函数将会让你的应用易于被定时攻击。为了避免泄露时间信息,使用'crypto/subtle'包中的函数(即,subtle.ConstantTimeCompare())。

  38、从Panic中恢复

  recover()函数可以用于获取/拦截panic。仅当在一个defer函数中被完成时,调用recover()将会完成这个小技巧。

  错误代码:

package main

import "fmt"

func main() {
    recover() // doesn't do anything
    panic("not good")
    recover() // won't be executed
    fmt.Println("ok")
}

  运行结果:

panic: not good

  正确代码:

package main

import "fmt"

func main() {
    defer func() {
        fmt.Println("recovered:", recover())
    }()
    panic("not good")
}

  运行结果:

recovered: not good

  recover()的调用仅当它在defer函数中被直接调用时才有效。

  错误代码:

package main

import "fmt"

func doRecover() {
    fmt.Println("recovered =>", recover()) //prints: recovered => <nil>
}
func main() {
    defer func() {
        doRecover() //panic is not recovered
    }()
    panic("not good")
}

  运行结果:

recovered => <nil>
panic: not good

  39、在Slice, Array, and Map "range"语句中更新引用元素的值

  在“range”语句中生成的数据的值是真实集合元素的拷贝。它们不是原有元素的引用。

  这意味着更新这些值将不会修改原来的数据。同时也意味着使用这些值的地址将不会得到原有数据的指针。

package main

import "fmt"

func main() {
    data := []int{1, 2, 3}
    for _, v := range data {
        v *= 10 //original item is not changed
    }
    fmt.Println("data:", data) //prints data: [1 2 3]
}

  运行结果:

data: [1 2 3]

  如果你需要更新原有集合中的数据,使用索引操作符来获得数据。

package main

import "fmt"

func main() {
    data := []int{1, 2, 3}
    for i, _ := range data {
        data[i] *= 10
    }
    fmt.Println("data:", data) //prints data: [10 20 30]
}

  运行结果:

data: [10 20 30]

  如果你的集合保存的是指针,那规则会稍有不同。

  如果要更新原有记录指向的数据,你依然需要使用索引操作,但你可以使用for range语句中的第二个值来更新存储在目标位置的数据。

&{10} &{20} &{30}

  40、在Slice中"隐藏"数据

  当你重新划分一个slice时,新的slice将引用原有slice的数组。如果你忘了这个行为的话,在你的应用分配大量临时的slice用于创建新的slice来引用原有数据的一小部分时,会导致难以预期的内存使用。

package main

import "fmt"

func get() []byte {
    raw := make([]byte, 10000)
    fmt.Println(len(raw), cap(raw), &raw[0]) //prints: 10000 10000 <byte_addr_x>
    return raw[:3]
}
func main() {
    data := get()
    fmt.Println(len(data), cap(data), &data[0]) //prints: 3 10000 <byte_addr_x>
}

  运行结果:

10000 10000 0xc420080000
3 10000 0xc420080000

  为了避免这个陷阱,你需要从临时的slice中拷贝数据(而不是重新划分slice)。

package main

import "fmt"

func get() []byte {
    raw := make([]byte, 10000)
    fmt.Println(len(raw), cap(raw), &raw[0]) //prints: 10000 10000 <byte_addr_x>
    res := make([]byte, 3)
    copy(res, raw[:3])
    return res
}
func main() {
    data := get()
    fmt.Println(len(data), cap(data), &data[0]) //prints: 3 3 <byte_addr_y>
}

  41、Slice的数据“毁坏”

  比如说你需要重新一个路径(在slice中保存)。你通过修改第一个文件夹的名字,然后把名字合并来创建新的路劲,来重新划分指向各个文件夹的路径。

package main

import (
    "bytes"
    "fmt"
)

func main() {
    path := []byte("AAAA/BBBBBBBBB")
    sepIndex := bytes.IndexByte(path, '/')
    dir1 := path[:sepIndex]
    dir2 := path[sepIndex+1:]
    fmt.Println("dir1 =>", string(dir1)) //prints: dir1 => AAAA
    fmt.Println("dir2 =>", string(dir2)) //prints: dir2 => BBBBBBBBB
    dir1 = append(dir1, "suffix"...)
    path = bytes.Join([][]byte{dir1, dir2}, []byte{'/'})
    fmt.Println("dir1 =>", string(dir1)) //prints: dir1 => AAAAsuffix
    fmt.Println("dir2 =>", string(dir2)) //prints: dir2 => uffixBBBB 
    fmt.Println("new path =>", string(path))
}

  运行结果:

dir1 => AAAA
dir2 => BBBBBBBBB
dir1 => AAAAsuffix
dir2 => uffixBBBB
new path => AAAAsuffix/uffixBBBB

  结果与你想的不一样。与"AAAAsuffix/BBBBBBBBB"相反,你将会得到"AAAAsuffix/uffixBBBB"。这个情况的发生是因为两个文件夹的slice都潜在的引用了同一个原始的路径slice。这意味着原始路径也被修改了。根据你的应用,这也许会是个问题。

  通过分配新的slice并拷贝需要的数据,你可以修复这个问题。另一个选择是使用完整的slice表达式。

package main

import (
    "bytes"
    "fmt"
)

func main() {
    path := []byte("AAAA/BBBBBBBBB")
    sepIndex := bytes.IndexByte(path, '/')
    dir1 := path[:sepIndex:sepIndex] //full slice expression
    dir2 := path[sepIndex+1:]
    fmt.Println("dir1 =>", string(dir1)) //prints: dir1 => AAAA
    fmt.Println("dir2 =>", string(dir2)) //prints: dir2 => BBBBBBBBB
    dir1 = append(dir1, "suffix"...)
    path = bytes.Join([][]byte{dir1, dir2}, []byte{'/'})
    fmt.Println("dir1 =>", string(dir1)) //prints: dir1 => AAAAsuffix
    fmt.Println("dir2 =>", string(dir2)) //prints: dir2 => BBBBBBBBB 
    fmt.Println("new path =>", string(path))
}

  运行结果:

dir1 => AAAA
dir2 => BBBBBBBBB
dir1 => AAAAsuffix
dir2 => BBBBBBBBB
new path => AAAAsuffix/BBBBBBBBB

  完整的slice表达式中的额外参数可以控制新的slice的容量。现在在那个slice后添加元素将会触发一个新的buffer分配,而不是覆盖第二个slice中的数据。

  42、陈旧的(Stale)Slices

  多个slice可以引用同一个数据。比如,当你从一个已有的slice创建一个新的slice时,这就会发生。如果你的应用功能需要这种行为,那么你将需要关注下“走味的”slice。

  在某些情况下,在一个slice中添加新的数据,在原有数组无法保持更多新的数据时,将导致分配一个新的数组。而现在其他的slice还指向老的数组(和老的数据)。

package main

import "fmt"

func main() {
    s1 := []int{1, 2, 3}
    fmt.Println(len(s1), cap(s1), s1) //prints 3 3 [1 2 3]
    s2 := s1[1:]
    fmt.Println(len(s2), cap(s2), s2) //prints 2 2 [2 3]
    for i := range s2 {
        s2[i] += 20
    }
    //still referencing the same array
    fmt.Println(s1) //prints [1 22 23]
    fmt.Println(s2) //prints [22 23]
    s2 = append(s2, 4)
    for i := range s2 {
        s2[i] += 10
    }
    //s1 is now "stale"
    fmt.Println(s1) //prints [1 22 23]
    fmt.Println(s2) //prints [32 33 14]
}

  运行结果:

3 3 [1 2 3]
2 2 [2 3]
[1 22 23]
[22 23]
[1 22 23]
[32 33 14]

  老男孩GO语言课程率行业之先开设,并进行多次迭代升级,以帮助学员学习到真正有用的知识,如有学习需求,可以关注“GO语言开发”课程。

  推荐阅读:

  Go语言错误总结(一)!老男孩GO语言学习周期

  老男孩教育-Go语言操作MySQL(一)

  Go语言项目实战(一)!老男孩GO语言入门

本文经授权发布,不代表老男孩教育立场。如若转载请联系原作者。