mirror of
https://github.com/yusing/godoxy.git
synced 2026-03-27 11:31:06 +01:00
refactor: move accesslog to logging/accesslog
This commit is contained in:
52
internal/logging/accesslog/status_code_range.go
Normal file
52
internal/logging/accesslog/status_code_range.go
Normal file
@@ -0,0 +1,52 @@
|
||||
package accesslog
|
||||
|
||||
import (
|
||||
"strconv"
|
||||
|
||||
"github.com/yusing/go-proxy/internal/gperr"
|
||||
"github.com/yusing/go-proxy/internal/utils/strutils"
|
||||
)
|
||||
|
||||
type StatusCodeRange struct {
|
||||
Start int
|
||||
End int
|
||||
}
|
||||
|
||||
var ErrInvalidStatusCodeRange = gperr.New("invalid status code range")
|
||||
|
||||
func (r *StatusCodeRange) Includes(code int) bool {
|
||||
return r.Start <= code && code <= r.End
|
||||
}
|
||||
|
||||
// Parse implements strutils.Parser.
|
||||
func (r *StatusCodeRange) Parse(v string) error {
|
||||
split := strutils.SplitRune(v, '-')
|
||||
switch len(split) {
|
||||
case 1:
|
||||
start, err := strconv.Atoi(split[0])
|
||||
if err != nil {
|
||||
return gperr.Wrap(err)
|
||||
}
|
||||
r.Start = start
|
||||
r.End = start
|
||||
return nil
|
||||
case 2:
|
||||
start, errStart := strconv.Atoi(split[0])
|
||||
end, errEnd := strconv.Atoi(split[1])
|
||||
if err := gperr.Join(errStart, errEnd); err != nil {
|
||||
return err
|
||||
}
|
||||
r.Start = start
|
||||
r.End = end
|
||||
return nil
|
||||
default:
|
||||
return ErrInvalidStatusCodeRange.Subject(v)
|
||||
}
|
||||
}
|
||||
|
||||
func (r *StatusCodeRange) String() string {
|
||||
if r.Start == r.End {
|
||||
return strconv.Itoa(r.Start)
|
||||
}
|
||||
return strconv.Itoa(r.Start) + "-" + strconv.Itoa(r.End)
|
||||
}
|
||||
Reference in New Issue
Block a user