map增加迭代实现

This commit is contained in:
白茶清欢 2023-03-07 17:34:02 +08:00
parent 7c1d72f0de
commit 8e29a6318f
6 changed files with 71 additions and 0 deletions

View File

@ -34,4 +34,6 @@ type EasyMap interface {
Exist(key interface{}) bool
GetAll() map[interface{}]interface{}
GetAllForMapKeyString() map[string]interface{}
// Iterator 对数据的迭代
Iterator(IteratorFunc)
}

11
define.go Normal file
View File

@ -0,0 +1,11 @@
// Package easymap ...
//
// Description : easymap ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 2023-03-07 17:27
package easymap
// IteratorFunc 迭代函数
type IteratorFunc func(key interface{}, value interface{})

View File

@ -286,3 +286,19 @@ func (n *normal) GetAllForMapKeyString() map[string]interface{} {
}
return finalData
}
// Iterator ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 17:32 2023/3/7
func (n *normal) Iterator(handleFunc IteratorFunc) {
if nil == handleFunc {
return
}
n.RLock()
defer n.RUnlock()
for key, val := range n.data {
handleFunc(key, val)
}
}

View File

@ -155,3 +155,19 @@ func (s *segment) GetAllForMapKeyString() map[string]interface{} {
}
return finalData
}
// Iterator ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 17:32 2023/3/7
func (s *segment) Iterator(handleFunc IteratorFunc) {
if nil == handleFunc {
return
}
for i := 0; i < s.segment; i++ {
for k, v := range s.dataTable[i].GetAll() {
handleFunc(k, v)
}
}
}

View File

@ -233,3 +233,18 @@ func (s *syncMap) GetAllForMapKeyString() map[string]interface{} {
}
return finalData
}
// Iterator 迭代数据
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 17:27 2023/3/7
func (s *syncMap) Iterator(handleFunc IteratorFunc) {
if nil == handleFunc {
return
}
s.data.Range(func(key, value interface{}) bool {
handleFunc(key, value)
return true
})
}

View File

@ -136,3 +136,14 @@ func (s *segmentSync) GetAllForMapKeyString() map[string]interface{} {
}
return finalData
}
func (s *segmentSync) Iterator(handleFunc IteratorFunc) {
if nil == handleFunc {
return
}
for i := 0; i < s.segment; i++ {
for k, v := range s.dataTable[i].GetAll() {
handleFunc(k, v)
}
}
}