@yuants/app-virtual-exchange 0.16.0 → 0.18.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.
Files changed (67) hide show
  1. package/dist/series-collector/backwards-interest-rate.js +29 -63
  2. package/dist/series-collector/backwards-interest-rate.js.map +1 -1
  3. package/dist/series-collector/backwards-ohlc.js +33 -68
  4. package/dist/series-collector/backwards-ohlc.js.map +1 -1
  5. package/dist/series-collector/discovery.js +62 -0
  6. package/dist/series-collector/discovery.js.map +1 -0
  7. package/dist/series-collector/forwards-interest-rate.js +31 -70
  8. package/dist/series-collector/forwards-interest-rate.js.map +1 -1
  9. package/dist/series-collector/forwards-ohlc.js +33 -68
  10. package/dist/series-collector/forwards-ohlc.js.map +1 -1
  11. package/dist/series-collector/index.js +2 -4
  12. package/dist/series-collector/index.js.map +1 -1
  13. package/dist/series-collector/interest-rate.js +49 -0
  14. package/dist/series-collector/interest-rate.js.map +1 -0
  15. package/dist/series-collector/ohlc.js +49 -0
  16. package/dist/series-collector/ohlc.js.map +1 -0
  17. package/dist/series-collector/patch-interest-rate.js +63 -0
  18. package/dist/series-collector/patch-interest-rate.js.map +1 -0
  19. package/dist/series-collector/patch-ohlc.js +67 -0
  20. package/dist/series-collector/patch-ohlc.js.map +1 -0
  21. package/dist/series-collector/sql-helpers.js +11 -0
  22. package/dist/series-collector/sql-helpers.js.map +1 -0
  23. package/lib/series-collector/backwards-interest-rate.d.ts +3 -1
  24. package/lib/series-collector/backwards-interest-rate.d.ts.map +1 -1
  25. package/lib/series-collector/backwards-interest-rate.js +31 -63
  26. package/lib/series-collector/backwards-interest-rate.js.map +1 -1
  27. package/lib/series-collector/backwards-ohlc.d.ts +2 -1
  28. package/lib/series-collector/backwards-ohlc.d.ts.map +1 -1
  29. package/lib/series-collector/backwards-ohlc.js +34 -67
  30. package/lib/series-collector/backwards-ohlc.js.map +1 -1
  31. package/lib/series-collector/discovery.d.ts +12 -0
  32. package/lib/series-collector/discovery.d.ts.map +1 -0
  33. package/lib/series-collector/discovery.js +67 -0
  34. package/lib/series-collector/discovery.js.map +1 -0
  35. package/lib/series-collector/forwards-interest-rate.d.ts +3 -1
  36. package/lib/series-collector/forwards-interest-rate.d.ts.map +1 -1
  37. package/lib/series-collector/forwards-interest-rate.js +33 -70
  38. package/lib/series-collector/forwards-interest-rate.js.map +1 -1
  39. package/lib/series-collector/forwards-ohlc.d.ts +2 -1
  40. package/lib/series-collector/forwards-ohlc.d.ts.map +1 -1
  41. package/lib/series-collector/forwards-ohlc.js +34 -67
  42. package/lib/series-collector/forwards-ohlc.js.map +1 -1
  43. package/lib/series-collector/index.d.ts +2 -4
  44. package/lib/series-collector/index.d.ts.map +1 -1
  45. package/lib/series-collector/index.js +2 -4
  46. package/lib/series-collector/index.js.map +1 -1
  47. package/lib/series-collector/interest-rate.d.ts +2 -0
  48. package/lib/series-collector/interest-rate.d.ts.map +1 -0
  49. package/lib/series-collector/interest-rate.js +51 -0
  50. package/lib/series-collector/interest-rate.js.map +1 -0
  51. package/lib/series-collector/ohlc.d.ts +2 -0
  52. package/lib/series-collector/ohlc.d.ts.map +1 -0
  53. package/lib/series-collector/ohlc.js +51 -0
  54. package/lib/series-collector/ohlc.js.map +1 -0
  55. package/lib/series-collector/patch-interest-rate.d.ts +4 -0
  56. package/lib/series-collector/patch-interest-rate.d.ts.map +1 -0
  57. package/lib/series-collector/patch-interest-rate.js +67 -0
  58. package/lib/series-collector/patch-interest-rate.js.map +1 -0
  59. package/lib/series-collector/patch-ohlc.d.ts +3 -0
  60. package/lib/series-collector/patch-ohlc.d.ts.map +1 -0
  61. package/lib/series-collector/patch-ohlc.js +71 -0
  62. package/lib/series-collector/patch-ohlc.js.map +1 -0
  63. package/lib/series-collector/sql-helpers.d.ts +11 -0
  64. package/lib/series-collector/sql-helpers.d.ts.map +1 -0
  65. package/lib/series-collector/sql-helpers.js +17 -0
  66. package/lib/series-collector/sql-helpers.js.map +1 -0
  67. package/package.json +1 -1
@@ -2,78 +2,43 @@
2
2
  // 该文件会定期扫描所有 Terminal 的 ServiceInfo,提取出所有支持 Forwards 拉取的序列。
3
3
  // 然后对每个序列,向对应的 IngestOHLC Service 发送拉取请求,补齐历史数据。
4
4
  // 使用 Token Bucket 控制每个数据源的请求速率,避免过载。
5
- import { decodeOHLCSeriesId, encodeOHLCSeriesId } from '@yuants/data-ohlc';
6
- import { parseOHLCServiceMetadataFromSchema, } from '@yuants/exchange';
5
+ import { decodeOHLCSeriesId } from '@yuants/data-ohlc';
7
6
  import { Terminal } from '@yuants/protocol';
8
7
  import { escapeSQL, requestSQL } from '@yuants/sql';
9
8
  import { decodePath, formatTime, tokenBucket } from '@yuants/utils';
10
- import { defer, repeat, retry } from 'rxjs';
11
9
  const terminal = Terminal.fromNodeEnv();
12
- const listForwardSeriesIds = async () => {
13
- const product_ids = await requestSQL(terminal, `select product_id from product`);
14
- console.time('[SeriesCollector][OHLC][Forwards] calc');
15
- const series_ids = new Set();
16
- for (const terminalInfo of terminal.terminalInfos) {
17
- for (const serviceInfo of Object.values(terminalInfo.serviceInfo || {})) {
18
- if (serviceInfo.method !== 'IngestOHLC')
19
- continue;
20
- try {
21
- const meta = parseOHLCServiceMetadataFromSchema(serviceInfo.schema);
22
- if (meta.direction !== 'forward')
23
- continue;
24
- for (const { product_id } of product_ids) {
25
- if (!product_id.startsWith(meta.product_id_prefix))
26
- continue;
27
- for (const duration of meta.duration_list) {
28
- const series_id = encodeOHLCSeriesId(product_id, duration);
29
- series_ids.add(series_id);
30
- }
31
- }
32
- }
33
- finally {
34
- }
35
- }
10
+ const ingestCounter = terminal.metrics
11
+ .counter('series_collector_ingest_count', '')
12
+ .labels({ terminal_id: terminal.terminal_id, type: 'ohlc', task: 'forward' });
13
+ export const handleIngestOHLCForward = async (series_id, direction, signal) => {
14
+ var _a, _b, _c, _d;
15
+ const { product_id, duration } = decodeOHLCSeriesId(series_id);
16
+ const [datasource_id] = decodePath(product_id);
17
+ // 控制速率:每个数据源每秒钟只能请求一次
18
+ await tokenBucket(`ohlc:forward:${datasource_id}`).acquire();
19
+ let req;
20
+ if (direction === 'forward') {
21
+ const [record] = await requestSQL(terminal, `select end_time from series_data_range where series_id = ${escapeSQL(series_id)} and table_name = 'ohlc_v2' order by end_time desc limit 1`);
22
+ const time = record ? new Date(record.end_time).getTime() : 0;
23
+ req = {
24
+ product_id,
25
+ duration,
26
+ direction: 'forward',
27
+ time,
28
+ };
36
29
  }
37
- console.timeEnd('[SeriesCollector][OHLC][Forwards] calc');
38
- return series_ids;
30
+ else {
31
+ // backward
32
+ req = {
33
+ product_id,
34
+ duration,
35
+ direction,
36
+ time: Date.now(),
37
+ };
38
+ }
39
+ console.info(formatTime(Date.now()), '[SeriesCollector][OHLC][Forward]', 'Request', `product_id=${req.product_id}, duration=${req.duration}, direction=${req.direction}, time=${formatTime(req.time)}`);
40
+ const res = await terminal.client.requestForResponseData('IngestOHLC', req);
41
+ ingestCounter.inc(res.wrote_count || 0);
42
+ console.info(formatTime(Date.now()), '[SeriesCollector][OHLC][Forward]', 'Response', `series_id=${series_id}, ingested_count=${res.wrote_count}, start_time=${formatTime((_b = (_a = res.range) === null || _a === void 0 ? void 0 : _a.start_time) !== null && _b !== void 0 ? _b : NaN)}, end_time=${formatTime((_d = (_c = res.range) === null || _c === void 0 ? void 0 : _c.end_time) !== null && _d !== void 0 ? _d : NaN)}`);
39
43
  };
40
- defer(async () => {
41
- const time = Date.now();
42
- const series_ids = await listForwardSeriesIds();
43
- console.log(`[SeriesCollector][OHLC][Forwards] Found ${series_ids.size} series to collect forwards data for. (${formatTime(Date.now() - time)})`);
44
- await Promise.all([...series_ids].map(async (series_id) => {
45
- var _a, _b, _c, _d;
46
- try {
47
- const { product_id, duration } = decodeOHLCSeriesId(series_id);
48
- const [datasource_id] = decodePath(product_id);
49
- // 控制速率:每个数据源每秒钟只能请求一次
50
- await tokenBucket(`ohlc_forwards_target_${datasource_id}`, {
51
- refillInterval: 1000,
52
- capacity: 1,
53
- }).acquire();
54
- {
55
- const [record] = await requestSQL(terminal, `select end_time from series_data_range where series_id = ${escapeSQL(series_id)} and table_name = 'ohlc_v2' order by end_time desc limit 1`);
56
- const time = record ? new Date(record.end_time).getTime() : 0;
57
- const req = {
58
- product_id,
59
- duration,
60
- direction: 'forward',
61
- time,
62
- };
63
- console.info(formatTime(Date.now()), 'DispatchIngestOHLC_Forwards', `product_id=${product_id}, duration=${duration}, time=${formatTime(time)}`);
64
- const res = await terminal.client.requestForResponseData('IngestOHLC', req);
65
- terminal.metrics
66
- .counter('series_collector_forwards_ingest_count', '')
67
- .labels({ terminal_id: terminal.terminal_id, type: 'ohlc' })
68
- .inc(res.wrote_count || 0);
69
- console.info(formatTime(Date.now()), 'DispatchIngestOHLCResult_Forwards', `series_id=${series_id}, ingested_count=${res.wrote_count}, start_time=${formatTime((_b = (_a = res.range) === null || _a === void 0 ? void 0 : _a.start_time) !== null && _b !== void 0 ? _b : NaN)}, end_time=${formatTime((_d = (_c = res.range) === null || _c === void 0 ? void 0 : _c.end_time) !== null && _d !== void 0 ? _d : NaN)}`);
70
- }
71
- }
72
- catch (e) {
73
- console.info(formatTime(Date.now()), 'DispatchIngestOHLCError_Forwards', `series_id=${series_id}`, e);
74
- }
75
- }));
76
- })
77
- .pipe(retry({ delay: 1000 }), repeat({ delay: 1000 }))
78
- .subscribe();
79
44
  //# sourceMappingURL=forwards-ohlc.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"forwards-ohlc.js","sourceRoot":"","sources":["../../src/series-collector/forwards-ohlc.ts"],"names":[],"mappings":"AAAA,yBAAyB;AACzB,4DAA4D;AAC5D,iDAAiD;AACjD,qCAAqC;AAErC,OAAO,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAC3E,OAAO,EAGL,kCAAkC,GACnC,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AACpE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,MAAM,CAAC;AAE5C,MAAM,QAAQ,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;AAExC,MAAM,oBAAoB,GAAG,KAAK,IAAI,EAAE;IACtC,MAAM,WAAW,GAAG,MAAM,UAAU,CAA2B,QAAQ,EAAE,gCAAgC,CAAC,CAAC;IAE3G,OAAO,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC;IAEvD,MAAM,UAAU,GAAG,IAAI,GAAG,EAAU,CAAC;IACrC,KAAK,MAAM,YAAY,IAAI,QAAQ,CAAC,aAAa,EAAE;QACjD,KAAK,MAAM,WAAW,IAAI,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,WAAW,IAAI,EAAE,CAAC,EAAE;YACvE,IAAI,WAAW,CAAC,MAAM,KAAK,YAAY;gBAAE,SAAS;YAClD,IAAI;gBACF,MAAM,IAAI,GAAG,kCAAkC,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;gBACpE,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS;oBAAE,SAAS;gBAE3C,KAAK,MAAM,EAAE,UAAU,EAAE,IAAI,WAAW,EAAE;oBACxC,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,iBAAiB,CAAC;wBAAE,SAAS;oBAC7D,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,aAAa,EAAE;wBACzC,MAAM,SAAS,GAAG,kBAAkB,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;wBAC3D,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;qBAC3B;iBACF;aACF;oBAAS;aACT;SACF;KACF;IAED,OAAO,CAAC,OAAO,CAAC,wCAAwC,CAAC,CAAC;IAE1D,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC;AAEF,KAAK,CAAC,KAAK,IAAI,EAAE;IACf,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACxB,MAAM,UAAU,GAAG,MAAM,oBAAoB,EAAE,CAAC;IAChD,OAAO,CAAC,GAAG,CACT,2CACE,UAAU,CAAC,IACb,0CAA0C,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,GAAG,CAC3E,CAAC;IAEF,MAAM,OAAO,CAAC,GAAG,CACf,CAAC,GAAG,UAAU,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE;;QACtC,IAAI;YACF,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;YAC/D,MAAM,CAAC,aAAa,CAAC,GAAG,UAAU,CAAC,UAAU,CAAC,CAAC;YAC/C,sBAAsB;YACtB,MAAM,WAAW,CAAC,wBAAwB,aAAa,EAAE,EAAE;gBACzD,cAAc,EAAE,IAAI;gBACpB,QAAQ,EAAE,CAAC;aACZ,CAAC,CAAC,OAAO,EAAE,CAAC;YACb;gBACE,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,UAAU,CAK/B,QAAQ,EACR,4DAA4D,SAAS,CACnE,SAAS,CACV,4DAA4D,CAC9D,CAAC;gBAEF,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;gBAE9D,MAAM,GAAG,GAAuB;oBAC9B,UAAU;oBACV,QAAQ;oBACR,SAAS,EAAE,SAAS;oBACpB,IAAI;iBACL,CAAC;gBAEF,OAAO,CAAC,IAAI,CACV,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,6BAA6B,EAC7B,cAAc,UAAU,cAAc,QAAQ,UAAU,UAAU,CAAC,IAAI,CAAC,EAAE,CAC3E,CAAC;gBAEF,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,sBAAsB,CACtD,YAAY,EACZ,GAAG,CACJ,CAAC;gBAEF,QAAQ,CAAC,OAAO;qBACb,OAAO,CAAC,wCAAwC,EAAE,EAAE,CAAC;qBACrD,MAAM,CAAC,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;qBAC3D,GAAG,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;gBAE7B,OAAO,CAAC,IAAI,CACV,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,mCAAmC,EACnC,aAAa,SAAS,oBAAoB,GAAG,CAAC,WAAW,gBAAgB,UAAU,CACjF,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,UAAU,mCAAI,GAAG,CAC7B,cAAc,UAAU,CAAC,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,QAAQ,mCAAI,GAAG,CAAC,EAAE,CACxD,CAAC;aACH;SACF;QAAC,OAAO,CAAC,EAAE;YACV,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,kCAAkC,EAAE,aAAa,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC;SACvG;IACH,CAAC,CAAC,CACH,CAAC;AACJ,CAAC,CAAC;KACC,IAAI,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;KACrD,SAAS,EAAE,CAAC","sourcesContent":["// 解决 Forwards 拉取历史数据的调度器\n// 该文件会定期扫描所有 Terminal 的 ServiceInfo,提取出所有支持 Forwards 拉取的序列。\n// 然后对每个序列,向对应的 IngestOHLC Service 发送拉取请求,补齐历史数据。\n// 使用 Token Bucket 控制每个数据源的请求速率,避免过载。\n\nimport { decodeOHLCSeriesId, encodeOHLCSeriesId } from '@yuants/data-ohlc';\nimport {\n IIngestOHLCRequest,\n ISeriesIngestResult,\n parseOHLCServiceMetadataFromSchema,\n} from '@yuants/exchange';\nimport { Terminal } from '@yuants/protocol';\nimport { escapeSQL, requestSQL } from '@yuants/sql';\nimport { decodePath, formatTime, tokenBucket } from '@yuants/utils';\nimport { defer, repeat, retry } from 'rxjs';\n\nconst terminal = Terminal.fromNodeEnv();\n\nconst listForwardSeriesIds = async () => {\n const product_ids = await requestSQL<{ product_id: string }[]>(terminal, `select product_id from product`);\n\n console.time('[SeriesCollector][OHLC][Forwards] calc');\n\n const series_ids = new Set<string>();\n for (const terminalInfo of terminal.terminalInfos) {\n for (const serviceInfo of Object.values(terminalInfo.serviceInfo || {})) {\n if (serviceInfo.method !== 'IngestOHLC') continue;\n try {\n const meta = parseOHLCServiceMetadataFromSchema(serviceInfo.schema);\n if (meta.direction !== 'forward') continue;\n\n for (const { product_id } of product_ids) {\n if (!product_id.startsWith(meta.product_id_prefix)) continue;\n for (const duration of meta.duration_list) {\n const series_id = encodeOHLCSeriesId(product_id, duration);\n series_ids.add(series_id);\n }\n }\n } finally {\n }\n }\n }\n\n console.timeEnd('[SeriesCollector][OHLC][Forwards] calc');\n\n return series_ids;\n};\n\ndefer(async () => {\n const time = Date.now();\n const series_ids = await listForwardSeriesIds();\n console.log(\n `[SeriesCollector][OHLC][Forwards] Found ${\n series_ids.size\n } series to collect forwards data for. (${formatTime(Date.now() - time)})`,\n );\n\n await Promise.all(\n [...series_ids].map(async (series_id) => {\n try {\n const { product_id, duration } = decodeOHLCSeriesId(series_id);\n const [datasource_id] = decodePath(product_id);\n // 控制速率:每个数据源每秒钟只能请求一次\n await tokenBucket(`ohlc_forwards_target_${datasource_id}`, {\n refillInterval: 1000,\n capacity: 1,\n }).acquire();\n {\n const [record] = await requestSQL<\n {\n end_time: string;\n }[]\n >(\n terminal,\n `select end_time from series_data_range where series_id = ${escapeSQL(\n series_id,\n )} and table_name = 'ohlc_v2' order by end_time desc limit 1`,\n );\n\n const time = record ? new Date(record.end_time).getTime() : 0;\n\n const req: IIngestOHLCRequest = {\n product_id,\n duration,\n direction: 'forward',\n time,\n };\n\n console.info(\n formatTime(Date.now()),\n 'DispatchIngestOHLC_Forwards',\n `product_id=${product_id}, duration=${duration}, time=${formatTime(time)}`,\n );\n\n const res = await terminal.client.requestForResponseData<IIngestOHLCRequest, ISeriesIngestResult>(\n 'IngestOHLC',\n req,\n );\n\n terminal.metrics\n .counter('series_collector_forwards_ingest_count', '')\n .labels({ terminal_id: terminal.terminal_id, type: 'ohlc' })\n .inc(res.wrote_count || 0);\n\n console.info(\n formatTime(Date.now()),\n 'DispatchIngestOHLCResult_Forwards',\n `series_id=${series_id}, ingested_count=${res.wrote_count}, start_time=${formatTime(\n res.range?.start_time ?? NaN,\n )}, end_time=${formatTime(res.range?.end_time ?? NaN)}`,\n );\n }\n } catch (e) {\n console.info(formatTime(Date.now()), 'DispatchIngestOHLCError_Forwards', `series_id=${series_id}`, e);\n }\n }),\n );\n})\n .pipe(retry({ delay: 1000 }), repeat({ delay: 1000 }))\n .subscribe();\n"]}
1
+ {"version":3,"file":"forwards-ohlc.js","sourceRoot":"","sources":["../../src/series-collector/forwards-ohlc.ts"],"names":[],"mappings":"AAAA,yBAAyB;AACzB,4DAA4D;AAC5D,iDAAiD;AACjD,qCAAqC;AAErC,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAEvD,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAEpE,MAAM,QAAQ,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;AAExC,MAAM,aAAa,GAAG,QAAQ,CAAC,OAAO;KACnC,OAAO,CAAC,+BAA+B,EAAE,EAAE,CAAC;KAC5C,MAAM,CAAC,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;AAEhF,MAAM,CAAC,MAAM,uBAAuB,GAAG,KAAK,EAC1C,SAAiB,EACjB,SAAiC,EACjC,MAAmB,EACnB,EAAE;;IACF,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;IAC/D,MAAM,CAAC,aAAa,CAAC,GAAG,UAAU,CAAC,UAAU,CAAC,CAAC;IAC/C,sBAAsB;IACtB,MAAM,WAAW,CAAC,gBAAgB,aAAa,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;IAC7D,IAAI,GAAuB,CAAC;IAC5B,IAAI,SAAS,KAAK,SAAS,EAAE;QAC3B,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,UAAU,CAK/B,QAAQ,EACR,4DAA4D,SAAS,CACnE,SAAS,CACV,4DAA4D,CAC9D,CAAC;QAEF,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAE9D,GAAG,GAAG;YACJ,UAAU;YACV,QAAQ;YACR,SAAS,EAAE,SAAS;YACpB,IAAI;SACL,CAAC;KACH;SAAM;QACL,WAAW;QACX,GAAG,GAAG;YACJ,UAAU;YACV,QAAQ;YACR,SAAS;YACT,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE;SACjB,CAAC;KACH;IAED,OAAO,CAAC,IAAI,CACV,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,kCAAkC,EAClC,SAAS,EACT,cAAc,GAAG,CAAC,UAAU,cAAc,GAAG,CAAC,QAAQ,eAAe,GAAG,CAAC,SAAS,UAAU,UAAU,CACpG,GAAG,CAAC,IAAI,CACT,EAAE,CACJ,CAAC;IAEF,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,sBAAsB,CACtD,YAAY,EACZ,GAAG,CACJ,CAAC;IAEF,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;IAExC,OAAO,CAAC,IAAI,CACV,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,kCAAkC,EAClC,UAAU,EACV,aAAa,SAAS,oBAAoB,GAAG,CAAC,WAAW,gBAAgB,UAAU,CACjF,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,UAAU,mCAAI,GAAG,CAC7B,cAAc,UAAU,CAAC,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,QAAQ,mCAAI,GAAG,CAAC,EAAE,CACxD,CAAC;AACJ,CAAC,CAAC","sourcesContent":["// 解决 Forwards 拉取历史数据的调度器\n// 该文件会定期扫描所有 Terminal 的 ServiceInfo,提取出所有支持 Forwards 拉取的序列。\n// 然后对每个序列,向对应的 IngestOHLC Service 发送拉取请求,补齐历史数据。\n// 使用 Token Bucket 控制每个数据源的请求速率,避免过载。\n\nimport { decodeOHLCSeriesId } from '@yuants/data-ohlc';\nimport { IIngestOHLCRequest, ISeriesIngestResult } from '@yuants/exchange';\nimport { Terminal } from '@yuants/protocol';\nimport { escapeSQL, requestSQL } from '@yuants/sql';\nimport { decodePath, formatTime, tokenBucket } from '@yuants/utils';\n\nconst terminal = Terminal.fromNodeEnv();\n\nconst ingestCounter = terminal.metrics\n .counter('series_collector_ingest_count', '')\n .labels({ terminal_id: terminal.terminal_id, type: 'ohlc', task: 'forward' });\n\nexport const handleIngestOHLCForward = async (\n series_id: string,\n direction: 'forward' | 'backward',\n signal: AbortSignal,\n) => {\n const { product_id, duration } = decodeOHLCSeriesId(series_id);\n const [datasource_id] = decodePath(product_id);\n // 控制速率:每个数据源每秒钟只能请求一次\n await tokenBucket(`ohlc:forward:${datasource_id}`).acquire();\n let req: IIngestOHLCRequest;\n if (direction === 'forward') {\n const [record] = await requestSQL<\n {\n end_time: string;\n }[]\n >(\n terminal,\n `select end_time from series_data_range where series_id = ${escapeSQL(\n series_id,\n )} and table_name = 'ohlc_v2' order by end_time desc limit 1`,\n );\n\n const time = record ? new Date(record.end_time).getTime() : 0;\n\n req = {\n product_id,\n duration,\n direction: 'forward',\n time,\n };\n } else {\n // backward\n req = {\n product_id,\n duration,\n direction,\n time: Date.now(),\n };\n }\n\n console.info(\n formatTime(Date.now()),\n '[SeriesCollector][OHLC][Forward]',\n 'Request',\n `product_id=${req.product_id}, duration=${req.duration}, direction=${req.direction}, time=${formatTime(\n req.time,\n )}`,\n );\n\n const res = await terminal.client.requestForResponseData<IIngestOHLCRequest, ISeriesIngestResult>(\n 'IngestOHLC',\n req,\n );\n\n ingestCounter.inc(res.wrote_count || 0);\n\n console.info(\n formatTime(Date.now()),\n '[SeriesCollector][OHLC][Forward]',\n 'Response',\n `series_id=${series_id}, ingested_count=${res.wrote_count}, start_time=${formatTime(\n res.range?.start_time ?? NaN,\n )}, end_time=${formatTime(res.range?.end_time ?? NaN)}`,\n );\n};\n"]}
@@ -1,5 +1,3 @@
1
- import './backwards-interest-rate';
2
- import './backwards-ohlc';
3
- import './forwards-interest-rate';
4
- import './forwards-ohlc';
1
+ import './interest-rate';
2
+ import './ohlc';
5
3
  //# sourceMappingURL=index.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/series-collector/index.ts"],"names":[],"mappings":"AAAA,OAAO,2BAA2B,CAAC;AACnC,OAAO,kBAAkB,CAAC;AAC1B,OAAO,0BAA0B,CAAC;AAClC,OAAO,iBAAiB,CAAC","sourcesContent":["import './backwards-interest-rate';\nimport './backwards-ohlc';\nimport './forwards-interest-rate';\nimport './forwards-ohlc';\n"]}
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/series-collector/index.ts"],"names":[],"mappings":"AAAA,OAAO,iBAAiB,CAAC;AACzB,OAAO,QAAQ,CAAC","sourcesContent":["import './interest-rate';\nimport './ohlc';\n"]}
@@ -0,0 +1,49 @@
1
+ import { formatTime, listWatch } from '@yuants/utils';
2
+ import { defer, map, Observable, repeat, retry, tap } from 'rxjs';
3
+ import { handleIngestInterestRateBackward } from './backwards-interest-rate';
4
+ import { listInterestRateSeriesIds } from './discovery';
5
+ import { handleIngestInterestRateForward } from './forwards-interest-rate';
6
+ import { handleInterestRatePatch } from './patch-interest-rate';
7
+ defer(() => listInterestRateSeriesIds())
8
+ .pipe(retry({ delay: 1000 }), repeat({ delay: 60000 }), map((x) => Array.from(x.entries())), listWatch((x) => x[0], ([product_id, meta]) => new Observable((sub) => {
9
+ // 处理每个利率品种任务: (forward / backward / patch),都需要独立调度
10
+ const abortController = new AbortController();
11
+ sub.add(() => {
12
+ abortController.abort();
13
+ });
14
+ // 先处理前向任务
15
+ const forwardTask = defer(async () => {
16
+ await handleIngestInterestRateForward(product_id, meta, abortController.signal);
17
+ })
18
+ .pipe(tap({
19
+ error: (err) => console.info(formatTime(Date.now()), `[SeriesCollector][InterestRate][Forward]`, 'Error', err),
20
+ }), retry(), repeat())
21
+ .subscribe();
22
+ sub.add(() => {
23
+ forwardTask.unsubscribe();
24
+ });
25
+ // 设置后向任务
26
+ const backwardTask = defer(async () => {
27
+ await handleIngestInterestRateBackward(product_id, meta, abortController.signal);
28
+ })
29
+ .pipe(tap({
30
+ error: (err) => console.info(formatTime(Date.now()), `[SeriesCollector][InterestRate][Backward]`, 'Error', err),
31
+ }), retry(), repeat())
32
+ .subscribe();
33
+ sub.add(() => {
34
+ backwardTask.unsubscribe();
35
+ });
36
+ // 设置补齐任务
37
+ const patchTask = defer(async () => {
38
+ await handleInterestRatePatch(product_id, meta, abortController.signal);
39
+ })
40
+ .pipe(tap({
41
+ error: (err) => console.info(formatTime(Date.now()), `[SeriesCollector][InterestRate][Patch]`, 'Error', err),
42
+ }), retry(), repeat())
43
+ .subscribe();
44
+ sub.add(() => {
45
+ patchTask.unsubscribe();
46
+ });
47
+ })))
48
+ .subscribe();
49
+ //# sourceMappingURL=interest-rate.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"interest-rate.js","sourceRoot":"","sources":["../../src/series-collector/interest-rate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AACtD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,MAAM,CAAC;AAClE,OAAO,EAAE,gCAAgC,EAAE,MAAM,2BAA2B,CAAC;AAC7E,OAAO,EAAE,yBAAyB,EAAE,MAAM,aAAa,CAAC;AACxD,OAAO,EAAE,+BAA+B,EAAE,MAAM,0BAA0B,CAAC;AAC3E,OAAO,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAEhE,KAAK,CAAC,GAAG,EAAE,CAAC,yBAAyB,EAAE,CAAC;KACrC,IAAI,CACH,KAAK,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EACtB,MAAM,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EACxB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,EACnC,SAAS,CACP,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,EACX,CAAC,CAAC,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE,CACrB,IAAI,UAAU,CAAC,CAAC,GAAG,EAAE,EAAE;IACrB,mDAAmD;IACnD,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;IAE9C,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE;QACX,eAAe,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,UAAU;IACV,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,IAAI,EAAE;QACnC,MAAM,+BAA+B,CAAC,UAAU,EAAE,IAAI,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;IAClF,CAAC,CAAC;SACC,IAAI,CACH,GAAG,CAAC;QACF,KAAK,EAAE,CAAC,GAAG,EAAE,EAAE,CACb,OAAO,CAAC,IAAI,CACV,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,0CAA0C,EAC1C,OAAO,EACP,GAAG,CACJ;KACJ,CAAC,EAEF,KAAK,EAAE,EACP,MAAM,EAAE,CACT;SACA,SAAS,EAAE,CAAC;IAEf,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE;QACX,WAAW,CAAC,WAAW,EAAE,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,SAAS;IACT,MAAM,YAAY,GAAG,KAAK,CAAC,KAAK,IAAI,EAAE;QACpC,MAAM,gCAAgC,CAAC,UAAU,EAAE,IAAI,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;IACnF,CAAC,CAAC;SACC,IAAI,CACH,GAAG,CAAC;QACF,KAAK,EAAE,CAAC,GAAG,EAAE,EAAE,CACb,OAAO,CAAC,IAAI,CACV,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,2CAA2C,EAC3C,OAAO,EACP,GAAG,CACJ;KACJ,CAAC,EACF,KAAK,EAAE,EACP,MAAM,EAAE,CACT;SACA,SAAS,EAAE,CAAC;IAEf,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE;QACX,YAAY,CAAC,WAAW,EAAE,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,SAAS;IACT,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,IAAI,EAAE;QACjC,MAAM,uBAAuB,CAAC,UAAU,EAAE,IAAI,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;IAC1E,CAAC,CAAC;SACC,IAAI,CACH,GAAG,CAAC;QACF,KAAK,EAAE,CAAC,GAAG,EAAE,EAAE,CACb,OAAO,CAAC,IAAI,CACV,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,wCAAwC,EACxC,OAAO,EACP,GAAG,CACJ;KACJ,CAAC,EACF,KAAK,EAAE,EACP,MAAM,EAAE,CACT;SACA,SAAS,EAAE,CAAC;IAEf,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE;QACX,SAAS,CAAC,WAAW,EAAE,CAAC;IAC1B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CACL,CACF;KACA,SAAS,EAAE,CAAC","sourcesContent":["import { formatTime, listWatch } from '@yuants/utils';\nimport { defer, map, Observable, repeat, retry, tap } from 'rxjs';\nimport { handleIngestInterestRateBackward } from './backwards-interest-rate';\nimport { listInterestRateSeriesIds } from './discovery';\nimport { handleIngestInterestRateForward } from './forwards-interest-rate';\nimport { handleInterestRatePatch } from './patch-interest-rate';\n\ndefer(() => listInterestRateSeriesIds())\n .pipe(\n retry({ delay: 1000 }),\n repeat({ delay: 60000 }),\n map((x) => Array.from(x.entries())),\n listWatch(\n (x) => x[0],\n ([product_id, meta]) =>\n new Observable((sub) => {\n // 处理每个利率品种任务: (forward / backward / patch),都需要独立调度\n const abortController = new AbortController();\n\n sub.add(() => {\n abortController.abort();\n });\n\n // 先处理前向任务\n const forwardTask = defer(async () => {\n await handleIngestInterestRateForward(product_id, meta, abortController.signal);\n })\n .pipe(\n tap({\n error: (err) =>\n console.info(\n formatTime(Date.now()),\n `[SeriesCollector][InterestRate][Forward]`,\n 'Error',\n err,\n ),\n }),\n\n retry(),\n repeat(),\n )\n .subscribe();\n\n sub.add(() => {\n forwardTask.unsubscribe();\n });\n\n // 设置后向任务\n const backwardTask = defer(async () => {\n await handleIngestInterestRateBackward(product_id, meta, abortController.signal);\n })\n .pipe(\n tap({\n error: (err) =>\n console.info(\n formatTime(Date.now()),\n `[SeriesCollector][InterestRate][Backward]`,\n 'Error',\n err,\n ),\n }),\n retry(),\n repeat(),\n )\n .subscribe();\n\n sub.add(() => {\n backwardTask.unsubscribe();\n });\n\n // 设置补齐任务\n const patchTask = defer(async () => {\n await handleInterestRatePatch(product_id, meta, abortController.signal);\n })\n .pipe(\n tap({\n error: (err) =>\n console.info(\n formatTime(Date.now()),\n `[SeriesCollector][InterestRate][Patch]`,\n 'Error',\n err,\n ),\n }),\n retry(),\n repeat(),\n )\n .subscribe();\n\n sub.add(() => {\n patchTask.unsubscribe();\n });\n }),\n ),\n )\n .subscribe();\n"]}
@@ -0,0 +1,49 @@
1
+ import { formatTime, listWatch } from '@yuants/utils';
2
+ import { defer, map, Observable, repeat, retry, tap } from 'rxjs';
3
+ import { listOHLCSeriesIds } from './discovery';
4
+ import { handleIngestOHLCForward } from './forwards-ohlc';
5
+ import { handleIngestOHLCBackward } from './backwards-ohlc';
6
+ import { handleOHLCPatch } from './patch-ohlc';
7
+ defer(() => listOHLCSeriesIds())
8
+ .pipe(retry({ delay: 1000 }), repeat({ delay: 60000 }), map((x) => Array.from(x.entries())), listWatch((x) => x[0], ([product_id, direction]) => new Observable((sub) => {
9
+ // 处理每个利率品种任务: (forward / backward / patch),都需要独立调度
10
+ const abortController = new AbortController();
11
+ sub.add(() => {
12
+ abortController.abort();
13
+ });
14
+ // 先处理前向任务
15
+ const forwardTask = defer(async () => {
16
+ await handleIngestOHLCForward(product_id, direction, abortController.signal);
17
+ })
18
+ .pipe(tap({
19
+ error: (err) => console.info(formatTime(Date.now()), `[SeriesCollector][OHLC][Forward]`, 'Error', err),
20
+ }), retry(), repeat())
21
+ .subscribe();
22
+ sub.add(() => {
23
+ forwardTask.unsubscribe();
24
+ });
25
+ // 设置后向任务
26
+ const backwardTask = defer(async () => {
27
+ await handleIngestOHLCBackward(product_id, direction, abortController.signal);
28
+ })
29
+ .pipe(tap({
30
+ error: (err) => console.info(formatTime(Date.now()), `[SeriesCollector][OHLC][Backward]`, 'Error', err),
31
+ }), retry(), repeat())
32
+ .subscribe();
33
+ sub.add(() => {
34
+ backwardTask.unsubscribe();
35
+ });
36
+ // 设置补齐任务
37
+ const patchTask = defer(async () => {
38
+ await handleOHLCPatch(product_id, direction, abortController.signal);
39
+ })
40
+ .pipe(tap({
41
+ error: (err) => console.info(formatTime(Date.now()), `[SeriesCollector][OHLC][Patch]`, 'Error', err),
42
+ }), retry(), repeat())
43
+ .subscribe();
44
+ sub.add(() => {
45
+ patchTask.unsubscribe();
46
+ });
47
+ })))
48
+ .subscribe();
49
+ //# sourceMappingURL=ohlc.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ohlc.js","sourceRoot":"","sources":["../../src/series-collector/ohlc.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AACtD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,MAAM,CAAC;AAElE,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAGhD,OAAO,EAAE,uBAAuB,EAAE,MAAM,iBAAiB,CAAC;AAC1D,OAAO,EAAE,wBAAwB,EAAE,MAAM,kBAAkB,CAAC;AAC5D,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAE/C,KAAK,CAAC,GAAG,EAAE,CAAC,iBAAiB,EAAE,CAAC;KAC7B,IAAI,CACH,KAAK,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EACtB,MAAM,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EACxB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,EACnC,SAAS,CACP,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,EACX,CAAC,CAAC,UAAU,EAAE,SAAS,CAAC,EAAE,EAAE,CAC1B,IAAI,UAAU,CAAC,CAAC,GAAG,EAAE,EAAE;IACrB,mDAAmD;IACnD,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;IAE9C,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE;QACX,eAAe,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,UAAU;IACV,MAAM,WAAW,GAAG,KAAK,CAAC,KAAK,IAAI,EAAE;QACnC,MAAM,uBAAuB,CAAC,UAAU,EAAE,SAAS,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;IAC/E,CAAC,CAAC;SACC,IAAI,CACH,GAAG,CAAC;QACF,KAAK,EAAE,CAAC,GAAG,EAAE,EAAE,CACb,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,kCAAkC,EAAE,OAAO,EAAE,GAAG,CAAC;KACzF,CAAC,EAEF,KAAK,EAAE,EACP,MAAM,EAAE,CACT;SACA,SAAS,EAAE,CAAC;IAEf,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE;QACX,WAAW,CAAC,WAAW,EAAE,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,SAAS;IACT,MAAM,YAAY,GAAG,KAAK,CAAC,KAAK,IAAI,EAAE;QACpC,MAAM,wBAAwB,CAAC,UAAU,EAAE,SAAS,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;IAChF,CAAC,CAAC;SACC,IAAI,CACH,GAAG,CAAC;QACF,KAAK,EAAE,CAAC,GAAG,EAAE,EAAE,CACb,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,mCAAmC,EAAE,OAAO,EAAE,GAAG,CAAC;KAC1F,CAAC,EACF,KAAK,EAAE,EACP,MAAM,EAAE,CACT;SACA,SAAS,EAAE,CAAC;IAEf,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE;QACX,YAAY,CAAC,WAAW,EAAE,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,SAAS;IACT,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,IAAI,EAAE;QACjC,MAAM,eAAe,CAAC,UAAU,EAAE,SAAS,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC;IACvE,CAAC,CAAC;SACC,IAAI,CACH,GAAG,CAAC;QACF,KAAK,EAAE,CAAC,GAAG,EAAE,EAAE,CACb,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,gCAAgC,EAAE,OAAO,EAAE,GAAG,CAAC;KACvF,CAAC,EACF,KAAK,EAAE,EACP,MAAM,EAAE,CACT;SACA,SAAS,EAAE,CAAC;IAEf,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE;QACX,SAAS,CAAC,WAAW,EAAE,CAAC;IAC1B,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CACL,CACF;KACA,SAAS,EAAE,CAAC","sourcesContent":["import { formatTime, listWatch } from '@yuants/utils';\nimport { defer, map, Observable, repeat, retry, tap } from 'rxjs';\nimport { handleIngestInterestRateBackward } from './backwards-interest-rate';\nimport { listOHLCSeriesIds } from './discovery';\nimport { handleIngestInterestRateForward } from './forwards-interest-rate';\nimport { handleInterestRatePatch } from './patch-interest-rate';\nimport { handleIngestOHLCForward } from './forwards-ohlc';\nimport { handleIngestOHLCBackward } from './backwards-ohlc';\nimport { handleOHLCPatch } from './patch-ohlc';\n\ndefer(() => listOHLCSeriesIds())\n .pipe(\n retry({ delay: 1000 }),\n repeat({ delay: 60000 }),\n map((x) => Array.from(x.entries())),\n listWatch(\n (x) => x[0],\n ([product_id, direction]) =>\n new Observable((sub) => {\n // 处理每个利率品种任务: (forward / backward / patch),都需要独立调度\n const abortController = new AbortController();\n\n sub.add(() => {\n abortController.abort();\n });\n\n // 先处理前向任务\n const forwardTask = defer(async () => {\n await handleIngestOHLCForward(product_id, direction, abortController.signal);\n })\n .pipe(\n tap({\n error: (err) =>\n console.info(formatTime(Date.now()), `[SeriesCollector][OHLC][Forward]`, 'Error', err),\n }),\n\n retry(),\n repeat(),\n )\n .subscribe();\n\n sub.add(() => {\n forwardTask.unsubscribe();\n });\n\n // 设置后向任务\n const backwardTask = defer(async () => {\n await handleIngestOHLCBackward(product_id, direction, abortController.signal);\n })\n .pipe(\n tap({\n error: (err) =>\n console.info(formatTime(Date.now()), `[SeriesCollector][OHLC][Backward]`, 'Error', err),\n }),\n retry(),\n repeat(),\n )\n .subscribe();\n\n sub.add(() => {\n backwardTask.unsubscribe();\n });\n\n // 设置补齐任务\n const patchTask = defer(async () => {\n await handleOHLCPatch(product_id, direction, abortController.signal);\n })\n .pipe(\n tap({\n error: (err) =>\n console.info(formatTime(Date.now()), `[SeriesCollector][OHLC][Patch]`, 'Error', err),\n }),\n retry(),\n repeat(),\n )\n .subscribe();\n\n sub.add(() => {\n patchTask.unsubscribe();\n });\n }),\n ),\n )\n .subscribe();\n"]}
@@ -0,0 +1,63 @@
1
+ import { Terminal } from '@yuants/protocol';
2
+ import { escapeSQL, requestSQL } from '@yuants/sql';
3
+ import { decodePath, formatTime, tokenBucket } from '@yuants/utils';
4
+ const terminal = Terminal.fromNodeEnv();
5
+ const ingestCounter = terminal.metrics
6
+ .counter('series_collector_ingest_count', '')
7
+ .labels({ terminal_id: terminal.terminal_id, type: 'interest_rate', task: 'forward' });
8
+ // Patch 任务:查找数据缺口并进行补齐
9
+ export const handleInterestRatePatch = async (product_id, meta, signal) => {
10
+ var _a, _b, _c, _d;
11
+ const [datasource_id] = decodePath(product_id);
12
+ await tokenBucket(`interest_rate:patch:${datasource_id}`).acquire(1, signal);
13
+ const [record] = await requestSQL(terminal, `
14
+ WITH reversed_ranges AS (
15
+ SELECT
16
+ start_time,
17
+ end_time,
18
+ LEAD(end_time) OVER (
19
+ PARTITION BY table_name, series_id
20
+ ORDER BY start_time DESC
21
+ ) AS next_end_time -- 注意:倒序时 LEAD 是前一个区间
22
+ FROM series_data_range
23
+ WHERE table_name = 'interest_rate'
24
+ AND series_id = ${escapeSQL(product_id)}
25
+ )
26
+ SELECT
27
+ next_end_time AS gap_start_time, -- 前一个区间的结束时间
28
+ start_time AS gap_end_time -- 当前区间的开始时间
29
+ FROM reversed_ranges
30
+ WHERE next_end_time IS NOT NULL
31
+ AND start_time > next_end_time -- 有空缺
32
+ ORDER BY start_time DESC -- 从最新开始
33
+ LIMIT 1;
34
+ `);
35
+ // no gap
36
+ if (!record)
37
+ return;
38
+ const gapStartTime = new Date(record.gap_start_time).getTime();
39
+ const gapEndTime = new Date(record.gap_end_time).getTime();
40
+ console.info(formatTime(Date.now()), '[SeriesCollector][InterestRate][Patch]', 'FindGap', `series=${product_id}, from=${formatTime(gapStartTime)}, to=${formatTime(gapEndTime)}`);
41
+ let req;
42
+ if (meta.direction === 'forward') {
43
+ // forward patch
44
+ req = {
45
+ product_id: product_id,
46
+ direction: 'forward',
47
+ time: gapStartTime,
48
+ };
49
+ }
50
+ else {
51
+ // backward patch
52
+ req = {
53
+ product_id: product_id,
54
+ direction: 'backward',
55
+ time: gapEndTime,
56
+ };
57
+ }
58
+ console.info(formatTime(Date.now()), '[SeriesCollector][InterestRate][Patch]', product_id, 'PatchRequest', `direction=${req.direction}, time=${formatTime(req.time)}`);
59
+ const res = await terminal.client.requestForResponseData('IngestInterestRate', req);
60
+ ingestCounter.inc(res.wrote_count || 0);
61
+ console.info(formatTime(Date.now()), '[SeriesCollector][InterestRate][Patch]', product_id, 'PatchBackwardResult', `ingested_count=${res.wrote_count}, start_time=${formatTime((_b = (_a = res.range) === null || _a === void 0 ? void 0 : _a.start_time) !== null && _b !== void 0 ? _b : NaN)}, end_time=${formatTime((_d = (_c = res.range) === null || _c === void 0 ? void 0 : _c.end_time) !== null && _d !== void 0 ? _d : NaN)}`);
62
+ };
63
+ //# sourceMappingURL=patch-interest-rate.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"patch-interest-rate.js","sourceRoot":"","sources":["../../src/series-collector/patch-interest-rate.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAEpE,MAAM,QAAQ,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;AAExC,MAAM,aAAa,GAAG,QAAQ,CAAC,OAAO;KACnC,OAAO,CAAC,+BAA+B,EAAE,EAAE,CAAC;KAC5C,MAAM,CAAC,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;AAEzF,uBAAuB;AACvB,MAAM,CAAC,MAAM,uBAAuB,GAAG,KAAK,EAC1C,UAAkB,EAClB,IAAkC,EAClC,MAAmB,EACnB,EAAE;;IACF,MAAM,CAAC,aAAa,CAAC,GAAG,UAAU,CAAC,UAAU,CAAC,CAAC;IAC/C,MAAM,WAAW,CAAC,uBAAuB,aAAa,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;IAC7E,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,UAAU,CAC/B,QAAQ,EACR;;;;;;;;;;;wBAWoB,SAAS,CAAC,UAAU,CAAC;;;;;;;;;;KAUxC,CACF,CAAC;IAEF,SAAS;IACT,IAAI,CAAC,MAAM;QAAE,OAAO;IAEpB,MAAM,YAAY,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,OAAO,EAAE,CAAC;IAC/D,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE,CAAC;IAE3D,OAAO,CAAC,IAAI,CACV,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,wCAAwC,EACxC,SAAS,EACT,UAAU,UAAU,UAAU,UAAU,CAAC,YAAY,CAAC,QAAQ,UAAU,CAAC,UAAU,CAAC,EAAE,CACvF,CAAC;IAEF,IAAI,GAA+B,CAAC;IAEpC,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS,EAAE;QAChC,gBAAgB;QAChB,GAAG,GAAG;YACJ,UAAU,EAAE,UAAU;YACtB,SAAS,EAAE,SAAkB;YAC7B,IAAI,EAAE,YAAY;SACnB,CAAC;KACH;SAAM;QACL,iBAAiB;QACjB,GAAG,GAAG;YACJ,UAAU,EAAE,UAAU;YACtB,SAAS,EAAE,UAAmB;YAC9B,IAAI,EAAE,UAAU;SACjB,CAAC;KACH;IAED,OAAO,CAAC,IAAI,CACV,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,wCAAwC,EACxC,UAAU,EACV,cAAc,EACd,aAAa,GAAG,CAAC,SAAS,UAAU,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAC3D,CAAC;IAEF,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,sBAAsB,CACtD,oBAAoB,EACpB,GAAG,CACJ,CAAC;IAEF,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;IAExC,OAAO,CAAC,IAAI,CACV,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,wCAAwC,EACxC,UAAU,EACV,qBAAqB,EACrB,kBAAkB,GAAG,CAAC,WAAW,gBAAgB,UAAU,CACzD,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,UAAU,mCAAI,GAAG,CAC7B,cAAc,UAAU,CAAC,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,QAAQ,mCAAI,GAAG,CAAC,EAAE,CACxD,CAAC;AACJ,CAAC,CAAC","sourcesContent":["import {\n IIngestInterestRateRequest,\n IInterestRateServiceMetadata,\n ISeriesIngestResult,\n} from '@yuants/exchange';\nimport { Terminal } from '@yuants/protocol';\nimport { escapeSQL, requestSQL } from '@yuants/sql';\nimport { decodePath, formatTime, tokenBucket } from '@yuants/utils';\n\nconst terminal = Terminal.fromNodeEnv();\n\nconst ingestCounter = terminal.metrics\n .counter('series_collector_ingest_count', '')\n .labels({ terminal_id: terminal.terminal_id, type: 'interest_rate', task: 'forward' });\n\n// Patch 任务:查找数据缺口并进行补齐\nexport const handleInterestRatePatch = async (\n product_id: string,\n meta: IInterestRateServiceMetadata,\n signal: AbortSignal,\n) => {\n const [datasource_id] = decodePath(product_id);\n await tokenBucket(`interest_rate:patch:${datasource_id}`).acquire(1, signal);\n const [record] = await requestSQL<{ gap_start_time: string; gap_end_time: string }[]>(\n terminal,\n `\nWITH reversed_ranges AS (\n SELECT \n start_time,\n end_time,\n LEAD(end_time) OVER (\n PARTITION BY table_name, series_id \n ORDER BY start_time DESC\n ) AS next_end_time -- 注意:倒序时 LEAD 是前一个区间\n FROM series_data_range\n WHERE table_name = 'interest_rate' \n AND series_id = ${escapeSQL(product_id)}\n)\nSELECT \n next_end_time AS gap_start_time, -- 前一个区间的结束时间\n start_time AS gap_end_time -- 当前区间的开始时间\nFROM reversed_ranges\nWHERE next_end_time IS NOT NULL \n AND start_time > next_end_time -- 有空缺\nORDER BY start_time DESC -- 从最新开始\nLIMIT 1;\n `,\n );\n\n // no gap\n if (!record) return;\n\n const gapStartTime = new Date(record.gap_start_time).getTime();\n const gapEndTime = new Date(record.gap_end_time).getTime();\n\n console.info(\n formatTime(Date.now()),\n '[SeriesCollector][InterestRate][Patch]',\n 'FindGap',\n `series=${product_id}, from=${formatTime(gapStartTime)}, to=${formatTime(gapEndTime)}`,\n );\n\n let req: IIngestInterestRateRequest;\n\n if (meta.direction === 'forward') {\n // forward patch\n req = {\n product_id: product_id,\n direction: 'forward' as const,\n time: gapStartTime,\n };\n } else {\n // backward patch\n req = {\n product_id: product_id,\n direction: 'backward' as const,\n time: gapEndTime,\n };\n }\n\n console.info(\n formatTime(Date.now()),\n '[SeriesCollector][InterestRate][Patch]',\n product_id,\n 'PatchRequest',\n `direction=${req.direction}, time=${formatTime(req.time)}`,\n );\n\n const res = await terminal.client.requestForResponseData<IIngestInterestRateRequest, ISeriesIngestResult>(\n 'IngestInterestRate',\n req,\n );\n\n ingestCounter.inc(res.wrote_count || 0);\n\n console.info(\n formatTime(Date.now()),\n '[SeriesCollector][InterestRate][Patch]',\n product_id,\n 'PatchBackwardResult',\n `ingested_count=${res.wrote_count}, start_time=${formatTime(\n res.range?.start_time ?? NaN,\n )}, end_time=${formatTime(res.range?.end_time ?? NaN)}`,\n );\n};\n"]}
@@ -0,0 +1,67 @@
1
+ import { decodeOHLCSeriesId } from '@yuants/data-ohlc';
2
+ import { Terminal } from '@yuants/protocol';
3
+ import { escapeSQL, requestSQL } from '@yuants/sql';
4
+ import { decodePath, formatTime, tokenBucket } from '@yuants/utils';
5
+ const terminal = Terminal.fromNodeEnv();
6
+ const ingestCounter = terminal.metrics
7
+ .counter('series_collector_ingest_count', '')
8
+ .labels({ terminal_id: terminal.terminal_id, type: 'ohlc', task: 'forward' });
9
+ // Patch 任务:查找数据缺口并进行补齐
10
+ export const handleOHLCPatch = async (series_id, direction, signal) => {
11
+ var _a, _b, _c, _d;
12
+ const [datasource_id] = decodePath(series_id);
13
+ await tokenBucket(`ohlc:patch:${datasource_id}`).acquire(1, signal);
14
+ const [record] = await requestSQL(terminal, `
15
+ WITH reversed_ranges AS (
16
+ SELECT
17
+ start_time,
18
+ end_time,
19
+ LEAD(end_time) OVER (
20
+ PARTITION BY table_name, series_id
21
+ ORDER BY start_time DESC
22
+ ) AS next_end_time -- 注意:倒序时 LEAD 是前一个区间
23
+ FROM series_data_range
24
+ WHERE table_name = 'ohlc_v2'
25
+ AND series_id = ${escapeSQL(series_id)}
26
+ )
27
+ SELECT
28
+ next_end_time AS gap_start_time, -- 前一个区间的结束时间
29
+ start_time AS gap_end_time -- 当前区间的开始时间
30
+ FROM reversed_ranges
31
+ WHERE next_end_time IS NOT NULL
32
+ AND start_time > next_end_time -- 有空缺
33
+ ORDER BY start_time DESC -- 从最新开始
34
+ LIMIT 1;
35
+ `);
36
+ // no gap
37
+ if (!record)
38
+ return;
39
+ const gapStartTime = new Date(record.gap_start_time).getTime();
40
+ const gapEndTime = new Date(record.gap_end_time).getTime();
41
+ console.info(formatTime(Date.now()), '[SeriesCollector][OHLC][Patch]', 'FindGap', `series=${series_id}, from=${formatTime(gapStartTime)}, to=${formatTime(gapEndTime)}`);
42
+ const { product_id, duration } = decodeOHLCSeriesId(series_id);
43
+ let req;
44
+ if (direction === 'forward') {
45
+ // forward patch
46
+ req = {
47
+ product_id: product_id,
48
+ duration,
49
+ direction: 'forward',
50
+ time: gapStartTime,
51
+ };
52
+ }
53
+ else {
54
+ // backward patch
55
+ req = {
56
+ product_id: product_id,
57
+ duration,
58
+ direction: 'backward',
59
+ time: gapEndTime,
60
+ };
61
+ }
62
+ console.info(formatTime(Date.now()), '[SeriesCollector][OHLC][Patch]', series_id, 'PatchRequest', `direction=${req.direction}, time=${formatTime(req.time)}`);
63
+ const res = await terminal.client.requestForResponseData('IngestOHLC', req);
64
+ ingestCounter.inc(res.wrote_count || 0);
65
+ console.info(formatTime(Date.now()), '[SeriesCollector][OHLC][Patch]', series_id, 'PatchBackwardResult', `ingested_count=${res.wrote_count}, start_time=${formatTime((_b = (_a = res.range) === null || _a === void 0 ? void 0 : _a.start_time) !== null && _b !== void 0 ? _b : NaN)}, end_time=${formatTime((_d = (_c = res.range) === null || _c === void 0 ? void 0 : _c.end_time) !== null && _d !== void 0 ? _d : NaN)}`);
66
+ };
67
+ //# sourceMappingURL=patch-ohlc.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"patch-ohlc.js","sourceRoot":"","sources":["../../src/series-collector/patch-ohlc.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAEvD,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAC5C,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACpD,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAEpE,MAAM,QAAQ,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;AAExC,MAAM,aAAa,GAAG,QAAQ,CAAC,OAAO;KACnC,OAAO,CAAC,+BAA+B,EAAE,EAAE,CAAC;KAC5C,MAAM,CAAC,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;AAEhF,uBAAuB;AACvB,MAAM,CAAC,MAAM,eAAe,GAAG,KAAK,EAClC,SAAiB,EACjB,SAAiC,EACjC,MAAmB,EACnB,EAAE;;IACF,MAAM,CAAC,aAAa,CAAC,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;IAC9C,MAAM,WAAW,CAAC,cAAc,aAAa,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;IACpE,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,UAAU,CAC/B,QAAQ,EACR;;;;;;;;;;;wBAWoB,SAAS,CAAC,SAAS,CAAC;;;;;;;;;;KAUvC,CACF,CAAC;IAEF,SAAS;IACT,IAAI,CAAC,MAAM;QAAE,OAAO;IAEpB,MAAM,YAAY,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,OAAO,EAAE,CAAC;IAC/D,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE,CAAC;IAE3D,OAAO,CAAC,IAAI,CACV,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,gCAAgC,EAChC,SAAS,EACT,UAAU,SAAS,UAAU,UAAU,CAAC,YAAY,CAAC,QAAQ,UAAU,CAAC,UAAU,CAAC,EAAE,CACtF,CAAC;IAEF,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,kBAAkB,CAAC,SAAS,CAAC,CAAC;IAE/D,IAAI,GAAuB,CAAC;IAE5B,IAAI,SAAS,KAAK,SAAS,EAAE;QAC3B,gBAAgB;QAChB,GAAG,GAAG;YACJ,UAAU,EAAE,UAAU;YACtB,QAAQ;YACR,SAAS,EAAE,SAAkB;YAC7B,IAAI,EAAE,YAAY;SACnB,CAAC;KACH;SAAM;QACL,iBAAiB;QACjB,GAAG,GAAG;YACJ,UAAU,EAAE,UAAU;YACtB,QAAQ;YACR,SAAS,EAAE,UAAmB;YAC9B,IAAI,EAAE,UAAU;SACjB,CAAC;KACH;IAED,OAAO,CAAC,IAAI,CACV,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,gCAAgC,EAChC,SAAS,EACT,cAAc,EACd,aAAa,GAAG,CAAC,SAAS,UAAU,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAC3D,CAAC;IAEF,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,sBAAsB,CACtD,YAAY,EACZ,GAAG,CACJ,CAAC;IAEF,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;IAExC,OAAO,CAAC,IAAI,CACV,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,gCAAgC,EAChC,SAAS,EACT,qBAAqB,EACrB,kBAAkB,GAAG,CAAC,WAAW,gBAAgB,UAAU,CACzD,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,UAAU,mCAAI,GAAG,CAC7B,cAAc,UAAU,CAAC,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,QAAQ,mCAAI,GAAG,CAAC,EAAE,CACxD,CAAC;AACJ,CAAC,CAAC","sourcesContent":["import { decodeOHLCSeriesId } from '@yuants/data-ohlc';\nimport { IIngestOHLCRequest, ISeriesIngestResult } from '@yuants/exchange';\nimport { Terminal } from '@yuants/protocol';\nimport { escapeSQL, requestSQL } from '@yuants/sql';\nimport { decodePath, formatTime, tokenBucket } from '@yuants/utils';\n\nconst terminal = Terminal.fromNodeEnv();\n\nconst ingestCounter = terminal.metrics\n .counter('series_collector_ingest_count', '')\n .labels({ terminal_id: terminal.terminal_id, type: 'ohlc', task: 'forward' });\n\n// Patch 任务:查找数据缺口并进行补齐\nexport const handleOHLCPatch = async (\n series_id: string,\n direction: 'forward' | 'backward',\n signal: AbortSignal,\n) => {\n const [datasource_id] = decodePath(series_id);\n await tokenBucket(`ohlc:patch:${datasource_id}`).acquire(1, signal);\n const [record] = await requestSQL<{ gap_start_time: string; gap_end_time: string }[]>(\n terminal,\n `\nWITH reversed_ranges AS (\n SELECT \n start_time,\n end_time,\n LEAD(end_time) OVER (\n PARTITION BY table_name, series_id \n ORDER BY start_time DESC\n ) AS next_end_time -- 注意:倒序时 LEAD 是前一个区间\n FROM series_data_range\n WHERE table_name = 'ohlc_v2' \n AND series_id = ${escapeSQL(series_id)}\n)\nSELECT \n next_end_time AS gap_start_time, -- 前一个区间的结束时间\n start_time AS gap_end_time -- 当前区间的开始时间\nFROM reversed_ranges\nWHERE next_end_time IS NOT NULL \n AND start_time > next_end_time -- 有空缺\nORDER BY start_time DESC -- 从最新开始\nLIMIT 1;\n `,\n );\n\n // no gap\n if (!record) return;\n\n const gapStartTime = new Date(record.gap_start_time).getTime();\n const gapEndTime = new Date(record.gap_end_time).getTime();\n\n console.info(\n formatTime(Date.now()),\n '[SeriesCollector][OHLC][Patch]',\n 'FindGap',\n `series=${series_id}, from=${formatTime(gapStartTime)}, to=${formatTime(gapEndTime)}`,\n );\n\n const { product_id, duration } = decodeOHLCSeriesId(series_id);\n\n let req: IIngestOHLCRequest;\n\n if (direction === 'forward') {\n // forward patch\n req = {\n product_id: product_id,\n duration,\n direction: 'forward' as const,\n time: gapStartTime,\n };\n } else {\n // backward patch\n req = {\n product_id: product_id,\n duration,\n direction: 'backward' as const,\n time: gapEndTime,\n };\n }\n\n console.info(\n formatTime(Date.now()),\n '[SeriesCollector][OHLC][Patch]',\n series_id,\n 'PatchRequest',\n `direction=${req.direction}, time=${formatTime(req.time)}`,\n );\n\n const res = await terminal.client.requestForResponseData<IIngestOHLCRequest, ISeriesIngestResult>(\n 'IngestOHLC',\n req,\n );\n\n ingestCounter.inc(res.wrote_count || 0);\n\n console.info(\n formatTime(Date.now()),\n '[SeriesCollector][OHLC][Patch]',\n series_id,\n 'PatchBackwardResult',\n `ingested_count=${res.wrote_count}, start_time=${formatTime(\n res.range?.start_time ?? NaN,\n )}, end_time=${formatTime(res.range?.end_time ?? NaN)}`,\n );\n};\n"]}
@@ -0,0 +1,11 @@
1
+ import { escapeSQL, requestSQL } from '@yuants/sql';
2
+ /**
3
+ * 查找某个利率品种系列ID的最新结束时间(for forward task)
4
+ * @param terminal
5
+ * @param product_id
6
+ * @returns
7
+ */
8
+ export const findInterestRateEndTimeForward = (terminal, product_id) => requestSQL(terminal, `select end_time from series_data_range where series_id = ${escapeSQL(product_id)} and table_name = 'interest_rate' order by end_time desc limit 1`).then((records) => { var _a; return (_a = records === null || records === void 0 ? void 0 : records[0]) === null || _a === void 0 ? void 0 : _a.end_time; });
9
+ export const findInterestRateStartTimeBackward = (terminal, product_id) => requestSQL(terminal, `select start_time from series_data_range where series_id = ${escapeSQL(product_id)} and table_name = 'interest_rate' order by start_time asc limit 1`).then((records) => { var _a; return (_a = records === null || records === void 0 ? void 0 : records[0]) === null || _a === void 0 ? void 0 : _a.start_time; });
10
+ export const findOHLCEndTimeForward = (terminal, series_id) => requestSQL(terminal, `select end_time from series_data_range where series_id = ${escapeSQL(series_id)} and table_name = 'ohlc_v2' order by end_time desc limit 1`).then((records) => { var _a; return (_a = records === null || records === void 0 ? void 0 : records[0]) === null || _a === void 0 ? void 0 : _a.end_time; });
11
+ //# sourceMappingURL=sql-helpers.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"sql-helpers.js","sourceRoot":"","sources":["../../src/series-collector/sql-helpers.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEpD;;;;;GAKG;AACH,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAAC,QAAkB,EAAE,UAAkB,EAAE,EAAE,CACvF,UAAU,CAKR,QAAQ,EACR,4DAA4D,SAAS,CACnE,UAAU,CACX,kEAAkE,CACpE,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,WAAC,OAAA,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAG,CAAC,CAAC,0CAAE,QAAQ,CAAA,EAAA,CAAC,CAAC;AAE9C,MAAM,CAAC,MAAM,iCAAiC,GAAG,CAAC,QAAkB,EAAE,UAAkB,EAAE,EAAE,CAC1F,UAAU,CAKR,QAAQ,EACR,8DAA8D,SAAS,CACrE,UAAU,CACX,mEAAmE,CACrE,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,WAAC,OAAA,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAG,CAAC,CAAC,0CAAE,UAAU,CAAA,EAAA,CAAC,CAAC;AAEhD,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,QAAkB,EAAE,SAAiB,EAAE,EAAE,CAC9E,UAAU,CAKR,QAAQ,EACR,4DAA4D,SAAS,CACnE,SAAS,CACV,4DAA4D,CAC9D,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,WAAC,OAAA,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAG,CAAC,CAAC,0CAAE,QAAQ,CAAA,EAAA,CAAC,CAAC","sourcesContent":["import { Terminal } from '@yuants/protocol';\nimport { escapeSQL, requestSQL } from '@yuants/sql';\n\n/**\n * 查找某个利率品种系列ID的最新结束时间(for forward task)\n * @param terminal\n * @param product_id\n * @returns\n */\nexport const findInterestRateEndTimeForward = (terminal: Terminal, product_id: string) =>\n requestSQL<\n {\n end_time: string;\n }[]\n >(\n terminal,\n `select end_time from series_data_range where series_id = ${escapeSQL(\n product_id,\n )} and table_name = 'interest_rate' order by end_time desc limit 1`,\n ).then((records) => records?.[0]?.end_time);\n\nexport const findInterestRateStartTimeBackward = (terminal: Terminal, product_id: string) =>\n requestSQL<\n {\n start_time: string;\n }[]\n >(\n terminal,\n `select start_time from series_data_range where series_id = ${escapeSQL(\n product_id,\n )} and table_name = 'interest_rate' order by start_time asc limit 1`,\n ).then((records) => records?.[0]?.start_time);\n\nexport const findOHLCEndTimeForward = (terminal: Terminal, series_id: string) =>\n requestSQL<\n {\n end_time: string;\n }[]\n >(\n terminal,\n `select end_time from series_data_range where series_id = ${escapeSQL(\n series_id,\n )} and table_name = 'ohlc_v2' order by end_time desc limit 1`,\n ).then((records) => records?.[0]?.end_time);\n"]}
@@ -1,2 +1,4 @@
1
- export {};
1
+ /// <reference types="node" />
2
+ import { IInterestRateServiceMetadata } from '@yuants/exchange';
3
+ export declare const handleIngestInterestRateBackward: (product_id: string, meta: IInterestRateServiceMetadata, signal: AbortSignal) => Promise<void>;
2
4
  //# sourceMappingURL=backwards-interest-rate.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"backwards-interest-rate.d.ts","sourceRoot":"","sources":["../../src/series-collector/backwards-interest-rate.ts"],"names":[],"mappings":""}
1
+ {"version":3,"file":"backwards-interest-rate.d.ts","sourceRoot":"","sources":["../../src/series-collector/backwards-interest-rate.ts"],"names":[],"mappings":";AAKA,OAAO,EAEL,4BAA4B,EAE7B,MAAM,kBAAkB,CAAC;AAU1B,eAAO,MAAM,gCAAgC,eAC/B,MAAM,QACZ,4BAA4B,UAC1B,WAAW,kBA6CpB,CAAC"}
@@ -4,72 +4,40 @@
4
4
  // 然后对每个序列,向对应的 IngestInterestRate Service 发送拉取请求,补齐历史数据。
5
5
  // 使用 Token Bucket 控制每个数据源的请求速率,避免过载。
6
6
  Object.defineProperty(exports, "__esModule", { value: true });
7
- const exchange_1 = require("@yuants/exchange");
7
+ exports.handleIngestInterestRateBackward = void 0;
8
8
  const protocol_1 = require("@yuants/protocol");
9
- const sql_1 = require("@yuants/sql");
10
9
  const utils_1 = require("@yuants/utils");
11
- const rxjs_1 = require("rxjs");
10
+ const sql_helpers_1 = require("./sql-helpers");
12
11
  const terminal = protocol_1.Terminal.fromNodeEnv();
13
- const listBackwardSeriesIds = async () => {
14
- const product_ids = await (0, sql_1.requestSQL)(terminal, `select product_id from product`);
15
- console.time('[SeriesCollector][InterestRate][Backwards] calc');
16
- const series_ids = new Set();
17
- for (const terminalInfo of terminal.terminalInfos) {
18
- for (const serviceInfo of Object.values(terminalInfo.serviceInfo || {})) {
19
- if (serviceInfo.method !== 'IngestInterestRate')
20
- continue;
21
- try {
22
- const meta = (0, exchange_1.parseInterestRateServiceMetadataFromSchema)(serviceInfo.schema);
23
- if (meta.direction !== 'backward')
24
- continue;
25
- for (const { product_id } of product_ids) {
26
- if (!product_id.startsWith(meta.product_id_prefix))
27
- continue;
28
- series_ids.add(product_id);
29
- }
30
- }
31
- finally {
32
- }
33
- }
12
+ const ingestCounter = terminal.metrics
13
+ .counter('series_collector_ingest_count', '')
14
+ .labels({ terminal_id: terminal.terminal_id, type: 'interest_rate', task: 'backward' });
15
+ const handleIngestInterestRateBackward = async (product_id, meta, signal) => {
16
+ var _a, _b, _c, _d;
17
+ const [datasource_id] = (0, utils_1.decodePath)(product_id);
18
+ // 控制速率:每个数据源每秒钟只能请求一次
19
+ await (0, utils_1.tokenBucket)(`interest_rate:backward:${datasource_id}`).acquire(1, signal);
20
+ let req;
21
+ if (meta.direction === 'backward') {
22
+ const startTime = await (0, sql_helpers_1.findInterestRateStartTimeBackward)(terminal, product_id);
23
+ const start_time = startTime ? new Date(startTime).getTime() : Date.now();
24
+ req = {
25
+ product_id: product_id,
26
+ direction: 'backward',
27
+ time: start_time,
28
+ };
34
29
  }
35
- console.timeEnd('[SeriesCollector][InterestRate][Backwards] calc');
36
- return series_ids;
30
+ else {
31
+ req = {
32
+ product_id: product_id,
33
+ direction: 'forward',
34
+ time: 0,
35
+ };
36
+ }
37
+ console.info((0, utils_1.formatTime)(Date.now()), '[SeriesCollector][InterestRate][Backward]', 'Request', `product_id=${req.product_id}, direction=${req.direction}, time=${(0, utils_1.formatTime)(req.time)}`);
38
+ const res = await terminal.client.requestForResponseData('IngestInterestRate', req);
39
+ ingestCounter.inc(res.wrote_count || 0);
40
+ console.info((0, utils_1.formatTime)(Date.now()), '[SeriesCollector][InterestRate][Backward]', 'Result', `series_id=${product_id}, ingested_count=${res.wrote_count}, start_time=${(0, utils_1.formatTime)((_b = (_a = res.range) === null || _a === void 0 ? void 0 : _a.start_time) !== null && _b !== void 0 ? _b : NaN)}, end_time=${(0, utils_1.formatTime)((_d = (_c = res.range) === null || _c === void 0 ? void 0 : _c.end_time) !== null && _d !== void 0 ? _d : NaN)}`);
37
41
  };
38
- (0, rxjs_1.defer)(async () => {
39
- const time = Date.now();
40
- const series_ids = await listBackwardSeriesIds();
41
- console.log(`[SeriesCollector][InterestRate][Backwards] Found ${series_ids.size} series to collect backwards data for. (${(0, utils_1.formatTime)(Date.now() - time)})`);
42
- await Promise.all([...series_ids].map(async (product_id) => {
43
- var _a, _b, _c, _d;
44
- try {
45
- const [datasource_id] = (0, utils_1.decodePath)(product_id);
46
- // 控制速率:每个数据源每秒钟只能请求一次
47
- await (0, utils_1.tokenBucket)(`interest_rate_backwards:${datasource_id}`, {
48
- refillInterval: 1000,
49
- capacity: 1,
50
- }).acquire();
51
- {
52
- const [record] = await (0, sql_1.requestSQL)(terminal, `select start_time from series_data_range where series_id = ${(0, sql_1.escapeSQL)(product_id)} and table_name = 'interest_rate' order by start_time limit 1`);
53
- const start_time = record ? new Date(record.start_time).getTime() : Date.now();
54
- const req = {
55
- product_id: product_id,
56
- direction: 'backward',
57
- time: start_time,
58
- };
59
- console.info((0, utils_1.formatTime)(Date.now()), 'DispatchIngestInterestRateRequest', `product_id=${product_id}, time=${(0, utils_1.formatTime)(start_time)}`);
60
- const res = await terminal.client.requestForResponseData('IngestInterestRate', req);
61
- terminal.metrics
62
- .counter('series_collector_backwards_ingest_count', '')
63
- .labels({ terminal_id: terminal.terminal_id, type: 'interest_rate' })
64
- .inc(res.wrote_count || 0);
65
- console.info((0, utils_1.formatTime)(Date.now()), 'DispatchIngestInterestRateResult', `series_id=${product_id}, ingested_count=${res.wrote_count}, start_time=${(0, utils_1.formatTime)((_b = (_a = res.range) === null || _a === void 0 ? void 0 : _a.start_time) !== null && _b !== void 0 ? _b : NaN)}, end_time=${(0, utils_1.formatTime)((_d = (_c = res.range) === null || _c === void 0 ? void 0 : _c.end_time) !== null && _d !== void 0 ? _d : NaN)}`);
66
- }
67
- }
68
- catch (e) {
69
- console.info((0, utils_1.formatTime)(Date.now()), 'DispatchIngestInterestRateError', `series_id=${product_id}`, e);
70
- }
71
- }));
72
- })
73
- .pipe((0, rxjs_1.retry)({ delay: 1000 }), (0, rxjs_1.repeat)({ delay: 1000 }))
74
- .subscribe();
42
+ exports.handleIngestInterestRateBackward = handleIngestInterestRateBackward;
75
43
  //# sourceMappingURL=backwards-interest-rate.js.map