AdguardTeam/AdGuardHome
1
1
Fork 0
mirror of https://github.com/AdguardTeam/AdGuardHome.git synced 2024-07-14 22:00:25 +03:00
Code Issues Packages Projects Releases Wiki Activity

client: imp code

Browse Source
This commit is contained in:
Stanislav Chzhen 2024-05-06 20:55:43 +03:00
parent 54212e975b
commit 7faddd8aad
2 changed files with 10 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
internal/client/index.go
View File

@ -337,14 +337,13 @@ func (ci *Index) Range(f func(c *Persistent) (cont bool)) {
}
}
// SortedRange is like [Index.Range] but sorts the keys before iterating
// ensuring a predictable order.
func (ci *Index) SortedRange(
s func(a, b *Persistent) (n int),
f func(c *Persistent) (cont bool),
) {
// RangeByName is like [Index.Range] but sorts the persistent clients by name
// before iterating ensuring a predictable order.
func (ci *Index) RangeByName(f func(c *Persistent) (cont bool)) {
cs := maps.Values(ci.uidToClient)
slices.SortFunc(cs, s)
slices.SortFunc(cs, func(a, b *Persistent) (n int) {
return strings.Compare(a.Name, b.Name)
})
for _, c := range cs {
if !f(c) {
@ -355,12 +354,8 @@ func (ci *Index) SortedRange(
// CloseUpstreams closes upstream configurations of persistent clients.
func (ci *Index) CloseUpstreams() (err error) {
sortFunc := func(a, b *Persistent) (n int) {
return strings.Compare(a.Name, b.Name)
}
var errs []error
ci.SortedRange(sortFunc, func(c *Persistent) (cont bool) {
ci.RangeByName(func(c *Persistent) (cont bool) {
err = c.CloseUpstreams()
if err != nil {
errs = append(errs, err)

8
internal/client/index_internal_test.go
View File

@ -125,14 +125,12 @@ func TestClientIndex(t *testing.T) {
})
t.Run("sorted_range", func(t *testing.T) {
sortFunc := func(a, b *Persistent) (n int) {
slices.SortFunc(clients, func(a, b *Persistent) (n int) {
return strings.Compare(a.Name, b.Name)
}
slices.SortFunc(clients, sortFunc)
})
got := []*Persistent{}
ci.SortedRange(sortFunc, func(c *Persistent) (cont bool) {
ci.RangeByName(func(c *Persistent) (cont bool) {
got = append(got, c)
return true