map迭代支持终止迭代控制

define.go

@@ -7,5 +7,5 @@
 // Date : 2023-03-07 17:27
 package easymap
 
-// IteratorFunc 迭代函数
-type IteratorFunc func(key interface{}, value interface{})
+// IteratorFunc 迭代函数， 返回值为是否继续迭代，true 继续迭代 false 终止后续迭代
+type IteratorFunc func(key interface{}, value interface{}) bool

normal.go

@@ -299,6 +299,9 @@ func (n *normal) Iterator(handleFunc IteratorFunc) {
 	n.RLock()
 	defer n.RUnlock()
 	for key, val := range n.data {
-		handleFunc(key, val)
+		if !handleFunc(key, val) {
+			// 终止迭代
+			break
+		}
 	}
 }

segment.go

@@ -165,9 +165,16 @@ func (s *segment) Iterator(handleFunc IteratorFunc) {
 	if nil == handleFunc {
 		return
 	}
+	isBreak := false
 	for i := 0; i < s.segment; i++ {
 		for k, v := range s.dataTable[i].GetAll() {
-			handleFunc(k, v)
+			if !handleFunc(k, v) {
+				isBreak = true
+				break
+			}
+		}
+		if isBreak {
+			break
 		}
 	}
 }

sync_normal.go

@@ -244,7 +244,6 @@ func (s *syncMap) Iterator(handleFunc IteratorFunc) {
 		return
 	}
 	s.data.Range(func(key, value interface{}) bool {
-		handleFunc(key, value)
-		return true
+		return handleFunc(key, value)
 	})
 }

sync_segment.go

@@ -141,9 +141,16 @@ func (s *segmentSync) Iterator(handleFunc IteratorFunc) {
 	if nil == handleFunc {
 		return
 	}
+	isBreak := false
 	for i := 0; i < s.segment; i++ {
 		for k, v := range s.dataTable[i].GetAll() {
-			handleFunc(k, v)
+			if !handleFunc(k, v) {
+				isBreak = true
+				break
+			}
+		}
+		if isBreak {
+			break
 		}
 	}
 }