mirror of
https://github.com/MHSanaei/3x-ui.git
synced 2026-06-28 00:24:19 +00:00
MHSanaei
6a032bcb2a
Local hot paths: - autoRenewClients: replace the O(clients x expired) inner scan with an email->traffic map lookup (quadratic at scale). - node traffic sync: scope the client_traffics email-membership query to the snapshot's emails instead of plucking the whole table every poll. - add a (expiry_time, reset) index for the per-tick auto-renew filter. - SQLite: add cache_size/mmap_size/temp_store pragmas (env-tunable); keep the single-file DELETE journal and synchronous=FULL defaults. - scale benchmarks now run on SQLite too via XUI_SCALE_TEST=1 (shared setupScaleDB/resetScaleTables helpers), not just Postgres. Node paths: - bulk add/delete/adjust on a node-attached inbound folded one HTTP RPC per client; above nodeBulkPushThreshold (32) mark the node dirty and let one ReconcileNode push converge it instead of O(M) sequential round-trips. Small ops keep the live per-client path. Also hoist nodePushPlan out of the per-email delete loop. - ReconcileNode skips inbounds whose wire payload is unchanged (per-tag fingerprint on Remote), guarded by node-side tag presence so a restarted node is still re-seeded. Tests: auto-renew multi-inbound correctness, node-path dispatch (large ops fold to dirty, small ops push live) via a manager runtime override seam, and reconcile delta-skip.
65 lines
2.0 KiB
Go
65 lines
2.0 KiB
Go
package service
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/mhsanaei/3x-ui/v3/internal/config"
|
|
"github.com/mhsanaei/3x-ui/v3/internal/database"
|
|
xuilogger "github.com/mhsanaei/3x-ui/v3/internal/logger"
|
|
|
|
"github.com/op/go-logging"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// setupScaleDB initializes the DB for a scale benchmark on either Postgres
|
|
// (XUI_DB_TYPE=postgres + XUI_DB_DSN) or SQLite (XUI_SCALE_TEST=1, temp file),
|
|
// and registers cleanup. Skips the test when neither backend is configured.
|
|
func setupScaleDB(t *testing.T) {
|
|
t.Helper()
|
|
xuilogger.InitLogger(logging.ERROR)
|
|
|
|
if os.Getenv("XUI_DB_TYPE") == "postgres" && strings.TrimSpace(os.Getenv("XUI_DB_DSN")) != "" {
|
|
if err := database.InitDB(""); err != nil {
|
|
t.Fatalf("InitDB(postgres): %v", err)
|
|
}
|
|
t.Cleanup(func() { _ = database.CloseDB() })
|
|
return
|
|
}
|
|
|
|
switch strings.ToLower(strings.TrimSpace(os.Getenv("XUI_SCALE_TEST"))) {
|
|
case "1", "true", "yes":
|
|
dbPath := filepath.Join(t.TempDir(), "scale.db")
|
|
if err := database.InitDB(dbPath); err != nil {
|
|
t.Fatalf("InitDB(sqlite): %v", err)
|
|
}
|
|
t.Cleanup(func() { _ = database.CloseDB() })
|
|
return
|
|
}
|
|
|
|
t.Skip("set XUI_SCALE_TEST=1 (sqlite) or XUI_DB_TYPE=postgres + XUI_DB_DSN (postgres) to run the scale benchmark")
|
|
}
|
|
|
|
// resetScaleTables empties the given tables between sub-sizes. Postgres uses a
|
|
// single TRUNCATE ... CASCADE; SQLite deletes per table and clears the
|
|
// autoincrement counters so ids restart like RESTART IDENTITY.
|
|
func resetScaleTables(t *testing.T, db *gorm.DB, tables ...string) {
|
|
t.Helper()
|
|
if config.GetDBKind() == "postgres" {
|
|
stmt := "TRUNCATE TABLE " + strings.Join(tables, ", ") + " RESTART IDENTITY CASCADE"
|
|
if err := db.Exec(stmt).Error; err != nil {
|
|
t.Fatalf("truncate: %v", err)
|
|
}
|
|
return
|
|
}
|
|
for _, tbl := range tables {
|
|
if err := db.Exec("DELETE FROM " + tbl).Error; err != nil {
|
|
t.Fatalf("delete %s: %v", tbl, err)
|
|
}
|
|
}
|
|
// Best-effort id reset; sqlite_sequence is absent until the first insert.
|
|
db.Exec("DELETE FROM sqlite_sequence")
|
|
}
|