malhuda/alice-lg
1
0
Fork 0
Code Issues 39 Pull Requests 11 Actions Packages Projects Releases 2 Wiki Activity

use pointer for neighbor filtering

Browse Source
This commit is contained in:
Annika Hannig 2022-11-25 11:43:19 +01:00
parent 9955f3feeb
commit e30d98f0cf
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pkg/store/backends/memory/routes_backend.go
View File

@ -6,6 +6,7 @@ import (
"sync"
"github.com/alice-lg/alice-lg/pkg/api"
"github.com/alice-lg/alice-lg/pkg/pools"
"github.com/alice-lg/alice-lg/pkg/sources"
)
@ -70,9 +71,18 @@ func (r *RoutesBackend) FindByNeighbors(
) (api.LookupRoutes, error) {
result := api.LookupRoutes{}
// Use neighbor ID pointers from pool
neighborPtrs := make([]*string, 0, len(neighborIDs))
for _, id := range neighborIDs {
ptr := pools.Neighbors.Get(id)
if ptr != nil {
neighborPtrs = append(neighborPtrs, ptr)
}
}
r.routes.Range(func(k, rs interface{}) bool {
for _, route := range rs.(api.LookupRoutes) {
if isMemberOf(neighborIDs, *route.NeighborID) {
if isMemberOf(neighborPtrs, route.NeighborID) {
result = append(result, route)
}
}
@ -104,7 +114,7 @@ func (r *RoutesBackend) FindByPrefix(
// isMemberOf checks if a key is present in
// a list of strings.
func isMemberOf(list []string, key string) bool {
func isMemberOf(list []*string, key *string) bool {
for _, v := range list {
if v == key {
return true