mirror of
https://github.com/buger/goreplay.git
synced 2024-04-21 12:32:02 +00:00
32 lines
583 B
Go
32 lines
583 B
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
type urlRewrite struct {
|
|
src *regexp.Regexp
|
|
target []byte
|
|
}
|
|
|
|
type UrlRewriteMap []urlRewrite
|
|
|
|
func (r *UrlRewriteMap) String() string {
|
|
return fmt.Sprint(*r)
|
|
}
|
|
|
|
func (r *UrlRewriteMap) Set(value string) error {
|
|
valArr := strings.SplitN(value, ":", 2)
|
|
if len(valArr) < 2 {
|
|
return errors.New("need both src and target, colon-delimited (ex. /a:/b).")
|
|
}
|
|
regexp, err := regexp.Compile(valArr[0])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*r = append(*r, urlRewrite{src: regexp, target: []byte(valArr[1]) })
|
|
return nil
|
|
} |