mirror of https://gitee.com/IrisVega/frp.git
137 lines
2.4 KiB
Go
137 lines
2.4 KiB
Go
package vhost
|
|
|
|
import (
|
|
"cmp"
|
|
"errors"
|
|
"slices"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
var ErrRouterConfigConflict = errors.New("router config conflict")
|
|
|
|
type routerByHTTPUser map[string][]*Router
|
|
|
|
type Routers struct {
|
|
indexByDomain map[string]routerByHTTPUser
|
|
|
|
mutex sync.RWMutex
|
|
}
|
|
|
|
type Router struct {
|
|
domain string
|
|
location string
|
|
httpUser string
|
|
|
|
// store any object here
|
|
payload interface{}
|
|
}
|
|
|
|
func NewRouters() *Routers {
|
|
return &Routers{
|
|
indexByDomain: make(map[string]routerByHTTPUser),
|
|
}
|
|
}
|
|
|
|
func (r *Routers) Add(domain, location, httpUser string, payload interface{}) error {
|
|
domain = strings.ToLower(domain)
|
|
|
|
r.mutex.Lock()
|
|
defer r.mutex.Unlock()
|
|
|
|
if _, exist := r.exist(domain, location, httpUser); exist {
|
|
return ErrRouterConfigConflict
|
|
}
|
|
|
|
routersByHTTPUser, found := r.indexByDomain[domain]
|
|
if !found {
|
|
routersByHTTPUser = make(map[string][]*Router)
|
|
}
|
|
vrs, found := routersByHTTPUser[httpUser]
|
|
if !found {
|
|
vrs = make([]*Router, 0, 1)
|
|
}
|
|
|
|
vr := &Router{
|
|
domain: domain,
|
|
location: location,
|
|
httpUser: httpUser,
|
|
payload: payload,
|
|
}
|
|
vrs = append(vrs, vr)
|
|
|
|
slices.SortFunc(vrs, func(a, b *Router) int {
|
|
return -cmp.Compare(a.location, b.location)
|
|
})
|
|
|
|
routersByHTTPUser[httpUser] = vrs
|
|
r.indexByDomain[domain] = routersByHTTPUser
|
|
return nil
|
|
}
|
|
|
|
func (r *Routers) Del(domain, location, httpUser string) {
|
|
domain = strings.ToLower(domain)
|
|
|
|
r.mutex.Lock()
|
|
defer r.mutex.Unlock()
|
|
|
|
routersByHTTPUser, found := r.indexByDomain[domain]
|
|
if !found {
|
|
return
|
|
}
|
|
|
|
vrs, found := routersByHTTPUser[httpUser]
|
|
if !found {
|
|
return
|
|
}
|
|
newVrs := make([]*Router, 0)
|
|
for _, vr := range vrs {
|
|
if vr.location != location {
|
|
newVrs = append(newVrs, vr)
|
|
}
|
|
}
|
|
routersByHTTPUser[httpUser] = newVrs
|
|
}
|
|
|
|
func (r *Routers) Get(host, path, httpUser string) (vr *Router, exist bool) {
|
|
host = strings.ToLower(host)
|
|
|
|
r.mutex.RLock()
|
|
defer r.mutex.RUnlock()
|
|
|
|
routersByHTTPUser, found := r.indexByDomain[host]
|
|
if !found {
|
|
return
|
|
}
|
|
|
|
vrs, found := routersByHTTPUser[httpUser]
|
|
if !found {
|
|
return
|
|
}
|
|
|
|
for _, vr = range vrs {
|
|
if strings.HasPrefix(path, vr.location) {
|
|
return vr, true
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func (r *Routers) exist(host, path, httpUser string) (route *Router, exist bool) {
|
|
routersByHTTPUser, found := r.indexByDomain[host]
|
|
if !found {
|
|
return
|
|
}
|
|
routers, found := routersByHTTPUser[httpUser]
|
|
if !found {
|
|
return
|
|
}
|
|
|
|
for _, route = range routers {
|
|
if path == route.location {
|
|
return route, true
|
|
}
|
|
}
|
|
return
|
|
}
|