@nahisaho/yata-scale 1.8.5
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 +66 -0
- package/dist/CacheManager.d.ts +107 -0
- package/dist/CacheManager.d.ts.map +1 -0
- package/dist/CacheManager.js +343 -0
- package/dist/CacheManager.js.map +1 -0
- package/dist/IndexManager.d.ts +103 -0
- package/dist/IndexManager.d.ts.map +1 -0
- package/dist/IndexManager.js +385 -0
- package/dist/IndexManager.js.map +1 -0
- package/dist/MigrationHelper.d.ts +91 -0
- package/dist/MigrationHelper.d.ts.map +1 -0
- package/dist/MigrationHelper.js +220 -0
- package/dist/MigrationHelper.js.map +1 -0
- package/dist/QueryCoordinator.d.ts +76 -0
- package/dist/QueryCoordinator.d.ts.map +1 -0
- package/dist/QueryCoordinator.js +298 -0
- package/dist/QueryCoordinator.js.map +1 -0
- package/dist/ShardManager.d.ts +48 -0
- package/dist/ShardManager.d.ts.map +1 -0
- package/dist/ShardManager.js +193 -0
- package/dist/ShardManager.js.map +1 -0
- package/dist/SyncController.d.ts +102 -0
- package/dist/SyncController.d.ts.map +1 -0
- package/dist/SyncController.js +286 -0
- package/dist/SyncController.js.map +1 -0
- package/dist/YataScaleManager.d.ts +75 -0
- package/dist/YataScaleManager.d.ts.map +1 -0
- package/dist/YataScaleManager.js +265 -0
- package/dist/YataScaleManager.js.map +1 -0
- package/dist/cache/CacheManager.d.ts +116 -0
- package/dist/cache/CacheManager.d.ts.map +1 -0
- package/dist/cache/CacheManager.js +286 -0
- package/dist/cache/CacheManager.js.map +1 -0
- package/dist/cache/InvalidationManager.d.ts +91 -0
- package/dist/cache/InvalidationManager.d.ts.map +1 -0
- package/dist/cache/InvalidationManager.js +155 -0
- package/dist/cache/InvalidationManager.js.map +1 -0
- package/dist/cache/L1Cache.d.ts +97 -0
- package/dist/cache/L1Cache.d.ts.map +1 -0
- package/dist/cache/L1Cache.js +225 -0
- package/dist/cache/L1Cache.js.map +1 -0
- package/dist/cache/L2Cache.d.ts +93 -0
- package/dist/cache/L2Cache.d.ts.map +1 -0
- package/dist/cache/L2Cache.js +248 -0
- package/dist/cache/L2Cache.js.map +1 -0
- package/dist/cache/L3Cache.d.ts +101 -0
- package/dist/cache/L3Cache.d.ts.map +1 -0
- package/dist/cache/L3Cache.js +229 -0
- package/dist/cache/L3Cache.js.map +1 -0
- package/dist/cache/index.d.ts +12 -0
- package/dist/cache/index.d.ts.map +1 -0
- package/dist/cache/index.js +11 -0
- package/dist/cache/index.js.map +1 -0
- package/dist/errors.d.ts +255 -0
- package/dist/errors.d.ts.map +1 -0
- package/dist/errors.js +515 -0
- package/dist/errors.js.map +1 -0
- package/dist/index/BPlusTreeIndex.d.ts +101 -0
- package/dist/index/BPlusTreeIndex.d.ts.map +1 -0
- package/dist/index/BPlusTreeIndex.js +310 -0
- package/dist/index/BPlusTreeIndex.js.map +1 -0
- package/dist/index/BloomFilter.d.ts +85 -0
- package/dist/index/BloomFilter.d.ts.map +1 -0
- package/dist/index/BloomFilter.js +215 -0
- package/dist/index/BloomFilter.js.map +1 -0
- package/dist/index/FullTextIndex.d.ts +87 -0
- package/dist/index/FullTextIndex.d.ts.map +1 -0
- package/dist/index/FullTextIndex.js +213 -0
- package/dist/index/FullTextIndex.js.map +1 -0
- package/dist/index/GraphIndex.d.ts +116 -0
- package/dist/index/GraphIndex.d.ts.map +1 -0
- package/dist/index/GraphIndex.js +308 -0
- package/dist/index/GraphIndex.js.map +1 -0
- package/dist/index/IndexManager.d.ts +105 -0
- package/dist/index/IndexManager.d.ts.map +1 -0
- package/dist/index/IndexManager.js +287 -0
- package/dist/index/IndexManager.js.map +1 -0
- package/dist/index/index.d.ts +12 -0
- package/dist/index/index.d.ts.map +1 -0
- package/dist/index/index.js +11 -0
- package/dist/index/index.js.map +1 -0
- package/dist/index.d.ts +12 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +12 -0
- package/dist/index.js.map +1 -0
- package/dist/query/DistributedExecutor.d.ts +78 -0
- package/dist/query/DistributedExecutor.d.ts.map +1 -0
- package/dist/query/DistributedExecutor.js +232 -0
- package/dist/query/DistributedExecutor.js.map +1 -0
- package/dist/query/QueryCoordinator.d.ts +98 -0
- package/dist/query/QueryCoordinator.d.ts.map +1 -0
- package/dist/query/QueryCoordinator.js +246 -0
- package/dist/query/QueryCoordinator.js.map +1 -0
- package/dist/query/QueryPlanner.d.ts +88 -0
- package/dist/query/QueryPlanner.d.ts.map +1 -0
- package/dist/query/QueryPlanner.js +235 -0
- package/dist/query/QueryPlanner.js.map +1 -0
- package/dist/query/WorkerPool.d.ts +108 -0
- package/dist/query/WorkerPool.d.ts.map +1 -0
- package/dist/query/WorkerPool.js +195 -0
- package/dist/query/WorkerPool.js.map +1 -0
- package/dist/query/index.d.ts +11 -0
- package/dist/query/index.d.ts.map +1 -0
- package/dist/query/index.js +10 -0
- package/dist/query/index.js.map +1 -0
- package/dist/shard/GraphPartitionStrategy.d.ts +65 -0
- package/dist/shard/GraphPartitionStrategy.d.ts.map +1 -0
- package/dist/shard/GraphPartitionStrategy.js +208 -0
- package/dist/shard/GraphPartitionStrategy.js.map +1 -0
- package/dist/shard/HashPartitionStrategy.d.ts +56 -0
- package/dist/shard/HashPartitionStrategy.d.ts.map +1 -0
- package/dist/shard/HashPartitionStrategy.js +169 -0
- package/dist/shard/HashPartitionStrategy.js.map +1 -0
- package/dist/shard/RangePartitionStrategy.d.ts +63 -0
- package/dist/shard/RangePartitionStrategy.d.ts.map +1 -0
- package/dist/shard/RangePartitionStrategy.js +157 -0
- package/dist/shard/RangePartitionStrategy.js.map +1 -0
- package/dist/shard/Rebalancer.d.ts +70 -0
- package/dist/shard/Rebalancer.d.ts.map +1 -0
- package/dist/shard/Rebalancer.js +184 -0
- package/dist/shard/Rebalancer.js.map +1 -0
- package/dist/shard/ShardManager.d.ts +123 -0
- package/dist/shard/ShardManager.d.ts.map +1 -0
- package/dist/shard/ShardManager.js +353 -0
- package/dist/shard/ShardManager.js.map +1 -0
- package/dist/shard/ShardRouter.d.ts +102 -0
- package/dist/shard/ShardRouter.d.ts.map +1 -0
- package/dist/shard/ShardRouter.js +181 -0
- package/dist/shard/ShardRouter.js.map +1 -0
- package/dist/shard/index.d.ts +13 -0
- package/dist/shard/index.d.ts.map +1 -0
- package/dist/shard/index.js +12 -0
- package/dist/shard/index.js.map +1 -0
- package/dist/sync/ConflictResolver.d.ts +71 -0
- package/dist/sync/ConflictResolver.d.ts.map +1 -0
- package/dist/sync/ConflictResolver.js +214 -0
- package/dist/sync/ConflictResolver.js.map +1 -0
- package/dist/sync/SyncController.d.ts +150 -0
- package/dist/sync/SyncController.d.ts.map +1 -0
- package/dist/sync/SyncController.js +304 -0
- package/dist/sync/SyncController.js.map +1 -0
- package/dist/sync/VectorClock.d.ts +102 -0
- package/dist/sync/VectorClock.d.ts.map +1 -0
- package/dist/sync/VectorClock.js +189 -0
- package/dist/sync/VectorClock.js.map +1 -0
- package/dist/sync/WALManager.d.ts +120 -0
- package/dist/sync/WALManager.d.ts.map +1 -0
- package/dist/sync/WALManager.js +240 -0
- package/dist/sync/WALManager.js.map +1 -0
- package/dist/sync/index.d.ts +11 -0
- package/dist/sync/index.d.ts.map +1 -0
- package/dist/sync/index.js +10 -0
- package/dist/sync/index.js.map +1 -0
- package/dist/types.d.ts +323 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +7 -0
- package/dist/types.js.map +1 -0
- package/package.json +86 -0
|
@@ -0,0 +1,220 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @nahisaho/yata-scale - Migration Helper
|
|
3
|
+
*
|
|
4
|
+
* Helps migrate from yata-local to yata-scale
|
|
5
|
+
*/
|
|
6
|
+
import { ok, err } from 'neverthrow';
|
|
7
|
+
import { InitializationError } from './errors.js';
|
|
8
|
+
/**
|
|
9
|
+
* Migration helper for transitioning to yata-scale
|
|
10
|
+
*/
|
|
11
|
+
export class MigrationHelper {
|
|
12
|
+
batchSize;
|
|
13
|
+
progress;
|
|
14
|
+
constructor(options) {
|
|
15
|
+
this.batchSize = options?.batchSize ?? 1000;
|
|
16
|
+
this.progress = {
|
|
17
|
+
totalEntities: 0,
|
|
18
|
+
migratedEntities: 0,
|
|
19
|
+
totalRelationships: 0,
|
|
20
|
+
migratedRelationships: 0,
|
|
21
|
+
errors: [],
|
|
22
|
+
startTime: new Date(),
|
|
23
|
+
currentPhase: 'entities',
|
|
24
|
+
};
|
|
25
|
+
}
|
|
26
|
+
/**
|
|
27
|
+
* Migrate data from source to target
|
|
28
|
+
*/
|
|
29
|
+
async migrate(source, sink, options) {
|
|
30
|
+
try {
|
|
31
|
+
// Get counts
|
|
32
|
+
this.progress = {
|
|
33
|
+
...this.progress,
|
|
34
|
+
totalEntities: await source.getEntityCount(),
|
|
35
|
+
totalRelationships: await source.getRelationshipCount(),
|
|
36
|
+
startTime: new Date(),
|
|
37
|
+
};
|
|
38
|
+
// Migrate entities
|
|
39
|
+
this.progress = { ...this.progress, currentPhase: 'entities' };
|
|
40
|
+
await this.migrateEntities(source, sink, options);
|
|
41
|
+
// Migrate relationships
|
|
42
|
+
this.progress = { ...this.progress, currentPhase: 'relationships' };
|
|
43
|
+
await this.migrateRelationships(source, sink, options);
|
|
44
|
+
// Flush
|
|
45
|
+
await sink.flush();
|
|
46
|
+
this.progress = { ...this.progress, currentPhase: 'complete' };
|
|
47
|
+
return ok(this.progress);
|
|
48
|
+
}
|
|
49
|
+
catch (error) {
|
|
50
|
+
return err(new InitializationError(`Migration failed: ${error instanceof Error ? error.message : String(error)}`));
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
/**
|
|
54
|
+
* Migrate entities
|
|
55
|
+
*/
|
|
56
|
+
async migrateEntities(source, sink, options) {
|
|
57
|
+
let batch = [];
|
|
58
|
+
for await (const entity of source.getEntities()) {
|
|
59
|
+
// Validate if requested
|
|
60
|
+
if (options?.validateData) {
|
|
61
|
+
const validation = this.validateEntity(entity);
|
|
62
|
+
if (!validation.valid) {
|
|
63
|
+
this.progress = {
|
|
64
|
+
...this.progress,
|
|
65
|
+
errors: [...this.progress.errors, `Entity ${entity.id}: ${validation.error}`],
|
|
66
|
+
};
|
|
67
|
+
continue;
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
batch.push(entity);
|
|
71
|
+
if (batch.length >= this.batchSize) {
|
|
72
|
+
await this.processBatch(batch, sink, 'entity');
|
|
73
|
+
batch = [];
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
// Process remaining
|
|
77
|
+
if (batch.length > 0) {
|
|
78
|
+
await this.processBatch(batch, sink, 'entity');
|
|
79
|
+
}
|
|
80
|
+
}
|
|
81
|
+
/**
|
|
82
|
+
* Migrate relationships
|
|
83
|
+
*/
|
|
84
|
+
async migrateRelationships(source, sink, options) {
|
|
85
|
+
let batch = [];
|
|
86
|
+
for await (const relationship of source.getRelationships()) {
|
|
87
|
+
// Validate if requested
|
|
88
|
+
if (options?.validateData) {
|
|
89
|
+
const validation = this.validateRelationship(relationship);
|
|
90
|
+
if (!validation.valid) {
|
|
91
|
+
this.progress = {
|
|
92
|
+
...this.progress,
|
|
93
|
+
errors: [
|
|
94
|
+
...this.progress.errors,
|
|
95
|
+
`Relationship ${relationship.id}: ${validation.error}`,
|
|
96
|
+
],
|
|
97
|
+
};
|
|
98
|
+
continue;
|
|
99
|
+
}
|
|
100
|
+
}
|
|
101
|
+
batch.push(relationship);
|
|
102
|
+
if (batch.length >= this.batchSize) {
|
|
103
|
+
await this.processBatch(batch, sink, 'relationship');
|
|
104
|
+
batch = [];
|
|
105
|
+
}
|
|
106
|
+
}
|
|
107
|
+
// Process remaining
|
|
108
|
+
if (batch.length > 0) {
|
|
109
|
+
await this.processBatch(batch, sink, 'relationship');
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
/**
|
|
113
|
+
* Process a batch
|
|
114
|
+
*/
|
|
115
|
+
async processBatch(batch, sink, type) {
|
|
116
|
+
for (const item of batch) {
|
|
117
|
+
try {
|
|
118
|
+
if (type === 'entity') {
|
|
119
|
+
await sink.putEntity(item);
|
|
120
|
+
this.progress = {
|
|
121
|
+
...this.progress,
|
|
122
|
+
migratedEntities: this.progress.migratedEntities + 1,
|
|
123
|
+
};
|
|
124
|
+
}
|
|
125
|
+
else {
|
|
126
|
+
await sink.putRelationship(item);
|
|
127
|
+
this.progress = {
|
|
128
|
+
...this.progress,
|
|
129
|
+
migratedRelationships: this.progress.migratedRelationships + 1,
|
|
130
|
+
};
|
|
131
|
+
}
|
|
132
|
+
}
|
|
133
|
+
catch (error) {
|
|
134
|
+
this.progress = {
|
|
135
|
+
...this.progress,
|
|
136
|
+
errors: [
|
|
137
|
+
...this.progress.errors,
|
|
138
|
+
`Failed to migrate ${type} ${item.id}: ${error instanceof Error ? error.message : String(error)}`,
|
|
139
|
+
],
|
|
140
|
+
};
|
|
141
|
+
}
|
|
142
|
+
}
|
|
143
|
+
}
|
|
144
|
+
/**
|
|
145
|
+
* Validate entity
|
|
146
|
+
*/
|
|
147
|
+
validateEntity(entity) {
|
|
148
|
+
if (!entity.id) {
|
|
149
|
+
return { valid: false, error: 'Missing ID' };
|
|
150
|
+
}
|
|
151
|
+
if (!entity.type) {
|
|
152
|
+
return { valid: false, error: 'Missing type' };
|
|
153
|
+
}
|
|
154
|
+
if (!entity.name) {
|
|
155
|
+
return { valid: false, error: 'Missing name' };
|
|
156
|
+
}
|
|
157
|
+
return { valid: true };
|
|
158
|
+
}
|
|
159
|
+
/**
|
|
160
|
+
* Validate relationship
|
|
161
|
+
*/
|
|
162
|
+
validateRelationship(rel) {
|
|
163
|
+
if (!rel.id) {
|
|
164
|
+
return { valid: false, error: 'Missing ID' };
|
|
165
|
+
}
|
|
166
|
+
if (!rel.sourceId) {
|
|
167
|
+
return { valid: false, error: 'Missing sourceId' };
|
|
168
|
+
}
|
|
169
|
+
if (!rel.targetId) {
|
|
170
|
+
return { valid: false, error: 'Missing targetId' };
|
|
171
|
+
}
|
|
172
|
+
if (!rel.type) {
|
|
173
|
+
return { valid: false, error: 'Missing type' };
|
|
174
|
+
}
|
|
175
|
+
return { valid: true };
|
|
176
|
+
}
|
|
177
|
+
/**
|
|
178
|
+
* Get current progress
|
|
179
|
+
*/
|
|
180
|
+
getProgress() {
|
|
181
|
+
return this.progress;
|
|
182
|
+
}
|
|
183
|
+
/**
|
|
184
|
+
* Get migration percentage
|
|
185
|
+
*/
|
|
186
|
+
getPercentage() {
|
|
187
|
+
const total = this.progress.totalEntities + this.progress.totalRelationships;
|
|
188
|
+
const done = this.progress.migratedEntities + this.progress.migratedRelationships;
|
|
189
|
+
return total > 0 ? (done / total) * 100 : 0;
|
|
190
|
+
}
|
|
191
|
+
/**
|
|
192
|
+
* Generate configuration for yata-scale based on data
|
|
193
|
+
*/
|
|
194
|
+
static generateConfig(entityCount, relationshipCount) {
|
|
195
|
+
// Calculate optimal shard count (target 100K entities per shard)
|
|
196
|
+
const shardCount = Math.max(1, Math.ceil(entityCount / 100000));
|
|
197
|
+
// Calculate cache sizes
|
|
198
|
+
const hotEntities = Math.min(entityCount * 0.1, 10000);
|
|
199
|
+
const warmSize = Math.min(entityCount * 0.3, 100 * 1024 * 1024);
|
|
200
|
+
return {
|
|
201
|
+
shards: {
|
|
202
|
+
count: shardCount,
|
|
203
|
+
replicationFactor: shardCount > 1 ? 2 : 1,
|
|
204
|
+
partitionStrategy: shardCount > 3 ? 'hash' : 'range',
|
|
205
|
+
},
|
|
206
|
+
cache: {
|
|
207
|
+
l1MaxEntries: Math.floor(hotEntities),
|
|
208
|
+
l2MaxSize: Math.floor(warmSize),
|
|
209
|
+
l3MaxEntries: entityCount,
|
|
210
|
+
ttlMs: 300000,
|
|
211
|
+
},
|
|
212
|
+
query: {
|
|
213
|
+
defaultTimeout: 30000,
|
|
214
|
+
maxConcurrency: 4,
|
|
215
|
+
maxResultSize: 10000,
|
|
216
|
+
},
|
|
217
|
+
};
|
|
218
|
+
}
|
|
219
|
+
}
|
|
220
|
+
//# sourceMappingURL=MigrationHelper.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"MigrationHelper.js","sourceRoot":"","sources":["../src/MigrationHelper.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,EAAE,EAAE,GAAG,EAAe,MAAM,YAAY,CAAC;AAElD,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AA4ClD;;GAEG;AACH,MAAM,OAAO,eAAe;IACT,SAAS,CAAS;IAC3B,QAAQ,CAAoB;IAEpC,YAAY,OAA0B;QACpC,IAAI,CAAC,SAAS,GAAG,OAAO,EAAE,SAAS,IAAI,IAAI,CAAC;QAC5C,IAAI,CAAC,QAAQ,GAAG;YACd,aAAa,EAAE,CAAC;YAChB,gBAAgB,EAAE,CAAC;YACnB,kBAAkB,EAAE,CAAC;YACrB,qBAAqB,EAAE,CAAC;YACxB,MAAM,EAAE,EAAE;YACV,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,YAAY,EAAE,UAAU;SACzB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CACX,MAAkB,EAClB,IAAc,EACd,OAA0B;QAE1B,IAAI,CAAC;YACH,aAAa;YACb,IAAI,CAAC,QAAQ,GAAG;gBACd,GAAG,IAAI,CAAC,QAAQ;gBAChB,aAAa,EAAE,MAAM,MAAM,CAAC,cAAc,EAAE;gBAC5C,kBAAkB,EAAE,MAAM,MAAM,CAAC,oBAAoB,EAAE;gBACvD,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB,CAAC;YAEF,mBAAmB;YACnB,IAAI,CAAC,QAAQ,GAAG,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC;YAC/D,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;YAElD,wBAAwB;YACxB,IAAI,CAAC,QAAQ,GAAG,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,YAAY,EAAE,eAAe,EAAE,CAAC;YACpE,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;YAEvD,QAAQ;YACR,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;YAEnB,IAAI,CAAC,QAAQ,GAAG,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC;YAE/D,OAAO,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC3B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,GAAG,CACR,IAAI,mBAAmB,CACrB,qBAAqB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAC9E,CACF,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,eAAe,CAC3B,MAAkB,EAClB,IAAc,EACd,OAA0B;QAE1B,IAAI,KAAK,GAAa,EAAE,CAAC;QAEzB,IAAI,KAAK,EAAE,MAAM,MAAM,IAAI,MAAM,CAAC,WAAW,EAAE,EAAE,CAAC;YAChD,wBAAwB;YACxB,IAAI,OAAO,EAAE,YAAY,EAAE,CAAC;gBAC1B,MAAM,UAAU,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;gBAC/C,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;oBACtB,IAAI,CAAC,QAAQ,GAAG;wBACd,GAAG,IAAI,CAAC,QAAQ;wBAChB,MAAM,EAAE,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,UAAU,MAAM,CAAC,EAAE,KAAK,UAAU,CAAC,KAAK,EAAE,CAAC;qBAC9E,CAAC;oBACF,SAAS;gBACX,CAAC;YACH,CAAC;YAED,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAEnB,IAAI,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnC,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;gBAC/C,KAAK,GAAG,EAAE,CAAC;YACb,CAAC;QACH,CAAC;QAED,oBAAoB;QACpB,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,oBAAoB,CAChC,MAAkB,EAClB,IAAc,EACd,OAA0B;QAE1B,IAAI,KAAK,GAAmB,EAAE,CAAC;QAE/B,IAAI,KAAK,EAAE,MAAM,YAAY,IAAI,MAAM,CAAC,gBAAgB,EAAE,EAAE,CAAC;YAC3D,wBAAwB;YACxB,IAAI,OAAO,EAAE,YAAY,EAAE,CAAC;gBAC1B,MAAM,UAAU,GAAG,IAAI,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;gBAC3D,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;oBACtB,IAAI,CAAC,QAAQ,GAAG;wBACd,GAAG,IAAI,CAAC,QAAQ;wBAChB,MAAM,EAAE;4BACN,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM;4BACvB,gBAAgB,YAAY,CAAC,EAAE,KAAK,UAAU,CAAC,KAAK,EAAE;yBACvD;qBACF,CAAC;oBACF,SAAS;gBACX,CAAC;YACH,CAAC;YAED,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAEzB,IAAI,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnC,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,IAAI,EAAE,cAAc,CAAC,CAAC;gBACrD,KAAK,GAAG,EAAE,CAAC;YACb,CAAC;QACH,CAAC;QAED,oBAAoB;QACpB,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,IAAI,EAAE,cAAc,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,YAAY,CACxB,KAAgC,EAChC,IAAc,EACd,IAA+B;QAE/B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;oBACtB,MAAM,IAAI,CAAC,SAAS,CAAC,IAAc,CAAC,CAAC;oBACrC,IAAI,CAAC,QAAQ,GAAG;wBACd,GAAG,IAAI,CAAC,QAAQ;wBAChB,gBAAgB,EAAE,IAAI,CAAC,QAAQ,CAAC,gBAAgB,GAAG,CAAC;qBACrD,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,CAAC,eAAe,CAAC,IAAoB,CAAC,CAAC;oBACjD,IAAI,CAAC,QAAQ,GAAG;wBACd,GAAG,IAAI,CAAC,QAAQ;wBAChB,qBAAqB,EAAE,IAAI,CAAC,QAAQ,CAAC,qBAAqB,GAAG,CAAC;qBAC/D,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,CAAC,QAAQ,GAAG;oBACd,GAAG,IAAI,CAAC,QAAQ;oBAChB,MAAM,EAAE;wBACN,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM;wBACvB,qBAAqB,IAAI,IAAK,IAAe,CAAC,EAAE,KAAK,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;qBAC9G;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,MAAc;QACnC,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;YACf,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;QAC/C,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;QACjD,CAAC;QACD,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACjB,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;QACjD,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;IACzB,CAAC;IAED;;OAEG;IACK,oBAAoB,CAAC,GAAiB;QAC5C,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACZ,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC;QAC/C,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAClB,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,kBAAkB,EAAE,CAAC;QACrD,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;YAClB,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,kBAAkB,EAAE,CAAC;QACrD,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;YACd,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;QACjD,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;IACzB,CAAC;IAED;;OAEG;IACH,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,aAAa;QACX,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QAC7E,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,CAAC,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,qBAAqB,CAAC;QAClF,OAAO,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,GAAG,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9C,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,cAAc,CACnB,WAAmB,EACnB,iBAAyB;QAEzB,iEAAiE;QACjE,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,CAAC,CAAC;QAEhE,wBAAwB;QACxB,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,WAAW,GAAG,GAAG,EAAE,KAAK,CAAC,CAAC;QACvD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,WAAW,GAAG,GAAG,EAAE,GAAG,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC;QAEhE,OAAO;YACL,MAAM,EAAE;gBACN,KAAK,EAAE,UAAU;gBACjB,iBAAiB,EAAE,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBACzC,iBAAiB,EAAE,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO;aACrD;YACD,KAAK,EAAE;gBACL,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC;gBACrC,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;gBAC/B,YAAY,EAAE,WAAW;gBACzB,KAAK,EAAE,MAAM;aACd;YACD,KAAK,EAAE;gBACL,cAAc,EAAE,KAAK;gBACrB,cAAc,EAAE,CAAC;gBACjB,aAAa,EAAE,KAAK;aACrB;SACF,CAAC;IACJ,CAAC;CACF"}
|
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @nahisaho/yata-scale - Query Coordinator
|
|
3
|
+
*
|
|
4
|
+
* Coordinates query execution across shards
|
|
5
|
+
*/
|
|
6
|
+
import { type Result } from 'neverthrow';
|
|
7
|
+
import type { GraphQuery, QueryResult, QueryPlan, Entity, Relationship } from './types.js';
|
|
8
|
+
import { QueryError } from './errors.js';
|
|
9
|
+
/**
|
|
10
|
+
* Query planner
|
|
11
|
+
*/
|
|
12
|
+
export declare class QueryPlanner {
|
|
13
|
+
plan(query: GraphQuery, shards: string[]): QueryPlan;
|
|
14
|
+
optimize(plan: QueryPlan): QueryPlan;
|
|
15
|
+
private generateCacheKey;
|
|
16
|
+
}
|
|
17
|
+
/**
|
|
18
|
+
* Worker pool for parallel execution
|
|
19
|
+
*/
|
|
20
|
+
export declare class WorkerPool {
|
|
21
|
+
private concurrency;
|
|
22
|
+
private active;
|
|
23
|
+
private pending;
|
|
24
|
+
constructor(concurrency: number);
|
|
25
|
+
map<T, R>(items: T[], fn: (item: T) => R | Promise<R>): Promise<R[]>;
|
|
26
|
+
filter<T>(items: T[], predicate: (item: T) => boolean | Promise<boolean>): Promise<T[]>;
|
|
27
|
+
reduce<T, R>(items: T[], fn: (acc: R, item: T) => R | Promise<R>, initial: R): Promise<R>;
|
|
28
|
+
get activeCount(): number;
|
|
29
|
+
get pendingCount(): number;
|
|
30
|
+
shutdown(): Promise<void>;
|
|
31
|
+
}
|
|
32
|
+
/**
|
|
33
|
+
* Distributed query executor
|
|
34
|
+
*/
|
|
35
|
+
export declare class DistributedExecutor {
|
|
36
|
+
private pool;
|
|
37
|
+
private timeout;
|
|
38
|
+
private stats;
|
|
39
|
+
constructor(concurrency: number, timeout: number);
|
|
40
|
+
execute(query: GraphQuery, shards: string[], executor: (shardId: string, query: GraphQuery) => Promise<{
|
|
41
|
+
entities: Entity[];
|
|
42
|
+
relationships: Relationship[];
|
|
43
|
+
}>): Promise<Result<QueryResult, QueryError>>;
|
|
44
|
+
private executeOnShards;
|
|
45
|
+
private timeoutPromise;
|
|
46
|
+
private mergeResults;
|
|
47
|
+
getStats(): {
|
|
48
|
+
totalQueries: number;
|
|
49
|
+
averageLatency: number;
|
|
50
|
+
};
|
|
51
|
+
shutdown(): Promise<void>;
|
|
52
|
+
}
|
|
53
|
+
/**
|
|
54
|
+
* Query coordinator
|
|
55
|
+
*/
|
|
56
|
+
export declare class QueryCoordinator {
|
|
57
|
+
private planner;
|
|
58
|
+
private executor;
|
|
59
|
+
private cache;
|
|
60
|
+
private cacheTtl;
|
|
61
|
+
private stats;
|
|
62
|
+
constructor(concurrency: number, cacheTtl: number, timeout: number);
|
|
63
|
+
query(query: GraphQuery, shards: string[], executor: (shardId: string, query: GraphQuery) => Promise<{
|
|
64
|
+
entities: Entity[];
|
|
65
|
+
relationships: Relationship[];
|
|
66
|
+
}>): Promise<Result<QueryResult, QueryError>>;
|
|
67
|
+
validate(query: GraphQuery): Result<void, QueryError>;
|
|
68
|
+
explain(query: GraphQuery, shards: string[]): string;
|
|
69
|
+
getStats(): {
|
|
70
|
+
cacheHits: number;
|
|
71
|
+
cacheMisses: number;
|
|
72
|
+
};
|
|
73
|
+
clearCache(): void;
|
|
74
|
+
shutdown(): Promise<void>;
|
|
75
|
+
}
|
|
76
|
+
//# sourceMappingURL=QueryCoordinator.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"QueryCoordinator.d.ts","sourceRoot":"","sources":["../src/QueryCoordinator.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAW,KAAK,MAAM,EAAE,MAAM,YAAY,CAAC;AAClD,OAAO,KAAK,EAAE,UAAU,EAAE,WAAW,EAAE,SAAS,EAAa,MAAM,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AACtG,OAAO,EAAE,UAAU,EAAqB,MAAM,aAAa,CAAC;AAE5D;;GAEG;AACH,qBAAa,YAAY;IACvB,IAAI,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,SAAS;IA2EpD,QAAQ,CAAC,IAAI,EAAE,SAAS,GAAG,SAAS;IAapC,OAAO,CAAC,gBAAgB;CAGzB;AAED;;GAEG;AACH,qBAAa,UAAU;IACrB,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,MAAM,CAAK;IACnB,OAAO,CAAC,OAAO,CAAkC;gBAErC,WAAW,EAAE,MAAM;IAIzB,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC;IAmBpE,MAAM,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,SAAS,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,CAAC,EAAE,CAAC;IAYvF,MAAM,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC,GAAG,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;IAQ/F,IAAI,WAAW,IAAI,MAAM,CAExB;IAED,IAAI,YAAY,IAAI,MAAM,CAEzB;IAEK,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAKhC;AAED;;GAEG;AACH,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,IAAI,CAAa;IACzB,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,KAAK,CAAqC;gBAEtC,WAAW,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM;IAK1C,OAAO,CACX,KAAK,EAAE,UAAU,EACjB,MAAM,EAAE,MAAM,EAAE,EAChB,QAAQ,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,KAAK,OAAO,CAAC;QAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;QAAC,aAAa,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC,GAC/G,OAAO,CAAC,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;YA+B7B,eAAe;IAmB7B,OAAO,CAAC,cAAc;IAItB,OAAO,CAAC,YAAY;IAqBpB,QAAQ,IAAI;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,cAAc,EAAE,MAAM,CAAA;KAAE;IAOtD,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAGhC;AAED;;GAEG;AACH,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,OAAO,CAAe;IAC9B,OAAO,CAAC,QAAQ,CAAsB;IACtC,OAAO,CAAC,KAAK,CAAmE;IAChF,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,KAAK,CAAoC;gBAErC,WAAW,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM;IAM5D,KAAK,CACT,KAAK,EAAE,UAAU,EACjB,MAAM,EAAE,MAAM,EAAE,EAChB,QAAQ,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,KAAK,OAAO,CAAC;QAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;QAAC,aAAa,EAAE,YAAY,EAAE,CAAA;KAAE,CAAC,GAC/G,OAAO,CAAC,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;IA2B3C,QAAQ,CAAC,KAAK,EAAE,UAAU,GAAG,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC;IAgBrD,OAAO,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM;IAkBpD,QAAQ,IAAI;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE;IAItD,UAAU,IAAI,IAAI;IAIZ,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;CAGhC"}
|
|
@@ -0,0 +1,298 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @nahisaho/yata-scale - Query Coordinator
|
|
3
|
+
*
|
|
4
|
+
* Coordinates query execution across shards
|
|
5
|
+
*/
|
|
6
|
+
import { ok, err } from 'neverthrow';
|
|
7
|
+
import { QueryError, QueryTimeoutError } from './errors.js';
|
|
8
|
+
/**
|
|
9
|
+
* Query planner
|
|
10
|
+
*/
|
|
11
|
+
export class QueryPlanner {
|
|
12
|
+
plan(query, shards) {
|
|
13
|
+
const steps = [];
|
|
14
|
+
let estimatedCost = 0;
|
|
15
|
+
if (query.type === 'lookup' && query.entityIds) {
|
|
16
|
+
steps.push({
|
|
17
|
+
type: 'index',
|
|
18
|
+
target: 'entity',
|
|
19
|
+
cost: query.entityIds.length * 10,
|
|
20
|
+
details: { ids: query.entityIds },
|
|
21
|
+
});
|
|
22
|
+
estimatedCost += query.entityIds.length * 10;
|
|
23
|
+
}
|
|
24
|
+
if (query.type === 'traverse' && query.startEntityIds) {
|
|
25
|
+
steps.push({
|
|
26
|
+
type: 'scan',
|
|
27
|
+
target: 'graph',
|
|
28
|
+
cost: query.startEntityIds.length * (query.maxDepth ?? 1) * 50,
|
|
29
|
+
details: { startIds: query.startEntityIds, depth: query.maxDepth },
|
|
30
|
+
});
|
|
31
|
+
estimatedCost += query.startEntityIds.length * (query.maxDepth ?? 1) * 50;
|
|
32
|
+
}
|
|
33
|
+
if (query.type === 'pattern' && query.pattern) {
|
|
34
|
+
steps.push({
|
|
35
|
+
type: 'scan',
|
|
36
|
+
target: 'pattern',
|
|
37
|
+
cost: 500,
|
|
38
|
+
details: { pattern: query.pattern },
|
|
39
|
+
});
|
|
40
|
+
estimatedCost += 500;
|
|
41
|
+
}
|
|
42
|
+
if (query.filters && query.filters.length > 0) {
|
|
43
|
+
steps.push({
|
|
44
|
+
type: 'filter',
|
|
45
|
+
target: 'result',
|
|
46
|
+
cost: query.filters.length * 20,
|
|
47
|
+
details: { filters: query.filters },
|
|
48
|
+
});
|
|
49
|
+
estimatedCost += query.filters.length * 20;
|
|
50
|
+
}
|
|
51
|
+
if (query.orderBy && query.orderBy.length > 0) {
|
|
52
|
+
steps.push({
|
|
53
|
+
type: 'sort',
|
|
54
|
+
target: 'result',
|
|
55
|
+
cost: 100,
|
|
56
|
+
details: { orderBy: query.orderBy },
|
|
57
|
+
});
|
|
58
|
+
estimatedCost += 100;
|
|
59
|
+
}
|
|
60
|
+
if (query.limit !== undefined) {
|
|
61
|
+
steps.push({
|
|
62
|
+
type: 'limit',
|
|
63
|
+
target: 'result',
|
|
64
|
+
cost: 10,
|
|
65
|
+
details: { limit: query.limit, offset: query.offset },
|
|
66
|
+
});
|
|
67
|
+
estimatedCost += 10;
|
|
68
|
+
}
|
|
69
|
+
return {
|
|
70
|
+
query,
|
|
71
|
+
steps,
|
|
72
|
+
estimatedCost,
|
|
73
|
+
targetShards: shards,
|
|
74
|
+
parallelizable: shards.length > 1,
|
|
75
|
+
cacheKey: this.generateCacheKey(query),
|
|
76
|
+
timeout: query.timeout,
|
|
77
|
+
};
|
|
78
|
+
}
|
|
79
|
+
optimize(plan) {
|
|
80
|
+
const optimizations = [];
|
|
81
|
+
// Push filters before sorts
|
|
82
|
+
const filterIdx = plan.steps.findIndex((s) => s.type === 'filter');
|
|
83
|
+
const sortIdx = plan.steps.findIndex((s) => s.type === 'sort');
|
|
84
|
+
if (sortIdx !== -1 && filterIdx > sortIdx) {
|
|
85
|
+
optimizations.push('filter-pushdown');
|
|
86
|
+
}
|
|
87
|
+
return { ...plan, optimizations };
|
|
88
|
+
}
|
|
89
|
+
generateCacheKey(query) {
|
|
90
|
+
return `query:${JSON.stringify(query)}`.substring(0, 100);
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
/**
|
|
94
|
+
* Worker pool for parallel execution
|
|
95
|
+
*/
|
|
96
|
+
export class WorkerPool {
|
|
97
|
+
concurrency;
|
|
98
|
+
active = 0;
|
|
99
|
+
pending = [];
|
|
100
|
+
constructor(concurrency) {
|
|
101
|
+
this.concurrency = concurrency;
|
|
102
|
+
}
|
|
103
|
+
async map(items, fn) {
|
|
104
|
+
const results = [];
|
|
105
|
+
for (const item of items) {
|
|
106
|
+
while (this.active >= this.concurrency) {
|
|
107
|
+
await new Promise((r) => setTimeout(r, 1));
|
|
108
|
+
}
|
|
109
|
+
this.active++;
|
|
110
|
+
try {
|
|
111
|
+
results.push(await fn(item));
|
|
112
|
+
}
|
|
113
|
+
finally {
|
|
114
|
+
this.active--;
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
return results;
|
|
118
|
+
}
|
|
119
|
+
async filter(items, predicate) {
|
|
120
|
+
const results = [];
|
|
121
|
+
for (const item of items) {
|
|
122
|
+
if (await predicate(item)) {
|
|
123
|
+
results.push(item);
|
|
124
|
+
}
|
|
125
|
+
}
|
|
126
|
+
return results;
|
|
127
|
+
}
|
|
128
|
+
async reduce(items, fn, initial) {
|
|
129
|
+
let result = initial;
|
|
130
|
+
for (const item of items) {
|
|
131
|
+
result = await fn(result, item);
|
|
132
|
+
}
|
|
133
|
+
return result;
|
|
134
|
+
}
|
|
135
|
+
get activeCount() {
|
|
136
|
+
return this.active;
|
|
137
|
+
}
|
|
138
|
+
get pendingCount() {
|
|
139
|
+
return this.pending.length;
|
|
140
|
+
}
|
|
141
|
+
async shutdown() {
|
|
142
|
+
while (this.active > 0) {
|
|
143
|
+
await new Promise((r) => setTimeout(r, 10));
|
|
144
|
+
}
|
|
145
|
+
}
|
|
146
|
+
}
|
|
147
|
+
/**
|
|
148
|
+
* Distributed query executor
|
|
149
|
+
*/
|
|
150
|
+
export class DistributedExecutor {
|
|
151
|
+
pool;
|
|
152
|
+
timeout;
|
|
153
|
+
stats = { totalQueries: 0, totalTime: 0 };
|
|
154
|
+
constructor(concurrency, timeout) {
|
|
155
|
+
this.pool = new WorkerPool(concurrency);
|
|
156
|
+
this.timeout = timeout;
|
|
157
|
+
}
|
|
158
|
+
async execute(query, shards, executor) {
|
|
159
|
+
const startTime = Date.now();
|
|
160
|
+
try {
|
|
161
|
+
const results = await Promise.race([
|
|
162
|
+
this.executeOnShards(query, shards, executor),
|
|
163
|
+
this.timeoutPromise(query.timeout ?? this.timeout),
|
|
164
|
+
]);
|
|
165
|
+
if (results === 'timeout') {
|
|
166
|
+
return err(new QueryTimeoutError(query.timeout ?? this.timeout));
|
|
167
|
+
}
|
|
168
|
+
const merged = this.mergeResults(results);
|
|
169
|
+
this.stats.totalQueries++;
|
|
170
|
+
this.stats.totalTime += Date.now() - startTime;
|
|
171
|
+
return ok({
|
|
172
|
+
entities: merged.entities,
|
|
173
|
+
relationships: merged.relationships,
|
|
174
|
+
totalCount: merged.entities.length + merged.relationships.length,
|
|
175
|
+
hasMore: false,
|
|
176
|
+
executionTimeMs: Date.now() - startTime,
|
|
177
|
+
shardsQueried: shards,
|
|
178
|
+
});
|
|
179
|
+
}
|
|
180
|
+
catch (error) {
|
|
181
|
+
return err(new QueryError(error instanceof Error ? error.message : String(error)));
|
|
182
|
+
}
|
|
183
|
+
}
|
|
184
|
+
async executeOnShards(query, shards, executor) {
|
|
185
|
+
const results = [];
|
|
186
|
+
for (const shard of shards) {
|
|
187
|
+
try {
|
|
188
|
+
const result = await executor(shard, query);
|
|
189
|
+
results.push(result);
|
|
190
|
+
}
|
|
191
|
+
catch {
|
|
192
|
+
// Continue with other shards
|
|
193
|
+
}
|
|
194
|
+
}
|
|
195
|
+
return results;
|
|
196
|
+
}
|
|
197
|
+
timeoutPromise(ms) {
|
|
198
|
+
return new Promise((resolve) => setTimeout(() => resolve('timeout'), ms));
|
|
199
|
+
}
|
|
200
|
+
mergeResults(results) {
|
|
201
|
+
const entityMap = new Map();
|
|
202
|
+
const relMap = new Map();
|
|
203
|
+
for (const result of results) {
|
|
204
|
+
for (const entity of result.entities) {
|
|
205
|
+
entityMap.set(entity.id, entity);
|
|
206
|
+
}
|
|
207
|
+
for (const rel of result.relationships) {
|
|
208
|
+
relMap.set(rel.id, rel);
|
|
209
|
+
}
|
|
210
|
+
}
|
|
211
|
+
return {
|
|
212
|
+
entities: Array.from(entityMap.values()),
|
|
213
|
+
relationships: Array.from(relMap.values()),
|
|
214
|
+
};
|
|
215
|
+
}
|
|
216
|
+
getStats() {
|
|
217
|
+
return {
|
|
218
|
+
totalQueries: this.stats.totalQueries,
|
|
219
|
+
averageLatency: this.stats.totalQueries > 0 ? this.stats.totalTime / this.stats.totalQueries : 0,
|
|
220
|
+
};
|
|
221
|
+
}
|
|
222
|
+
async shutdown() {
|
|
223
|
+
await this.pool.shutdown();
|
|
224
|
+
}
|
|
225
|
+
}
|
|
226
|
+
/**
|
|
227
|
+
* Query coordinator
|
|
228
|
+
*/
|
|
229
|
+
export class QueryCoordinator {
|
|
230
|
+
planner;
|
|
231
|
+
executor;
|
|
232
|
+
cache = new Map();
|
|
233
|
+
cacheTtl;
|
|
234
|
+
stats = { cacheHits: 0, cacheMisses: 0 };
|
|
235
|
+
constructor(concurrency, cacheTtl, timeout) {
|
|
236
|
+
this.planner = new QueryPlanner();
|
|
237
|
+
this.executor = new DistributedExecutor(concurrency, timeout);
|
|
238
|
+
this.cacheTtl = cacheTtl;
|
|
239
|
+
}
|
|
240
|
+
async query(query, shards, executor) {
|
|
241
|
+
const plan = this.planner.plan(query, shards);
|
|
242
|
+
// Check cache
|
|
243
|
+
if (plan.cacheKey && query.options?.useCache !== false) {
|
|
244
|
+
const cached = this.cache.get(plan.cacheKey);
|
|
245
|
+
if (cached && cached.expiry > Date.now()) {
|
|
246
|
+
this.stats.cacheHits++;
|
|
247
|
+
return ok(cached.result);
|
|
248
|
+
}
|
|
249
|
+
}
|
|
250
|
+
this.stats.cacheMisses++;
|
|
251
|
+
const result = await this.executor.execute(query, shards, executor);
|
|
252
|
+
// Cache result
|
|
253
|
+
if (result.isOk() && plan.cacheKey) {
|
|
254
|
+
this.cache.set(plan.cacheKey, {
|
|
255
|
+
result: result.value,
|
|
256
|
+
expiry: Date.now() + (query.options?.cacheTtl ?? this.cacheTtl),
|
|
257
|
+
});
|
|
258
|
+
}
|
|
259
|
+
return result;
|
|
260
|
+
}
|
|
261
|
+
validate(query) {
|
|
262
|
+
if (!['lookup', 'traverse', 'pattern', 'aggregate'].includes(query.type)) {
|
|
263
|
+
return err(new QueryError(`Invalid query type: ${query.type}`));
|
|
264
|
+
}
|
|
265
|
+
if (query.type === 'lookup' && (!query.entityIds || query.entityIds.length === 0)) {
|
|
266
|
+
return err(new QueryError('lookup query requires entityIds'));
|
|
267
|
+
}
|
|
268
|
+
if (query.type === 'traverse' && (!query.startEntityIds || query.startEntityIds.length === 0)) {
|
|
269
|
+
return err(new QueryError('traverse query requires startEntityIds'));
|
|
270
|
+
}
|
|
271
|
+
return ok(undefined);
|
|
272
|
+
}
|
|
273
|
+
explain(query, shards) {
|
|
274
|
+
const plan = this.planner.plan(query, shards);
|
|
275
|
+
const lines = [
|
|
276
|
+
`Query Type: ${query.type}`,
|
|
277
|
+
`Estimated Cost: ${plan.estimatedCost}`,
|
|
278
|
+
`Target Shards: ${plan.targetShards.join(', ')}`,
|
|
279
|
+
`Parallelizable: ${plan.parallelizable}`,
|
|
280
|
+
'',
|
|
281
|
+
'Execution Steps:',
|
|
282
|
+
];
|
|
283
|
+
for (const step of plan.steps) {
|
|
284
|
+
lines.push(` - ${step.type} on ${step.target} (cost: ${step.cost})`);
|
|
285
|
+
}
|
|
286
|
+
return lines.join('\n');
|
|
287
|
+
}
|
|
288
|
+
getStats() {
|
|
289
|
+
return { ...this.stats };
|
|
290
|
+
}
|
|
291
|
+
clearCache() {
|
|
292
|
+
this.cache.clear();
|
|
293
|
+
}
|
|
294
|
+
async shutdown() {
|
|
295
|
+
await this.executor.shutdown();
|
|
296
|
+
}
|
|
297
|
+
}
|
|
298
|
+
//# sourceMappingURL=QueryCoordinator.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"QueryCoordinator.js","sourceRoot":"","sources":["../src/QueryCoordinator.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,EAAE,EAAE,GAAG,EAAe,MAAM,YAAY,CAAC;AAElD,OAAO,EAAE,UAAU,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAE5D;;GAEG;AACH,MAAM,OAAO,YAAY;IACvB,IAAI,CAAC,KAAiB,EAAE,MAAgB;QACtC,MAAM,KAAK,GAAgB,EAAE,CAAC;QAC9B,IAAI,aAAa,GAAG,CAAC,CAAC;QAEtB,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YAC/C,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,OAAO;gBACb,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,EAAE;gBACjC,OAAO,EAAE,EAAE,GAAG,EAAE,KAAK,CAAC,SAAS,EAAE;aAClC,CAAC,CAAC;YACH,aAAa,IAAI,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,EAAE,CAAC;QAC/C,CAAC;QAED,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,KAAK,CAAC,cAAc,EAAE,CAAC;YACtD,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,MAAM;gBACZ,MAAM,EAAE,OAAO;gBACf,IAAI,EAAE,KAAK,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,CAAC,CAAC,GAAG,EAAE;gBAC9D,OAAO,EAAE,EAAE,QAAQ,EAAE,KAAK,CAAC,cAAc,EAAE,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE;aACnE,CAAC,CAAC;YACH,aAAa,IAAI,KAAK,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,CAAC,CAAC,GAAG,EAAE,CAAC;QAC5E,CAAC;QAED,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAC9C,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,MAAM;gBACZ,MAAM,EAAE,SAAS;gBACjB,IAAI,EAAE,GAAG;gBACT,OAAO,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE;aACpC,CAAC,CAAC;YACH,aAAa,IAAI,GAAG,CAAC;QACvB,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9C,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,QAAQ;gBACd,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,EAAE;gBAC/B,OAAO,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE;aACpC,CAAC,CAAC;YACH,aAAa,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,EAAE,CAAC;QAC7C,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC9C,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,MAAM;gBACZ,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,GAAG;gBACT,OAAO,EAAE,EAAE,OAAO,EAAE,KAAK,CAAC,OAAO,EAAE;aACpC,CAAC,CAAC;YACH,aAAa,IAAI,GAAG,CAAC;QACvB,CAAC;QAED,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAC9B,KAAK,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,OAAO;gBACb,MAAM,EAAE,QAAQ;gBAChB,IAAI,EAAE,EAAE;gBACR,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE;aACtD,CAAC,CAAC;YACH,aAAa,IAAI,EAAE,CAAC;QACtB,CAAC;QAED,OAAO;YACL,KAAK;YACL,KAAK;YACL,aAAa;YACb,YAAY,EAAE,MAAM;YACpB,cAAc,EAAE,MAAM,CAAC,MAAM,GAAG,CAAC;YACjC,QAAQ,EAAE,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC;YACtC,OAAO,EAAE,KAAK,CAAC,OAAO;SACvB,CAAC;IACJ,CAAC;IAED,QAAQ,CAAC,IAAe;QACtB,MAAM,aAAa,GAAa,EAAE,CAAC;QAEnC,4BAA4B;QAC5B,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QACnE,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QAC/D,IAAI,OAAO,KAAK,CAAC,CAAC,IAAI,SAAS,GAAG,OAAO,EAAE,CAAC;YAC1C,aAAa,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACxC,CAAC;QAED,OAAO,EAAE,GAAG,IAAI,EAAE,aAAa,EAAE,CAAC;IACpC,CAAC;IAEO,gBAAgB,CAAC,KAAiB;QACxC,OAAO,SAAS,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IAC5D,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,UAAU;IACb,WAAW,CAAS;IACpB,MAAM,GAAG,CAAC,CAAC;IACX,OAAO,GAA+B,EAAE,CAAC;IAEjD,YAAY,WAAmB;QAC7B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,GAAG,CAAO,KAAU,EAAE,EAA+B;QACzD,MAAM,OAAO,GAAQ,EAAE,CAAC;QAExB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,OAAO,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;gBACvC,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAC7C,CAAC;YAED,IAAI,CAAC,MAAM,EAAE,CAAC;YACd,IAAI,CAAC;gBACH,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC;YAC/B,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,MAAM,CAAI,KAAU,EAAE,SAAkD;QAC5E,MAAM,OAAO,GAAQ,EAAE,CAAC;QAExB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,MAAM,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC1B,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,KAAK,CAAC,MAAM,CAAO,KAAU,EAAE,EAAuC,EAAE,OAAU;QAChF,IAAI,MAAM,GAAG,OAAO,CAAC;QACrB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,GAAG,MAAM,EAAE,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAClC,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,OAAO,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,mBAAmB;IACtB,IAAI,CAAa;IACjB,OAAO,CAAS;IAChB,KAAK,GAAG,EAAE,YAAY,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;IAElD,YAAY,WAAmB,EAAE,OAAe;QAC9C,IAAI,CAAC,IAAI,GAAG,IAAI,UAAU,CAAC,WAAW,CAAC,CAAC;QACxC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,KAAK,CAAC,OAAO,CACX,KAAiB,EACjB,MAAgB,EAChB,QAAgH;QAEhH,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;gBACjC,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC;gBAC7C,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC;aACnD,CAAC,CAAC;YAEH,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;gBAC1B,OAAO,GAAG,CAAC,IAAI,iBAAiB,CAAC,KAAK,CAAC,OAAO,IAAI,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;YACnE,CAAC;YAED,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,OAAuE,CAAC,CAAC;YAE1G,IAAI,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC;YAC1B,IAAI,CAAC,KAAK,CAAC,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAE/C,OAAO,EAAE,CAAC;gBACR,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,aAAa,EAAE,MAAM,CAAC,aAAa;gBACnC,UAAU,EAAE,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,MAAM,CAAC,aAAa,CAAC,MAAM;gBAChE,OAAO,EAAE,KAAK;gBACd,eAAe,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBACvC,aAAa,EAAE,MAAM;aACtB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,GAAG,CAAC,IAAI,UAAU,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACrF,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,eAAe,CAC3B,KAAiB,EACjB,MAAgB,EAChB,QAAgH;QAEhH,MAAM,OAAO,GAAiE,EAAE,CAAC;QAEjF,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;gBAC5C,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACvB,CAAC;YAAC,MAAM,CAAC;gBACP,6BAA6B;YAC/B,CAAC;QACH,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,cAAc,CAAC,EAAU;QAC/B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAC5E,CAAC;IAEO,YAAY,CAClB,OAAqE;QAErE,MAAM,SAAS,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC5C,MAAM,MAAM,GAAG,IAAI,GAAG,EAAwB,CAAC;QAE/C,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,KAAK,MAAM,MAAM,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACrC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;YACnC,CAAC;YACD,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;gBACvC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QAED,OAAO;YACL,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC;YACxC,aAAa,EAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;SAC3C,CAAC;IACJ,CAAC;IAED,QAAQ;QACN,OAAO;YACL,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY;YACrC,cAAc,EAAE,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;SACjG,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;IAC7B,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,gBAAgB;IACnB,OAAO,CAAe;IACtB,QAAQ,CAAsB;IAC9B,KAAK,GAAyD,IAAI,GAAG,EAAE,CAAC;IACxE,QAAQ,CAAS;IACjB,KAAK,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;IAEjD,YAAY,WAAmB,EAAE,QAAgB,EAAE,OAAe;QAChE,IAAI,CAAC,OAAO,GAAG,IAAI,YAAY,EAAE,CAAC;QAClC,IAAI,CAAC,QAAQ,GAAG,IAAI,mBAAmB,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QAC9D,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,KAAK,CAAC,KAAK,CACT,KAAiB,EACjB,MAAgB,EAChB,QAAgH;QAEhH,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAE9C,cAAc;QACd,IAAI,IAAI,CAAC,QAAQ,IAAI,KAAK,CAAC,OAAO,EAAE,QAAQ,KAAK,KAAK,EAAE,CAAC;YACvD,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC7C,IAAI,MAAM,IAAI,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC;gBACzC,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC;gBACvB,OAAO,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;QAEzB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QAEpE,eAAe;QACf,IAAI,MAAM,CAAC,IAAI,EAAE,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE;gBAC5B,MAAM,EAAE,MAAM,CAAC,KAAK;gBACpB,MAAM,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC;aAChE,CAAC,CAAC;QACL,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,QAAQ,CAAC,KAAiB;QACxB,IAAI,CAAC,CAAC,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,WAAW,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;YACzE,OAAO,GAAG,CAAC,IAAI,UAAU,CAAC,uBAAuB,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAClE,CAAC;QAED,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC,EAAE,CAAC;YAClF,OAAO,GAAG,CAAC,IAAI,UAAU,CAAC,iCAAiC,CAAC,CAAC,CAAC;QAChE,CAAC;QAED,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,IAAI,CAAC,CAAC,KAAK,CAAC,cAAc,IAAI,KAAK,CAAC,cAAc,CAAC,MAAM,KAAK,CAAC,CAAC,EAAE,CAAC;YAC9F,OAAO,GAAG,CAAC,IAAI,UAAU,CAAC,wCAAwC,CAAC,CAAC,CAAC;QACvE,CAAC;QAED,OAAO,EAAE,CAAC,SAAS,CAAC,CAAC;IACvB,CAAC;IAED,OAAO,CAAC,KAAiB,EAAE,MAAgB;QACzC,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC9C,MAAM,KAAK,GAAG;YACZ,eAAe,KAAK,CAAC,IAAI,EAAE;YAC3B,mBAAmB,IAAI,CAAC,aAAa,EAAE;YACvC,kBAAkB,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;YAChD,mBAAmB,IAAI,CAAC,cAAc,EAAE;YACxC,EAAE;YACF,kBAAkB;SACnB,CAAC;QAEF,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,KAAK,CAAC,IAAI,CAAC,OAAO,IAAI,CAAC,IAAI,OAAO,IAAI,CAAC,MAAM,WAAW,IAAI,CAAC,IAAI,GAAG,CAAC,CAAC;QACxE,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED,QAAQ;QACN,OAAO,EAAE,GAAG,IAAI,CAAC,KAAK,EAAE,CAAC;IAC3B,CAAC;IAED,UAAU;QACR,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;IACjC,CAAC;CACF"}
|
|
@@ -0,0 +1,48 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* @nahisaho/yata-scale - Shard Manager
|
|
3
|
+
*
|
|
4
|
+
* Manages sharding for distributed knowledge graph
|
|
5
|
+
*/
|
|
6
|
+
import { type Result } from 'neverthrow';
|
|
7
|
+
import type { ShardConfig, ShardInfo, ShardStatus, RebalancePlan } from './types.js';
|
|
8
|
+
import { ShardNotFoundError, ShardError } from './errors.js';
|
|
9
|
+
/**
|
|
10
|
+
* Hash-based partition strategy using consistent hashing
|
|
11
|
+
*/
|
|
12
|
+
export declare class HashPartitionStrategy {
|
|
13
|
+
private readonly virtualNodes;
|
|
14
|
+
private ring;
|
|
15
|
+
private sortedKeys;
|
|
16
|
+
private shards;
|
|
17
|
+
constructor(virtualNodes?: number);
|
|
18
|
+
private hash;
|
|
19
|
+
addShard(shardId: string): void;
|
|
20
|
+
removeShard(shardId: string): void;
|
|
21
|
+
getShardForEntity(entityId: string): string;
|
|
22
|
+
getAllShards(): string[];
|
|
23
|
+
}
|
|
24
|
+
/**
|
|
25
|
+
* Shard manager for distributed knowledge graph
|
|
26
|
+
*/
|
|
27
|
+
export declare class ShardManager {
|
|
28
|
+
private shards;
|
|
29
|
+
private strategy;
|
|
30
|
+
constructor();
|
|
31
|
+
createShard(shardId: string, _config: ShardConfig): Result<ShardInfo, ShardError>;
|
|
32
|
+
getShard(shardId: string): Result<ShardInfo, ShardNotFoundError>;
|
|
33
|
+
deleteShard(shardId: string): Result<void, ShardNotFoundError>;
|
|
34
|
+
updateShardStatus(shardId: string, status: ShardStatus): Result<void, ShardNotFoundError>;
|
|
35
|
+
getShardForEntity(entityId: string): string;
|
|
36
|
+
getAllShards(): string[];
|
|
37
|
+
getStats(): {
|
|
38
|
+
totalShards: number;
|
|
39
|
+
totalEntities: number;
|
|
40
|
+
healthyShards: number;
|
|
41
|
+
};
|
|
42
|
+
checkHealth(): {
|
|
43
|
+
healthy: boolean;
|
|
44
|
+
unhealthyShards: string[];
|
|
45
|
+
};
|
|
46
|
+
planRebalance(loads: Map<string, number>): RebalancePlan;
|
|
47
|
+
}
|
|
48
|
+
//# sourceMappingURL=ShardManager.d.ts.map
|