golang中使用sync.Map的方法

181 阅读 0 评论 120 点赞

我是靠谱客的博主寂寞大炮，最近开发中收集的这篇文章主要介绍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)
//循环读取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
	}
}

总结