mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-09-07 02:37:14 +00:00
perf(clients): push a bulk client change to every node at once
63b46cd6made a multi-inbound client create apply its inbounds concurrently, andd34ec97fdid the same for the single-client update, delete and detach. The bulk operations were never converted, so they still walked their inbounds in a plain sequential loop with a node RPC in each iteration — and those are what the panel actually calls for a multi-select delete or an enable/disable, which is why editing and deleting still felt slow on a master with several nodes. Measured with a node runtime injecting 100ms per RPC, one client per node: nodes=1 update=101ms bulkSetEnable=101ms bulkAdjust=101ms bulkDelete=101ms nodes=3 update=102ms bulkSetEnable=302ms bulkAdjust=304ms bulkDelete=303ms nodes=5 update=203ms bulkSetEnable=504ms bulkAdjust=504ms bulkDelete=504ms after, all of them track the single-client ops: nodes=3 bulkSetEnable=103ms bulkAdjust=101ms bulkDelete=101ms nodes=5 bulkSetEnable=202ms bulkAdjust=202ms bulkDelete=202ms Generalize the fanout into fanoutInboundResults over an arbitrary per-inbound result type and route six loops through it: BulkDelete, BulkSetEnable, BulkAdjust, BulkDetach, BulkAttach, BulkCreate, plus applyClientFieldByEmail — the field edit behind the Telegram bot's enable/limit/expiry buttons and the LDAP job. Each keeps its preparation sequential and overlaps only the node pushes, inheriting the same concurrency cap and per-inbound panic recovery. Two ordering details the sequential loops got for free and the fanout must do itself: the three loops that ranged a map now walk sortedInboundIds, so which inbound wins a per-email skip reason is the lowest id instead of whatever the map yielded; and BulkAttach de-duplicates a repeated inbound id up front, because the second pass used to see the client the first pass had just added. The allocating paths stay serial when a tunnel inbound is involved. WireGuard and AmneziaWG pick a free peer address by reading every inbound's used-set before they write, so two overlapping allocations hand out the same address and the in-transaction re-check refuses the loser — a bulk create of two clients onto two wg inbounds returned created=1. addFanoutLimit drops those batches back to one at a time; every other protocol keeps the full cap. Eight tests: seven barrier tests that a sequential caller cannot satisfy (peak pushes in flight is 1 without the change, 4 with it), and one that pins the tunnel allocation.
This commit is contained in:
@@ -61,12 +61,28 @@ func (s *ClientService) BulkAttach(inboundSvc *InboundService, emails []string,
|
||||
}
|
||||
|
||||
needRestart := false
|
||||
// Prepared in order first, as in Create: fillProtocolDefaults mints the
|
||||
// shared credentials, so only the node pushes below may overlap.
|
||||
attachIds := make([]int, 0, len(inboundIds))
|
||||
attachPayloads := make([]string, 0, len(inboundIds))
|
||||
attachClients := make([][]model.Client, 0, len(inboundIds))
|
||||
attachAnyTunnel := false
|
||||
// A repeated id used to be caught by the second pass seeing the client
|
||||
// already attached; the applies no longer run before the next prep.
|
||||
seenInbound := make(map[int]struct{}, len(inboundIds))
|
||||
for _, ibId := range inboundIds {
|
||||
if _, dup := seenInbound[ibId]; dup {
|
||||
continue
|
||||
}
|
||||
seenInbound[ibId] = struct{}{}
|
||||
inbound, err := inboundSvc.GetInbound(ibId)
|
||||
if err != nil {
|
||||
recordErr("inbound %d: %v", ibId, err)
|
||||
continue
|
||||
}
|
||||
if inbound.Protocol == model.WireGuard || inbound.Protocol == model.AmneziaWG {
|
||||
attachAnyTunnel = true
|
||||
}
|
||||
existingClients, err := inboundSvc.GetClients(inbound)
|
||||
if err != nil {
|
||||
recordErr("inbound %d: %v", ibId, err)
|
||||
@@ -101,15 +117,31 @@ func (s *ClientService) BulkAttach(inboundSvc *InboundService, emails []string,
|
||||
recordErr("inbound %d: %v", ibId, err)
|
||||
continue
|
||||
}
|
||||
nr, err := s.AddInboundClient(inboundSvc, &model.Inbound{Id: ibId, Settings: string(payload)})
|
||||
attachIds = append(attachIds, ibId)
|
||||
attachPayloads = append(attachPayloads, string(payload))
|
||||
attachClients = append(attachClients, clientsToAdd)
|
||||
}
|
||||
|
||||
attachResults, attachPanics := fanoutInboundResults(attachIds, addFanoutLimit(attachAnyTunnel), func(i int) inboundApplyOutcome {
|
||||
nr, err := s.AddInboundClient(inboundSvc, &model.Inbound{Id: attachIds[i], Settings: attachPayloads[i]})
|
||||
return inboundApplyOutcome{needRestart: nr, err: err}
|
||||
})
|
||||
for i, out := range attachResults {
|
||||
err := out.err
|
||||
if attachPanics[i] != nil {
|
||||
// The apply may already have committed, so ask for the restart the
|
||||
// lost return value can no longer report.
|
||||
needRestart = true
|
||||
err = attachPanics[i]
|
||||
}
|
||||
if err != nil {
|
||||
recordErr("inbound %d: %v", ibId, err)
|
||||
recordErr("inbound %d: %v", attachIds[i], err)
|
||||
continue
|
||||
}
|
||||
if nr {
|
||||
if out.needRestart {
|
||||
needRestart = true
|
||||
}
|
||||
for _, c := range clientsToAdd {
|
||||
for _, c := range attachClients[i] {
|
||||
result.Attached = append(result.Attached, c.Email)
|
||||
}
|
||||
}
|
||||
@@ -187,21 +219,38 @@ func (s *ClientService) BulkDetach(inboundSvc *InboundService, emails []string,
|
||||
}
|
||||
|
||||
needRestart := false
|
||||
// Ordered and de-duplicated up front: the sequential loop dropped each map
|
||||
// entry as it went, which the concurrent applies can no longer do.
|
||||
detachIds := make([]int, 0, len(recsByInbound))
|
||||
detachRecs := make([][]*model.ClientRecord, 0, len(recsByInbound))
|
||||
for _, ibId := range inboundIds {
|
||||
recs, ok := recsByInbound[ibId]
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
delete(recsByInbound, ibId)
|
||||
nr, err := s.delInboundClients(inboundSvc, ibId, recs, true)
|
||||
detachIds = append(detachIds, ibId)
|
||||
detachRecs = append(detachRecs, recs)
|
||||
}
|
||||
detachResults, detachPanics := fanoutInboundResults(detachIds, inboundFanoutConcurrency, func(i int) inboundApplyOutcome {
|
||||
nr, err := s.delInboundClients(inboundSvc, detachIds[i], detachRecs[i], true)
|
||||
return inboundApplyOutcome{needRestart: nr, err: err}
|
||||
})
|
||||
for i, out := range detachResults {
|
||||
err := out.err
|
||||
if detachPanics[i] != nil {
|
||||
// See BulkAttach: a panicking apply may already have committed.
|
||||
needRestart = true
|
||||
err = detachPanics[i]
|
||||
}
|
||||
if err != nil {
|
||||
recordErr("inbound %d: %v", ibId, err)
|
||||
for _, rec := range recs {
|
||||
recordErr("inbound %d: %v", detachIds[i], err)
|
||||
for _, rec := range detachRecs[i] {
|
||||
emailFailed[strings.ToLower(rec.Email)] = true
|
||||
}
|
||||
continue
|
||||
}
|
||||
if nr {
|
||||
if out.needRestart {
|
||||
needRestart = true
|
||||
}
|
||||
}
|
||||
@@ -216,6 +265,12 @@ func (s *ClientService) BulkDetach(inboundSvc *InboundService, emails []string,
|
||||
return result, needRestart, nil
|
||||
}
|
||||
|
||||
// inboundApplyOutcome carries one inbound's apply result out of the fanout.
|
||||
type inboundApplyOutcome struct {
|
||||
needRestart bool
|
||||
err error
|
||||
}
|
||||
|
||||
// BulkAdjustResult is returned by BulkAdjust to report how many clients were
|
||||
// successfully updated and which were skipped (typically because the field
|
||||
// being adjusted was unlimited for that client) or failed.
|
||||
@@ -404,8 +459,21 @@ func (s *ClientService) BulkAdjust(inboundSvc *InboundService, emails []string,
|
||||
flowHonored := map[string]bool{}
|
||||
flowIneligible := map[string]bool{}
|
||||
execFailed := map[string]bool{}
|
||||
for inboundId, ibEmails := range emailsByInbound {
|
||||
ibRes := s.bulkAdjustInboundClients(inboundSvc, inboundId, ibEmails, plan, flow)
|
||||
adjustIds := sortedInboundIds(emailsByInbound)
|
||||
adjustResults, adjustPanics := fanoutInboundResults(adjustIds, inboundFanoutConcurrency, func(i int) bulkInboundAdjustResult {
|
||||
return s.bulkAdjustInboundClients(inboundSvc, adjustIds[i], emailsByInbound[adjustIds[i]], plan, flow)
|
||||
})
|
||||
for i, ibRes := range adjustResults {
|
||||
if adjustPanics[i] != nil {
|
||||
needRestart = true
|
||||
for _, email := range emailsByInbound[adjustIds[i]] {
|
||||
execFailed[email] = true
|
||||
if _, already := skippedReasons[email]; !already {
|
||||
skippedReasons[email] = adjustPanics[i].Error()
|
||||
}
|
||||
}
|
||||
continue
|
||||
}
|
||||
if ibRes.needRestart {
|
||||
needRestart = true
|
||||
}
|
||||
@@ -796,8 +864,20 @@ func (s *ClientService) BulkDelete(inboundSvc *InboundService, emails []string,
|
||||
}
|
||||
|
||||
needRestart := false
|
||||
for inboundId, ibEmails := range emailsByInbound {
|
||||
ibResult := s.bulkDelInboundClients(inboundSvc, inboundId, ibEmails, recordsByEmail, keepTraffic)
|
||||
delIds := sortedInboundIds(emailsByInbound)
|
||||
delResults, delPanics := fanoutInboundResults(delIds, inboundFanoutConcurrency, func(i int) bulkInboundDeleteResult {
|
||||
return s.bulkDelInboundClients(inboundSvc, delIds[i], emailsByInbound[delIds[i]], recordsByEmail, keepTraffic)
|
||||
})
|
||||
for i, ibResult := range delResults {
|
||||
if delPanics[i] != nil {
|
||||
needRestart = true
|
||||
for _, email := range emailsByInbound[delIds[i]] {
|
||||
if _, already := skippedReasons[email]; !already {
|
||||
skippedReasons[email] = delPanics[i].Error()
|
||||
}
|
||||
}
|
||||
continue
|
||||
}
|
||||
if ibResult.needRestart {
|
||||
needRestart = true
|
||||
}
|
||||
@@ -1234,6 +1314,7 @@ func (s *ClientService) BulkCreate(inboundSvc *InboundService, payloads []Client
|
||||
inboundOrder := make([]int, 0)
|
||||
failed := make([]bool, len(prep))
|
||||
reason := make([]string, len(prep))
|
||||
createAnyTunnel := false
|
||||
|
||||
for idx := range prep {
|
||||
le := strings.ToLower(prep[idx].client.Email)
|
||||
@@ -1271,6 +1352,9 @@ func (s *ClientService) BulkCreate(inboundSvc *InboundService, payloads []Client
|
||||
ok = false
|
||||
break
|
||||
}
|
||||
if ib.Protocol == model.WireGuard || ib.Protocol == model.AmneziaWG {
|
||||
createAnyTunnel = true
|
||||
}
|
||||
if e := s.fillProtocolDefaults(&prep[idx].client, ib); e != nil {
|
||||
failed[idx] = true
|
||||
reason[idx] = e.Error()
|
||||
@@ -1292,22 +1376,33 @@ func (s *ClientService) BulkCreate(inboundSvc *InboundService, payloads []Client
|
||||
}
|
||||
|
||||
needRestart := false
|
||||
for _, ibId := range inboundOrder {
|
||||
createResults, createPanics := fanoutInboundResults(inboundOrder, addFanoutLimit(createAnyTunnel), func(i int) inboundApplyOutcome {
|
||||
ibId := inboundOrder[i]
|
||||
payload, e := json.Marshal(map[string][]model.Client{"clients": byInbound[ibId]})
|
||||
if e == nil {
|
||||
var nr bool
|
||||
nr, e = s.AddInboundClient(inboundSvc, &model.Inbound{Id: ibId, Settings: string(payload)})
|
||||
if e == nil && nr {
|
||||
needRestart = true
|
||||
}
|
||||
if e != nil {
|
||||
return inboundApplyOutcome{err: e}
|
||||
}
|
||||
nr, e := s.AddInboundClient(inboundSvc, &model.Inbound{Id: ibId, Settings: string(payload)})
|
||||
return inboundApplyOutcome{needRestart: nr, err: e}
|
||||
})
|
||||
for i, out := range createResults {
|
||||
e := out.err
|
||||
if createPanics[i] != nil {
|
||||
// See BulkAttach: a panicking apply may already have committed.
|
||||
needRestart = true
|
||||
e = createPanics[i]
|
||||
}
|
||||
if e != nil {
|
||||
for _, idx := range idxByInbound[ibId] {
|
||||
for _, idx := range idxByInbound[inboundOrder[i]] {
|
||||
failed[idx] = true
|
||||
if reason[idx] == "" {
|
||||
reason[idx] = e.Error()
|
||||
}
|
||||
}
|
||||
continue
|
||||
}
|
||||
if out.needRestart {
|
||||
needRestart = true
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1440,8 +1535,20 @@ func (s *ClientService) BulkSetEnable(inboundSvc *InboundService, emails []strin
|
||||
}
|
||||
|
||||
needRestart := false
|
||||
for inboundId, ibEmails := range emailsByInbound {
|
||||
ibRes := s.bulkSetEnableInboundClients(inboundSvc, inboundId, ibEmails, enable)
|
||||
enableIds := sortedInboundIds(emailsByInbound)
|
||||
enableResults, enablePanics := fanoutInboundResults(enableIds, inboundFanoutConcurrency, func(i int) bulkSetEnableInboundResult {
|
||||
return s.bulkSetEnableInboundClients(inboundSvc, enableIds[i], emailsByInbound[enableIds[i]], enable)
|
||||
})
|
||||
for i, ibRes := range enableResults {
|
||||
if enablePanics[i] != nil {
|
||||
needRestart = true
|
||||
for _, email := range emailsByInbound[enableIds[i]] {
|
||||
if _, already := skippedReasons[email]; !already {
|
||||
skippedReasons[email] = enablePanics[i].Error()
|
||||
}
|
||||
}
|
||||
continue
|
||||
}
|
||||
if ibRes.needRestart {
|
||||
needRestart = true
|
||||
}
|
||||
|
||||
@@ -0,0 +1,228 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/mhsanaei/3x-ui/v3/internal/database"
|
||||
"github.com/mhsanaei/3x-ui/v3/internal/database/model"
|
||||
)
|
||||
|
||||
// The bulk ops walked their inbounds one node round-trip at a time, so a client
|
||||
// spanning several nodes cost the SUM of every node's latency. Each test below
|
||||
// times out on the barrier unless the pushes overlap.
|
||||
|
||||
func TestBulkDeleteAcrossNodesPushesConcurrently(t *testing.T) {
|
||||
setupBulkDB(t)
|
||||
startSerializedWriter(t)
|
||||
|
||||
const nodes = inboundFanoutConcurrency + 1
|
||||
bar := newApplyBarrier(inboundFanoutConcurrency)
|
||||
seedClientAcrossNodes(t, bar, nodes, 46101, "bulkdel@x", "aaaaaaaa-1111-2222-3333-444444444444")
|
||||
|
||||
bar.arm()
|
||||
if _, _, err := (&ClientService{}).BulkDelete(&InboundService{}, []string{"bulkdel@x"}, false); err != nil {
|
||||
t.Fatalf("BulkDelete across %d node inbounds: %v", nodes, err)
|
||||
}
|
||||
if got := bar.deleteClient.Load(); got == 0 {
|
||||
t.Fatalf("no node push reached the barrier at all")
|
||||
}
|
||||
if got := bar.maxPar.Load(); got != inboundFanoutConcurrency {
|
||||
t.Fatalf("peak node pushes in flight = %d, want overlap at the %d cap (barrier timed out: %v)",
|
||||
got, inboundFanoutConcurrency, bar.expired.Load())
|
||||
}
|
||||
}
|
||||
|
||||
func TestBulkSetEnableAcrossNodesPushesConcurrently(t *testing.T) {
|
||||
setupBulkDB(t)
|
||||
startSerializedWriter(t)
|
||||
|
||||
const nodes = inboundFanoutConcurrency + 1
|
||||
bar := newApplyBarrier(inboundFanoutConcurrency)
|
||||
seedClientAcrossNodes(t, bar, nodes, 46201, "bulkena@x", "bbbbbbbb-1111-2222-3333-444444444444")
|
||||
|
||||
bar.arm()
|
||||
if _, _, err := (&ClientService{}).BulkSetEnable(&InboundService{}, []string{"bulkena@x"}, false); err != nil {
|
||||
t.Fatalf("BulkSetEnable across %d node inbounds: %v", nodes, err)
|
||||
}
|
||||
if got := bar.updateUser.Load(); got == 0 {
|
||||
t.Fatalf("no node push reached the barrier at all")
|
||||
}
|
||||
if got := bar.maxPar.Load(); got != inboundFanoutConcurrency {
|
||||
t.Fatalf("peak node pushes in flight = %d, want overlap at the %d cap (barrier timed out: %v)",
|
||||
got, inboundFanoutConcurrency, bar.expired.Load())
|
||||
}
|
||||
}
|
||||
|
||||
func TestBulkAdjustAcrossNodesPushesConcurrently(t *testing.T) {
|
||||
setupBulkDB(t)
|
||||
startSerializedWriter(t)
|
||||
|
||||
const nodes = inboundFanoutConcurrency + 1
|
||||
const email = "bulkadj@x"
|
||||
bar := newApplyBarrier(inboundFanoutConcurrency)
|
||||
mgr := useTestRuntimeManager(t)
|
||||
ids := fanoutNodeInbounds(t, mgr, bar, nodes, 46301)
|
||||
// An expiry to extend, or BulkAdjust reports the client ineligible and
|
||||
// never reaches a node at all.
|
||||
if _, err := (&ClientService{}).Create(&InboundService{}, &ClientCreatePayload{
|
||||
Client: model.Client{
|
||||
Email: email, ID: "cccccccc-1111-2222-3333-444444444444", SubID: "sub-" + email,
|
||||
Enable: true, ExpiryTime: time.Now().Add(24 * time.Hour).UnixMilli(),
|
||||
},
|
||||
InboundIds: ids,
|
||||
}); err != nil {
|
||||
t.Fatalf("seed Create across %d node inbounds: %v", nodes, err)
|
||||
}
|
||||
|
||||
bar.arm()
|
||||
if _, _, err := (&ClientService{}).BulkAdjust(&InboundService{}, []string{email}, 1, 0, ""); err != nil {
|
||||
t.Fatalf("BulkAdjust across %d node inbounds: %v", nodes, err)
|
||||
}
|
||||
if got := bar.updateUser.Load(); got == 0 {
|
||||
t.Fatalf("no node push reached the barrier at all")
|
||||
}
|
||||
if got := bar.maxPar.Load(); got != inboundFanoutConcurrency {
|
||||
t.Fatalf("peak node pushes in flight = %d, want overlap at the %d cap (barrier timed out: %v)",
|
||||
got, inboundFanoutConcurrency, bar.expired.Load())
|
||||
}
|
||||
}
|
||||
|
||||
func TestBulkAttachAcrossNodesPushesConcurrently(t *testing.T) {
|
||||
setupBulkDB(t)
|
||||
startSerializedWriter(t)
|
||||
|
||||
const nodes = inboundFanoutConcurrency + 1
|
||||
const email = "bulkatt@x"
|
||||
bar := newApplyBarrier(inboundFanoutConcurrency)
|
||||
mgr := useTestRuntimeManager(t)
|
||||
// Seeded on the first inbound only, so the other nodes are all attach work.
|
||||
ids := fanoutNodeInbounds(t, mgr, bar, nodes, 46501)
|
||||
if _, err := (&ClientService{}).Create(&InboundService{}, &ClientCreatePayload{
|
||||
Client: model.Client{Email: email, ID: "eeeeeeee-1111-2222-3333-444444444444", SubID: "sub-" + email, Enable: true},
|
||||
InboundIds: ids[:1],
|
||||
}); err != nil {
|
||||
t.Fatalf("seed Create: %v", err)
|
||||
}
|
||||
|
||||
bar.arm()
|
||||
if _, _, err := (&ClientService{}).BulkAttach(&InboundService{}, []string{email}, ids[1:]); err != nil {
|
||||
t.Fatalf("BulkAttach across %d node inbounds: %v", nodes-1, err)
|
||||
}
|
||||
if got := bar.addClient.Load(); got == 0 {
|
||||
t.Fatalf("no node push reached the barrier at all")
|
||||
}
|
||||
if got := bar.maxPar.Load(); got != inboundFanoutConcurrency {
|
||||
t.Fatalf("peak node pushes in flight = %d, want overlap at the %d cap (barrier timed out: %v)",
|
||||
got, inboundFanoutConcurrency, bar.expired.Load())
|
||||
}
|
||||
}
|
||||
|
||||
func TestBulkCreateAcrossNodesPushesConcurrently(t *testing.T) {
|
||||
setupBulkDB(t)
|
||||
startSerializedWriter(t)
|
||||
|
||||
const nodes = inboundFanoutConcurrency + 1
|
||||
bar := newApplyBarrier(inboundFanoutConcurrency)
|
||||
mgr := useTestRuntimeManager(t)
|
||||
ids := fanoutNodeInbounds(t, mgr, bar, nodes, 46601)
|
||||
|
||||
bar.arm()
|
||||
if _, _, err := (&ClientService{}).BulkCreate(&InboundService{}, []ClientCreatePayload{{
|
||||
Client: model.Client{Email: "bulknew@x", ID: "ffffffff-1111-2222-3333-444444444444", SubID: "sub-bulknew", Enable: true},
|
||||
InboundIds: ids,
|
||||
}}); err != nil {
|
||||
t.Fatalf("BulkCreate across %d node inbounds: %v", nodes, err)
|
||||
}
|
||||
if got := bar.addClient.Load(); got == 0 {
|
||||
t.Fatalf("no node push reached the barrier at all")
|
||||
}
|
||||
if got := bar.maxPar.Load(); got != inboundFanoutConcurrency {
|
||||
t.Fatalf("peak node pushes in flight = %d, want overlap at the %d cap (barrier timed out: %v)",
|
||||
got, inboundFanoutConcurrency, bar.expired.Load())
|
||||
}
|
||||
}
|
||||
|
||||
func TestBulkDetachAcrossNodesPushesConcurrently(t *testing.T) {
|
||||
setupBulkDB(t)
|
||||
startSerializedWriter(t)
|
||||
|
||||
const nodes = inboundFanoutConcurrency + 1
|
||||
const email = "bulkdet@x"
|
||||
bar := newApplyBarrier(inboundFanoutConcurrency)
|
||||
recID := seedClientAcrossNodes(t, bar, nodes, 46401, email, "dddddddd-1111-2222-3333-444444444444")
|
||||
ids, err := (&ClientService{}).GetInboundIdsForRecord(recID)
|
||||
if err != nil {
|
||||
t.Fatalf("GetInboundIdsForRecord: %v", err)
|
||||
}
|
||||
|
||||
bar.arm()
|
||||
if _, _, err := (&ClientService{}).BulkDetach(&InboundService{}, []string{email}, ids); err != nil {
|
||||
t.Fatalf("BulkDetach across %d node inbounds: %v", nodes, err)
|
||||
}
|
||||
if got := bar.deleteUser.Load(); got == 0 {
|
||||
t.Fatalf("no node push reached the barrier at all")
|
||||
}
|
||||
if got := bar.maxPar.Load(); got != inboundFanoutConcurrency {
|
||||
t.Fatalf("peak node pushes in flight = %d, want overlap at the %d cap (barrier timed out: %v)",
|
||||
got, inboundFanoutConcurrency, bar.expired.Load())
|
||||
}
|
||||
}
|
||||
|
||||
// TestApplyClientFieldAcrossNodesPushesConcurrently covers the field-edit path
|
||||
// the Telegram bot and the LDAP job use (enable toggle, ip/expiry/traffic reset).
|
||||
func TestApplyClientFieldAcrossNodesPushesConcurrently(t *testing.T) {
|
||||
setupBulkDB(t)
|
||||
startSerializedWriter(t)
|
||||
|
||||
const nodes = inboundFanoutConcurrency + 1
|
||||
const email = "fieldedit@x"
|
||||
bar := newApplyBarrier(inboundFanoutConcurrency)
|
||||
seedClientAcrossNodes(t, bar, nodes, 46701, email, "99999999-1111-2222-3333-444444444444")
|
||||
|
||||
bar.arm()
|
||||
if _, err := (&ClientService{}).ResetClientIpLimitByEmail(&InboundService{}, email, 3); err != nil {
|
||||
t.Fatalf("ResetClientIpLimitByEmail across %d node inbounds: %v", nodes, err)
|
||||
}
|
||||
if got := bar.updateUser.Load(); got == 0 {
|
||||
t.Fatalf("no node push reached the barrier at all")
|
||||
}
|
||||
if got := bar.maxPar.Load(); got != inboundFanoutConcurrency {
|
||||
t.Fatalf("peak node pushes in flight = %d, want overlap at the %d cap (barrier timed out: %v)",
|
||||
got, inboundFanoutConcurrency, bar.expired.Load())
|
||||
}
|
||||
}
|
||||
|
||||
// TestBulkCreateSerializesTunnelAddressAllocation pins that a bulk create over
|
||||
// WireGuard inbounds does not overlap: allocation reads every inbound's used-set
|
||||
// before writing, so two concurrent picks collide and the second is refused.
|
||||
func TestBulkCreateSerializesTunnelAddressAllocation(t *testing.T) {
|
||||
setupBulkDB(t)
|
||||
startSerializedWriter(t)
|
||||
db := database.GetDB()
|
||||
|
||||
ids := make([]int, 0, 2)
|
||||
for i := range 2 {
|
||||
ib := &model.Inbound{
|
||||
UserId: 1, Enable: true, Port: 51820 + i,
|
||||
Tag: fmt.Sprintf("wg-%d", i), Protocol: model.WireGuard,
|
||||
Settings: `{"clients":[],"mtu":1420,"secretKey":"QO3O1V0m0Sm1yQ0hVvJ0kM0kQe0mYq0Wc0Zk0Xs0Zm8=","peers":[]}`,
|
||||
}
|
||||
if err := db.Create(ib).Error; err != nil {
|
||||
t.Fatalf("create wg inbound: %v", err)
|
||||
}
|
||||
ids = append(ids, ib.Id)
|
||||
}
|
||||
|
||||
res, _, err := (&ClientService{}).BulkCreate(&InboundService{}, []ClientCreatePayload{
|
||||
{Client: model.Client{Email: "a@wg", SubID: "sa", Enable: true}, InboundIds: []int{ids[0]}},
|
||||
{Client: model.Client{Email: "b@wg", SubID: "sb", Enable: true}, InboundIds: []int{ids[1]}},
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("BulkCreate over two wg inbounds: %v", err)
|
||||
}
|
||||
if res.Created != 2 {
|
||||
t.Fatalf("created = %d, want 2 — concurrent allocation handed out one address twice: %+v", res.Created, res.Skipped)
|
||||
}
|
||||
}
|
||||
@@ -7,6 +7,7 @@ import (
|
||||
"fmt"
|
||||
"net/netip"
|
||||
"runtime/debug"
|
||||
"slices"
|
||||
"strings"
|
||||
"sync"
|
||||
"sync/atomic"
|
||||
@@ -297,6 +298,60 @@ func (s *ClientService) fanoutInboundClientAdds(inboundSvc *InboundService, adds
|
||||
return fanoutInboundApplies(applies)
|
||||
}
|
||||
|
||||
// fanoutInboundResults runs one job per inbound with the node pushes
|
||||
// overlapping, so a bulk op costs one RPC round-trip instead of one per node.
|
||||
// limit is the caller's own cap: an op that allocates tunnel addresses passes 1,
|
||||
// because allocation reads a cross-inbound used-set before it writes.
|
||||
func fanoutInboundResults[T any](inboundIds []int, limit int, run func(i int) T) ([]T, []error) {
|
||||
if limit < 1 {
|
||||
limit = 1
|
||||
}
|
||||
out := make([]T, len(inboundIds))
|
||||
errs := make([]error, len(inboundIds))
|
||||
sem := make(chan struct{}, limit)
|
||||
var wg sync.WaitGroup
|
||||
for i := range inboundIds {
|
||||
wg.Add(1)
|
||||
sem <- struct{}{}
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
defer func() { <-sem }()
|
||||
// Off the request goroutine gin's Recovery no longer covers this,
|
||||
// so an unrecovered panic here would take the whole panel down.
|
||||
defer func() {
|
||||
if r := recover(); r != nil {
|
||||
errs[i] = fmt.Errorf("inbound %d: panic: %v", inboundIds[i], r)
|
||||
logger.Errorf("panic applying bulk client change to inbound %d: %v\n%s", inboundIds[i], r, debug.Stack())
|
||||
}
|
||||
}()
|
||||
out[i] = run(i)
|
||||
}()
|
||||
}
|
||||
wg.Wait()
|
||||
return out, errs
|
||||
}
|
||||
|
||||
// addFanoutLimit serializes an add that touches a tunnel inbound. WireGuard and
|
||||
// AmneziaWG pick a free peer address by reading every inbound's used-set first,
|
||||
// so two overlapping allocations hand out the same one and the second is refused.
|
||||
func addFanoutLimit(anyTunnel bool) int {
|
||||
if anyTunnel {
|
||||
return 1
|
||||
}
|
||||
return inboundFanoutConcurrency
|
||||
}
|
||||
|
||||
// sortedInboundIds gives the fanout a stable order, so which inbound wins a
|
||||
// per-email report no longer depends on Go's map iteration order.
|
||||
func sortedInboundIds[V any](byInbound map[int]V) []int {
|
||||
ids := make([]int, 0, len(byInbound))
|
||||
for id := range byInbound {
|
||||
ids = append(ids, id)
|
||||
}
|
||||
slices.Sort(ids)
|
||||
return ids
|
||||
}
|
||||
|
||||
// markInboundNodesDirty makes a half-applied client edit unobservable to a node
|
||||
// snapshot merge, which skips a node whose config is already flagged dirty.
|
||||
func markInboundNodesDirty(inboundIds []int) error {
|
||||
|
||||
@@ -1343,6 +1343,9 @@ func (s *ClientService) applyClientFieldByEmail(inboundSvc *InboundService, clie
|
||||
|
||||
needRestart := false
|
||||
found := false
|
||||
// Built before any inbound is written, as in Update: only the applies
|
||||
// overlap, so one node's round-trip no longer waits on the previous one.
|
||||
applies := make([]inboundApply, 0, len(inboundIds))
|
||||
for _, ibId := range inboundIds {
|
||||
inbound, gErr := inboundSvc.GetInbound(ibId)
|
||||
if gErr != nil {
|
||||
@@ -1379,17 +1382,17 @@ func (s *ClientService) applyClientFieldByEmail(inboundSvc *InboundService, clie
|
||||
return needRestart, mErr
|
||||
}
|
||||
inbound.Settings = string(modifiedSettings)
|
||||
nr, uErr := s.UpdateInboundClient(inboundSvc, inbound, clientEmail)
|
||||
if uErr != nil {
|
||||
return needRestart, uErr
|
||||
}
|
||||
needRestart = needRestart || nr
|
||||
data := inbound
|
||||
applies = append(applies, inboundApply{id: ibId, run: func() (bool, error) {
|
||||
return s.UpdateInboundClient(inboundSvc, data, clientEmail)
|
||||
}})
|
||||
}
|
||||
|
||||
if !found {
|
||||
return needRestart, common.NewError("Client Not Found For Email:", clientEmail)
|
||||
}
|
||||
return needRestart, nil
|
||||
nr, applyErr := fanoutInboundApplies(applies)
|
||||
return needRestart || nr, applyErr
|
||||
}
|
||||
|
||||
func (s *ClientService) ResetClientIpLimitByEmail(inboundSvc *InboundService, clientEmail string, count int) (bool, error) {
|
||||
|
||||
@@ -65,6 +65,11 @@ func (b *applyBarrierRuntime) UpdateUser(ctx context.Context, ib *model.Inbound,
|
||||
return b.fakeNodeRuntime.UpdateUser(ctx, ib, oldEmail, c)
|
||||
}
|
||||
|
||||
func (b *applyBarrierRuntime) AddClient(ctx context.Context, ib *model.Inbound, c model.Client) error {
|
||||
b.wait()
|
||||
return b.fakeNodeRuntime.AddClient(ctx, ib, c)
|
||||
}
|
||||
|
||||
func (b *applyBarrierRuntime) DeleteClient(ctx context.Context, email string) error {
|
||||
b.wait()
|
||||
return b.fakeNodeRuntime.DeleteClient(ctx, email)
|
||||
|
||||
Reference in New Issue
Block a user