atomic-queues 2.3.0 → 3.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +297 -382
- package/dist/cli/generators/classes.d.ts +1 -1
- package/dist/cli/generators/json-schema.d.ts +1 -1
- package/dist/cli/generators/typescript.d.ts +1 -1
- package/dist/cli/index.js +147 -5
- package/dist/cli/index.js.map +1 -1
- package/dist/cluster/cluster-discovery.service.d.ts +91 -0
- package/dist/cluster/cluster-discovery.service.d.ts.map +1 -0
- package/dist/cluster/cluster-discovery.service.js +423 -0
- package/dist/cluster/cluster-discovery.service.js.map +1 -0
- package/dist/cluster/grpc-peer-monitor.service.d.ts +31 -0
- package/dist/cluster/grpc-peer-monitor.service.d.ts.map +1 -0
- package/dist/cluster/grpc-peer-monitor.service.js +192 -0
- package/dist/cluster/grpc-peer-monitor.service.js.map +1 -0
- package/dist/cluster/index.d.ts +7 -0
- package/dist/cluster/index.d.ts.map +1 -0
- package/dist/cluster/index.js +23 -0
- package/dist/cluster/index.js.map +1 -0
- package/dist/cluster/leader-election.service.d.ts +38 -0
- package/dist/cluster/leader-election.service.d.ts.map +1 -0
- package/dist/cluster/leader-election.service.js +184 -0
- package/dist/cluster/leader-election.service.js.map +1 -0
- package/dist/cluster/master-coordinator.d.ts +50 -0
- package/dist/cluster/master-coordinator.d.ts.map +1 -0
- package/dist/cluster/master-coordinator.js +307 -0
- package/dist/cluster/master-coordinator.js.map +1 -0
- package/dist/cluster/redis-health-monitor.service.d.ts +23 -0
- package/dist/cluster/redis-health-monitor.service.d.ts.map +1 -0
- package/dist/cluster/redis-health-monitor.service.js +100 -0
- package/dist/cluster/redis-health-monitor.service.js.map +1 -0
- package/dist/cluster/server-ring.service.d.ts +48 -0
- package/dist/cluster/server-ring.service.d.ts.map +1 -0
- package/dist/cluster/server-ring.service.js +136 -0
- package/dist/cluster/server-ring.service.js.map +1 -0
- package/dist/decorators/entity.decorators.d.ts +16 -24
- package/dist/decorators/entity.decorators.d.ts.map +1 -1
- package/dist/decorators/entity.decorators.js +0 -39
- package/dist/decorators/entity.decorators.js.map +1 -1
- package/dist/decorators/interfaces.d.ts +10 -10
- package/dist/decorators/interfaces.d.ts.map +1 -1
- package/dist/decorators/job.decorators.d.ts +4 -52
- package/dist/decorators/job.decorators.d.ts.map +1 -1
- package/dist/decorators/job.decorators.js +6 -54
- package/dist/decorators/job.decorators.js.map +1 -1
- package/dist/decorators/metadata-readers.d.ts +4 -2
- package/dist/decorators/metadata-readers.d.ts.map +1 -1
- package/dist/decorators/metadata-readers.js +2 -0
- package/dist/decorators/metadata-readers.js.map +1 -1
- package/dist/decorators/schema.decorators.d.ts +1 -1
- package/dist/decorators/schema.decorators.d.ts.map +1 -1
- package/dist/decorators/schema.decorators.js.map +1 -1
- package/dist/decorators/utils.d.ts +1 -1
- package/dist/decorators/utils.d.ts.map +1 -1
- package/dist/decorators/utils.js +5 -1
- package/dist/decorators/utils.js.map +1 -1
- package/dist/domain/interfaces/config.interfaces.d.ts +92 -29
- package/dist/domain/interfaces/config.interfaces.d.ts.map +1 -1
- package/dist/domain/interfaces/index.d.ts +1 -0
- package/dist/domain/interfaces/index.d.ts.map +1 -1
- package/dist/domain/interfaces/index.js +1 -0
- package/dist/domain/interfaces/index.js.map +1 -1
- package/dist/{services/registry → domain/interfaces}/registry.types.d.ts.map +1 -1
- package/dist/domain/interfaces/registry.types.js.map +1 -0
- package/dist/grpc/grpc-client-pool.service.d.ts +71 -0
- package/dist/grpc/grpc-client-pool.service.d.ts.map +1 -0
- package/dist/grpc/grpc-client-pool.service.js +307 -0
- package/dist/grpc/grpc-client-pool.service.js.map +1 -0
- package/dist/grpc/grpc-server.service.d.ts +47 -0
- package/dist/grpc/grpc-server.service.d.ts.map +1 -0
- package/dist/grpc/grpc-server.service.js +494 -0
- package/dist/grpc/grpc-server.service.js.map +1 -0
- package/dist/grpc/index.d.ts +3 -0
- package/dist/grpc/index.d.ts.map +1 -0
- package/dist/{services/executor-pool → grpc}/index.js +2 -1
- package/dist/grpc/index.js.map +1 -0
- package/dist/index.d.ts +4 -0
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +4 -0
- package/dist/index.js.map +1 -1
- package/dist/module/atomic-queues.module.d.ts +1 -0
- package/dist/module/atomic-queues.module.d.ts.map +1 -1
- package/dist/module/atomic-queues.module.js +59 -10
- package/dist/module/atomic-queues.module.js.map +1 -1
- package/dist/services/command-discovery/command-discovery.service.js +2 -2
- package/dist/services/command-discovery/command-discovery.service.js.map +1 -1
- package/dist/services/index.d.ts +2 -8
- package/dist/services/index.d.ts.map +1 -1
- package/dist/services/index.js +2 -8
- package/dist/services/index.js.map +1 -1
- package/dist/services/message-router/index.d.ts +2 -0
- package/dist/services/message-router/index.d.ts.map +1 -0
- package/dist/services/{actor-system → message-router}/index.js +1 -1
- package/dist/services/message-router/index.js.map +1 -0
- package/dist/services/message-router/message-router.service.d.ts +53 -0
- package/dist/services/message-router/message-router.service.d.ts.map +1 -0
- package/dist/services/message-router/message-router.service.js +519 -0
- package/dist/services/message-router/message-router.service.js.map +1 -0
- package/dist/services/queue-bus/cluster-contracts.d.ts +1 -1
- package/dist/services/queue-bus/cluster-contracts.d.ts.map +1 -1
- package/dist/services/queue-bus/cluster-contracts.js.map +1 -1
- package/dist/services/queue-bus/queue-bus.service.d.ts +3 -21
- package/dist/services/queue-bus/queue-bus.service.d.ts.map +1 -1
- package/dist/services/queue-bus/queue-bus.service.js +15 -119
- package/dist/services/queue-bus/queue-bus.service.js.map +1 -1
- package/dist/utils/id.utils.d.ts +3 -0
- package/dist/utils/id.utils.d.ts.map +1 -0
- package/dist/utils/id.utils.js +14 -0
- package/dist/utils/id.utils.js.map +1 -0
- package/dist/utils/index.d.ts +1 -0
- package/dist/utils/index.d.ts.map +1 -1
- package/dist/utils/index.js +1 -0
- package/dist/utils/index.js.map +1 -1
- package/dist/wal/index.d.ts +4 -0
- package/dist/wal/index.d.ts.map +1 -0
- package/dist/{services/gate → wal}/index.js +3 -1
- package/dist/wal/index.js.map +1 -0
- package/dist/wal/wal.scripts.d.ts +51 -0
- package/dist/wal/wal.scripts.d.ts.map +1 -0
- package/dist/wal/wal.scripts.js +84 -0
- package/dist/wal/wal.scripts.js.map +1 -0
- package/dist/wal/wal.service.d.ts +46 -0
- package/dist/wal/wal.service.d.ts.map +1 -0
- package/dist/wal/wal.service.js +243 -0
- package/dist/wal/wal.service.js.map +1 -0
- package/dist/wal/wal.types.d.ts +23 -0
- package/dist/wal/wal.types.d.ts.map +1 -0
- package/dist/wal/wal.types.js +3 -0
- package/dist/wal/wal.types.js.map +1 -0
- package/dist/workers/consistent-hash.d.ts +97 -0
- package/dist/workers/consistent-hash.d.ts.map +1 -0
- package/dist/workers/consistent-hash.js +231 -0
- package/dist/workers/consistent-hash.js.map +1 -0
- package/dist/workers/entity-worker-manager.d.ts +35 -0
- package/dist/workers/entity-worker-manager.d.ts.map +1 -0
- package/dist/workers/entity-worker-manager.js +237 -0
- package/dist/workers/entity-worker-manager.js.map +1 -0
- package/dist/workers/entity-worker.d.ts +54 -0
- package/dist/workers/entity-worker.d.ts.map +1 -0
- package/dist/workers/entity-worker.js +142 -0
- package/dist/workers/entity-worker.js.map +1 -0
- package/dist/workers/index.d.ts +4 -0
- package/dist/workers/index.d.ts.map +1 -0
- package/dist/{services/log → workers}/index.js +3 -1
- package/dist/workers/index.js.map +1 -0
- package/package.json +17 -4
- package/dist/services/actor-system/actor-system.service.d.ts +0 -19
- package/dist/services/actor-system/actor-system.service.d.ts.map +0 -1
- package/dist/services/actor-system/actor-system.service.js +0 -86
- package/dist/services/actor-system/actor-system.service.js.map +0 -1
- package/dist/services/actor-system/index.d.ts +0 -2
- package/dist/services/actor-system/index.d.ts.map +0 -1
- package/dist/services/actor-system/index.js.map +0 -1
- package/dist/services/executor-pool/executor-pool.service.d.ts +0 -38
- package/dist/services/executor-pool/executor-pool.service.d.ts.map +0 -1
- package/dist/services/executor-pool/executor-pool.service.js +0 -166
- package/dist/services/executor-pool/executor-pool.service.js.map +0 -1
- package/dist/services/executor-pool/index.d.ts +0 -2
- package/dist/services/executor-pool/index.d.ts.map +0 -1
- package/dist/services/executor-pool/index.js.map +0 -1
- package/dist/services/gate/gate.service.d.ts +0 -17
- package/dist/services/gate/gate.service.d.ts.map +0 -1
- package/dist/services/gate/gate.service.js +0 -81
- package/dist/services/gate/gate.service.js.map +0 -1
- package/dist/services/gate/index.d.ts +0 -2
- package/dist/services/gate/index.d.ts.map +0 -1
- package/dist/services/gate/index.js.map +0 -1
- package/dist/services/log/index.d.ts +0 -2
- package/dist/services/log/index.d.ts.map +0 -1
- package/dist/services/log/index.js.map +0 -1
- package/dist/services/log/log.service.d.ts +0 -21
- package/dist/services/log/log.service.d.ts.map +0 -1
- package/dist/services/log/log.service.js +0 -92
- package/dist/services/log/log.service.js.map +0 -1
- package/dist/services/registry/index.d.ts +0 -4
- package/dist/services/registry/index.d.ts.map +0 -1
- package/dist/services/registry/index.js +0 -20
- package/dist/services/registry/index.js.map +0 -1
- package/dist/services/registry/registry.service.d.ts +0 -43
- package/dist/services/registry/registry.service.d.ts.map +0 -1
- package/dist/services/registry/registry.service.js +0 -367
- package/dist/services/registry/registry.service.js.map +0 -1
- package/dist/services/registry/registry.types.js.map +0 -1
- package/dist/services/registry/schema-converter.d.ts +0 -2
- package/dist/services/registry/schema-converter.d.ts.map +0 -1
- package/dist/services/registry/schema-converter.js +0 -27
- package/dist/services/registry/schema-converter.js.map +0 -1
- package/dist/services/result-collector/index.d.ts +0 -2
- package/dist/services/result-collector/index.d.ts.map +0 -1
- package/dist/services/result-collector/index.js +0 -18
- package/dist/services/result-collector/index.js.map +0 -1
- package/dist/services/result-collector/result-collector.service.d.ts +0 -17
- package/dist/services/result-collector/result-collector.service.d.ts.map +0 -1
- package/dist/services/result-collector/result-collector.service.js +0 -92
- package/dist/services/result-collector/result-collector.service.js.map +0 -1
- package/dist/services/scheduler/index.d.ts +0 -2
- package/dist/services/scheduler/index.d.ts.map +0 -1
- package/dist/services/scheduler/index.js +0 -18
- package/dist/services/scheduler/index.js.map +0 -1
- package/dist/services/scheduler/scheduler.service.d.ts +0 -17
- package/dist/services/scheduler/scheduler.service.d.ts.map +0 -1
- package/dist/services/scheduler/scheduler.service.js +0 -140
- package/dist/services/scheduler/scheduler.service.js.map +0 -1
- /package/dist/{services/registry → domain/interfaces}/registry.types.d.ts +0 -0
- /package/dist/{services/registry → domain/interfaces}/registry.types.js +0 -0
|
@@ -0,0 +1,423 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
var __metadata = (this && this.__metadata) || function (k, v) {
|
|
9
|
+
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
|
10
|
+
};
|
|
11
|
+
var __param = (this && this.__param) || function (paramIndex, decorator) {
|
|
12
|
+
return function (target, key) { decorator(target, key, paramIndex); }
|
|
13
|
+
};
|
|
14
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
15
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
16
|
+
};
|
|
17
|
+
var ClusterDiscoveryService_1;
|
|
18
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
19
|
+
exports.ClusterDiscoveryService = void 0;
|
|
20
|
+
const common_1 = require("@nestjs/common");
|
|
21
|
+
const ioredis_1 = __importDefault(require("ioredis"));
|
|
22
|
+
const uuid_1 = require("uuid");
|
|
23
|
+
const utils_1 = require("../utils");
|
|
24
|
+
const constants_1 = require("../services/constants");
|
|
25
|
+
const grpc_peer_monitor_service_1 = require("./grpc-peer-monitor.service");
|
|
26
|
+
const redis_health_monitor_service_1 = require("./redis-health-monitor.service");
|
|
27
|
+
const HEARTBEAT_SCRIPT = `
|
|
28
|
+
redis.call("HSET", KEYS[1], "heartbeat_at", ARGV[1])
|
|
29
|
+
redis.call("PEXPIRE", KEYS[1], ARGV[2])
|
|
30
|
+
return redis.call("HGET", KEYS[1], "ring_version")
|
|
31
|
+
`;
|
|
32
|
+
/**
|
|
33
|
+
* Cluster Discovery Service.
|
|
34
|
+
*
|
|
35
|
+
* Manages server heartbeats, ring membership, and ring change events
|
|
36
|
+
* using Redis as the source of truth.
|
|
37
|
+
*
|
|
38
|
+
* Only active when `config.grpc.enabled` is true.
|
|
39
|
+
*/
|
|
40
|
+
let ClusterDiscoveryService = ClusterDiscoveryService_1 = class ClusterDiscoveryService {
|
|
41
|
+
constructor(redis, config, peerMonitor, redisHealthMonitor) {
|
|
42
|
+
this.redis = redis;
|
|
43
|
+
this.config = config;
|
|
44
|
+
this.peerMonitor = peerMonitor;
|
|
45
|
+
this.redisHealthMonitor = redisHealthMonitor;
|
|
46
|
+
this.logger = new common_1.Logger(ClusterDiscoveryService_1.name);
|
|
47
|
+
this.instanceId = (0, uuid_1.v4)();
|
|
48
|
+
this.lastSeenRingVersion = 0;
|
|
49
|
+
this.heartbeatTimer = null;
|
|
50
|
+
this.reconcileTimer = null;
|
|
51
|
+
this.eventReconcileTimer = null;
|
|
52
|
+
this.subscriber = null;
|
|
53
|
+
this.redisAvailable = true;
|
|
54
|
+
this.unsubscribePeerMonitor = null;
|
|
55
|
+
this.unsubscribeRedisHealth = null;
|
|
56
|
+
this.changeListeners = [];
|
|
57
|
+
this.keyPrefix = (0, utils_1.resolveKeyPrefix)(config);
|
|
58
|
+
this.enabled = config.grpc?.enabled ?? false;
|
|
59
|
+
this.serverId = config.grpc?.serverId ?? 'unknown';
|
|
60
|
+
this.serviceGroup = config.grpc?.serviceGroup ?? 'default';
|
|
61
|
+
this.grpcAddress = config.grpc?.advertisedAddress ?? '0.0.0.0:50051';
|
|
62
|
+
this.heartbeatIntervalMs = config.grpc?.heartbeatMs ?? 400;
|
|
63
|
+
this.reconcileIntervalMs = config.grpc?.reconcileIntervalMs ?? 2000;
|
|
64
|
+
this.nodeTTL = config.grpc?.nodeTTLMs ?? 1500;
|
|
65
|
+
}
|
|
66
|
+
async onModuleInit() {
|
|
67
|
+
if (!this.enabled)
|
|
68
|
+
return;
|
|
69
|
+
// Clean up stale keys from a previous instance of this serverId
|
|
70
|
+
const oldPattern = `${this.keyPrefix}:cluster:nodes:${this.serverId}:*`;
|
|
71
|
+
const oldKeys = await this.scanKeys(oldPattern);
|
|
72
|
+
if (oldKeys.length > 0) {
|
|
73
|
+
await this.redis.del(...oldKeys);
|
|
74
|
+
const indexKey = this.getNodeIndexKey();
|
|
75
|
+
const pipeline = this.redis.pipeline();
|
|
76
|
+
for (const key of oldKeys) {
|
|
77
|
+
const member = key.replace(`${this.keyPrefix}:cluster:nodes:`, '');
|
|
78
|
+
pipeline.srem(indexKey, member);
|
|
79
|
+
}
|
|
80
|
+
await pipeline.exec();
|
|
81
|
+
this.logger.log(`Cleaned ${oldKeys.length} stale node key(s) from previous instance`);
|
|
82
|
+
}
|
|
83
|
+
// Register this node
|
|
84
|
+
await this.registerNode();
|
|
85
|
+
// Increment ring version
|
|
86
|
+
await this.incrementRingVersion();
|
|
87
|
+
// Publish join event
|
|
88
|
+
await this.publishEvent('join');
|
|
89
|
+
// Start heartbeat
|
|
90
|
+
this.heartbeatTimer = setInterval(() => {
|
|
91
|
+
this.heartbeat().catch((err) => {
|
|
92
|
+
this.logger.error(`Heartbeat failed: ${err.message}`);
|
|
93
|
+
});
|
|
94
|
+
}, this.heartbeatIntervalMs);
|
|
95
|
+
// Start ring reconciliation (slower than heartbeat — consistency fallback)
|
|
96
|
+
this.reconcileTimer = setInterval(() => {
|
|
97
|
+
this.reconcile().catch((err) => {
|
|
98
|
+
this.logger.error(`Reconciliation failed: ${err.message}`);
|
|
99
|
+
});
|
|
100
|
+
}, this.reconcileIntervalMs);
|
|
101
|
+
// Subscribe to ring events
|
|
102
|
+
this.subscriber = this.redis.duplicate();
|
|
103
|
+
const channel = this.getEventsChannel();
|
|
104
|
+
await this.subscriber.subscribe(channel);
|
|
105
|
+
this.subscriber.on('message', (_ch, payload) => {
|
|
106
|
+
this.handleEvent(payload);
|
|
107
|
+
});
|
|
108
|
+
// Subscribe to gRPC peer state changes for fast failure detection
|
|
109
|
+
if (this.peerMonitor) {
|
|
110
|
+
this.unsubscribePeerMonitor = this.peerMonitor.onPeerStateChange((peerId, state) => {
|
|
111
|
+
if (state === 'suspected-dead') {
|
|
112
|
+
this.logger.warn(`Peer ${peerId} suspected dead via gRPC — triggering reconcile`);
|
|
113
|
+
this.reconcile().catch((err) => {
|
|
114
|
+
this.logger.error(`Triggered reconcile failed: ${err.message}`);
|
|
115
|
+
});
|
|
116
|
+
}
|
|
117
|
+
});
|
|
118
|
+
}
|
|
119
|
+
// Subscribe to Redis health changes for voluntary step-down
|
|
120
|
+
if (this.redisHealthMonitor) {
|
|
121
|
+
this.unsubscribeRedisHealth = this.redisHealthMonitor.onHealthChange((healthy) => {
|
|
122
|
+
if (!healthy) {
|
|
123
|
+
this.handleRedisLost();
|
|
124
|
+
}
|
|
125
|
+
else {
|
|
126
|
+
this.handleRedisRecovered();
|
|
127
|
+
}
|
|
128
|
+
});
|
|
129
|
+
}
|
|
130
|
+
this.logger.log(`Cluster discovery started: serverId=${this.serverId}, group=${this.serviceGroup}`);
|
|
131
|
+
}
|
|
132
|
+
async onApplicationShutdown() {
|
|
133
|
+
if (!this.enabled)
|
|
134
|
+
return;
|
|
135
|
+
if (this.heartbeatTimer)
|
|
136
|
+
clearInterval(this.heartbeatTimer);
|
|
137
|
+
if (this.reconcileTimer)
|
|
138
|
+
clearInterval(this.reconcileTimer);
|
|
139
|
+
if (this.eventReconcileTimer)
|
|
140
|
+
clearTimeout(this.eventReconcileTimer);
|
|
141
|
+
if (this.unsubscribePeerMonitor)
|
|
142
|
+
this.unsubscribePeerMonitor();
|
|
143
|
+
if (this.unsubscribeRedisHealth)
|
|
144
|
+
this.unsubscribeRedisHealth();
|
|
145
|
+
// Remove this node from the ring
|
|
146
|
+
await this.redis.del(this.getNodeKey());
|
|
147
|
+
await this.redis.srem(this.getNodeIndexKey(), `${this.serverId}:${this.instanceId}`);
|
|
148
|
+
await this.incrementRingVersion();
|
|
149
|
+
await this.publishEvent('leave');
|
|
150
|
+
if (this.subscriber) {
|
|
151
|
+
await this.subscriber.unsubscribe();
|
|
152
|
+
await this.subscriber.quit();
|
|
153
|
+
}
|
|
154
|
+
this.logger.log('Cluster discovery stopped');
|
|
155
|
+
}
|
|
156
|
+
// =========================================================================
|
|
157
|
+
// PUBLIC API
|
|
158
|
+
// =========================================================================
|
|
159
|
+
/**
|
|
160
|
+
* Get all live nodes in the cluster.
|
|
161
|
+
*/
|
|
162
|
+
async getNodes() {
|
|
163
|
+
const indexKey = this.getNodeIndexKey();
|
|
164
|
+
const members = await this.redis.smembers(indexKey);
|
|
165
|
+
const nodes = [];
|
|
166
|
+
const staleMembers = [];
|
|
167
|
+
for (const member of members) {
|
|
168
|
+
const nodeKey = `${this.keyPrefix}:cluster:nodes:${member}`;
|
|
169
|
+
const data = await this.redis.hgetall(nodeKey);
|
|
170
|
+
if (data && data.server_id) {
|
|
171
|
+
nodes.push(this.parseNodeData(data));
|
|
172
|
+
}
|
|
173
|
+
else {
|
|
174
|
+
staleMembers.push(member);
|
|
175
|
+
}
|
|
176
|
+
}
|
|
177
|
+
if (staleMembers.length > 0) {
|
|
178
|
+
const pipeline = this.redis.pipeline();
|
|
179
|
+
for (const member of staleMembers) {
|
|
180
|
+
pipeline.srem(indexKey, member);
|
|
181
|
+
}
|
|
182
|
+
pipeline.exec().catch(() => { });
|
|
183
|
+
}
|
|
184
|
+
return nodes;
|
|
185
|
+
}
|
|
186
|
+
/**
|
|
187
|
+
* Get the current ring version.
|
|
188
|
+
*/
|
|
189
|
+
async getRingVersion() {
|
|
190
|
+
const versionKey = `${this.keyPrefix}:cluster:ring:version`;
|
|
191
|
+
const version = await this.redis.get(versionKey);
|
|
192
|
+
return version ? parseInt(version, 10) : 0;
|
|
193
|
+
}
|
|
194
|
+
/**
|
|
195
|
+
* Register a listener for ring change events.
|
|
196
|
+
*/
|
|
197
|
+
onRingChange(listener) {
|
|
198
|
+
this.changeListeners.push(listener);
|
|
199
|
+
return () => {
|
|
200
|
+
const idx = this.changeListeners.indexOf(listener);
|
|
201
|
+
if (idx >= 0)
|
|
202
|
+
this.changeListeners.splice(idx, 1);
|
|
203
|
+
};
|
|
204
|
+
}
|
|
205
|
+
/**
|
|
206
|
+
* Whether the cluster is healthy (Redis reachable).
|
|
207
|
+
* When false, this node has stepped down and should not accept new work.
|
|
208
|
+
*/
|
|
209
|
+
isClusterHealthy() {
|
|
210
|
+
return this.redisAvailable;
|
|
211
|
+
}
|
|
212
|
+
/**
|
|
213
|
+
* Get this server's ID.
|
|
214
|
+
*/
|
|
215
|
+
getServerId() {
|
|
216
|
+
return this.serverId;
|
|
217
|
+
}
|
|
218
|
+
/**
|
|
219
|
+
* Resolve which service group owns an entity type (via Redis registry).
|
|
220
|
+
*/
|
|
221
|
+
async resolveServiceGroup(entityType) {
|
|
222
|
+
const registryKey = `${this.keyPrefix}:cluster:entity-registry:${entityType}`;
|
|
223
|
+
return this.redis.get(registryKey);
|
|
224
|
+
}
|
|
225
|
+
// =========================================================================
|
|
226
|
+
// INTERNAL — Registration
|
|
227
|
+
// =========================================================================
|
|
228
|
+
async registerNode() {
|
|
229
|
+
const key = this.getNodeKey();
|
|
230
|
+
const indexKey = this.getNodeIndexKey();
|
|
231
|
+
const entityTypes = Object.keys(this.config.entities ?? {});
|
|
232
|
+
const data = {
|
|
233
|
+
server_id: this.serverId,
|
|
234
|
+
instance_id: this.instanceId,
|
|
235
|
+
grpc_address: this.grpcAddress,
|
|
236
|
+
service_group: this.serviceGroup,
|
|
237
|
+
entity_types: entityTypes.join(','),
|
|
238
|
+
ring_version: '0',
|
|
239
|
+
started_at: Date.now().toString(),
|
|
240
|
+
heartbeat_at: Date.now().toString(),
|
|
241
|
+
};
|
|
242
|
+
await this.redis.hset(key, data);
|
|
243
|
+
await this.redis.pexpire(key, this.nodeTTL);
|
|
244
|
+
await this.redis.sadd(indexKey, `${this.serverId}:${this.instanceId}`);
|
|
245
|
+
// Register entity type → service group mapping for cross-service routing
|
|
246
|
+
for (const et of entityTypes) {
|
|
247
|
+
const registryKey = `${this.keyPrefix}:cluster:entity-registry:${et}`;
|
|
248
|
+
await this.redis.set(registryKey, this.serviceGroup, 'PX', this.nodeTTL * 2);
|
|
249
|
+
}
|
|
250
|
+
}
|
|
251
|
+
async heartbeat() {
|
|
252
|
+
const key = this.getNodeKey();
|
|
253
|
+
const entityTypes = Object.keys(this.config.entities ?? {});
|
|
254
|
+
const pipeline = this.redis.pipeline();
|
|
255
|
+
pipeline.eval(HEARTBEAT_SCRIPT, 1, key, Date.now().toString(), this.nodeTTL.toString());
|
|
256
|
+
for (const et of entityTypes) {
|
|
257
|
+
pipeline.pexpire(`${this.keyPrefix}:cluster:entity-registry:${et}`, this.nodeTTL * 2);
|
|
258
|
+
}
|
|
259
|
+
await pipeline.exec();
|
|
260
|
+
}
|
|
261
|
+
async reconcile() {
|
|
262
|
+
if (!this.redisAvailable)
|
|
263
|
+
return;
|
|
264
|
+
const currentVersion = await this.getRingVersion();
|
|
265
|
+
if (currentVersion > this.lastSeenRingVersion) {
|
|
266
|
+
this.lastSeenRingVersion = currentVersion;
|
|
267
|
+
}
|
|
268
|
+
const nodes = await this.getNodes();
|
|
269
|
+
// Sync gRPC peer monitor with current Redis-known nodes
|
|
270
|
+
if (this.peerMonitor) {
|
|
271
|
+
this.peerMonitor.syncPeers(nodes
|
|
272
|
+
.filter((n) => n.serverId !== this.serverId)
|
|
273
|
+
.map((n) => ({ serverId: n.serverId, address: n.grpcAddress })));
|
|
274
|
+
}
|
|
275
|
+
this.notifyListeners(nodes);
|
|
276
|
+
}
|
|
277
|
+
// =========================================================================
|
|
278
|
+
// INTERNAL — Events
|
|
279
|
+
// =========================================================================
|
|
280
|
+
async publishEvent(action) {
|
|
281
|
+
const channel = this.getEventsChannel();
|
|
282
|
+
const ringVersion = await this.getRingVersion();
|
|
283
|
+
await this.redis.publish(channel, JSON.stringify({
|
|
284
|
+
action,
|
|
285
|
+
serverId: this.serverId,
|
|
286
|
+
serviceGroup: this.serviceGroup,
|
|
287
|
+
timestamp: Date.now(),
|
|
288
|
+
ringVersion,
|
|
289
|
+
}));
|
|
290
|
+
}
|
|
291
|
+
handleEvent(payload) {
|
|
292
|
+
try {
|
|
293
|
+
const event = JSON.parse(payload);
|
|
294
|
+
if (event.serverId === this.serverId)
|
|
295
|
+
return;
|
|
296
|
+
this.logger.log(`Ring event: ${event.action} from ${event.serverId}`);
|
|
297
|
+
const receivedVersion = event.ringVersion;
|
|
298
|
+
const gapDetected = receivedVersion !== undefined &&
|
|
299
|
+
this.lastSeenRingVersion > 0 &&
|
|
300
|
+
receivedVersion > this.lastSeenRingVersion + 1;
|
|
301
|
+
if (receivedVersion !== undefined && receivedVersion > this.lastSeenRingVersion) {
|
|
302
|
+
this.lastSeenRingVersion = receivedVersion;
|
|
303
|
+
}
|
|
304
|
+
if (gapDetected) {
|
|
305
|
+
this.logger.warn(`Ring version gap detected (expected <= ${this.lastSeenRingVersion}, got ${receivedVersion}) — immediate reconcile`);
|
|
306
|
+
if (this.eventReconcileTimer)
|
|
307
|
+
clearTimeout(this.eventReconcileTimer);
|
|
308
|
+
this.eventReconcileTimer = null;
|
|
309
|
+
this.reconcile().catch((err) => this.logger.error(`Gap-triggered reconcile failed: ${err.message}`));
|
|
310
|
+
return;
|
|
311
|
+
}
|
|
312
|
+
if (this.eventReconcileTimer)
|
|
313
|
+
clearTimeout(this.eventReconcileTimer);
|
|
314
|
+
this.eventReconcileTimer = setTimeout(() => {
|
|
315
|
+
this.eventReconcileTimer = null;
|
|
316
|
+
this.reconcile().catch((err) => this.logger.error(`Event-triggered reconcile failed: ${err.message}`));
|
|
317
|
+
}, 200);
|
|
318
|
+
}
|
|
319
|
+
catch {
|
|
320
|
+
// Ignore malformed events
|
|
321
|
+
}
|
|
322
|
+
}
|
|
323
|
+
notifyListeners(nodes) {
|
|
324
|
+
for (const listener of this.changeListeners) {
|
|
325
|
+
try {
|
|
326
|
+
listener(nodes);
|
|
327
|
+
}
|
|
328
|
+
catch (err) {
|
|
329
|
+
this.logger.error(`Ring change listener error: ${err.message}`);
|
|
330
|
+
}
|
|
331
|
+
}
|
|
332
|
+
}
|
|
333
|
+
// =========================================================================
|
|
334
|
+
// INTERNAL — Redis health step-down / recovery
|
|
335
|
+
// =========================================================================
|
|
336
|
+
handleRedisLost() {
|
|
337
|
+
this.redisAvailable = false;
|
|
338
|
+
this.logger.error('Redis connectivity lost — stepping down from cluster');
|
|
339
|
+
if (this.heartbeatTimer) {
|
|
340
|
+
clearInterval(this.heartbeatTimer);
|
|
341
|
+
this.heartbeatTimer = null;
|
|
342
|
+
}
|
|
343
|
+
if (this.reconcileTimer) {
|
|
344
|
+
clearInterval(this.reconcileTimer);
|
|
345
|
+
this.reconcileTimer = null;
|
|
346
|
+
}
|
|
347
|
+
// Notify with empty node list — triggers leader resignation and worker cleanup
|
|
348
|
+
this.notifyListeners([]);
|
|
349
|
+
}
|
|
350
|
+
handleRedisRecovered() {
|
|
351
|
+
this.redisAvailable = true;
|
|
352
|
+
this.logger.log('Redis connectivity restored — rejoining cluster');
|
|
353
|
+
this.registerNode()
|
|
354
|
+
.then(() => this.incrementRingVersion())
|
|
355
|
+
.then(() => this.publishEvent('join'))
|
|
356
|
+
.catch((err) => {
|
|
357
|
+
this.logger.error(`Failed to rejoin cluster: ${err.message}`);
|
|
358
|
+
});
|
|
359
|
+
if (!this.heartbeatTimer) {
|
|
360
|
+
this.heartbeatTimer = setInterval(() => {
|
|
361
|
+
this.heartbeat().catch((err) => {
|
|
362
|
+
this.logger.error(`Heartbeat failed: ${err.message}`);
|
|
363
|
+
});
|
|
364
|
+
}, this.heartbeatIntervalMs);
|
|
365
|
+
}
|
|
366
|
+
if (!this.reconcileTimer) {
|
|
367
|
+
this.reconcileTimer = setInterval(() => {
|
|
368
|
+
this.reconcile().catch((err) => {
|
|
369
|
+
this.logger.error(`Reconciliation failed: ${err.message}`);
|
|
370
|
+
});
|
|
371
|
+
}, this.reconcileIntervalMs);
|
|
372
|
+
}
|
|
373
|
+
}
|
|
374
|
+
// =========================================================================
|
|
375
|
+
// INTERNAL — Helpers
|
|
376
|
+
// =========================================================================
|
|
377
|
+
getNodeKey() {
|
|
378
|
+
return `${this.keyPrefix}:cluster:nodes:${this.serverId}:${this.instanceId}`;
|
|
379
|
+
}
|
|
380
|
+
getNodeIndexKey() {
|
|
381
|
+
return `${this.keyPrefix}:cluster:node-index`;
|
|
382
|
+
}
|
|
383
|
+
getEventsChannel() {
|
|
384
|
+
return `${this.keyPrefix}:cluster:events`;
|
|
385
|
+
}
|
|
386
|
+
async incrementRingVersion() {
|
|
387
|
+
const versionKey = `${this.keyPrefix}:cluster:ring:version`;
|
|
388
|
+
return this.redis.incr(versionKey);
|
|
389
|
+
}
|
|
390
|
+
parseNodeData(data) {
|
|
391
|
+
return {
|
|
392
|
+
serverId: data.server_id,
|
|
393
|
+
instanceId: data.instance_id || undefined,
|
|
394
|
+
grpcAddress: data.grpc_address,
|
|
395
|
+
serviceGroup: data.service_group,
|
|
396
|
+
entityTypes: data.entity_types ? data.entity_types.split(',').filter(Boolean) : [],
|
|
397
|
+
ringVersion: parseInt(data.ring_version || '0', 10),
|
|
398
|
+
startedAt: parseInt(data.started_at || '0', 10),
|
|
399
|
+
heartbeatAt: parseInt(data.heartbeat_at || '0', 10),
|
|
400
|
+
};
|
|
401
|
+
}
|
|
402
|
+
async scanKeys(pattern) {
|
|
403
|
+
let cursor = '0';
|
|
404
|
+
const keys = [];
|
|
405
|
+
do {
|
|
406
|
+
const [nextCursor, foundKeys] = await this.redis.scan(cursor, 'MATCH', pattern, 'COUNT', 100);
|
|
407
|
+
cursor = nextCursor;
|
|
408
|
+
keys.push(...foundKeys);
|
|
409
|
+
} while (cursor !== '0');
|
|
410
|
+
return keys;
|
|
411
|
+
}
|
|
412
|
+
};
|
|
413
|
+
exports.ClusterDiscoveryService = ClusterDiscoveryService;
|
|
414
|
+
exports.ClusterDiscoveryService = ClusterDiscoveryService = ClusterDiscoveryService_1 = __decorate([
|
|
415
|
+
(0, common_1.Injectable)(),
|
|
416
|
+
__param(0, (0, common_1.Inject)(constants_1.ATOMIC_QUEUES_REDIS)),
|
|
417
|
+
__param(1, (0, common_1.Inject)(constants_1.ATOMIC_QUEUES_CONFIG)),
|
|
418
|
+
__param(2, (0, common_1.Optional)()),
|
|
419
|
+
__param(3, (0, common_1.Optional)()),
|
|
420
|
+
__metadata("design:paramtypes", [ioredis_1.default, Object, grpc_peer_monitor_service_1.GrpcPeerMonitor,
|
|
421
|
+
redis_health_monitor_service_1.RedisHealthMonitor])
|
|
422
|
+
], ClusterDiscoveryService);
|
|
423
|
+
//# sourceMappingURL=cluster-discovery.service.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cluster-discovery.service.js","sourceRoot":"","sources":["../../src/cluster/cluster-discovery.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;AAAA,2CAOwB;AACxB,sDAA4B;AAC5B,+BAAoC;AAEpC,oCAA4C;AAC5C,qDAAkF;AAClF,2EAA8D;AAC9D,iFAAoE;AAapE,MAAM,gBAAgB,GAAG;;;;CAIxB,CAAC;AAEF;;;;;;;GAOG;AAEI,IAAM,uBAAuB,+BAA7B,MAAM,uBAAuB;IAwBlC,YAC+B,KAA6B,EAC5B,MAAkD,EACpE,WAA8C,EAC9C,kBAAwD;QAHtB,UAAK,GAAL,KAAK,CAAO;QACX,WAAM,GAAN,MAAM,CAA2B;QACnD,gBAAW,GAAX,WAAW,CAAkB;QAC7B,uBAAkB,GAAlB,kBAAkB,CAAqB;QA3BrD,WAAM,GAAG,IAAI,eAAM,CAAC,yBAAuB,CAAC,IAAI,CAAC,CAAC;QAUlD,eAAU,GAAW,IAAA,SAAM,GAAE,CAAC;QACvC,wBAAmB,GAAG,CAAC,CAAC;QAExB,mBAAc,GAA0B,IAAI,CAAC;QAC7C,mBAAc,GAA0B,IAAI,CAAC;QAC7C,wBAAmB,GAA0B,IAAI,CAAC;QAClD,eAAU,GAAiB,IAAI,CAAC;QAChC,mBAAc,GAAG,IAAI,CAAC;QACtB,2BAAsB,GAAwB,IAAI,CAAC;QACnD,2BAAsB,GAAwB,IAAI,CAAC;QAE1C,oBAAe,GAA0C,EAAE,CAAC;QAQ3E,IAAI,CAAC,SAAS,GAAG,IAAA,wBAAgB,EAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,OAAO,IAAI,KAAK,CAAC;QAC7C,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,IAAI,EAAE,QAAQ,IAAI,SAAS,CAAC;QACnD,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,IAAI,EAAE,YAAY,IAAI,SAAS,CAAC;QAC3D,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,IAAI,EAAE,iBAAiB,IAAI,eAAe,CAAC;QACrE,IAAI,CAAC,mBAAmB,GAAG,MAAM,CAAC,IAAI,EAAE,WAAW,IAAI,GAAG,CAAC;QAC3D,IAAI,CAAC,mBAAmB,GAAG,MAAM,CAAC,IAAI,EAAE,mBAAmB,IAAI,IAAI,CAAC;QACpE,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,SAAS,IAAI,IAAI,CAAC;IAChD,CAAC;IAED,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,OAAO;QAE1B,gEAAgE;QAChE,MAAM,UAAU,GAAG,GAAG,IAAI,CAAC,SAAS,kBAAkB,IAAI,CAAC,QAAQ,IAAI,CAAC;QACxE,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QAChD,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC;YACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;YACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;YACvC,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;gBAC1B,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,SAAS,iBAAiB,EAAE,EAAE,CAAC,CAAC;gBACnE,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAClC,CAAC;YACD,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACtB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,WAAW,OAAO,CAAC,MAAM,2CAA2C,CAAC,CAAC;QACxF,CAAC;QAED,qBAAqB;QACrB,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;QAE1B,yBAAyB;QACzB,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAElC,qBAAqB;QACrB,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAEhC,kBAAkB;QAClB,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;YACrC,IAAI,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC7B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qBAAsB,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;YACnE,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAE7B,2EAA2E;QAC3E,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;YACrC,IAAI,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBAC7B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA2B,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;YACxE,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAE7B,2BAA2B;QAC3B,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxC,MAAM,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACzC,IAAI,CAAC,UAAU,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,GAAW,EAAE,OAAe,EAAE,EAAE;YAC7D,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,kEAAkE;QAClE,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;gBACjF,IAAI,KAAK,KAAK,gBAAgB,EAAE,CAAC;oBAC/B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,MAAM,iDAAiD,CAAC,CAAC;oBAClF,IAAI,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;wBAC7B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+BAAgC,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;oBAC7E,CAAC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAED,4DAA4D;QAC5D,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC5B,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC/E,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,IAAI,CAAC,eAAe,EAAE,CAAC;gBACzB,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,oBAAoB,EAAE,CAAC;gBAC9B,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,uCAAuC,IAAI,CAAC,QAAQ,WAAW,IAAI,CAAC,YAAY,EAAE,CACnF,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,qBAAqB;QACzB,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,OAAO;QAE1B,IAAI,IAAI,CAAC,cAAc;YAAE,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC5D,IAAI,IAAI,CAAC,cAAc;YAAE,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAC5D,IAAI,IAAI,CAAC,mBAAmB;YAAE,YAAY,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACrE,IAAI,IAAI,CAAC,sBAAsB;YAAE,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAC/D,IAAI,IAAI,CAAC,sBAAsB;YAAE,IAAI,CAAC,sBAAsB,EAAE,CAAC;QAE/D,iCAAiC;QACjC,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;QACxC,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE,GAAG,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;QACrF,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAClC,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAEjC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,MAAM,IAAI,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC;YACpC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;QAC/B,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;IAC/C,CAAC;IAED,4EAA4E;IAC5E,aAAa;IACb,4EAA4E;IAE5E;;OAEG;IACH,KAAK,CAAC,QAAQ;QACZ,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QACxC,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAEpD,MAAM,KAAK,GAAkB,EAAE,CAAC;QAChC,MAAM,YAAY,GAAa,EAAE,CAAC;QAElC,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,OAAO,GAAG,GAAG,IAAI,CAAC,SAAS,kBAAkB,MAAM,EAAE,CAAC;YAC5D,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC/C,IAAI,IAAI,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBAC3B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;YACvC,CAAC;iBAAM,CAAC;gBACN,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;QAED,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;YACvC,KAAK,MAAM,MAAM,IAAI,YAAY,EAAE,CAAC;gBAClC,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAClC,CAAC;YACD,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAClC,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc;QAClB,MAAM,UAAU,GAAG,GAAG,IAAI,CAAC,SAAS,uBAAuB,CAAC;QAC5D,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACjD,OAAO,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7C,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,QAAwC;QACnD,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACpC,OAAO,GAAG,EAAE;YACV,MAAM,GAAG,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YACnD,IAAI,GAAG,IAAI,CAAC;gBAAE,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,gBAAgB;QACd,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,mBAAmB,CAAC,UAAkB;QAC1C,MAAM,WAAW,GAAG,GAAG,IAAI,CAAC,SAAS,4BAA4B,UAAU,EAAE,CAAC;QAC9E,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IACrC,CAAC;IAED,4EAA4E;IAC5E,0BAA0B;IAC1B,4EAA4E;IAEpE,KAAK,CAAC,YAAY;QACxB,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC9B,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;QACxC,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,IAAI,GAA2B;YACnC,SAAS,EAAE,IAAI,CAAC,QAAQ;YACxB,WAAW,EAAE,IAAI,CAAC,UAAU;YAC5B,YAAY,EAAE,IAAI,CAAC,WAAW;YAC9B,aAAa,EAAE,IAAI,CAAC,YAAY;YAChC,YAAY,EAAE,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC;YACnC,YAAY,EAAE,GAAG;YACjB,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;YACjC,YAAY,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;SACpC,CAAC;QAEF,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QACjC,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5C,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC;QAEvE,yEAAyE;QACzE,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;YAC7B,MAAM,WAAW,GAAG,GAAG,IAAI,CAAC,SAAS,4BAA4B,EAAE,EAAE,CAAC;YACtE,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,WAAW,EAAE,IAAI,CAAC,YAAY,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC;QAC/E,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,SAAS;QACrB,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC9B,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;QAC5D,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;QAEvC,QAAQ,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;QACxF,KAAK,MAAM,EAAE,IAAI,WAAW,EAAE,CAAC;YAC7B,QAAQ,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,SAAS,4BAA4B,EAAE,EAAE,EAAE,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC;QACxF,CAAC;QAED,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IACxB,CAAC;IAEO,KAAK,CAAC,SAAS;QACrB,IAAI,CAAC,IAAI,CAAC,cAAc;YAAE,OAAO;QAEjC,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QACnD,IAAI,cAAc,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC9C,IAAI,CAAC,mBAAmB,GAAG,cAAc,CAAC;QAC5C,CAAC;QAED,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QAEpC,wDAAwD;QACxD,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,IAAI,CAAC,WAAW,CAAC,SAAS,CACxB,KAAK;iBACF,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,KAAK,IAAI,CAAC,QAAQ,CAAC;iBAC3C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,CAClE,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC9B,CAAC;IAED,4EAA4E;IAC5E,oBAAoB;IACpB,4EAA4E;IAEpE,KAAK,CAAC,YAAY,CAAC,MAAwB;QACjD,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAChD,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CACtB,OAAO,EACP,IAAI,CAAC,SAAS,CAAC;YACb,MAAM;YACN,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;YACrB,WAAW;SACZ,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,WAAW,CAAC,OAAe;QACjC,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAClC,IAAI,KAAK,CAAC,QAAQ,KAAK,IAAI,CAAC,QAAQ;gBAAE,OAAO;YAE7C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,eAAe,KAAK,CAAC,MAAM,SAAS,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;YAEtE,MAAM,eAAe,GAAG,KAAK,CAAC,WAAiC,CAAC;YAChE,MAAM,WAAW,GACf,eAAe,KAAK,SAAS;gBAC7B,IAAI,CAAC,mBAAmB,GAAG,CAAC;gBAC5B,eAAe,GAAG,IAAI,CAAC,mBAAmB,GAAG,CAAC,CAAC;YAEjD,IAAI,eAAe,KAAK,SAAS,IAAI,eAAe,GAAG,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAChF,IAAI,CAAC,mBAAmB,GAAG,eAAe,CAAC;YAC7C,CAAC;YAED,IAAI,WAAW,EAAE,CAAC;gBAChB,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,0CAA0C,IAAI,CAAC,mBAAmB,SAAS,eAAe,yBAAyB,CACpH,CAAC;gBACF,IAAI,IAAI,CAAC,mBAAmB;oBAAE,YAAY,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;gBACrE,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC;gBAChC,IAAI,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAC7B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mCAAoC,GAAa,CAAC,OAAO,EAAE,CAAC,CAC/E,CAAC;gBACF,OAAO;YACT,CAAC;YAED,IAAI,IAAI,CAAC,mBAAmB;gBAAE,YAAY,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YACrE,IAAI,CAAC,mBAAmB,GAAG,UAAU,CAAC,GAAG,EAAE;gBACzC,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC;gBAChC,IAAI,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAC7B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qCAAsC,GAAa,CAAC,OAAO,EAAE,CAAC,CACjF,CAAC;YACJ,CAAC,EAAE,GAAG,CAAC,CAAC;QACV,CAAC;QAAC,MAAM,CAAC;YACP,0BAA0B;QAC5B,CAAC;IACH,CAAC;IAEO,eAAe,CAAC,KAAoB;QAC1C,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YAC5C,IAAI,CAAC;gBACH,QAAQ,CAAC,KAAK,CAAC,CAAC;YAClB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,+BAAgC,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;YAC7E,CAAC;QACH,CAAC;IACH,CAAC;IAED,4EAA4E;IAC5E,+CAA+C;IAC/C,4EAA4E;IAEpE,eAAe;QACrB,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;QAC5B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sDAAsD,CAAC,CAAC;QAE1E,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QACD,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;QAED,+EAA+E;QAC/E,IAAI,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;IAC3B,CAAC;IAEO,oBAAoB;QAC1B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC;QAEnE,IAAI,CAAC,YAAY,EAAE;aAChB,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,CAAC;aACvC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;aACrC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA8B,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;QAC3E,CAAC,CAAC,CAAC;QAEL,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;gBACrC,IAAI,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;oBAC7B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qBAAsB,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;gBACnE,CAAC,CAAC,CAAC;YACL,CAAC,EAAE,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC/B,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,IAAI,CAAC,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE;gBACrC,IAAI,CAAC,SAAS,EAAE,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;oBAC7B,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA2B,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;gBACxE,CAAC,CAAC,CAAC;YACL,CAAC,EAAE,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED,4EAA4E;IAC5E,qBAAqB;IACrB,4EAA4E;IAEpE,UAAU;QAChB,OAAO,GAAG,IAAI,CAAC,SAAS,kBAAkB,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;IAC/E,CAAC;IAEO,eAAe;QACrB,OAAO,GAAG,IAAI,CAAC,SAAS,qBAAqB,CAAC;IAChD,CAAC;IAEO,gBAAgB;QACtB,OAAO,GAAG,IAAI,CAAC,SAAS,iBAAiB,CAAC;IAC5C,CAAC;IAEO,KAAK,CAAC,oBAAoB;QAChC,MAAM,UAAU,GAAG,GAAG,IAAI,CAAC,SAAS,uBAAuB,CAAC;QAC5D,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACrC,CAAC;IAEO,aAAa,CAAC,IAA4B;QAChD,OAAO;YACL,QAAQ,EAAE,IAAI,CAAC,SAAS;YACxB,UAAU,EAAE,IAAI,CAAC,WAAW,IAAI,SAAS;YACzC,WAAW,EAAE,IAAI,CAAC,YAAY;YAC9B,YAAY,EAAE,IAAI,CAAC,aAAa;YAChC,WAAW,EAAE,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE;YAClF,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,YAAY,IAAI,GAAG,EAAE,EAAE,CAAC;YACnD,SAAS,EAAE,QAAQ,CAAC,IAAI,CAAC,UAAU,IAAI,GAAG,EAAE,EAAE,CAAC;YAC/C,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,YAAY,IAAI,GAAG,EAAE,EAAE,CAAC;SACpD,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,QAAQ,CAAC,OAAe;QACpC,IAAI,MAAM,GAAG,GAAG,CAAC;QACjB,MAAM,IAAI,GAAa,EAAE,CAAC;QAC1B,GAAG,CAAC;YACF,MAAM,CAAC,UAAU,EAAE,SAAS,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC;YAC9F,MAAM,GAAG,UAAU,CAAC;YACpB,IAAI,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,CAAC;QAC1B,CAAC,QAAQ,MAAM,KAAK,GAAG,EAAE;QACzB,OAAO,IAAI,CAAC;IACd,CAAC;CACF,CAAA;AA9bY,0DAAuB;kCAAvB,uBAAuB;IADnC,IAAA,mBAAU,GAAE;IA0BR,WAAA,IAAA,eAAM,EAAC,+BAAmB,CAAC,CAAA;IAC3B,WAAA,IAAA,eAAM,EAAC,gCAAoB,CAAC,CAAA;IAC5B,WAAA,IAAA,iBAAQ,GAAE,CAAA;IACV,WAAA,IAAA,iBAAQ,GAAE,CAAA;qCAH0C,iBAAK,UAEf,2CAAe;QACR,iDAAkB;GA5B3D,uBAAuB,CA8bnC"}
|
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
import { OnModuleInit, OnApplicationShutdown } from '@nestjs/common';
|
|
2
|
+
import { IAtomicQueuesModuleConfig } from '../domain';
|
|
3
|
+
export type PeerLivenessState = 'alive' | 'suspected-dead' | 'unknown';
|
|
4
|
+
export declare class GrpcPeerMonitor implements OnModuleInit, OnApplicationShutdown {
|
|
5
|
+
private readonly config;
|
|
6
|
+
private readonly logger;
|
|
7
|
+
private readonly enabled;
|
|
8
|
+
private readonly debounceMs;
|
|
9
|
+
private readonly keepaliveTimeMs;
|
|
10
|
+
private readonly keepaliveTimeoutMs;
|
|
11
|
+
private readonly connectivityWatchMs;
|
|
12
|
+
private grpcModule;
|
|
13
|
+
private readonly peers;
|
|
14
|
+
private readonly stateChangeListeners;
|
|
15
|
+
constructor(config: IAtomicQueuesModuleConfig);
|
|
16
|
+
onModuleInit(): Promise<void>;
|
|
17
|
+
onApplicationShutdown(): Promise<void>;
|
|
18
|
+
getPeerState(serverId: string): PeerLivenessState;
|
|
19
|
+
onPeerStateChange(listener: (serverId: string, state: PeerLivenessState) => void): () => void;
|
|
20
|
+
watchPeer(serverId: string, address: string): void;
|
|
21
|
+
unwatchPeer(serverId: string): void;
|
|
22
|
+
syncPeers(peers: Array<{
|
|
23
|
+
serverId: string;
|
|
24
|
+
address: string;
|
|
25
|
+
}>): void;
|
|
26
|
+
private startWatchLoop;
|
|
27
|
+
private handleStateTransition;
|
|
28
|
+
private cleanupPeer;
|
|
29
|
+
private notifyListeners;
|
|
30
|
+
}
|
|
31
|
+
//# sourceMappingURL=grpc-peer-monitor.service.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"grpc-peer-monitor.service.d.ts","sourceRoot":"","sources":["../../src/cluster/grpc-peer-monitor.service.ts"],"names":[],"mappings":"AAAA,OAAO,EAA8B,YAAY,EAAE,qBAAqB,EAAE,MAAM,gBAAgB,CAAC;AACjG,OAAO,EAAE,yBAAyB,EAAE,MAAM,WAAW,CAAC;AAGtD,MAAM,MAAM,iBAAiB,GAAG,OAAO,GAAG,gBAAgB,GAAG,SAAS,CAAC;AAmCvE,qBACa,eAAgB,YAAW,YAAY,EAAE,qBAAqB;IAc/B,OAAO,CAAC,QAAQ,CAAC,MAAM;IAbjE,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAoC;IAC3D,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAU;IAClC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAS;IACzC,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAS;IAC5C,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAS;IAE7C,OAAO,CAAC,UAAU,CAA2B;IAC7C,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAgC;IACtD,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAE9B;gBAEoD,MAAM,EAAE,yBAAyB;IAQtF,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;IAY7B,qBAAqB,IAAI,OAAO,CAAC,IAAI,CAAC;IAW5C,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,iBAAiB;IAIjD,iBAAiB,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,iBAAiB,KAAK,IAAI,GAAG,MAAM,IAAI;IAQ7F,SAAS,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAgClD,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAOnC,SAAS,CAAC,KAAK,EAAE,KAAK,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC,GAAG,IAAI;IAgBpE,OAAO,CAAC,cAAc;IAuBtB,OAAO,CAAC,qBAAqB;IAsC7B,OAAO,CAAC,WAAW;IAanB,OAAO,CAAC,eAAe;CASxB"}
|
|
@@ -0,0 +1,192 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
3
|
+
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
4
|
+
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
5
|
+
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
|
6
|
+
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
7
|
+
};
|
|
8
|
+
var __metadata = (this && this.__metadata) || function (k, v) {
|
|
9
|
+
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
|
10
|
+
};
|
|
11
|
+
var __param = (this && this.__param) || function (paramIndex, decorator) {
|
|
12
|
+
return function (target, key) { decorator(target, key, paramIndex); }
|
|
13
|
+
};
|
|
14
|
+
var GrpcPeerMonitor_1;
|
|
15
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
16
|
+
exports.GrpcPeerMonitor = void 0;
|
|
17
|
+
const common_1 = require("@nestjs/common");
|
|
18
|
+
const constants_1 = require("../services/constants");
|
|
19
|
+
let GrpcPeerMonitor = GrpcPeerMonitor_1 = class GrpcPeerMonitor {
|
|
20
|
+
constructor(config) {
|
|
21
|
+
this.config = config;
|
|
22
|
+
this.logger = new common_1.Logger(GrpcPeerMonitor_1.name);
|
|
23
|
+
this.grpcModule = null;
|
|
24
|
+
this.peers = new Map();
|
|
25
|
+
this.stateChangeListeners = [];
|
|
26
|
+
this.enabled = (config.grpc?.enabled ?? false) && (config.grpc?.peerMonitorEnabled ?? true);
|
|
27
|
+
this.debounceMs = config.grpc?.peerSuspectDebounceMs ?? 500;
|
|
28
|
+
this.keepaliveTimeMs = config.grpc?.keepaliveTimeMs ?? 10000;
|
|
29
|
+
this.keepaliveTimeoutMs = config.grpc?.keepaliveTimeoutMs ?? 5000;
|
|
30
|
+
this.connectivityWatchMs = config.grpc?.deadlines?.connectivityWatchMs ?? 30000;
|
|
31
|
+
}
|
|
32
|
+
async onModuleInit() {
|
|
33
|
+
if (!this.enabled)
|
|
34
|
+
return;
|
|
35
|
+
try {
|
|
36
|
+
// eslint-disable-next-line @typescript-eslint/no-var-requires
|
|
37
|
+
this.grpcModule = require('@grpc/grpc-js');
|
|
38
|
+
this.logger.log('gRPC peer monitor initialized');
|
|
39
|
+
}
|
|
40
|
+
catch {
|
|
41
|
+
this.logger.warn('gRPC peer monitor disabled: @grpc/grpc-js not installed');
|
|
42
|
+
}
|
|
43
|
+
}
|
|
44
|
+
async onApplicationShutdown() {
|
|
45
|
+
for (const [, peer] of this.peers) {
|
|
46
|
+
this.cleanupPeer(peer);
|
|
47
|
+
}
|
|
48
|
+
this.peers.clear();
|
|
49
|
+
}
|
|
50
|
+
// =========================================================================
|
|
51
|
+
// PUBLIC API
|
|
52
|
+
// =========================================================================
|
|
53
|
+
getPeerState(serverId) {
|
|
54
|
+
return this.peers.get(serverId)?.state ?? 'unknown';
|
|
55
|
+
}
|
|
56
|
+
onPeerStateChange(listener) {
|
|
57
|
+
this.stateChangeListeners.push(listener);
|
|
58
|
+
return () => {
|
|
59
|
+
const idx = this.stateChangeListeners.indexOf(listener);
|
|
60
|
+
if (idx >= 0)
|
|
61
|
+
this.stateChangeListeners.splice(idx, 1);
|
|
62
|
+
};
|
|
63
|
+
}
|
|
64
|
+
watchPeer(serverId, address) {
|
|
65
|
+
if (!this.grpcModule)
|
|
66
|
+
return;
|
|
67
|
+
const existing = this.peers.get(serverId);
|
|
68
|
+
if (existing && existing.address === address)
|
|
69
|
+
return;
|
|
70
|
+
if (existing)
|
|
71
|
+
this.cleanupPeer(existing);
|
|
72
|
+
const channelOptions = {
|
|
73
|
+
'grpc.keepalive_time_ms': this.keepaliveTimeMs,
|
|
74
|
+
'grpc.keepalive_timeout_ms': this.keepaliveTimeoutMs,
|
|
75
|
+
'grpc.keepalive_permit_without_calls': 1,
|
|
76
|
+
};
|
|
77
|
+
const client = new this.grpcModule.Client(address, this.grpcModule.credentials.createInsecure(), channelOptions);
|
|
78
|
+
const entry = {
|
|
79
|
+
serverId,
|
|
80
|
+
address,
|
|
81
|
+
state: 'unknown',
|
|
82
|
+
client,
|
|
83
|
+
suspectTimer: null,
|
|
84
|
+
watchActive: true,
|
|
85
|
+
};
|
|
86
|
+
this.peers.set(serverId, entry);
|
|
87
|
+
this.startWatchLoop(entry);
|
|
88
|
+
}
|
|
89
|
+
unwatchPeer(serverId) {
|
|
90
|
+
const peer = this.peers.get(serverId);
|
|
91
|
+
if (!peer)
|
|
92
|
+
return;
|
|
93
|
+
this.cleanupPeer(peer);
|
|
94
|
+
this.peers.delete(serverId);
|
|
95
|
+
}
|
|
96
|
+
syncPeers(peers) {
|
|
97
|
+
const incoming = new Set(peers.map((p) => p.serverId));
|
|
98
|
+
for (const [id] of this.peers) {
|
|
99
|
+
if (!incoming.has(id))
|
|
100
|
+
this.unwatchPeer(id);
|
|
101
|
+
}
|
|
102
|
+
for (const p of peers) {
|
|
103
|
+
this.watchPeer(p.serverId, p.address);
|
|
104
|
+
}
|
|
105
|
+
}
|
|
106
|
+
// =========================================================================
|
|
107
|
+
// INTERNAL — connectivity state watch loop
|
|
108
|
+
// =========================================================================
|
|
109
|
+
startWatchLoop(entry) {
|
|
110
|
+
if (!this.grpcModule || !entry.watchActive)
|
|
111
|
+
return;
|
|
112
|
+
const channel = entry.client.getChannel();
|
|
113
|
+
const currentState = channel.getConnectivityState(true);
|
|
114
|
+
this.handleStateTransition(entry, currentState);
|
|
115
|
+
const deadline = new Date(Date.now() + this.connectivityWatchMs);
|
|
116
|
+
channel.watchConnectivityState(currentState, deadline, (err) => {
|
|
117
|
+
if (!entry.watchActive)
|
|
118
|
+
return;
|
|
119
|
+
if (err) {
|
|
120
|
+
// Deadline expired without state change — re-arm
|
|
121
|
+
this.startWatchLoop(entry);
|
|
122
|
+
return;
|
|
123
|
+
}
|
|
124
|
+
// State changed — process and re-arm
|
|
125
|
+
this.startWatchLoop(entry);
|
|
126
|
+
});
|
|
127
|
+
}
|
|
128
|
+
handleStateTransition(entry, grpcState) {
|
|
129
|
+
if (!this.grpcModule)
|
|
130
|
+
return;
|
|
131
|
+
const { connectivityState } = this.grpcModule;
|
|
132
|
+
const READY = connectivityState.READY;
|
|
133
|
+
const TRANSIENT_FAILURE = connectivityState.TRANSIENT_FAILURE;
|
|
134
|
+
const SHUTDOWN = connectivityState.SHUTDOWN;
|
|
135
|
+
if (grpcState === READY) {
|
|
136
|
+
if (entry.suspectTimer) {
|
|
137
|
+
clearTimeout(entry.suspectTimer);
|
|
138
|
+
entry.suspectTimer = null;
|
|
139
|
+
}
|
|
140
|
+
if (entry.state !== 'alive') {
|
|
141
|
+
entry.state = 'alive';
|
|
142
|
+
this.notifyListeners(entry.serverId, 'alive');
|
|
143
|
+
}
|
|
144
|
+
}
|
|
145
|
+
else if (grpcState === TRANSIENT_FAILURE || grpcState === SHUTDOWN) {
|
|
146
|
+
if (entry.state === 'alive' && !entry.suspectTimer) {
|
|
147
|
+
entry.suspectTimer = setTimeout(() => {
|
|
148
|
+
entry.suspectTimer = null;
|
|
149
|
+
if (entry.state !== 'suspected-dead') {
|
|
150
|
+
entry.state = 'suspected-dead';
|
|
151
|
+
this.logger.warn(`Peer ${entry.serverId} suspected dead (gRPC ${grpcState === SHUTDOWN ? 'SHUTDOWN' : 'TRANSIENT_FAILURE'})`);
|
|
152
|
+
this.notifyListeners(entry.serverId, 'suspected-dead');
|
|
153
|
+
}
|
|
154
|
+
}, this.debounceMs);
|
|
155
|
+
}
|
|
156
|
+
}
|
|
157
|
+
// IDLE and CONNECTING: no action, wait for READY or TRANSIENT_FAILURE
|
|
158
|
+
}
|
|
159
|
+
// =========================================================================
|
|
160
|
+
// INTERNAL — helpers
|
|
161
|
+
// =========================================================================
|
|
162
|
+
cleanupPeer(peer) {
|
|
163
|
+
peer.watchActive = false;
|
|
164
|
+
if (peer.suspectTimer) {
|
|
165
|
+
clearTimeout(peer.suspectTimer);
|
|
166
|
+
peer.suspectTimer = null;
|
|
167
|
+
}
|
|
168
|
+
try {
|
|
169
|
+
peer.client.close();
|
|
170
|
+
}
|
|
171
|
+
catch {
|
|
172
|
+
// ignore
|
|
173
|
+
}
|
|
174
|
+
}
|
|
175
|
+
notifyListeners(serverId, state) {
|
|
176
|
+
for (const listener of this.stateChangeListeners) {
|
|
177
|
+
try {
|
|
178
|
+
listener(serverId, state);
|
|
179
|
+
}
|
|
180
|
+
catch (err) {
|
|
181
|
+
this.logger.error(`Peer state change listener error: ${err.message}`);
|
|
182
|
+
}
|
|
183
|
+
}
|
|
184
|
+
}
|
|
185
|
+
};
|
|
186
|
+
exports.GrpcPeerMonitor = GrpcPeerMonitor;
|
|
187
|
+
exports.GrpcPeerMonitor = GrpcPeerMonitor = GrpcPeerMonitor_1 = __decorate([
|
|
188
|
+
(0, common_1.Injectable)(),
|
|
189
|
+
__param(0, (0, common_1.Inject)(constants_1.ATOMIC_QUEUES_CONFIG)),
|
|
190
|
+
__metadata("design:paramtypes", [Object])
|
|
191
|
+
], GrpcPeerMonitor);
|
|
192
|
+
//# sourceMappingURL=grpc-peer-monitor.service.js.map
|