api/router: support pcre and google.api pattern matching (#1549)
* api/router: support pcre and google.api pattern matching Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
parent
6db825838a
commit
5821c18193
48
static.go
48
static.go
@ -21,6 +21,7 @@ type endpoint struct {
|
|||||||
apiep *api.Endpoint
|
apiep *api.Endpoint
|
||||||
hostregs []*regexp.Regexp
|
hostregs []*regexp.Regexp
|
||||||
pathregs []util.Pattern
|
pathregs []util.Pattern
|
||||||
|
pcreregs []*regexp.Regexp
|
||||||
}
|
}
|
||||||
|
|
||||||
// router is the default router
|
// router is the default router
|
||||||
@ -94,6 +95,7 @@ func (r *staticRouter) Register(ep *api.Endpoint) error {
|
|||||||
|
|
||||||
var pathregs []util.Pattern
|
var pathregs []util.Pattern
|
||||||
var hostregs []*regexp.Regexp
|
var hostregs []*regexp.Regexp
|
||||||
|
var pcreregs []*regexp.Regexp
|
||||||
|
|
||||||
for _, h := range ep.Host {
|
for _, h := range ep.Host {
|
||||||
if h == "" || h == "*" {
|
if h == "" || h == "*" {
|
||||||
@ -107,9 +109,18 @@ func (r *staticRouter) Register(ep *api.Endpoint) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, p := range ep.Path {
|
for _, p := range ep.Path {
|
||||||
|
var pcreok bool
|
||||||
|
pcrereg, err := regexp.CompilePOSIX(p)
|
||||||
|
if err == nil {
|
||||||
|
pcreregs = append(pcreregs, pcrereg)
|
||||||
|
pcreok = true
|
||||||
|
}
|
||||||
|
|
||||||
rule, err := util.Parse(p)
|
rule, err := util.Parse(p)
|
||||||
if err != nil {
|
if err != nil && !pcreok {
|
||||||
return err
|
return err
|
||||||
|
} else if err != nil && pcreok {
|
||||||
|
continue
|
||||||
}
|
}
|
||||||
tpl := rule.Compile()
|
tpl := rule.Compile()
|
||||||
pathreg, err := util.NewPattern(tpl.Version, tpl.OpCodes, tpl.Pool, "")
|
pathreg, err := util.NewPattern(tpl.Version, tpl.OpCodes, tpl.Pool, "")
|
||||||
@ -120,7 +131,12 @@ func (r *staticRouter) Register(ep *api.Endpoint) error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
r.Lock()
|
r.Lock()
|
||||||
r.eps[ep.Name] = &endpoint{apiep: ep, pathregs: pathregs, hostregs: hostregs}
|
r.eps[ep.Name] = &endpoint{
|
||||||
|
apiep: ep,
|
||||||
|
pcreregs: pcreregs,
|
||||||
|
pathregs: pathregs,
|
||||||
|
hostregs: hostregs,
|
||||||
|
}
|
||||||
r.Unlock()
|
r.Unlock()
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
@ -219,11 +235,10 @@ func (r *staticRouter) endpoint(req *http.Request) (*endpoint, error) {
|
|||||||
var mMatch, hMatch, pMatch bool
|
var mMatch, hMatch, pMatch bool
|
||||||
|
|
||||||
// 1. try method
|
// 1. try method
|
||||||
methodLoop:
|
|
||||||
for _, m := range ep.apiep.Method {
|
for _, m := range ep.apiep.Method {
|
||||||
if m == req.Method {
|
if m == req.Method {
|
||||||
mMatch = true
|
mMatch = true
|
||||||
break methodLoop
|
break
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if !mMatch {
|
if !mMatch {
|
||||||
@ -237,15 +252,14 @@ func (r *staticRouter) endpoint(req *http.Request) (*endpoint, error) {
|
|||||||
if len(ep.apiep.Host) == 0 {
|
if len(ep.apiep.Host) == 0 {
|
||||||
hMatch = true
|
hMatch = true
|
||||||
} else {
|
} else {
|
||||||
hostLoop:
|
|
||||||
for idx, h := range ep.apiep.Host {
|
for idx, h := range ep.apiep.Host {
|
||||||
if h == "" || h == "*" {
|
if h == "" || h == "*" {
|
||||||
hMatch = true
|
hMatch = true
|
||||||
break hostLoop
|
break
|
||||||
} else {
|
} else {
|
||||||
if ep.hostregs[idx].MatchString(req.URL.Host) {
|
if ep.hostregs[idx].MatchString(req.URL.Host) {
|
||||||
hMatch = true
|
hMatch = true
|
||||||
break hostLoop
|
break
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -257,13 +271,12 @@ func (r *staticRouter) endpoint(req *http.Request) (*endpoint, error) {
|
|||||||
logger.Debugf("api host match %s", req.URL.Host)
|
logger.Debugf("api host match %s", req.URL.Host)
|
||||||
}
|
}
|
||||||
|
|
||||||
// 3. try path
|
// 3. try google.api path
|
||||||
pathLoop:
|
|
||||||
for _, pathreg := range ep.pathregs {
|
for _, pathreg := range ep.pathregs {
|
||||||
matches, err := pathreg.Match(path, "")
|
matches, err := pathreg.Match(path, "")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if logger.V(logger.DebugLevel, logger.DefaultLogger) {
|
if logger.V(logger.DebugLevel, logger.DefaultLogger) {
|
||||||
logger.Debugf("api path not match %s != %v", path, pathreg)
|
logger.Debugf("api gpath not match %s != %v", path, pathreg)
|
||||||
}
|
}
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
@ -278,8 +291,21 @@ func (r *staticRouter) endpoint(req *http.Request) (*endpoint, error) {
|
|||||||
}
|
}
|
||||||
md["x-api-body"] = ep.apiep.Body
|
md["x-api-body"] = ep.apiep.Body
|
||||||
*req = *req.Clone(metadata.NewContext(ctx, md))
|
*req = *req.Clone(metadata.NewContext(ctx, md))
|
||||||
break pathLoop
|
break
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 4. try path via pcre path matching
|
||||||
|
for _, pathreg := range ep.pcreregs {
|
||||||
|
if !pathreg.MatchString(req.URL.Path) {
|
||||||
|
if logger.V(logger.DebugLevel, logger.DefaultLogger) {
|
||||||
|
logger.Debugf("api pcre path not match %s != %v", req.URL.Path, pathreg)
|
||||||
|
}
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
pMatch = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
if !pMatch {
|
if !pMatch {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user