refactor(query): replace anonymous functions with sequence and for loop

This commit is contained in:
yusing 2025-06-03 20:51:00 +08:00
parent e737737415
commit 58db228e25
2 changed files with 12 additions and 21 deletions

View file

@ -2,36 +2,25 @@ package config
import (
config "github.com/yusing/go-proxy/internal/config/types"
"github.com/yusing/go-proxy/internal/route"
"github.com/yusing/go-proxy/internal/route/provider"
)
func (cfg *Config) DumpRoutes() map[string]*route.Route {
entries := make(map[string]*route.Route)
cfg.providers.RangeAll(func(_ string, p *provider.Provider) {
p.RangeRoutes(func(alias string, r *route.Route) {
entries[alias] = r
})
})
return entries
}
func (cfg *Config) DumpRouteProviders() map[string]*provider.Provider {
entries := make(map[string]*provider.Provider)
cfg.providers.RangeAll(func(_ string, p *provider.Provider) {
entries := make(map[string]*provider.Provider, cfg.providers.Size())
for _, p := range cfg.providers.Range {
entries[p.ShortName()] = p
})
}
return entries
}
func (cfg *Config) RouteProviderList() []config.RouteProviderListResponse {
var list []config.RouteProviderListResponse
cfg.providers.RangeAll(func(_ string, p *provider.Provider) {
list := make([]config.RouteProviderListResponse, 0, cfg.providers.Size())
for _, p := range cfg.providers.Range {
list = append(list, config.RouteProviderListResponse{
ShortName: p.ShortName(),
FullName: p.String(),
})
})
}
return list
}
@ -40,13 +29,13 @@ func (cfg *Config) Statistics() map[string]any {
var total uint16
providerStats := make(map[string]provider.ProviderStats)
cfg.providers.RangeAll(func(_ string, p *provider.Provider) {
for _, p := range cfg.providers.Range {
stats := p.Statistics()
providerStats[p.ShortName()] = stats
rps.AddOther(stats.RPs)
streams.AddOther(stats.Streams)
total += stats.RPs.Total + stats.Streams.Total
})
}
return map[string]any{
"total": total,

View file

@ -137,9 +137,11 @@ func (p *Provider) Start(parent task.Parent) gperr.Error {
return nil
}
func (p *Provider) RangeRoutes(do func(string, *route.Route)) {
func (p *Provider) IterRoutes(yield func(string, *route.Route) bool) {
for alias, r := range p.routes {
do(alias, r)
if !yield(alias, r) {
break
}
}
}