42 lines
1.1 KiB
Go
42 lines
1.1 KiB
Go
package types
|
|
|
|
import (
|
|
"slices"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
// HTTPCodeRanges holds HTTP code ranges.
|
|
type HTTPCodeRanges [][2]int
|
|
|
|
// NewHTTPCodeRanges creates HTTPCodeRanges from a given []string.
|
|
// Break out the http status code ranges into a low int and high int
|
|
// for ease of use at runtime.
|
|
func NewHTTPCodeRanges(strBlocks []string) (HTTPCodeRanges, error) {
|
|
var blocks HTTPCodeRanges
|
|
for _, block := range strBlocks {
|
|
codes := strings.Split(block, "-")
|
|
// if only a single HTTP code was configured, assume the best and create the correct configuration on the user's behalf
|
|
if len(codes) == 1 {
|
|
codes = append(codes, codes[0])
|
|
}
|
|
lowCode, err := strconv.Atoi(codes[0])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
highCode, err := strconv.Atoi(codes[1])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
blocks = append(blocks, [2]int{lowCode, highCode})
|
|
}
|
|
return blocks, nil
|
|
}
|
|
|
|
// Contains tests whether the passed status code is within one of its HTTP code ranges.
|
|
func (h HTTPCodeRanges) Contains(statusCode int) bool {
|
|
return slices.ContainsFunc(h, func(block [2]int) bool {
|
|
return statusCode >= block[0] && statusCode <= block[1]
|
|
})
|
|
}
|