@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
@@ -1 +1 @@
1
- {"version":3,"file":"backwards-interest-rate.js","sourceRoot":"","sources":["../../src/series-collector/backwards-interest-rate.ts"],"names":[],"mappings":";AAAA,0BAA0B;AAC1B,6DAA6D;AAC7D,yDAAyD;AACzD,qCAAqC;;AAErC,+CAI0B;AAC1B,+CAA4C;AAC5C,qCAAoD;AACpD,yCAAoE;AACpE,+BAA4C;AAE5C,MAAM,QAAQ,GAAG,mBAAQ,CAAC,WAAW,EAAE,CAAC;AAExC,MAAM,qBAAqB,GAAG,KAAK,IAAI,EAAE;IACvC,MAAM,WAAW,GAAG,MAAM,IAAA,gBAAU,EAA2B,QAAQ,EAAE,gCAAgC,CAAC,CAAC;IAE3G,OAAO,CAAC,IAAI,CAAC,iDAAiD,CAAC,CAAC;IAEhE,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,oBAAoB;gBAAE,SAAS;YAC1D,IAAI;gBACF,MAAM,IAAI,GAAG,IAAA,qDAA0C,EAAC,WAAW,CAAC,MAAM,CAAC,CAAC;gBAC5E,IAAI,IAAI,CAAC,SAAS,KAAK,UAAU;oBAAE,SAAS;gBAE5C,KAAK,MAAM,EAAE,UAAU,EAAE,IAAI,WAAW,EAAE;oBACxC,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,iBAAiB,CAAC;wBAAE,SAAS;oBAC7D,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;iBAC5B;aACF;oBAAS;aACT;SACF;KACF;IAED,OAAO,CAAC,OAAO,CAAC,iDAAiD,CAAC,CAAC;IAEnE,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC;AAEF,IAAA,YAAK,EAAC,KAAK,IAAI,EAAE;IACf,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACxB,MAAM,UAAU,GAAG,MAAM,qBAAqB,EAAE,CAAC;IACjD,OAAO,CAAC,GAAG,CACT,oDACE,UAAU,CAAC,IACb,2CAA2C,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,GAAG,CAC5E,CAAC;IAEF,MAAM,OAAO,CAAC,GAAG,CACf,CAAC,GAAG,UAAU,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE;;QACvC,IAAI;YACF,MAAM,CAAC,aAAa,CAAC,GAAG,IAAA,kBAAU,EAAC,UAAU,CAAC,CAAC;YAC/C,sBAAsB;YACtB,MAAM,IAAA,mBAAW,EAAC,2BAA2B,aAAa,EAAE,EAAE;gBAC5D,cAAc,EAAE,IAAI;gBACpB,QAAQ,EAAE,CAAC;aACZ,CAAC,CAAC,OAAO,EAAE,CAAC;YACb;gBACE,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,IAAA,gBAAU,EAK/B,QAAQ,EACR,8DAA8D,IAAA,eAAS,EACrE,UAAU,CACX,+DAA+D,CACjE,CAAC;gBACF,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;gBAE/E,MAAM,GAAG,GAA+B;oBACtC,UAAU,EAAE,UAAU;oBACtB,SAAS,EAAE,UAAU;oBACrB,IAAI,EAAE,UAAU;iBACjB,CAAC;gBAEF,OAAO,CAAC,IAAI,CACV,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,mCAAmC,EACnC,cAAc,UAAU,UAAU,IAAA,kBAAU,EAAC,UAAU,CAAC,EAAE,CAC3D,CAAC;gBAEF,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,sBAAsB,CAGtD,oBAAoB,EAAE,GAAG,CAAC,CAAC;gBAE7B,QAAQ,CAAC,OAAO;qBACb,OAAO,CAAC,yCAAyC,EAAE,EAAE,CAAC;qBACtD,MAAM,CAAC,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;qBACpE,GAAG,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;gBAE7B,OAAO,CAAC,IAAI,CACV,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,kCAAkC,EAClC,aAAa,UAAU,oBAAoB,GAAG,CAAC,WAAW,gBAAgB,IAAA,kBAAU,EAClF,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,UAAU,mCAAI,GAAG,CAC7B,cAAc,IAAA,kBAAU,EAAC,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,QAAQ,mCAAI,GAAG,CAAC,EAAE,CACxD,CAAC;aACH;SACF;QAAC,OAAO,CAAC,EAAE;YACV,OAAO,CAAC,IAAI,CAAC,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,iCAAiC,EAAE,aAAa,UAAU,EAAE,EAAE,CAAC,CAAC,CAAC;SACvG;IACH,CAAC,CAAC,CACH,CAAC;AACJ,CAAC,CAAC;KACC,IAAI,CAAC,IAAA,YAAK,EAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,IAAA,aAAM,EAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;KACrD,SAAS,EAAE,CAAC","sourcesContent":["// 解决 Backwards 拉取历史数据的调度器\n// 该文件会定期扫描所有 Terminal 的 ServiceInfo,提取出所有支持 Backwards 拉取的序列。\n// 然后对每个序列,向对应的 IngestInterestRate Service 发送拉取请求,补齐历史数据。\n// 使用 Token Bucket 控制每个数据源的请求速率,避免过载。\n\nimport {\n IIngestInterestRateRequest,\n ISeriesIngestResult,\n parseInterestRateServiceMetadataFromSchema,\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 listBackwardSeriesIds = async () => {\n const product_ids = await requestSQL<{ product_id: string }[]>(terminal, `select product_id from product`);\n\n console.time('[SeriesCollector][InterestRate][Backwards] 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 !== 'IngestInterestRate') continue;\n try {\n const meta = parseInterestRateServiceMetadataFromSchema(serviceInfo.schema);\n if (meta.direction !== 'backward') continue;\n\n for (const { product_id } of product_ids) {\n if (!product_id.startsWith(meta.product_id_prefix)) continue;\n series_ids.add(product_id);\n }\n } finally {\n }\n }\n }\n\n console.timeEnd('[SeriesCollector][InterestRate][Backwards] calc');\n\n return series_ids;\n};\n\ndefer(async () => {\n const time = Date.now();\n const series_ids = await listBackwardSeriesIds();\n console.log(\n `[SeriesCollector][InterestRate][Backwards] Found ${\n series_ids.size\n } series to collect backwards data for. (${formatTime(Date.now() - time)})`,\n );\n\n await Promise.all(\n [...series_ids].map(async (product_id) => {\n try {\n const [datasource_id] = decodePath(product_id);\n // 控制速率:每个数据源每秒钟只能请求一次\n await tokenBucket(`interest_rate_backwards:${datasource_id}`, {\n refillInterval: 1000,\n capacity: 1,\n }).acquire();\n {\n const [record] = await 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 limit 1`,\n );\n const start_time = record ? new Date(record.start_time).getTime() : Date.now();\n\n const req: IIngestInterestRateRequest = {\n product_id: product_id,\n direction: 'backward',\n time: start_time,\n };\n\n console.info(\n formatTime(Date.now()),\n 'DispatchIngestInterestRateRequest',\n `product_id=${product_id}, time=${formatTime(start_time)}`,\n );\n\n const res = await terminal.client.requestForResponseData<\n IIngestInterestRateRequest,\n ISeriesIngestResult\n >('IngestInterestRate', req);\n\n terminal.metrics\n .counter('series_collector_backwards_ingest_count', '')\n .labels({ terminal_id: terminal.terminal_id, type: 'interest_rate' })\n .inc(res.wrote_count || 0);\n\n console.info(\n formatTime(Date.now()),\n 'DispatchIngestInterestRateResult',\n `series_id=${product_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()), 'DispatchIngestInterestRateError', `series_id=${product_id}`, e);\n }\n }),\n );\n})\n .pipe(retry({ delay: 1000 }), repeat({ delay: 1000 }))\n .subscribe();\n"]}
1
+ {"version":3,"file":"backwards-interest-rate.js","sourceRoot":"","sources":["../../src/series-collector/backwards-interest-rate.ts"],"names":[],"mappings":";AAAA,0BAA0B;AAC1B,6DAA6D;AAC7D,yDAAyD;AACzD,qCAAqC;;;AAOrC,+CAA4C;AAC5C,yCAAoE;AACpE,+CAAkE;AAElE,MAAM,QAAQ,GAAG,mBAAQ,CAAC,WAAW,EAAE,CAAC;AACxC,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,UAAU,EAAE,CAAC,CAAC;AAEnF,MAAM,gCAAgC,GAAG,KAAK,EACnD,UAAkB,EAClB,IAAkC,EAClC,MAAmB,EACnB,EAAE;;IACF,MAAM,CAAC,aAAa,CAAC,GAAG,IAAA,kBAAU,EAAC,UAAU,CAAC,CAAC;IAC/C,sBAAsB;IACtB,MAAM,IAAA,mBAAW,EAAC,0BAA0B,aAAa,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;IAChF,IAAI,GAA+B,CAAC;IACpC,IAAI,IAAI,CAAC,SAAS,KAAK,UAAU,EAAE;QACjC,MAAM,SAAS,GAAG,MAAM,IAAA,+CAAiC,EAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAChF,MAAM,UAAU,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;QAE1E,GAAG,GAAG;YACJ,UAAU,EAAE,UAAU;YACtB,SAAS,EAAE,UAAU;YACrB,IAAI,EAAE,UAAU;SACjB,CAAC;KACH;SAAM;QACL,GAAG,GAAG;YACJ,UAAU,EAAE,UAAU;YACtB,SAAS,EAAE,SAAS;YACpB,IAAI,EAAE,CAAC;SACR,CAAC;KACH;IAED,OAAO,CAAC,IAAI,CACV,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,2CAA2C,EAC3C,SAAS,EACT,cAAc,GAAG,CAAC,UAAU,eAAe,GAAG,CAAC,SAAS,UAAU,IAAA,kBAAU,EAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CACzF,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,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,2CAA2C,EAC3C,QAAQ,EACR,aAAa,UAAU,oBAAoB,GAAG,CAAC,WAAW,gBAAgB,IAAA,kBAAU,EAClF,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,UAAU,mCAAI,GAAG,CAC7B,cAAc,IAAA,kBAAU,EAAC,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,QAAQ,mCAAI,GAAG,CAAC,EAAE,CACxD,CAAC;AACJ,CAAC,CAAC;AAhDW,QAAA,gCAAgC,oCAgD3C","sourcesContent":["// 解决 Backwards 拉取历史数据的调度器\n// 该文件会定期扫描所有 Terminal 的 ServiceInfo,提取出所有支持 Backwards 拉取的序列。\n// 然后对每个序列,向对应的 IngestInterestRate Service 发送拉取请求,补齐历史数据。\n// 使用 Token Bucket 控制每个数据源的请求速率,避免过载。\n\nimport {\n IIngestInterestRateRequest,\n IInterestRateServiceMetadata,\n ISeriesIngestResult,\n} from '@yuants/exchange';\nimport { Terminal } from '@yuants/protocol';\nimport { decodePath, formatTime, tokenBucket } from '@yuants/utils';\nimport { findInterestRateStartTimeBackward } from './sql-helpers';\n\nconst terminal = Terminal.fromNodeEnv();\nconst ingestCounter = terminal.metrics\n .counter('series_collector_ingest_count', '')\n .labels({ terminal_id: terminal.terminal_id, type: 'interest_rate', task: 'backward' });\n\nexport const handleIngestInterestRateBackward = async (\n product_id: string,\n meta: IInterestRateServiceMetadata,\n signal: AbortSignal,\n) => {\n const [datasource_id] = decodePath(product_id);\n // 控制速率:每个数据源每秒钟只能请求一次\n await tokenBucket(`interest_rate:backward:${datasource_id}`).acquire(1, signal);\n let req: IIngestInterestRateRequest;\n if (meta.direction === 'backward') {\n const startTime = await findInterestRateStartTimeBackward(terminal, product_id);\n const start_time = startTime ? new Date(startTime).getTime() : Date.now();\n\n req = {\n product_id: product_id,\n direction: 'backward',\n time: start_time,\n };\n } else {\n req = {\n product_id: product_id,\n direction: 'forward',\n time: 0,\n };\n }\n\n console.info(\n formatTime(Date.now()),\n '[SeriesCollector][InterestRate][Backward]',\n 'Request',\n `product_id=${req.product_id}, 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][Backward]',\n 'Result',\n `series_id=${product_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,2 +1,3 @@
1
- export {};
1
+ /// <reference types="node" />
2
+ export declare const handleIngestOHLCBackward: (series_id: string, direction: 'forward' | 'backward', signal: AbortSignal) => Promise<void>;
2
3
  //# sourceMappingURL=backwards-ohlc.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"backwards-ohlc.d.ts","sourceRoot":"","sources":["../../src/series-collector/backwards-ohlc.ts"],"names":[],"mappings":""}
1
+ {"version":3,"file":"backwards-ohlc.d.ts","sourceRoot":"","sources":["../../src/series-collector/backwards-ohlc.ts"],"names":[],"mappings":";AAiBA,eAAO,MAAM,wBAAwB,cACxB,MAAM,aACN,SAAS,GAAG,UAAU,UACzB,WAAW,kBA6DpB,CAAC"}
@@ -4,78 +4,45 @@
4
4
  // 然后对每个序列,向对应的 IngestOHLC Service 发送拉取请求,补齐历史数据。
5
5
  // 使用 Token Bucket 控制每个数据源的请求速率,避免过载。
6
6
  Object.defineProperty(exports, "__esModule", { value: true });
7
+ exports.handleIngestOHLCBackward = void 0;
7
8
  const data_ohlc_1 = require("@yuants/data-ohlc");
8
- const exchange_1 = require("@yuants/exchange");
9
9
  const protocol_1 = require("@yuants/protocol");
10
10
  const sql_1 = require("@yuants/sql");
11
11
  const utils_1 = require("@yuants/utils");
12
- const rxjs_1 = require("rxjs");
13
12
  const terminal = protocol_1.Terminal.fromNodeEnv();
14
- const listBackwardSeriesIds = async () => {
15
- const product_ids = await (0, sql_1.requestSQL)(terminal, `select product_id from product`);
16
- console.time('[SeriesCollector][Backwards] calc');
17
- const series_ids = new Set();
18
- for (const terminalInfo of terminal.terminalInfos) {
19
- for (const serviceInfo of Object.values(terminalInfo.serviceInfo || {})) {
20
- if (serviceInfo.method !== 'IngestOHLC')
21
- continue;
22
- try {
23
- const meta = (0, exchange_1.parseOHLCServiceMetadataFromSchema)(serviceInfo.schema);
24
- if (meta.direction !== 'backward')
25
- continue;
26
- for (const { product_id } of product_ids) {
27
- if (!product_id.startsWith(meta.product_id_prefix))
28
- continue;
29
- for (const duration of meta.duration_list) {
30
- const series_id = (0, data_ohlc_1.encodeOHLCSeriesId)(product_id, duration);
31
- series_ids.add(series_id);
32
- }
33
- }
34
- }
35
- finally {
36
- }
37
- }
13
+ const ingestCounter = terminal.metrics
14
+ .counter('series_collector_ingest_count', '')
15
+ .labels({ terminal_id: terminal.terminal_id, type: 'ohlc', task: 'backward' });
16
+ const handleIngestOHLCBackward = async (series_id, direction, signal) => {
17
+ var _a, _b, _c, _d;
18
+ const { product_id, duration } = (0, data_ohlc_1.decodeOHLCSeriesId)(series_id);
19
+ const [datasource_id] = (0, utils_1.decodePath)(product_id);
20
+ // 控制速率:每个数据源每秒钟只能请求一次
21
+ await (0, utils_1.tokenBucket)(`ohlc:backward:${datasource_id}`).acquire(1, signal);
22
+ let req;
23
+ if (direction === 'backward') {
24
+ const [record] = await (0, sql_1.requestSQL)(terminal, `select start_time from series_data_range where series_id = ${(0, sql_1.escapeSQL)(series_id)} and table_name = 'ohlc_v2' order by start_time limit 1`);
25
+ const start_time = record ? new Date(record.start_time).getTime() : Date.now();
26
+ req = {
27
+ product_id,
28
+ duration,
29
+ direction: 'backward',
30
+ time: start_time,
31
+ };
38
32
  }
39
- console.timeEnd('[SeriesCollector][Backwards] calc');
40
- return series_ids;
33
+ else {
34
+ // forward
35
+ req = {
36
+ product_id,
37
+ duration,
38
+ direction,
39
+ time: 0,
40
+ };
41
+ }
42
+ console.info((0, utils_1.formatTime)(Date.now()), '[SeriesCollector][OHLC][Backward]', 'Request', `product_id=${req.product_id}, duration=${req.duration}, direction=${req.direction}, time=${(0, utils_1.formatTime)(req.time)}`);
43
+ const res = await terminal.client.requestForResponseData('IngestOHLC', req);
44
+ ingestCounter.inc(res.wrote_count || 0);
45
+ console.info((0, utils_1.formatTime)(Date.now()), '[SeriesCollector][OHLC][Backward]', 'Response', `series_id=${series_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)}`);
41
46
  };
42
- (0, rxjs_1.defer)(async () => {
43
- const time = Date.now();
44
- const series_ids = await listBackwardSeriesIds();
45
- console.log(`[SeriesCollector][Backwards] Found ${series_ids.size} series to collect backwards data for. (${(0, utils_1.formatTime)(Date.now() - time)})`);
46
- await Promise.all([...series_ids].map(async (series_id) => {
47
- var _a, _b, _c, _d;
48
- try {
49
- const { product_id, duration } = (0, data_ohlc_1.decodeOHLCSeriesId)(series_id);
50
- const [datasource_id] = (0, utils_1.decodePath)(product_id);
51
- // 控制速率:每个数据源每秒钟只能请求一次
52
- await (0, utils_1.tokenBucket)(`backwards_target_${datasource_id}`, {
53
- refillInterval: 1000,
54
- capacity: 1,
55
- }).acquire();
56
- {
57
- const [record] = await (0, sql_1.requestSQL)(terminal, `select start_time from series_data_range where series_id = ${(0, sql_1.escapeSQL)(series_id)} and table_name = 'ohlc_v2' order by start_time limit 1`);
58
- const start_time = record ? new Date(record.start_time).getTime() : Date.now();
59
- const req = {
60
- product_id,
61
- duration,
62
- direction: 'backward',
63
- time: start_time,
64
- };
65
- console.info((0, utils_1.formatTime)(Date.now()), 'DispatchIngestOHLC', `product_id=${product_id}, duration=${duration}, time=${(0, utils_1.formatTime)(start_time)}`);
66
- const res = await terminal.client.requestForResponseData('IngestOHLC', req);
67
- terminal.metrics
68
- .counter('series_collector_backwards_ingest_count', '')
69
- .labels({ terminal_id: terminal.terminal_id, type: 'ohlc' })
70
- .inc(res.wrote_count || 0);
71
- console.info((0, utils_1.formatTime)(Date.now()), 'DispatchIngestOHLCResult', `series_id=${series_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)}`);
72
- }
73
- }
74
- catch (e) {
75
- console.info((0, utils_1.formatTime)(Date.now()), 'DispatchIngestOHLCError', `series_id=${series_id}`, e);
76
- }
77
- }));
78
- })
79
- .pipe((0, rxjs_1.retry)({ delay: 1000 }), (0, rxjs_1.repeat)({ delay: 1000 }))
80
- .subscribe();
47
+ exports.handleIngestOHLCBackward = handleIngestOHLCBackward;
81
48
  //# sourceMappingURL=backwards-ohlc.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"backwards-ohlc.js","sourceRoot":"","sources":["../../src/series-collector/backwards-ohlc.ts"],"names":[],"mappings":";AAAA,0BAA0B;AAC1B,6DAA6D;AAC7D,iDAAiD;AACjD,qCAAqC;;AAErC,iDAA2E;AAC3E,+CAI0B;AAC1B,+CAA4C;AAC5C,qCAAoD;AACpD,yCAAoE;AACpE,+BAA4C;AAE5C,MAAM,QAAQ,GAAG,mBAAQ,CAAC,WAAW,EAAE,CAAC;AAExC,MAAM,qBAAqB,GAAG,KAAK,IAAI,EAAE;IACvC,MAAM,WAAW,GAAG,MAAM,IAAA,gBAAU,EAA2B,QAAQ,EAAE,gCAAgC,CAAC,CAAC;IAE3G,OAAO,CAAC,IAAI,CAAC,mCAAmC,CAAC,CAAC;IAElD,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,IAAA,6CAAkC,EAAC,WAAW,CAAC,MAAM,CAAC,CAAC;gBACpE,IAAI,IAAI,CAAC,SAAS,KAAK,UAAU;oBAAE,SAAS;gBAE5C,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,IAAA,8BAAkB,EAAC,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,mCAAmC,CAAC,CAAC;IAErD,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC;AAEF,IAAA,YAAK,EAAC,KAAK,IAAI,EAAE;IACf,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACxB,MAAM,UAAU,GAAG,MAAM,qBAAqB,EAAE,CAAC;IACjD,OAAO,CAAC,GAAG,CACT,sCACE,UAAU,CAAC,IACb,2CAA2C,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,GAAG,CAC5E,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,IAAA,8BAAkB,EAAC,SAAS,CAAC,CAAC;YAC/D,MAAM,CAAC,aAAa,CAAC,GAAG,IAAA,kBAAU,EAAC,UAAU,CAAC,CAAC;YAC/C,sBAAsB;YACtB,MAAM,IAAA,mBAAW,EAAC,oBAAoB,aAAa,EAAE,EAAE;gBACrD,cAAc,EAAE,IAAI;gBACpB,QAAQ,EAAE,CAAC;aACZ,CAAC,CAAC,OAAO,EAAE,CAAC;YACb;gBACE,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,IAAA,gBAAU,EAK/B,QAAQ,EACR,8DAA8D,IAAA,eAAS,EACrE,SAAS,CACV,yDAAyD,CAC3D,CAAC;gBACF,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;gBAE/E,MAAM,GAAG,GAAuB;oBAC9B,UAAU;oBACV,QAAQ;oBACR,SAAS,EAAE,UAAU;oBACrB,IAAI,EAAE,UAAU;iBACjB,CAAC;gBAEF,OAAO,CAAC,IAAI,CACV,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,oBAAoB,EACpB,cAAc,UAAU,cAAc,QAAQ,UAAU,IAAA,kBAAU,EAAC,UAAU,CAAC,EAAE,CACjF,CAAC;gBAEF,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,sBAAsB,CACtD,YAAY,EACZ,GAAG,CACJ,CAAC;gBAEF,QAAQ,CAAC,OAAO;qBACb,OAAO,CAAC,yCAAyC,EAAE,EAAE,CAAC;qBACtD,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,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,0BAA0B,EAC1B,aAAa,SAAS,oBAAoB,GAAG,CAAC,WAAW,gBAAgB,IAAA,kBAAU,EACjF,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,UAAU,mCAAI,GAAG,CAC7B,cAAc,IAAA,kBAAU,EAAC,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,QAAQ,mCAAI,GAAG,CAAC,EAAE,CACxD,CAAC;aACH;SACF;QAAC,OAAO,CAAC,EAAE;YACV,OAAO,CAAC,IAAI,CAAC,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,yBAAyB,EAAE,aAAa,SAAS,EAAE,EAAE,CAAC,CAAC,CAAC;SAC9F;IACH,CAAC,CAAC,CACH,CAAC;AACJ,CAAC,CAAC;KACC,IAAI,CAAC,IAAA,YAAK,EAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,IAAA,aAAM,EAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;KACrD,SAAS,EAAE,CAAC","sourcesContent":["// 解决 Backwards 拉取历史数据的调度器\n// 该文件会定期扫描所有 Terminal 的 ServiceInfo,提取出所有支持 Backwards 拉取的序列。\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 listBackwardSeriesIds = async () => {\n const product_ids = await requestSQL<{ product_id: string }[]>(terminal, `select product_id from product`);\n\n console.time('[SeriesCollector][Backwards] 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 !== 'backward') 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][Backwards] calc');\n\n return series_ids;\n};\n\ndefer(async () => {\n const time = Date.now();\n const series_ids = await listBackwardSeriesIds();\n console.log(\n `[SeriesCollector][Backwards] Found ${\n series_ids.size\n } series to collect backwards 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(`backwards_target_${datasource_id}`, {\n refillInterval: 1000,\n capacity: 1,\n }).acquire();\n {\n const [record] = await requestSQL<\n {\n start_time: string;\n }[]\n >(\n terminal,\n `select start_time from series_data_range where series_id = ${escapeSQL(\n series_id,\n )} and table_name = 'ohlc_v2' order by start_time limit 1`,\n );\n const start_time = record ? new Date(record.start_time).getTime() : Date.now();\n\n const req: IIngestOHLCRequest = {\n product_id,\n duration,\n direction: 'backward',\n time: start_time,\n };\n\n console.info(\n formatTime(Date.now()),\n 'DispatchIngestOHLC',\n `product_id=${product_id}, duration=${duration}, time=${formatTime(start_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_backwards_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',\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', `series_id=${series_id}`, e);\n }\n }),\n );\n})\n .pipe(retry({ delay: 1000 }), repeat({ delay: 1000 }))\n .subscribe();\n"]}
1
+ {"version":3,"file":"backwards-ohlc.js","sourceRoot":"","sources":["../../src/series-collector/backwards-ohlc.ts"],"names":[],"mappings":";AAAA,0BAA0B;AAC1B,6DAA6D;AAC7D,iDAAiD;AACjD,qCAAqC;;;AAErC,iDAAuD;AAEvD,+CAA4C;AAC5C,qCAAoD;AACpD,yCAAoE;AAEpE,MAAM,QAAQ,GAAG,mBAAQ,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,UAAU,EAAE,CAAC,CAAC;AAE1E,MAAM,wBAAwB,GAAG,KAAK,EAC3C,SAAiB,EACjB,SAAiC,EACjC,MAAmB,EACnB,EAAE;;IACF,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,GAAG,IAAA,8BAAkB,EAAC,SAAS,CAAC,CAAC;IAC/D,MAAM,CAAC,aAAa,CAAC,GAAG,IAAA,kBAAU,EAAC,UAAU,CAAC,CAAC;IAC/C,sBAAsB;IACtB,MAAM,IAAA,mBAAW,EAAC,iBAAiB,aAAa,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;IAEvE,IAAI,GAAuB,CAAC;IAC5B,IAAI,SAAS,KAAK,UAAU,EAAE;QAC5B,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,IAAA,gBAAU,EAK/B,QAAQ,EACR,8DAA8D,IAAA,eAAS,EACrE,SAAS,CACV,yDAAyD,CAC3D,CAAC;QACF,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC;QAE/E,GAAG,GAAG;YACJ,UAAU;YACV,QAAQ;YACR,SAAS,EAAE,UAAU;YACrB,IAAI,EAAE,UAAU;SACjB,CAAC;KACH;SAAM;QACL,UAAU;QACV,GAAG,GAAG;YACJ,UAAU;YACV,QAAQ;YACR,SAAS;YACT,IAAI,EAAE,CAAC;SACR,CAAC;KACH;IAED,OAAO,CAAC,IAAI,CACV,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,mCAAmC,EACnC,SAAS,EACT,cAAc,GAAG,CAAC,UAAU,cAAc,GAAG,CAAC,QAAQ,eAAe,GAAG,CAAC,SAAS,UAAU,IAAA,kBAAU,EACpG,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,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,mCAAmC,EACnC,UAAU,EACV,aAAa,SAAS,oBAAoB,GAAG,CAAC,WAAW,gBAAgB,IAAA,kBAAU,EACjF,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,UAAU,mCAAI,GAAG,CAC7B,cAAc,IAAA,kBAAU,EAAC,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,QAAQ,mCAAI,GAAG,CAAC,EAAE,CACxD,CAAC;AACJ,CAAC,CAAC;AAhEW,QAAA,wBAAwB,4BAgEnC","sourcesContent":["// 解决 Backwards 拉取历史数据的调度器\n// 该文件会定期扫描所有 Terminal 的 ServiceInfo,提取出所有支持 Backwards 拉取的序列。\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: 'backward' });\n\nexport const handleIngestOHLCBackward = 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:backward:${datasource_id}`).acquire(1, signal);\n\n let req: IIngestOHLCRequest;\n if (direction === 'backward') {\n const [record] = await requestSQL<\n {\n start_time: string;\n }[]\n >(\n terminal,\n `select start_time from series_data_range where series_id = ${escapeSQL(\n series_id,\n )} and table_name = 'ohlc_v2' order by start_time limit 1`,\n );\n const start_time = record ? new Date(record.start_time).getTime() : Date.now();\n\n req = {\n product_id,\n duration,\n direction: 'backward',\n time: start_time,\n };\n } else {\n // forward\n req = {\n product_id,\n duration,\n direction,\n time: 0,\n };\n }\n\n console.info(\n formatTime(Date.now()),\n '[SeriesCollector][OHLC][Backward]',\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][Backward]',\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"]}
@@ -0,0 +1,12 @@
1
+ import { IInterestRateServiceMetadata } from '@yuants/exchange';
2
+ /**
3
+ * 列出所有支持利率的品种系列ID以及对应的 Service Metadata
4
+ * @returns
5
+ */
6
+ export declare const listInterestRateSeriesIds: () => Promise<Map<string, IInterestRateServiceMetadata>>;
7
+ /**
8
+ *
9
+ * @returns
10
+ */
11
+ export declare const listOHLCSeriesIds: () => Promise<Map<string, "backward" | "forward">>;
12
+ //# sourceMappingURL=discovery.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"discovery.d.ts","sourceRoot":"","sources":["../../src/series-collector/discovery.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,4BAA4B,EAG7B,MAAM,kBAAkB,CAAC;AAM1B;;;GAGG;AACH,eAAO,MAAM,yBAAyB,0DAwBrC,CAAC;AAEF;;;GAGG;AACH,eAAO,MAAM,iBAAiB,oDAuB7B,CAAC"}
@@ -0,0 +1,67 @@
1
+ "use strict";
2
+ // 发现所有支持利率的品种系列ID
3
+ Object.defineProperty(exports, "__esModule", { value: true });
4
+ exports.listOHLCSeriesIds = exports.listInterestRateSeriesIds = void 0;
5
+ const data_ohlc_1 = require("@yuants/data-ohlc");
6
+ const exchange_1 = require("@yuants/exchange");
7
+ const protocol_1 = require("@yuants/protocol");
8
+ const sql_1 = require("@yuants/sql");
9
+ const terminal = protocol_1.Terminal.fromNodeEnv();
10
+ /**
11
+ * 列出所有支持利率的品种系列ID以及对应的 Service Metadata
12
+ * @returns
13
+ */
14
+ const listInterestRateSeriesIds = async () => {
15
+ const product_ids = await (0, sql_1.requestSQL)(terminal,
16
+ // 必须是支持利率的品种
17
+ `select product_id from product where no_interest_rate = false`);
18
+ const series_ids = new Map();
19
+ for (const terminalInfo of terminal.terminalInfos) {
20
+ for (const serviceInfo of Object.values(terminalInfo.serviceInfo || {})) {
21
+ if (serviceInfo.method !== 'IngestInterestRate')
22
+ continue;
23
+ try {
24
+ const meta = (0, exchange_1.parseInterestRateServiceMetadataFromSchema)(serviceInfo.schema);
25
+ for (const { product_id } of product_ids) {
26
+ if (!product_id.startsWith(meta.product_id_prefix))
27
+ continue;
28
+ series_ids.set(product_id, meta);
29
+ }
30
+ }
31
+ finally {
32
+ }
33
+ }
34
+ }
35
+ return series_ids;
36
+ };
37
+ exports.listInterestRateSeriesIds = listInterestRateSeriesIds;
38
+ /**
39
+ *
40
+ * @returns
41
+ */
42
+ const listOHLCSeriesIds = async () => {
43
+ const product_ids = await (0, sql_1.requestSQL)(terminal, `select product_id from product`);
44
+ const series_ids = new Map();
45
+ for (const terminalInfo of terminal.terminalInfos) {
46
+ for (const serviceInfo of Object.values(terminalInfo.serviceInfo || {})) {
47
+ if (serviceInfo.method !== 'IngestOHLC')
48
+ continue;
49
+ try {
50
+ const meta = (0, exchange_1.parseOHLCServiceMetadataFromSchema)(serviceInfo.schema);
51
+ for (const { product_id } of product_ids) {
52
+ if (!product_id.startsWith(meta.product_id_prefix))
53
+ continue;
54
+ for (const duration of meta.duration_list) {
55
+ const series_id = (0, data_ohlc_1.encodeOHLCSeriesId)(product_id, duration);
56
+ series_ids.set(series_id, meta.direction);
57
+ }
58
+ }
59
+ }
60
+ finally {
61
+ }
62
+ }
63
+ }
64
+ return series_ids;
65
+ };
66
+ exports.listOHLCSeriesIds = listOHLCSeriesIds;
67
+ //# sourceMappingURL=discovery.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"discovery.js","sourceRoot":"","sources":["../../src/series-collector/discovery.ts"],"names":[],"mappings":";AAAA,kBAAkB;;;AAElB,iDAAuD;AACvD,+CAI0B;AAC1B,+CAA4C;AAC5C,qCAAyC;AAEzC,MAAM,QAAQ,GAAG,mBAAQ,CAAC,WAAW,EAAE,CAAC;AAExC;;;GAGG;AACI,MAAM,yBAAyB,GAAG,KAAK,IAAI,EAAE;IAClD,MAAM,WAAW,GAAG,MAAM,IAAA,gBAAU,EAClC,QAAQ;IACR,aAAa;IACb,+DAA+D,CAChE,CAAC;IAEF,MAAM,UAAU,GAAG,IAAI,GAAG,EAAwC,CAAC;IACnE,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,oBAAoB;gBAAE,SAAS;YAC1D,IAAI;gBACF,MAAM,IAAI,GAAG,IAAA,qDAA0C,EAAC,WAAW,CAAC,MAAM,CAAC,CAAC;gBAE5E,KAAK,MAAM,EAAE,UAAU,EAAE,IAAI,WAAW,EAAE;oBACxC,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,iBAAiB,CAAC;wBAAE,SAAS;oBAC7D,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;iBAClC;aACF;oBAAS;aACT;SACF;KACF;IAED,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC;AAxBW,QAAA,yBAAyB,6BAwBpC;AAEF;;;GAGG;AACI,MAAM,iBAAiB,GAAG,KAAK,IAAI,EAAE;IAC1C,MAAM,WAAW,GAAG,MAAM,IAAA,gBAAU,EAA2B,QAAQ,EAAE,gCAAgC,CAAC,CAAC;IAE3G,MAAM,UAAU,GAAG,IAAI,GAAG,EAAkC,CAAC;IAC7D,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,IAAA,6CAAkC,EAAC,WAAW,CAAC,MAAM,CAAC,CAAC;gBAEpE,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,IAAA,8BAAkB,EAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;wBAC3D,UAAU,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;qBAC3C;iBACF;aACF;oBAAS;aACT;SACF;KACF;IAED,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC;AAvBW,QAAA,iBAAiB,qBAuB5B","sourcesContent":["// 发现所有支持利率的品种系列ID\n\nimport { encodeOHLCSeriesId } from '@yuants/data-ohlc';\nimport {\n IInterestRateServiceMetadata,\n parseInterestRateServiceMetadataFromSchema,\n parseOHLCServiceMetadataFromSchema,\n} from '@yuants/exchange';\nimport { Terminal } from '@yuants/protocol';\nimport { requestSQL } from '@yuants/sql';\n\nconst terminal = Terminal.fromNodeEnv();\n\n/**\n * 列出所有支持利率的品种系列ID以及对应的 Service Metadata\n * @returns\n */\nexport const listInterestRateSeriesIds = async () => {\n const product_ids = await requestSQL<{ product_id: string }[]>(\n terminal,\n // 必须是支持利率的品种\n `select product_id from product where no_interest_rate = false`,\n );\n\n const series_ids = new Map<string, IInterestRateServiceMetadata>();\n for (const terminalInfo of terminal.terminalInfos) {\n for (const serviceInfo of Object.values(terminalInfo.serviceInfo || {})) {\n if (serviceInfo.method !== 'IngestInterestRate') continue;\n try {\n const meta = parseInterestRateServiceMetadataFromSchema(serviceInfo.schema);\n\n for (const { product_id } of product_ids) {\n if (!product_id.startsWith(meta.product_id_prefix)) continue;\n series_ids.set(product_id, meta);\n }\n } finally {\n }\n }\n }\n\n return series_ids;\n};\n\n/**\n *\n * @returns\n */\nexport const listOHLCSeriesIds = async () => {\n const product_ids = await requestSQL<{ product_id: string }[]>(terminal, `select product_id from product`);\n\n const series_ids = new Map<string, 'forward' | 'backward'>();\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\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.set(series_id, meta.direction);\n }\n }\n } finally {\n }\n }\n }\n\n return series_ids;\n};\n"]}
@@ -1,2 +1,4 @@
1
- export {};
1
+ /// <reference types="node" />
2
+ import { IInterestRateServiceMetadata } from '@yuants/exchange';
3
+ export declare const handleIngestInterestRateForward: (product_id: string, meta: IInterestRateServiceMetadata, signal: AbortSignal) => Promise<void>;
2
4
  //# sourceMappingURL=forwards-interest-rate.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"forwards-interest-rate.d.ts","sourceRoot":"","sources":["../../src/series-collector/forwards-interest-rate.ts"],"names":[],"mappings":""}
1
+ {"version":3,"file":"forwards-interest-rate.d.ts","sourceRoot":"","sources":["../../src/series-collector/forwards-interest-rate.ts"],"names":[],"mappings":";AAKA,OAAO,EAEL,4BAA4B,EAE7B,MAAM,kBAAkB,CAAC;AAW1B,eAAO,MAAM,+BAA+B,eAC9B,MAAM,QACZ,4BAA4B,UAC1B,WAAW,kBAgDpB,CAAC"}
@@ -4,80 +4,43 @@
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.handleIngestInterestRateForward = 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][Forwards] calc');
16
- const series_ids = new Map();
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
- for (const { product_id } of product_ids) {
24
- if (!product_id.startsWith(meta.product_id_prefix))
25
- continue;
26
- series_ids.set(product_id, meta);
27
- }
28
- }
29
- finally {
30
- }
12
+ const ingestCounter = terminal.metrics
13
+ .counter('series_collector_ingest_count', '')
14
+ .labels({ terminal_id: terminal.terminal_id, type: 'interest_rate', task: 'forward' });
15
+ const handleIngestInterestRateForward = 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:forwards:${datasource_id}`).acquire(1, signal);
20
+ {
21
+ let req;
22
+ if (meta.direction === 'forward') {
23
+ const endTime = await (0, sql_helpers_1.findInterestRateEndTimeForward)(terminal, product_id);
24
+ const time = endTime ? new Date(endTime).getTime() : 0;
25
+ req = {
26
+ product_id: product_id,
27
+ direction: 'forward',
28
+ time,
29
+ };
31
30
  }
31
+ else {
32
+ // backward
33
+ req = {
34
+ product_id,
35
+ direction: 'backward',
36
+ time: Date.now(),
37
+ };
38
+ }
39
+ console.info((0, utils_1.formatTime)(Date.now()), '[SeriesCollector][InterestRate][Forward]', 'Request', `product_id=${req.product_id}, direction=${req.direction}, time=${(0, utils_1.formatTime)(req.time)}`);
40
+ const res = await terminal.client.requestForResponseData('IngestInterestRate', req);
41
+ ingestCounter.inc(res.wrote_count || 0);
42
+ console.info((0, utils_1.formatTime)(Date.now()), '[SeriesCollector][InterestRate][Forward]', '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)}`);
32
43
  }
33
- console.timeEnd('[SeriesCollector][InterestRate][Forwards] calc');
34
- return series_ids;
35
44
  };
36
- (0, rxjs_1.defer)(async () => {
37
- const series_ids = await listBackwardSeriesIds();
38
- console.log(`[SeriesCollector][InterestRate][Forwards] Found ${series_ids.size} series to collect forwards data for. `);
39
- await Promise.all([...series_ids].map(async ([product_id, meta]) => {
40
- var _a, _b, _c, _d;
41
- try {
42
- const [datasource_id] = (0, utils_1.decodePath)(product_id);
43
- // 控制速率:每个数据源每秒钟只能请求一次
44
- await (0, utils_1.tokenBucket)(`interest_rate_forwards:${datasource_id}`, {
45
- refillInterval: 1000,
46
- capacity: 1,
47
- }).acquire();
48
- {
49
- let req;
50
- if (meta.direction === 'forward') {
51
- const [record] = await (0, sql_1.requestSQL)(terminal, `select end_time from series_data_range where series_id = ${(0, sql_1.escapeSQL)(product_id)} and table_name = 'interest_rate' order by end_time desc limit 1`);
52
- const time = record ? new Date(record.end_time).getTime() : 0;
53
- req = {
54
- product_id: product_id,
55
- direction: 'forward',
56
- time,
57
- };
58
- }
59
- else {
60
- // backward
61
- req = {
62
- product_id,
63
- direction: 'backward',
64
- time: Date.now(),
65
- };
66
- }
67
- console.info((0, utils_1.formatTime)(Date.now()), 'DispatchIngestInterestRateRequestForwardTask', `product_id=${req.product_id}, direction=${req.direction}, time=${(0, utils_1.formatTime)(req.time)}`);
68
- const res = await terminal.client.requestForResponseData('IngestInterestRate', req);
69
- terminal.metrics
70
- .counter('series_collector_forwards_ingest_count', '')
71
- .labels({ terminal_id: terminal.terminal_id, type: 'interest_rate' })
72
- .inc(res.wrote_count || 0);
73
- console.info((0, utils_1.formatTime)(Date.now()), 'DispatchIngestInterestRateRequestForwardTaskResult', `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)}`);
74
- }
75
- }
76
- catch (e) {
77
- console.info((0, utils_1.formatTime)(Date.now()), 'DispatchIngestInterestRateErrorForwards', `series_id=${product_id}`, e);
78
- }
79
- }));
80
- })
81
- .pipe((0, rxjs_1.retry)({ delay: 1000 }), (0, rxjs_1.repeat)({ delay: 1000 }))
82
- .subscribe();
45
+ exports.handleIngestInterestRateForward = handleIngestInterestRateForward;
83
46
  //# sourceMappingURL=forwards-interest-rate.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"forwards-interest-rate.js","sourceRoot":"","sources":["../../src/series-collector/forwards-interest-rate.ts"],"names":[],"mappings":";AAAA,yBAAyB;AACzB,4DAA4D;AAC5D,yDAAyD;AACzD,qCAAqC;;AAErC,+CAK0B;AAC1B,+CAA4C;AAC5C,qCAAoD;AACpD,yCAAoE;AACpE,+BAA4C;AAE5C,MAAM,QAAQ,GAAG,mBAAQ,CAAC,WAAW,EAAE,CAAC;AAExC,MAAM,qBAAqB,GAAG,KAAK,IAAI,EAAE;IACvC,MAAM,WAAW,GAAG,MAAM,IAAA,gBAAU,EAA2B,QAAQ,EAAE,gCAAgC,CAAC,CAAC;IAE3G,OAAO,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAC;IAE/D,MAAM,UAAU,GAAG,IAAI,GAAG,EAAwC,CAAC;IACnE,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,oBAAoB;gBAAE,SAAS;YAC1D,IAAI;gBACF,MAAM,IAAI,GAAG,IAAA,qDAA0C,EAAC,WAAW,CAAC,MAAM,CAAC,CAAC;gBAE5E,KAAK,MAAM,EAAE,UAAU,EAAE,IAAI,WAAW,EAAE;oBACxC,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,iBAAiB,CAAC;wBAAE,SAAS;oBAC7D,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;iBAClC;aACF;oBAAS;aACT;SACF;KACF;IAED,OAAO,CAAC,OAAO,CAAC,gDAAgD,CAAC,CAAC;IAElE,OAAO,UAAU,CAAC;AACpB,CAAC,CAAC;AAEF,IAAA,YAAK,EAAC,KAAK,IAAI,EAAE;IACf,MAAM,UAAU,GAAG,MAAM,qBAAqB,EAAE,CAAC;IACjD,OAAO,CAAC,GAAG,CACT,mDAAmD,UAAU,CAAC,IAAI,wCAAwC,CAC3G,CAAC;IAEF,MAAM,OAAO,CAAC,GAAG,CACf,CAAC,GAAG,UAAU,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;;QAC/C,IAAI;YACF,MAAM,CAAC,aAAa,CAAC,GAAG,IAAA,kBAAU,EAAC,UAAU,CAAC,CAAC;YAC/C,sBAAsB;YACtB,MAAM,IAAA,mBAAW,EAAC,0BAA0B,aAAa,EAAE,EAAE;gBAC3D,cAAc,EAAE,IAAI;gBACpB,QAAQ,EAAE,CAAC;aACZ,CAAC,CAAC,OAAO,EAAE,CAAC;YACb;gBACE,IAAI,GAA+B,CAAC;gBACpC,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS,EAAE;oBAChC,MAAM,CAAC,MAAM,CAAC,GAAG,MAAM,IAAA,gBAAU,EAK/B,QAAQ,EACR,4DAA4D,IAAA,eAAS,EACnE,UAAU,CACX,kEAAkE,CACpE,CAAC;oBACF,MAAM,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;oBAE9D,GAAG,GAAG;wBACJ,UAAU,EAAE,UAAU;wBACtB,SAAS,EAAE,SAAS;wBACpB,IAAI;qBACL,CAAC;iBACH;qBAAM;oBACL,WAAW;oBACX,GAAG,GAAG;wBACJ,UAAU;wBACV,SAAS,EAAE,UAAU;wBACrB,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE;qBACjB,CAAC;iBACH;gBAED,OAAO,CAAC,IAAI,CACV,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,8CAA8C,EAC9C,cAAc,GAAG,CAAC,UAAU,eAAe,GAAG,CAAC,SAAS,UAAU,IAAA,kBAAU,EAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CACzF,CAAC;gBACF,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,sBAAsB,CAGtD,oBAAoB,EAAE,GAAG,CAAC,CAAC;gBAE7B,QAAQ,CAAC,OAAO;qBACb,OAAO,CAAC,wCAAwC,EAAE,EAAE,CAAC;qBACrD,MAAM,CAAC,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC;qBACpE,GAAG,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;gBAE7B,OAAO,CAAC,IAAI,CACV,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,oDAAoD,EACpD,aAAa,UAAU,oBAAoB,GAAG,CAAC,WAAW,gBAAgB,IAAA,kBAAU,EAClF,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,UAAU,mCAAI,GAAG,CAC7B,cAAc,IAAA,kBAAU,EAAC,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,QAAQ,mCAAI,GAAG,CAAC,EAAE,CACxD,CAAC;aACH;SACF;QAAC,OAAO,CAAC,EAAE;YACV,OAAO,CAAC,IAAI,CACV,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,yCAAyC,EACzC,aAAa,UAAU,EAAE,EACzB,CAAC,CACF,CAAC;SACH;IACH,CAAC,CAAC,CACH,CAAC;AACJ,CAAC,CAAC;KACC,IAAI,CAAC,IAAA,YAAK,EAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,IAAA,aAAM,EAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;KACrD,SAAS,EAAE,CAAC","sourcesContent":["// 解决 Forwards 拉取历史数据的调度器\n// 该文件会定期扫描所有 Terminal 的 ServiceInfo,提取出所有支持 Forwards 拉取的序列。\n// 然后对每个序列,向对应的 IngestInterestRate Service 发送拉取请求,补齐历史数据。\n// 使用 Token Bucket 控制每个数据源的请求速率,避免过载。\n\nimport {\n IIngestInterestRateRequest,\n IInterestRateServiceMetadata,\n ISeriesIngestResult,\n parseInterestRateServiceMetadataFromSchema,\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 listBackwardSeriesIds = async () => {\n const product_ids = await requestSQL<{ product_id: string }[]>(terminal, `select product_id from product`);\n\n console.time('[SeriesCollector][InterestRate][Forwards] calc');\n\n const series_ids = new Map<string, IInterestRateServiceMetadata>();\n for (const terminalInfo of terminal.terminalInfos) {\n for (const serviceInfo of Object.values(terminalInfo.serviceInfo || {})) {\n if (serviceInfo.method !== 'IngestInterestRate') continue;\n try {\n const meta = parseInterestRateServiceMetadataFromSchema(serviceInfo.schema);\n\n for (const { product_id } of product_ids) {\n if (!product_id.startsWith(meta.product_id_prefix)) continue;\n series_ids.set(product_id, meta);\n }\n } finally {\n }\n }\n }\n\n console.timeEnd('[SeriesCollector][InterestRate][Forwards] calc');\n\n return series_ids;\n};\n\ndefer(async () => {\n const series_ids = await listBackwardSeriesIds();\n console.log(\n `[SeriesCollector][InterestRate][Forwards] Found ${series_ids.size} series to collect forwards data for. `,\n );\n\n await Promise.all(\n [...series_ids].map(async ([product_id, meta]) => {\n try {\n const [datasource_id] = decodePath(product_id);\n // 控制速率:每个数据源每秒钟只能请求一次\n await tokenBucket(`interest_rate_forwards:${datasource_id}`, {\n refillInterval: 1000,\n capacity: 1,\n }).acquire();\n {\n let req: IIngestInterestRateRequest;\n if (meta.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 product_id,\n )} and table_name = 'interest_rate' order by end_time desc limit 1`,\n );\n const time = record ? new Date(record.end_time).getTime() : 0;\n\n req = {\n product_id: product_id,\n direction: 'forward',\n time,\n };\n } else {\n // backward\n req = {\n product_id,\n direction: 'backward',\n time: Date.now(),\n };\n }\n\n console.info(\n formatTime(Date.now()),\n 'DispatchIngestInterestRateRequestForwardTask',\n `product_id=${req.product_id}, direction=${req.direction}, time=${formatTime(req.time)}`,\n );\n const res = await terminal.client.requestForResponseData<\n IIngestInterestRateRequest,\n ISeriesIngestResult\n >('IngestInterestRate', req);\n\n terminal.metrics\n .counter('series_collector_forwards_ingest_count', '')\n .labels({ terminal_id: terminal.terminal_id, type: 'interest_rate' })\n .inc(res.wrote_count || 0);\n\n console.info(\n formatTime(Date.now()),\n 'DispatchIngestInterestRateRequestForwardTaskResult',\n `series_id=${product_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(\n formatTime(Date.now()),\n 'DispatchIngestInterestRateErrorForwards',\n `series_id=${product_id}`,\n e,\n );\n }\n }),\n );\n})\n .pipe(retry({ delay: 1000 }), repeat({ delay: 1000 }))\n .subscribe();\n"]}
1
+ {"version":3,"file":"forwards-interest-rate.js","sourceRoot":"","sources":["../../src/series-collector/forwards-interest-rate.ts"],"names":[],"mappings":";AAAA,yBAAyB;AACzB,4DAA4D;AAC5D,yDAAyD;AACzD,qCAAqC;;;AAOrC,+CAA4C;AAC5C,yCAAoE;AACpE,+CAA+D;AAE/D,MAAM,QAAQ,GAAG,mBAAQ,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;AAElF,MAAM,+BAA+B,GAAG,KAAK,EAClD,UAAkB,EAClB,IAAkC,EAClC,MAAmB,EACnB,EAAE;;IACF,MAAM,CAAC,aAAa,CAAC,GAAG,IAAA,kBAAU,EAAC,UAAU,CAAC,CAAC;IAC/C,sBAAsB;IACtB,MAAM,IAAA,mBAAW,EAAC,0BAA0B,aAAa,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,CAAC,CAAC;IAEhF;QACE,IAAI,GAA+B,CAAC;QACpC,IAAI,IAAI,CAAC,SAAS,KAAK,SAAS,EAAE;YAChC,MAAM,OAAO,GAAG,MAAM,IAAA,4CAA8B,EAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YAC3E,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAEvD,GAAG,GAAG;gBACJ,UAAU,EAAE,UAAU;gBACtB,SAAS,EAAE,SAAS;gBACpB,IAAI;aACL,CAAC;SACH;aAAM;YACL,WAAW;YACX,GAAG,GAAG;gBACJ,UAAU;gBACV,SAAS,EAAE,UAAU;gBACrB,IAAI,EAAE,IAAI,CAAC,GAAG,EAAE;aACjB,CAAC;SACH;QAED,OAAO,CAAC,IAAI,CACV,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,0CAA0C,EAC1C,SAAS,EACT,cAAc,GAAG,CAAC,UAAU,eAAe,GAAG,CAAC,SAAS,UAAU,IAAA,kBAAU,EAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CACzF,CAAC;QACF,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,sBAAsB,CACtD,oBAAoB,EACpB,GAAG,CACJ,CAAC;QAEF,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,IAAI,CAAC,CAAC,CAAC;QAExC,OAAO,CAAC,IAAI,CACV,IAAA,kBAAU,EAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EACtB,0CAA0C,EAC1C,QAAQ,EACR,aAAa,UAAU,oBAAoB,GAAG,CAAC,WAAW,gBAAgB,IAAA,kBAAU,EAClF,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,UAAU,mCAAI,GAAG,CAC7B,cAAc,IAAA,kBAAU,EAAC,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,QAAQ,mCAAI,GAAG,CAAC,EAAE,CACxD,CAAC;KACH;AACH,CAAC,CAAC;AAnDW,QAAA,+BAA+B,mCAmD1C","sourcesContent":["// 解决 Forwards 拉取历史数据的调度器\n// 该文件会定期扫描所有 Terminal 的 ServiceInfo,提取出所有支持 Forwards 拉取的序列。\n// 然后对每个序列,向对应的 IngestInterestRate Service 发送拉取请求,补齐历史数据。\n// 使用 Token Bucket 控制每个数据源的请求速率,避免过载。\n\nimport {\n IIngestInterestRateRequest,\n IInterestRateServiceMetadata,\n ISeriesIngestResult,\n} from '@yuants/exchange';\nimport { Terminal } from '@yuants/protocol';\nimport { decodePath, formatTime, tokenBucket } from '@yuants/utils';\nimport { findInterestRateEndTimeForward } from './sql-helpers';\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\nexport const handleIngestInterestRateForward = async (\n product_id: string,\n meta: IInterestRateServiceMetadata,\n signal: AbortSignal,\n) => {\n const [datasource_id] = decodePath(product_id);\n // 控制速率:每个数据源每秒钟只能请求一次\n await tokenBucket(`interest_rate:forwards:${datasource_id}`).acquire(1, signal);\n\n {\n let req: IIngestInterestRateRequest;\n if (meta.direction === 'forward') {\n const endTime = await findInterestRateEndTimeForward(terminal, product_id);\n const time = endTime ? new Date(endTime).getTime() : 0;\n\n req = {\n product_id: product_id,\n direction: 'forward',\n time,\n };\n } else {\n // backward\n req = {\n product_id,\n direction: 'backward',\n time: Date.now(),\n };\n }\n\n console.info(\n formatTime(Date.now()),\n '[SeriesCollector][InterestRate][Forward]',\n 'Request',\n `product_id=${req.product_id}, direction=${req.direction}, time=${formatTime(req.time)}`,\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][Forward]',\n 'Result',\n `series_id=${product_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};\n"]}
@@ -1,2 +1,3 @@
1
- export {};
1
+ /// <reference types="node" />
2
+ export declare const handleIngestOHLCForward: (series_id: string, direction: 'forward' | 'backward', signal: AbortSignal) => Promise<void>;
2
3
  //# sourceMappingURL=forwards-ohlc.d.ts.map
@@ -1 +1 @@
1
- {"version":3,"file":"forwards-ohlc.d.ts","sourceRoot":"","sources":["../../src/series-collector/forwards-ohlc.ts"],"names":[],"mappings":""}
1
+ {"version":3,"file":"forwards-ohlc.d.ts","sourceRoot":"","sources":["../../src/series-collector/forwards-ohlc.ts"],"names":[],"mappings":";AAiBA,eAAO,MAAM,uBAAuB,cACvB,MAAM,aACN,SAAS,GAAG,UAAU,UACzB,WAAW,kBA6DpB,CAAC"}
@@ -4,78 +4,45 @@
4
4
  // 然后对每个序列,向对应的 IngestOHLC Service 发送拉取请求,补齐历史数据。
5
5
  // 使用 Token Bucket 控制每个数据源的请求速率,避免过载。
6
6
  Object.defineProperty(exports, "__esModule", { value: true });
7
+ exports.handleIngestOHLCForward = void 0;
7
8
  const data_ohlc_1 = require("@yuants/data-ohlc");
8
- const exchange_1 = require("@yuants/exchange");
9
9
  const protocol_1 = require("@yuants/protocol");
10
10
  const sql_1 = require("@yuants/sql");
11
11
  const utils_1 = require("@yuants/utils");
12
- const rxjs_1 = require("rxjs");
13
12
  const terminal = protocol_1.Terminal.fromNodeEnv();
14
- const listForwardSeriesIds = async () => {
15
- const product_ids = await (0, sql_1.requestSQL)(terminal, `select product_id from product`);
16
- console.time('[SeriesCollector][OHLC][Forwards] calc');
17
- const series_ids = new Set();
18
- for (const terminalInfo of terminal.terminalInfos) {
19
- for (const serviceInfo of Object.values(terminalInfo.serviceInfo || {})) {
20
- if (serviceInfo.method !== 'IngestOHLC')
21
- continue;
22
- try {
23
- const meta = (0, exchange_1.parseOHLCServiceMetadataFromSchema)(serviceInfo.schema);
24
- if (meta.direction !== 'forward')
25
- continue;
26
- for (const { product_id } of product_ids) {
27
- if (!product_id.startsWith(meta.product_id_prefix))
28
- continue;
29
- for (const duration of meta.duration_list) {
30
- const series_id = (0, data_ohlc_1.encodeOHLCSeriesId)(product_id, duration);
31
- series_ids.add(series_id);
32
- }
33
- }
34
- }
35
- finally {
36
- }
37
- }
13
+ const ingestCounter = terminal.metrics
14
+ .counter('series_collector_ingest_count', '')
15
+ .labels({ terminal_id: terminal.terminal_id, type: 'ohlc', task: 'forward' });
16
+ const handleIngestOHLCForward = async (series_id, direction, signal) => {
17
+ var _a, _b, _c, _d;
18
+ const { product_id, duration } = (0, data_ohlc_1.decodeOHLCSeriesId)(series_id);
19
+ const [datasource_id] = (0, utils_1.decodePath)(product_id);
20
+ // 控制速率:每个数据源每秒钟只能请求一次
21
+ await (0, utils_1.tokenBucket)(`ohlc:forward:${datasource_id}`).acquire();
22
+ let req;
23
+ if (direction === 'forward') {
24
+ const [record] = await (0, sql_1.requestSQL)(terminal, `select end_time from series_data_range where series_id = ${(0, sql_1.escapeSQL)(series_id)} and table_name = 'ohlc_v2' order by end_time desc limit 1`);
25
+ const time = record ? new Date(record.end_time).getTime() : 0;
26
+ req = {
27
+ product_id,
28
+ duration,
29
+ direction: 'forward',
30
+ time,
31
+ };
38
32
  }
39
- console.timeEnd('[SeriesCollector][OHLC][Forwards] calc');
40
- return series_ids;
33
+ else {
34
+ // backward
35
+ req = {
36
+ product_id,
37
+ duration,
38
+ direction,
39
+ time: Date.now(),
40
+ };
41
+ }
42
+ console.info((0, utils_1.formatTime)(Date.now()), '[SeriesCollector][OHLC][Forward]', 'Request', `product_id=${req.product_id}, duration=${req.duration}, direction=${req.direction}, time=${(0, utils_1.formatTime)(req.time)}`);
43
+ const res = await terminal.client.requestForResponseData('IngestOHLC', req);
44
+ ingestCounter.inc(res.wrote_count || 0);
45
+ console.info((0, utils_1.formatTime)(Date.now()), '[SeriesCollector][OHLC][Forward]', 'Response', `series_id=${series_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)}`);
41
46
  };
42
- (0, rxjs_1.defer)(async () => {
43
- const time = Date.now();
44
- const series_ids = await listForwardSeriesIds();
45
- console.log(`[SeriesCollector][OHLC][Forwards] Found ${series_ids.size} series to collect forwards data for. (${(0, utils_1.formatTime)(Date.now() - time)})`);
46
- await Promise.all([...series_ids].map(async (series_id) => {
47
- var _a, _b, _c, _d;
48
- try {
49
- const { product_id, duration } = (0, data_ohlc_1.decodeOHLCSeriesId)(series_id);
50
- const [datasource_id] = (0, utils_1.decodePath)(product_id);
51
- // 控制速率:每个数据源每秒钟只能请求一次
52
- await (0, utils_1.tokenBucket)(`ohlc_forwards_target_${datasource_id}`, {
53
- refillInterval: 1000,
54
- capacity: 1,
55
- }).acquire();
56
- {
57
- const [record] = await (0, sql_1.requestSQL)(terminal, `select end_time from series_data_range where series_id = ${(0, sql_1.escapeSQL)(series_id)} and table_name = 'ohlc_v2' order by end_time desc limit 1`);
58
- const time = record ? new Date(record.end_time).getTime() : 0;
59
- const req = {
60
- product_id,
61
- duration,
62
- direction: 'forward',
63
- time,
64
- };
65
- console.info((0, utils_1.formatTime)(Date.now()), 'DispatchIngestOHLC_Forwards', `product_id=${product_id}, duration=${duration}, time=${(0, utils_1.formatTime)(time)}`);
66
- const res = await terminal.client.requestForResponseData('IngestOHLC', req);
67
- terminal.metrics
68
- .counter('series_collector_forwards_ingest_count', '')
69
- .labels({ terminal_id: terminal.terminal_id, type: 'ohlc' })
70
- .inc(res.wrote_count || 0);
71
- console.info((0, utils_1.formatTime)(Date.now()), 'DispatchIngestOHLCResult_Forwards', `series_id=${series_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)}`);
72
- }
73
- }
74
- catch (e) {
75
- console.info((0, utils_1.formatTime)(Date.now()), 'DispatchIngestOHLCError_Forwards', `series_id=${series_id}`, e);
76
- }
77
- }));
78
- })
79
- .pipe((0, rxjs_1.retry)({ delay: 1000 }), (0, rxjs_1.repeat)({ delay: 1000 }))
80
- .subscribe();
47
+ exports.handleIngestOHLCForward = handleIngestOHLCForward;
81
48
  //# sourceMappingURL=forwards-ohlc.js.map