Peter Kurfer
4c2aa968d2
- support config file - support reading checks, server definition and endpoint definitions from env
57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package config
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
var (
|
|
ErrServerNameAlreadyRegistered = errors.New("server name is already registered")
|
|
ErrNoSuchServer = errors.New("no known server with given name")
|
|
|
|
_ ServerLookup = (*ServerRegister)(nil)
|
|
)
|
|
|
|
type ServerLookup interface {
|
|
Lookup(name string) (*Server, error)
|
|
}
|
|
|
|
func NewServerRegister() *ServerRegister {
|
|
return &ServerRegister{
|
|
servers: make(map[string]Server),
|
|
}
|
|
}
|
|
|
|
type ServerRegister struct {
|
|
lock sync.RWMutex
|
|
servers map[string]Server
|
|
}
|
|
|
|
func (l *ServerRegister) Register(name string, srv Server) error {
|
|
l.lock.Lock()
|
|
defer l.lock.Unlock()
|
|
|
|
name = strings.ToLower(name)
|
|
|
|
if _, ok := l.servers[name]; ok {
|
|
return fmt.Errorf("%w: %s", ErrServerNameAlreadyRegistered, name)
|
|
}
|
|
|
|
l.servers[name] = srv
|
|
return nil
|
|
}
|
|
|
|
func (l *ServerRegister) Lookup(name string) (*Server, error) {
|
|
l.lock.RLock()
|
|
defer l.lock.RUnlock()
|
|
|
|
name = strings.ToLower(name)
|
|
|
|
match, ok := l.servers[name]
|
|
if !ok {
|
|
return nil, fmt.Errorf("%w: %s", ErrNoSuchServer, name)
|
|
}
|
|
return &match, nil
|
|
}
|