mirror of https://github.com/v2fly/v2ray-core.git
70 lines
1.5 KiB
Go
70 lines
1.5 KiB
Go
package proxyman
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"v2ray.com/core/app"
|
|
"v2ray.com/core/proxy"
|
|
)
|
|
|
|
const (
|
|
APP_ID_INBOUND_MANAGER = app.ID(4)
|
|
APP_ID_OUTBOUND_MANAGER = app.ID(6)
|
|
)
|
|
|
|
type InboundHandlerManager interface {
|
|
GetHandler(tag string) (proxy.InboundHandler, int)
|
|
}
|
|
|
|
type OutboundHandlerManager interface {
|
|
GetHandler(tag string) proxy.OutboundHandler
|
|
GetDefaultHandler() proxy.OutboundHandler
|
|
}
|
|
|
|
type DefaultOutboundHandlerManager struct {
|
|
sync.RWMutex
|
|
defaultHandler proxy.OutboundHandler
|
|
taggedHandler map[string]proxy.OutboundHandler
|
|
}
|
|
|
|
func NewDefaultOutboundHandlerManager() *DefaultOutboundHandlerManager {
|
|
return &DefaultOutboundHandlerManager{
|
|
taggedHandler: make(map[string]proxy.OutboundHandler),
|
|
}
|
|
}
|
|
|
|
func (this *DefaultOutboundHandlerManager) Release() {
|
|
|
|
}
|
|
|
|
func (this *DefaultOutboundHandlerManager) GetDefaultHandler() proxy.OutboundHandler {
|
|
this.RLock()
|
|
defer this.RUnlock()
|
|
if this.defaultHandler == nil {
|
|
return nil
|
|
}
|
|
return this.defaultHandler
|
|
}
|
|
|
|
func (this *DefaultOutboundHandlerManager) SetDefaultHandler(handler proxy.OutboundHandler) {
|
|
this.Lock()
|
|
defer this.Unlock()
|
|
this.defaultHandler = handler
|
|
}
|
|
|
|
func (this *DefaultOutboundHandlerManager) GetHandler(tag string) proxy.OutboundHandler {
|
|
this.RLock()
|
|
defer this.RUnlock()
|
|
if handler, found := this.taggedHandler[tag]; found {
|
|
return handler
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (this *DefaultOutboundHandlerManager) SetHandler(tag string, handler proxy.OutboundHandler) {
|
|
this.Lock()
|
|
defer this.Unlock()
|
|
|
|
this.taggedHandler[tag] = handler
|
|
}
|