go技巧分享（二）

這是一個系列文章，主要分享go的使用建議和技巧，每次分享3點，希望你能有所收穫。

1 併發訪問map

package main

import "sync"

type Map struct {
    sync.RWMutex
    Data map[int]int
}

func main() {
    m := Map{}
    m.Data = make(map[int]int)

    for i := 0; i < 100000; i++ {
        go m.Write(i, i)
        go m.Read(i)
    }
}

func (m *Map) Read(key int) int {
    m.RLock()
    value := m.Data[key]
    m.RUnlock()
    return value
}

func (m *Map) Write(key int, value int) {
    m.Lock()
    m.Data[key] = value
    m.Unlock()
}

註釋掉加鎖和解鎖的代碼：

func (m *Map) Read(key int) int {
    // m.RLock()
    value := m.Data[key]
    // m.RUnlock()
    return value
}

func (m *Map) Write(key int, value int) {
    // m.Lock()
    m.Data[key] = value
    // m.Unlock()
}

/*
fatal error: concurrent map read and map write
或者
fatal error: concurrent map writes
*/

多協程併發訪問map時，有可能會導致程序退出，錯誤信息爲：fatal error: concurrent map read and map write 或者 fatal error: concurrent map writes，所以如果希望在多協程中併發訪問map，必須提供某種同步機制，一般情況下通過讀寫鎖sync.RWMutex實現對map的併發訪問控制，將map和sync.RWMutex封裝一下，可以實現對map的安全併發訪問。示例中，如果註釋掉加鎖和解鎖的代碼，運行時就會出現併發訪問map錯誤。

2 結構體轉換成json字符串時忽略字段

package main

import (
    "encoding/json"
    "fmt"
)

type User struct {
    Name string
    Age  int
    Addr string `json:"-"`
    // addr string
}

func main() {
    jsonStr, _ := json.Marshal(User{"user1", 12, "addr"})
    fmt.Printf("%s\n", jsonStr)
}

/*
{"Name":"user1","Age":12}
*/

將結構體轉換成json字符串時，如果想忽略某些字段，可以通過在該字段後面添加`json:”-“`實現，添加該tag後，轉換成json字符串時會忽略該字段。當然，你也可以通過將該字段首字母改成小寫實現，只是這樣的話該字段對其他包也隱藏了，兩種方法各有利弊，權衡使用。

3 判斷2個結構體是否相等

package main

import (
    "fmt"
    "reflect"
)

type User struct {
    Name string
    Age  int
    Addr string
}

func NewUser(name string, age int) User {
    u := User{}
    u.Name = name
    u.Age = age
    return u
}

func main() {
    u1 := NewUser("user1", 12)
    u2 := NewUser("user1", 12)
    fmt.Println(reflect.DeepEqual(u1, u2))
    fmt.Println(reflect.DeepEqual([]int{1, 2}, []int{1, 2}))
    fmt.Println(reflect.DeepEqual([2]int{1, 2}, [2]int{1, 2}))
    fmt.Println(reflect.DeepEqual(map[int]int{1: 1, 2: 2}, map[int]int{1: 1, 2: 2}))
}

/*
true
true
true
true
*/

在使用go語言過程中，有時需要比較2個結構體是否“相等”，即結構體字段和字段值都相同。如果自己實現比較函數，當結構體字段多時比較麻煩。通過使用reflect.DeepEqual可以實現判斷2個結構體是否相等，但會有一定的性能影響。另外，reflect.DeepEqual不但可以比較結構體，像切片、數組和字典等都可以比較。