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.
57 lines
1.2 KiB
57 lines
1.2 KiB
2 years ago
|
package matcher
|
||
|
|
|
||
|
|
import (
|
||
|
|
"git.diulo.com/mogfee/kit/middleware"
|
||
|
|
"sort"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Matcher interface {
|
||
|
|
Use(ms ...middleware.Middleware)
|
||
|
|
Add(selector string, ms ...middleware.Middleware)
|
||
|
|
Match(operation string) []middleware.Middleware
|
||
|
|
}
|
||
|
|
|
||
|
|
func New() Matcher {
|
||
|
|
return &matcher{
|
||
|
|
matchs: make(map[string][]middleware.Middleware),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type matcher struct {
|
||
|
|
prefix []string
|
||
|
|
defaults []middleware.Middleware
|
||
|
|
matchs map[string][]middleware.Middleware
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *matcher) Use(ms ...middleware.Middleware) {
|
||
|
|
m.defaults = ms
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *matcher) Add(selector string, ms ...middleware.Middleware) {
|
||
|
|
if strings.HasSuffix(selector, "*") {
|
||
|
|
selector = strings.TrimSuffix(selector, "*")
|
||
|
|
m.prefix = append(m.prefix, selector)
|
||
|
|
sort.Slice(m.prefix, func(i, j int) bool {
|
||
|
|
return m.prefix[i] > m.prefix[j]
|
||
|
|
})
|
||
|
|
}
|
||
|
|
m.matchs[selector] = ms
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *matcher) Match(operation string) []middleware.Middleware {
|
||
|
|
ms := make([]middleware.Middleware, 0, len(m.defaults))
|
||
|
|
if len(m.defaults) > 0 {
|
||
|
|
ms = append(ms, m.defaults...)
|
||
|
|
}
|
||
|
|
if next, ok := m.matchs[operation]; ok {
|
||
|
|
ms = append(ms, next...)
|
||
|
|
}
|
||
|
|
for _, prefix := range m.prefix {
|
||
|
|
if strings.HasPrefix(operation, prefix) {
|
||
|
|
return append(ms, m.matchs[prefix]...)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return ms
|
||
|
|
}
|