You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
1.9 KiB
82 lines
1.9 KiB
// Copyright 2019 HAProxy Technologies
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
|
|
package runtime
|
|
|
|
import (
|
|
"strconv"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
//nolint:gochecknoglobals
|
|
var (
|
|
possibleStates map[string]struct{}
|
|
oncePossibleStates sync.Once
|
|
)
|
|
|
|
//nolint:gochecknoglobals
|
|
var (
|
|
possibleHealths map[string]struct{}
|
|
oncePossibleHealths sync.Once
|
|
)
|
|
|
|
// ServerStateValid checks if server state is valid
|
|
func ServerStateValid(state string) bool {
|
|
oncePossibleStates.Do(func() {
|
|
possibleStates = map[string]struct{}{
|
|
"ready": {},
|
|
"drain": {},
|
|
"maint": {},
|
|
}
|
|
})
|
|
_, ok := possibleStates[state]
|
|
return ok
|
|
}
|
|
|
|
// ServerHealthValid checks if server health is valid
|
|
func ServerHealthValid(health string) bool {
|
|
oncePossibleHealths.Do(func() {
|
|
possibleHealths = map[string]struct{}{
|
|
"up": {},
|
|
"stopping": {},
|
|
"down": {},
|
|
}
|
|
})
|
|
_, ok := possibleHealths[health]
|
|
return ok
|
|
}
|
|
|
|
// ServerWeightValid checks if server state is valid
|
|
func ServerWeightValid(weight string) bool {
|
|
var n int64
|
|
var err error
|
|
if strings.HasSuffix(weight, "%") {
|
|
percent := strings.TrimSuffix(weight, "%")
|
|
if n, err = strconv.ParseInt(percent, 10, 64); err != nil {
|
|
return false
|
|
}
|
|
if n > -1 && n < 101 {
|
|
return true
|
|
}
|
|
}
|
|
if n, err = strconv.ParseInt(weight, 10, 64); err != nil {
|
|
return false
|
|
}
|
|
if n > -1 && n < 257 {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|