mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-08-17 16:50:58 +00:00
feat(node): per node outbound routing (#5275)
* feat: add per-node outbound routing for panel-to-node connections * feat(ui): add outbound tag selector to node form with i18n * fix(xray): avoid potential overflow warning in node egress rule allocation * chore: run "npm run gen" * fix --------- Co-authored-by: Sanaei <ho3ein.sanaei@gmail.com>
This commit is contained in:
@@ -8,11 +8,16 @@ import (
|
||||
"github.com/mhsanaei/3x-ui/v3/internal/database/model"
|
||||
)
|
||||
|
||||
type NodeEgressResolver interface {
|
||||
NodeEgressProxyURL(nodeID int) string
|
||||
}
|
||||
|
||||
type Manager struct {
|
||||
local Runtime
|
||||
|
||||
mu sync.RWMutex
|
||||
remotes map[int]*Remote
|
||||
mu sync.RWMutex
|
||||
remotes map[int]*Remote
|
||||
egressResolver NodeEgressResolver
|
||||
}
|
||||
|
||||
func NewManager(localDeps LocalDeps) *Manager {
|
||||
@@ -22,6 +27,21 @@ func NewManager(localDeps LocalDeps) *Manager {
|
||||
}
|
||||
}
|
||||
|
||||
func (m *Manager) SetNodeEgressResolver(r NodeEgressResolver) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
m.egressResolver = r
|
||||
}
|
||||
|
||||
func (m *Manager) NodeEgressProxyURL(nodeID int) string {
|
||||
m.mu.RLock()
|
||||
defer m.mu.RUnlock()
|
||||
if m.egressResolver == nil {
|
||||
return ""
|
||||
}
|
||||
return m.egressResolver.NodeEgressProxyURL(nodeID)
|
||||
}
|
||||
|
||||
func (m *Manager) RuntimeFor(nodeID *int) (Runtime, error) {
|
||||
if nodeID == nil {
|
||||
return m.local, nil
|
||||
@@ -45,7 +65,7 @@ func (m *Manager) RuntimeFor(nodeID *int) (Runtime, error) {
|
||||
if !n.Enable {
|
||||
return nil, errors.New("node " + n.Name + " is disabled")
|
||||
}
|
||||
rt := NewRemote(n)
|
||||
rt := NewRemote(n, m.egressResolver)
|
||||
m.remotes[*nodeID] = rt
|
||||
return rt, nil
|
||||
}
|
||||
@@ -68,7 +88,7 @@ func (m *Manager) RemoteFor(node *model.Node) (*Remote, error) {
|
||||
if rt, ok := m.remotes[node.Id]; ok {
|
||||
return rt, nil
|
||||
}
|
||||
rt := NewRemote(node)
|
||||
rt := NewRemote(node, m.egressResolver)
|
||||
m.remotes[node.Id] = rt
|
||||
return rt, nil
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user