go技巧分享(二)

作者: songleo | 来源:发表于2018-05-11 14:32 被阅读70次

    这是一个系列文章,主要分享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不但可以比较结构体,像切片、数组和字典等都可以比较。

    相关文章

      网友评论

        本文标题:go技巧分享(二)

        本文链接:https://www.haomeiwen.com/subject/utpfdftx.html