55 lines
1.1 KiB
Go
55 lines
1.1 KiB
Go
|
package gateway
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/traefik/traefik/v3/pkg/config/label"
|
||
|
|
)
|
||
|
|
|
||
|
|
const annotationsPrefix = "traefik.io/"
|
||
|
|
|
||
|
|
// ServiceConfig is the service's root configuration from annotations.
|
||
|
|
type ServiceConfig struct {
|
||
|
|
Service Service `json:"service"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// Service is the service's configuration from annotations.
|
||
|
|
type Service struct {
|
||
|
|
NativeLB bool `json:"nativeLB"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func parseServiceAnnotations(annotations map[string]string) (ServiceConfig, error) {
|
||
|
|
var svcConf ServiceConfig
|
||
|
|
|
||
|
|
labels := convertAnnotations(annotations)
|
||
|
|
if len(labels) == 0 {
|
||
|
|
return svcConf, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := label.Decode(labels, &svcConf, "traefik.service."); err != nil {
|
||
|
|
return svcConf, fmt.Errorf("decoding labels: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return svcConf, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func convertAnnotations(annotations map[string]string) map[string]string {
|
||
|
|
if len(annotations) == 0 {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
result := make(map[string]string)
|
||
|
|
|
||
|
|
for key, value := range annotations {
|
||
|
|
if !strings.HasPrefix(key, annotationsPrefix) {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
newKey := strings.ReplaceAll(key, "io/", "")
|
||
|
|
result[newKey] = value
|
||
|
|
}
|
||
|
|
|
||
|
|
return result
|
||
|
|
}
|