100 lines
3 KiB
Go
100 lines
3 KiB
Go
|
package runtime
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/containous/traefik/v2/pkg/config/dynamic"
|
||
|
|
"github.com/containous/traefik/v2/pkg/log"
|
||
|
|
)
|
||
|
|
|
||
|
|
// GetTCPRoutersByEntryPoints returns all the tcp routers by entry points name and routers name
|
||
|
|
func (c *Configuration) GetTCPRoutersByEntryPoints(ctx context.Context, entryPoints []string) map[string]map[string]*TCPRouterInfo {
|
||
|
|
entryPointsRouters := make(map[string]map[string]*TCPRouterInfo)
|
||
|
|
|
||
|
|
for rtName, rt := range c.TCPRouters {
|
||
|
|
eps := rt.EntryPoints
|
||
|
|
if len(eps) == 0 {
|
||
|
|
eps = entryPoints
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, entryPointName := range eps {
|
||
|
|
if !contains(entryPoints, entryPointName) {
|
||
|
|
log.FromContext(log.With(ctx, log.Str(log.EntryPointName, entryPointName))).
|
||
|
|
Errorf("entryPoint %q doesn't exist", entryPointName)
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, ok := entryPointsRouters[entryPointName]; !ok {
|
||
|
|
entryPointsRouters[entryPointName] = make(map[string]*TCPRouterInfo)
|
||
|
|
}
|
||
|
|
|
||
|
|
entryPointsRouters[entryPointName][rtName] = rt
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return entryPointsRouters
|
||
|
|
}
|
||
|
|
|
||
|
|
// TCPRouterInfo holds information about a currently running TCP router
|
||
|
|
type TCPRouterInfo struct {
|
||
|
|
*dynamic.TCPRouter // dynamic configuration
|
||
|
|
Err []string `json:"error,omitempty"` // initialization error
|
||
|
|
// Status reports whether the router is disabled, in a warning state, or all good (enabled).
|
||
|
|
// If not in "enabled" state, the reason for it should be in the list of Err.
|
||
|
|
// It is the caller's responsibility to set the initial status.
|
||
|
|
Status string `json:"status,omitempty"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// AddError adds err to r.Err, if it does not already exist.
|
||
|
|
// If critical is set, r is marked as disabled.
|
||
|
|
func (r *TCPRouterInfo) AddError(err error, critical bool) {
|
||
|
|
for _, value := range r.Err {
|
||
|
|
if value == err.Error() {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
r.Err = append(r.Err, err.Error())
|
||
|
|
if critical {
|
||
|
|
r.Status = StatusDisabled
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// only set it to "warning" if not already in a worse state
|
||
|
|
if r.Status != StatusDisabled {
|
||
|
|
r.Status = StatusWarning
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// TCPServiceInfo holds information about a currently running TCP service
|
||
|
|
type TCPServiceInfo struct {
|
||
|
|
*dynamic.TCPService // dynamic configuration
|
||
|
|
Err []string `json:"error,omitempty"` // initialization error
|
||
|
|
// Status reports whether the service is disabled, in a warning state, or all good (enabled).
|
||
|
|
// If not in "enabled" state, the reason for it should be in the list of Err.
|
||
|
|
// It is the caller's responsibility to set the initial status.
|
||
|
|
Status string `json:"status,omitempty"`
|
||
|
|
UsedBy []string `json:"usedBy,omitempty"` // list of routers using that service
|
||
|
|
}
|
||
|
|
|
||
|
|
// AddError adds err to s.Err, if it does not already exist.
|
||
|
|
// If critical is set, s is marked as disabled.
|
||
|
|
func (s *TCPServiceInfo) AddError(err error, critical bool) {
|
||
|
|
for _, value := range s.Err {
|
||
|
|
if value == err.Error() {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
s.Err = append(s.Err, err.Error())
|
||
|
|
if critical {
|
||
|
|
s.Status = StatusDisabled
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// only set it to "warning" if not already in a worse state
|
||
|
|
if s.Status != StatusDisabled {
|
||
|
|
s.Status = StatusWarning
|
||
|
|
}
|
||
|
|
}
|