about blog github
‹
›

11 May 2018
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不但可以比较结构体，像切片、数组和字典等都可以比较。



LEo at 00:05

about blog github