perf: group catalog source reads by identity

2026-09-28 16:23:58 +03:00
parent 7d2262fb43
commit e76eb545d7
2 changed files with 58 additions and 2 deletions
@@ -178,6 +178,50 @@ describe('local stock/bond search (SQLite HTTP)', () => {
  });
  afterAll(async () => app?.close());

+  it('reads only requested source identities when batching UIDs, types and periods', async () => {
+    const repository = app.get(CatalogDiscoveryRepository);
+    const fetchedAt = new Date();
+    for (const uid of ['a', 'b']) {
+      await prisma.tbankInstrument.create({
+        data: { provider: 'tbank', uid, instrumentType: 'share', metadata: {}, fetchedAt },
+      });
+    }
+    const expected = [
+      { instrumentUid: 'a', source: 'tbank', kind: 'detail', rangeKey: 'share' },
+      { instrumentUid: 'b', source: 'tbank', kind: 'detail', rangeKey: 'share' },
+      { instrumentUid: 'a', source: 'tbank', kind: 'dividends', rangeKey: '2026' },
+      { instrumentUid: 'b', source: 'tbank', kind: 'dividends', rangeKey: '2027' },
+      { instrumentUid: 'b', source: 'moex', kind: 'bond-events', rangeKey: '' },
+      { instrumentUid: 'a', source: 'ncr', kind: 'rating', rangeKey: '' },
+    ];
+    const distractors = [
+      { instrumentUid: 'a', source: 'tbank', kind: 'detail', rangeKey: 'bond' },
+      { instrumentUid: 'b', source: 'tbank', kind: 'dividends', rangeKey: '2026' },
+      { instrumentUid: 'a', source: 'tbank', kind: 'dividends', rangeKey: '2027' },
+      { instrumentUid: 'a', source: 'moex', kind: 'bond-events', rangeKey: '' },
+      { instrumentUid: 'b', source: 'tbank', kind: 'bond-events', rangeKey: '' },
+      { instrumentUid: 'a', source: 'tbank', kind: 'rating', rangeKey: '' },
+    ];
+    await prisma.tbankInstrumentSource.createMany({
+      data: [...expected, ...distractors].map((identity) => ({
+        ...identity, provider: 'tbank', fetchedAt, payload: identity,
+      })),
+    });
+    const rows = await repository.readSources([
+      { uid: 'a', instrumentType: 'share', kind: 'detail' },
+      { uid: 'b', instrumentType: 'share', kind: 'detail' },
+      { uid: 'a', instrumentType: 'share', kind: 'detail' },
+      { uid: 'a', instrumentType: 'share', kind: 'dividends', rangeKey: '2026' },
+      { uid: 'b', instrumentType: 'share', kind: 'dividends', rangeKey: '2027' },
+      { uid: 'b', instrumentType: 'bond', kind: 'bond-events' },
+      { uid: 'a', instrumentType: 'bond', kind: 'rating' },
+      { uid: 'missing', instrumentType: 'share', kind: 'detail' },
+    ]);
+    expect(rows.map((row) => row.payload)).toEqual(expect.arrayContaining(expected));
+    expect(rows).toHaveLength(expected.length);
+    expect(await repository.readSources([])).toEqual([]);
+  });
+
  it('returns missing without waiting for discovery, then partial while enrichment is blocked; observe never launches work', async () => {
    const found = deferred<(typeof share)[]>();
    const detail = deferred<typeof share>();