TyleoDvDelaware3217/GoDoxy
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2025-06-07 12:02:34 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

27
internal/config/query.go
View file

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

6
internal/route/provider/provider.go
View file

@ -137,9 +137,11 @@ func (p *Provider) Start(parent task.Parent) gperr.Error {
return nil 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 { for alias, r := range p.routes {
do(alias, r) if !yield(alias, r) {
break
}
} }
} }