mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-05-13 09:36:05 +00:00
8e7d215b4a
- Traffic-writer single-consumer queue (web/service/traffic_writer.go) serialises every DB write that touches up/down/all_time/last_online (AddTraffic, SetRemoteTraffic, Reset*, UpdateClientTrafficByEmail) so overlapping goroutines can no longer clobber each other's column-scoped Updates with a stale tx.Save. - DB pool: WAL + busy_timeout=10s + synchronous=NORMAL + _txlock= immediate, MaxOpenConns=8 / MaxIdleConns=4. The immediate-tx PRAGMA fixes residual "database is locked [0ms]" cases where deferred-tx writer-upgrade conflicts bypass busy_timeout. - SetRemoteTraffic full-mirrors node-authoritative state into central: settings JSON, remark, listen, port, total, expiry, all_time, enable, plus per-client total/expiry/reset/all_time. Inbounds and client_traffics rows present on node but missing from central are created; rows missing from snap are deleted (with cascading client_traffics removal). - NodeTrafficSyncJob detects structural changes from the mirror and broadcasts invalidate(inbounds) so open central UIs re-fetch via REST on node-side add/del/edit without manual refresh. - XrayTrafficJob broadcasts invalidate(inbounds) when auto-disable flips client_traffics.enable so the per-client toggle reflects depletion without manual refresh. - Frontend: inbounds page now subscribes to the BroadcastInbounds 'inbounds' WS event (full-list pushes from add/del/update controllers were silently dropped). Fixes invalidate payload field (dataType -> type). Restart- panel modal switched from Promise-wrap to onOk-only so Cancel actually cancels. - Node files trimmed of stale prose-comments; cron cadence dropped 10s -> 5s to match the inbounds page UX. - README badges and Go module path bumped v2 -> v3 to match module rename. Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
106 lines
1.8 KiB
Go
106 lines
1.8 KiB
Go
package runtime
|
|
|
|
import (
|
|
"errors"
|
|
"sync"
|
|
|
|
"github.com/mhsanaei/3x-ui/v3/database"
|
|
"github.com/mhsanaei/3x-ui/v3/database/model"
|
|
)
|
|
|
|
type Manager struct {
|
|
local Runtime
|
|
|
|
mu sync.RWMutex
|
|
remotes map[int]*Remote
|
|
}
|
|
|
|
func NewManager(localDeps LocalDeps) *Manager {
|
|
return &Manager{
|
|
local: NewLocal(localDeps),
|
|
remotes: make(map[int]*Remote),
|
|
}
|
|
}
|
|
|
|
func (m *Manager) RuntimeFor(nodeID *int) (Runtime, error) {
|
|
if nodeID == nil {
|
|
return m.local, nil
|
|
}
|
|
m.mu.RLock()
|
|
if rt, ok := m.remotes[*nodeID]; ok {
|
|
m.mu.RUnlock()
|
|
return rt, nil
|
|
}
|
|
m.mu.RUnlock()
|
|
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
if rt, ok := m.remotes[*nodeID]; ok {
|
|
return rt, nil
|
|
}
|
|
n, err := loadNode(*nodeID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if !n.Enable {
|
|
return nil, errors.New("node " + n.Name + " is disabled")
|
|
}
|
|
rt := NewRemote(n)
|
|
m.remotes[*nodeID] = rt
|
|
return rt, nil
|
|
}
|
|
|
|
func (m *Manager) Local() Runtime { return m.local }
|
|
|
|
func (m *Manager) RemoteFor(node *model.Node) (*Remote, error) {
|
|
if node == nil {
|
|
return nil, errors.New("node is nil")
|
|
}
|
|
m.mu.RLock()
|
|
if rt, ok := m.remotes[node.Id]; ok {
|
|
m.mu.RUnlock()
|
|
return rt, nil
|
|
}
|
|
m.mu.RUnlock()
|
|
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
if rt, ok := m.remotes[node.Id]; ok {
|
|
return rt, nil
|
|
}
|
|
rt := NewRemote(node)
|
|
m.remotes[node.Id] = rt
|
|
return rt, nil
|
|
}
|
|
|
|
func (m *Manager) InvalidateNode(nodeID int) {
|
|
m.mu.Lock()
|
|
defer m.mu.Unlock()
|
|
delete(m.remotes, nodeID)
|
|
}
|
|
|
|
func loadNode(id int) (*model.Node, error) {
|
|
db := database.GetDB()
|
|
n := &model.Node{}
|
|
if err := db.Model(model.Node{}).Where("id = ?", id).First(n).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return n, nil
|
|
}
|
|
|
|
var (
|
|
managerMu sync.RWMutex
|
|
manager *Manager
|
|
)
|
|
|
|
func SetManager(m *Manager) {
|
|
managerMu.Lock()
|
|
defer managerMu.Unlock()
|
|
manager = m
|
|
}
|
|
|
|
func GetManager() *Manager {
|
|
managerMu.RLock()
|
|
defer managerMu.RUnlock()
|
|
return manager
|
|
}
|