add Group.GetElements() method

werbenhu · Apr 6, 2023 · 1412bf7 · 1412bf7
1 parent 936e907
commit 1412bf7
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/group.go b/group.go
@@ -113,3 +113,15 @@ func (b *Group) Match(key string) (string, []byte, error) {
  }
  return "", nil, ErrNoResultMatched
 }
+
+func (b *Group) GetElements() []*Element {
+ b.RLock()
+ defer b.RUnlock()
+
+ els := make([]*Element, 0)
+ for _, e := range b.Elements {
+ els = append(els, e)
+ }
+
+ return els
+}
diff --git a/group_test.go b/group_test.go
@@ -115,6 +115,24 @@ func TestGroupAll(t *testing.T) {
  }
 }
 
+func TestGroupGetElements(t *testing.T) {
+ group := NewGroup("test", 10000)
+ group.Insert("192.168.1.100:1883", []byte("werbenhu100"))
+ group.Insert("192.168.1.101:1883", []byte("werbenhu101"))
+
+ els := group.GetElements()
+ assert.Equal(t, []*Element{
+ {
+ Key: "192.168.1.100:1883",
+ Payload: []byte("werbenhu100"),
+ },
+ {
+ Key: "192.168.1.101:1883",
+ Payload: []byte("werbenhu101"),
+ },
+ }, els)
+}
+
 func BenchmarkGroupHash(b *testing.B) {
  b.ResetTimer()
  key := "192.168.1.100:1883"