mirror of
https://github.com/duke-git/lancet.git
synced 2026-02-04 12:52:28 +08:00
feat: add Pop for set
This commit is contained in:
@@ -181,3 +181,15 @@ func (s Set[T]) EachWithBreak(iteratee func(item T) bool) {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Pop delete the top element of set then return it, if set is empty, return nil-value of T and false.
|
||||
func (s Set[T]) Pop() (v T, ok bool) {
|
||||
if len(s) > 0 {
|
||||
items := s.Values()
|
||||
item := items[len(s)-1]
|
||||
delete(s, item)
|
||||
return item, true
|
||||
}
|
||||
|
||||
return v, false
|
||||
}
|
||||
|
||||
@@ -209,3 +209,23 @@ func TestEachWithBreak(t *testing.T) {
|
||||
assert := internal.NewAssert(t, "TestEachWithBreak")
|
||||
assert.Equal(6, sum)
|
||||
}
|
||||
|
||||
func TestPop(t *testing.T) {
|
||||
assert := internal.NewAssert(t, "TestPop")
|
||||
|
||||
s := NewSet[int]()
|
||||
|
||||
val, ok := s.Pop()
|
||||
assert.Equal(0, val)
|
||||
assert.Equal(false, ok)
|
||||
|
||||
s.Add(1)
|
||||
s.Add(2)
|
||||
s.Add(3)
|
||||
|
||||
// s = NewSet(1, 2, 3, 4, 5)
|
||||
|
||||
val, ok = s.Pop()
|
||||
assert.Equal(3, val)
|
||||
assert.Equal(true, ok)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user