1Panel/backend/utils/nginx/components/upstream_server.go

78 lines
1.5 KiB
Go
Raw Normal View History

package components
import (
"fmt"
"sort"
"strings"
)
type UpstreamServer struct {
Comment string
Address string
Flags []string
Parameters map[string]string
}
func (uss *UpstreamServer) GetName() string {
return "server"
}
func (uss *UpstreamServer) GetBlock() IBlock {
return nil
}
func (uss *UpstreamServer) GetParameters() []string {
return uss.GetDirective().Parameters
}
func (uss *UpstreamServer) GetComment() string {
return uss.Comment
}
func (uss *UpstreamServer) GetDirective() *Directive {
directive := &Directive{
Name: "server",
Parameters: make([]string, 0),
Block: nil,
}
directive.Parameters = append(directive.Parameters, uss.Address)
paramNames := make([]string, 0)
for k := range uss.Parameters {
paramNames = append(paramNames, k)
}
sort.Strings(paramNames)
for _, k := range paramNames {
directive.Parameters = append(directive.Parameters, fmt.Sprintf("%s=%s", k, uss.Parameters[k]))
}
directive.Parameters = append(directive.Parameters, uss.Flags...)
return directive
}
func NewUpstreamServer(directive IDirective) *UpstreamServer {
uss := &UpstreamServer{
Comment: directive.GetComment(),
Flags: make([]string, 0),
Parameters: make(map[string]string, 0),
}
for i, parameter := range directive.GetParameters() {
if i == 0 {
uss.Address = parameter
continue
}
if strings.Contains(parameter, "=") {
s := strings.SplitN(parameter, "=", 2)
uss.Parameters[s[0]] = s[1]
} else {
uss.Flags = append(uss.Flags, parameter)
}
}
return uss
}