golang中使用sync.Map的方法

更新時間：2020年06月16日 22:39:11 作者：IT技術小小鳥

這篇文章主要介紹了golang中使用sync.Map的方法，本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下

背景

go中map數據結構不是線程安全的，即多個goroutine同時操作一個map，則會報錯，因此go1.9之后誕生了sync.Map

sync.Map思路來自java的ConcurrentHashMap

接口

sync.map就是1.9版本帶的線程安全map，主要有如下幾種方法：

Load(key interface{}) (value interface{}, ok bool)
//通過提供一個鍵key，查找對應的值value，如果不存在，則返回nil。ok的結果表示是否在map中找到值

Store(key, value interface{})
//這個相當于是寫map（更新或新增），第一個參數是key，第二個參數是value

LoadOrStore(key, value interface{}) (actual interface{}, loaded bool)
//通過提供一個鍵key，查找對應的值value，如果存在返回鍵的現有值，否則存儲并返回給定的值，如果是讀取則返回true，如果是存儲返回false

Delete(key interface{})
//通過提供一個鍵key，刪除鍵對應的值

Range(f func(key, value interface{}) bool)
//循環(huán)讀取map中的值。
//因為for ... range map是內置的語言特性，所以沒有辦法使用for range遍歷sync.Map, 但是可以使用它的Range方法，通過回調的方式遍

實踐

package main

import (
	"fmt"
	"sync"
)

var num = 0
var addTest *AddTest

func init() {
	addTest = &AddTest{}
}

type AddTest struct {
	m sync.Mutex
}

func (at *AddTest) increment(wg *sync.WaitGroup) {
	//互斥鎖
	at.m.Lock() //當有線程進去進行加鎖
	num++
	at.m.Unlock() //出來后解鎖，其他線程才可以進去
	wg.Done()
}

func (at *AddTest) decrement(wg *sync.WaitGroup) {
	//互斥鎖
	at.m.Lock() //當有線程進去進行加鎖
	num--
	at.m.Unlock() //出來后解鎖，其他線程才可以進去
	wg.Done()
}

var w sync.WaitGroup

var aa map[int]int

func main() {
	var bb sync.Map
	var wg sync.WaitGroup
	//aa = make(map[int]int)
	wg.Add(2)
	go func() {
		//wg.Add(1)
		for i:=0 ;i <100; i++{
			//aa[i] = i+1
			//fmt.Println("a")
			bb.Store(i, i+1)
		}
		wg.Done()
	}()

	go func() {

		for i:=0 ;i <100; i++{
			//aa[i] = i+1
			//fmt.Println("a")
			bb.Store(i, i+1)
		}
		wg.Done()
	}()
	wg.Wait()
	bb.Range(func(k, v interface{}) bool {
		fmt.Println("iterate:", k, v)
		return true
	}
}

總結