add web api

This commit is contained in:
ginuerzh
2022-02-08 23:46:54 +08:00
parent 0fba6d2500
commit f1eaef8d69
22 changed files with 1209 additions and 33 deletions

View File

@ -32,21 +32,30 @@ func (r *resolverRegistry) Unregister(name string) {
r.m.Delete(name)
}
func (r *resolverRegistry) IsRegistered(name string) bool {
_, ok := r.m.Load(name)
return ok
}
func (r *resolverRegistry) Get(name string) resolver.Resolver {
if _, ok := r.m.Load(name); !ok {
return nil
}
return &resolverWrapper{name: name}
}
func (r *resolverRegistry) get(name string) resolver.Resolver {
if v, ok := r.m.Load(name); ok {
return v.(resolver.Resolver)
}
return nil
}
type resolverWrapper struct {
name string
}
func (w *resolverWrapper) Resolve(ctx context.Context, network, host string) ([]net.IP, error) {
r := Resolver().Get(w.name)
r := Resolver().get(w.name)
if r == nil {
return nil, ErrNotFound
return nil, resolver.ErrInvalid
}
return r.Resolve(ctx, network, host)
}