TRKJ/EdgeCommon
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeCommon.git synced 2025-11-04 13:10:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复DataMap无法在多个节点之间共享的问题

Browse Source
This commit is contained in:
GoEdgeLab
2023-03-19 10:23:32 +08:00
parent dab1a86d09
commit 8509551d62
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pkg/serverconfigs/shared/data_map.go
View File

@@ -6,6 +6,7 @@ import (
"bytes" "bytes"
"crypto/md5" "crypto/md5"
"fmt" "fmt"
"sync"
) )
var dataMapPrefix = []byte("GOEDGE_DATA_MAP:") var dataMapPrefix = []byte("GOEDGE_DATA_MAP:")
@@ -14,6 +15,7 @@ var dataMapPrefix = []byte("GOEDGE_DATA_MAP:")
// 用来减少相同数据占用的空间和内存 // 用来减少相同数据占用的空间和内存
type DataMap struct { type DataMap struct {
Map map[string][]byte Map map[string][]byte
locker sync.Mutex
} }
// NewDataMap 构建对象 // NewDataMap 构建对象
@@ -23,6 +25,8 @@ func NewDataMap() *DataMap {
// Put 放入数据 // Put 放入数据
func (this *DataMap) Put(data []byte) (keyData []byte) { func (this *DataMap) Put(data []byte) (keyData []byte) {
this.locker.Lock()
defer this.locker.Unlock()
var key = string(dataMapPrefix) + fmt.Sprintf("%x", md5.Sum(data)) var key = string(dataMapPrefix) + fmt.Sprintf("%x", md5.Sum(data))
this.Map[key] = data this.Map[key] = data
return []byte(key) return []byte(key)
@@ -30,6 +34,8 @@ func (this *DataMap) Put(data []byte) (keyData []byte) {
// Read 读取数据 // Read 读取数据
func (this *DataMap) Read(key []byte) []byte { func (this *DataMap) Read(key []byte) []byte {
this.locker.Lock()
defer this.locker.Unlock()
if bytes.HasPrefix(key, dataMapPrefix) { if bytes.HasPrefix(key, dataMapPrefix) {
return this.Map[string(key)] return this.Map[string(key)]
} }