@filoz/repair-cli 0.1.1 → 0.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/package.json +3 -2
- package/dist/src/cli.js +2 -0
- package/dist/src/cli.js.map +1 -1
- package/dist/src/commands/datasets.d.ts +16 -1
- package/dist/src/commands/datasets.d.ts.map +1 -1
- package/dist/src/commands/datasets.js +107 -3
- package/dist/src/commands/datasets.js.map +1 -1
- package/dist/src/commands/providers.d.ts +16 -1
- package/dist/src/commands/providers.d.ts.map +1 -1
- package/dist/src/commands/providers.js.map +1 -1
- package/dist/src/commands/repair.d.ts +16 -1
- package/dist/src/commands/repair.d.ts.map +1 -1
- package/dist/src/commands/repair.js +7 -8
- package/dist/src/commands/repair.js.map +1 -1
- package/dist/src/commands/replicate.d.ts +24 -0
- package/dist/src/commands/replicate.d.ts.map +1 -0
- package/dist/src/commands/replicate.js +171 -0
- package/dist/src/commands/replicate.js.map +1 -0
- package/dist/src/commands/setup.d.ts.map +1 -1
- package/dist/src/commands/setup.js +15 -0
- package/dist/src/commands/setup.js.map +1 -1
- package/dist/src/commands/wallet.d.ts +16 -1
- package/dist/src/commands/wallet.d.ts.map +1 -1
- package/dist/src/db/dedupe-cids.d.ts +22 -0
- package/dist/src/db/dedupe-cids.d.ts.map +1 -0
- package/dist/src/db/dedupe-cids.js +28 -0
- package/dist/src/db/dedupe-cids.js.map +1 -0
- package/dist/src/db/find-providers-by-cid.d.ts +9 -0
- package/dist/src/db/find-providers-by-cid.d.ts.map +1 -0
- package/dist/src/db/{get-providers-by-cid.js → find-providers-by-cid.js} +4 -6
- package/dist/src/db/find-providers-by-cid.js.map +1 -0
- package/dist/src/db/find-repair-dataset.d.ts +10 -0
- package/dist/src/db/find-repair-dataset.d.ts.map +1 -0
- package/dist/src/db/{get-repair-dataset.js → find-repair-dataset.js} +8 -6
- package/dist/src/db/find-repair-dataset.js.map +1 -0
- package/dist/src/db/get-pieces.d.ts +16 -0
- package/dist/src/db/get-pieces.d.ts.map +1 -1
- package/dist/src/db/get-pieces.js +44 -3
- package/dist/src/db/get-pieces.js.map +1 -1
- package/dist/src/db/repair-create.d.ts.map +1 -1
- package/dist/src/db/repair-create.js +1 -0
- package/dist/src/db/repair-create.js.map +1 -1
- package/dist/src/db/repair-delete.d.ts.map +1 -1
- package/dist/src/db/repair-delete.js +0 -5
- package/dist/src/db/repair-delete.js.map +1 -1
- package/dist/src/db/replicate-create.d.ts +7 -0
- package/dist/src/db/replicate-create.d.ts.map +1 -0
- package/dist/src/db/replicate-create.js +78 -0
- package/dist/src/db/replicate-create.js.map +1 -0
- package/dist/src/db/upsert-operations.js +1 -1
- package/dist/src/db/upsert-operations.js.map +1 -1
- package/dist/src/local-schema.d.ts +19 -0
- package/dist/src/local-schema.d.ts.map +1 -1
- package/dist/src/local-schema.js +1 -0
- package/dist/src/local-schema.js.map +1 -1
- package/dist/src/middleware.d.ts +32 -2
- package/dist/src/middleware.d.ts.map +1 -1
- package/dist/src/middleware.js +10 -1
- package/dist/src/middleware.js.map +1 -1
- package/dist/src/pipeline/add-pieces.d.ts +12 -0
- package/dist/src/pipeline/add-pieces.d.ts.map +1 -0
- package/dist/src/pipeline/add-pieces.js +142 -0
- package/dist/src/pipeline/add-pieces.js.map +1 -0
- package/dist/src/pipeline/create-datasets.d.ts +3 -1
- package/dist/src/pipeline/create-datasets.d.ts.map +1 -1
- package/dist/src/pipeline/create-datasets.js +57 -5
- package/dist/src/pipeline/create-datasets.js.map +1 -1
- package/dist/src/utils.d.ts +166 -120
- package/dist/src/utils.d.ts.map +1 -1
- package/dist/src/utils.js +50 -25
- package/dist/src/utils.js.map +1 -1
- package/package.json +3 -2
- package/readme.md +110 -7
- package/src/cli.ts +2 -0
- package/src/commands/datasets.ts +109 -3
- package/src/commands/providers.ts +0 -1
- package/src/commands/repair.ts +12 -8
- package/src/commands/replicate.ts +183 -0
- package/src/commands/setup.ts +16 -0
- package/src/db/dedupe-cids.ts +49 -0
- package/src/db/{get-providers-by-cid.ts → find-providers-by-cid.ts} +5 -10
- package/src/db/{get-repair-dataset.ts → find-repair-dataset.ts} +12 -7
- package/src/db/get-pieces.ts +105 -3
- package/src/db/get-target-dataset.ts +1 -1
- package/src/db/repair-create.ts +1 -0
- package/src/db/repair-delete.ts +0 -5
- package/src/db/replicate-create.ts +106 -0
- package/src/db/upsert-operations.ts +1 -1
- package/src/local-schema.ts +1 -0
- package/src/middleware.ts +12 -1
- package/src/pipeline/add-pieces.ts +214 -0
- package/src/pipeline/create-datasets.ts +71 -11
- package/src/utils.ts +64 -32
- package/dist/src/db/get-providers-by-cid.d.ts +0 -10
- package/dist/src/db/get-providers-by-cid.d.ts.map +0 -1
- package/dist/src/db/get-providers-by-cid.js.map +0 -1
- package/dist/src/db/get-repair-dataset.d.ts +0 -9
- package/dist/src/db/get-repair-dataset.d.ts.map +0 -1
- package/dist/src/db/get-repair-dataset.js.map +0 -1
- package/dist/src/db/sync-pieces-onchain.d.ts +0 -10
- package/dist/src/db/sync-pieces-onchain.d.ts.map +0 -1
- package/dist/src/db/sync-pieces-onchain.js +0 -35
- package/dist/src/db/sync-pieces-onchain.js.map +0 -1
- package/dist/src/pipeline/pull.d.ts +0 -30
- package/dist/src/pipeline/pull.d.ts.map +0 -1
- package/dist/src/pipeline/pull.js +0 -169
- package/dist/src/pipeline/pull.js.map +0 -1
- package/src/db/sync-pieces-onchain.ts +0 -53
- package/src/pipeline/pull.ts +0 -255
package/dist/src/middleware.d.ts
CHANGED
|
@@ -5,15 +5,45 @@ import type { IndexerDatabase, LocalDatabase } from './types.ts';
|
|
|
5
5
|
export declare const contextSchema: z.ZodObject<{
|
|
6
6
|
indexerDb: z.ZodCustom<IndexerDatabase, IndexerDatabase>;
|
|
7
7
|
localDb: z.ZodCustom<LocalDatabase, LocalDatabase>;
|
|
8
|
-
config: z.ZodCustom<import("conf").
|
|
8
|
+
config: z.ZodCustom<import("iso-conf").Conf<z.ZodObject<{
|
|
9
|
+
privateKey: z.ZodOptional<z.ZodString>;
|
|
10
|
+
indexerMainnetUrl: z.ZodOptional<z.ZodURL>;
|
|
11
|
+
indexerCalibrationUrl: z.ZodOptional<z.ZodURL>;
|
|
12
|
+
chainId: z.ZodOptional<z.ZodNumber>;
|
|
13
|
+
dbPath: z.ZodOptional<z.ZodString>;
|
|
14
|
+
source: z.ZodOptional<z.ZodString>;
|
|
15
|
+
}, z.core.$strip>>, import("iso-conf").Conf<z.ZodObject<{
|
|
16
|
+
privateKey: z.ZodOptional<z.ZodString>;
|
|
17
|
+
indexerMainnetUrl: z.ZodOptional<z.ZodURL>;
|
|
18
|
+
indexerCalibrationUrl: z.ZodOptional<z.ZodURL>;
|
|
19
|
+
chainId: z.ZodOptional<z.ZodNumber>;
|
|
20
|
+
dbPath: z.ZodOptional<z.ZodString>;
|
|
21
|
+
source: z.ZodOptional<z.ZodString>;
|
|
22
|
+
}, z.core.$strip>>>;
|
|
9
23
|
client: z.ZodCustom<Client<Transport, Chain, Account>, Client<Transport, Chain, Account>>;
|
|
10
24
|
chain: z.ZodCustom<Chain, Chain>;
|
|
25
|
+
source: z.ZodString;
|
|
11
26
|
}, z.core.$strip>;
|
|
12
27
|
export declare const contextMiddleware: import("incur").MiddlewareHandler<z.ZodObject<{
|
|
13
28
|
indexerDb: z.ZodCustom<IndexerDatabase, IndexerDatabase>;
|
|
14
29
|
localDb: z.ZodCustom<LocalDatabase, LocalDatabase>;
|
|
15
|
-
config: z.ZodCustom<import("conf").
|
|
30
|
+
config: z.ZodCustom<import("iso-conf").Conf<z.ZodObject<{
|
|
31
|
+
privateKey: z.ZodOptional<z.ZodString>;
|
|
32
|
+
indexerMainnetUrl: z.ZodOptional<z.ZodURL>;
|
|
33
|
+
indexerCalibrationUrl: z.ZodOptional<z.ZodURL>;
|
|
34
|
+
chainId: z.ZodOptional<z.ZodNumber>;
|
|
35
|
+
dbPath: z.ZodOptional<z.ZodString>;
|
|
36
|
+
source: z.ZodOptional<z.ZodString>;
|
|
37
|
+
}, z.core.$strip>>, import("iso-conf").Conf<z.ZodObject<{
|
|
38
|
+
privateKey: z.ZodOptional<z.ZodString>;
|
|
39
|
+
indexerMainnetUrl: z.ZodOptional<z.ZodURL>;
|
|
40
|
+
indexerCalibrationUrl: z.ZodOptional<z.ZodURL>;
|
|
41
|
+
chainId: z.ZodOptional<z.ZodNumber>;
|
|
42
|
+
dbPath: z.ZodOptional<z.ZodString>;
|
|
43
|
+
source: z.ZodOptional<z.ZodString>;
|
|
44
|
+
}, z.core.$strip>>>;
|
|
16
45
|
client: z.ZodCustom<Client<Transport, Chain, Account>, Client<Transport, Chain, Account>>;
|
|
17
46
|
chain: z.ZodCustom<Chain, Chain>;
|
|
47
|
+
source: z.ZodString;
|
|
18
48
|
}, z.core.$strip>, undefined>;
|
|
19
49
|
//# sourceMappingURL=middleware.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../../src/middleware.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,4BAA4B,CAAA;AAEvD,OAAO,EAAc,CAAC,EAAE,MAAM,OAAO,CAAA;AACrC,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,MAAM,CAAA;AAEtD,OAAO,KAAK,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,YAAY,CAAA;AAGhE,eAAO,MAAM,aAAa
|
|
1
|
+
{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../../src/middleware.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,4BAA4B,CAAA;AAEvD,OAAO,EAAc,CAAC,EAAE,MAAM,OAAO,CAAA;AACrC,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,MAAM,CAAA;AAEtD,OAAO,KAAK,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,YAAY,CAAA;AAGhE,eAAO,MAAM,aAAa;;;;;;;;;;;;;;;;;;;;;iBAOxB,CAAA;AAEF,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;;;;;;;;6BA2B5B,CAAA"}
|
package/dist/src/middleware.js
CHANGED
|
@@ -8,9 +8,17 @@ export const contextSchema = z.object({
|
|
|
8
8
|
config: z.custom(),
|
|
9
9
|
client: z.custom(),
|
|
10
10
|
chain: z.custom(),
|
|
11
|
+
source: z.string(),
|
|
11
12
|
});
|
|
12
13
|
export const contextMiddleware = middleware(async (c, next) => {
|
|
13
|
-
const { dbPath, chainId, indexerMainnetUrl, indexerCalibrationUrl } = config.store;
|
|
14
|
+
const { dbPath, chainId, indexerMainnetUrl, indexerCalibrationUrl, source } = config.store;
|
|
15
|
+
if (!dbPath || !chainId || !indexerMainnetUrl || !indexerCalibrationUrl || !source) {
|
|
16
|
+
return c.error({
|
|
17
|
+
code: 'CONFIG_NOT_SET',
|
|
18
|
+
message: 'Config not set. Please run `repair setup` first.',
|
|
19
|
+
retryable: false,
|
|
20
|
+
});
|
|
21
|
+
}
|
|
14
22
|
const localDb = await createLocalDatabase(dbPath);
|
|
15
23
|
const indexerDb = drizzlePostgres(chainId === 314 ? indexerMainnetUrl : indexerCalibrationUrl, {
|
|
16
24
|
schema: indexerSchema,
|
|
@@ -21,6 +29,7 @@ export const contextMiddleware = middleware(async (c, next) => {
|
|
|
21
29
|
c.set('config', config);
|
|
22
30
|
c.set('client', client);
|
|
23
31
|
c.set('chain', chain);
|
|
32
|
+
c.set('source', source);
|
|
24
33
|
await next();
|
|
25
34
|
localDb.$client.close();
|
|
26
35
|
await indexerDb.$client.end();
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../../src/middleware.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,IAAI,eAAe,EAAE,MAAM,2BAA2B,CAAA;AACtE,OAAO,EAAE,UAAU,EAAE,CAAC,EAAE,MAAM,OAAO,CAAA;AAErC,OAAO,KAAK,aAAa,MAAM,qBAAqB,CAAA;AAEpD,OAAO,EAAE,MAAM,EAAE,mBAAmB,EAAE,SAAS,EAAE,MAAM,YAAY,CAAA;AAEnE,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC;IACpC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAmB;IACtC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAiB;IAClC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAiB;IACjC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAqC;IACrD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAS;
|
|
1
|
+
{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../../src/middleware.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,IAAI,eAAe,EAAE,MAAM,2BAA2B,CAAA;AACtE,OAAO,EAAE,UAAU,EAAE,CAAC,EAAE,MAAM,OAAO,CAAA;AAErC,OAAO,KAAK,aAAa,MAAM,qBAAqB,CAAA;AAEpD,OAAO,EAAE,MAAM,EAAE,mBAAmB,EAAE,SAAS,EAAE,MAAM,YAAY,CAAA;AAEnE,MAAM,CAAC,MAAM,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC;IACpC,SAAS,EAAE,CAAC,CAAC,MAAM,EAAmB;IACtC,OAAO,EAAE,CAAC,CAAC,MAAM,EAAiB;IAClC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAiB;IACjC,MAAM,EAAE,CAAC,CAAC,MAAM,EAAqC;IACrD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAS;IACxB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;CACnB,CAAC,CAAA;AAEF,MAAM,CAAC,MAAM,iBAAiB,GAAG,UAAU,CAAuB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE;IAClF,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC,KAAK,CAAA;IAE1F,IAAI,CAAC,MAAM,IAAI,CAAC,OAAO,IAAI,CAAC,iBAAiB,IAAI,CAAC,qBAAqB,IAAI,CAAC,MAAM,EAAE,CAAC;QACnF,OAAO,CAAC,CAAC,KAAK,CAAC;YACb,IAAI,EAAE,gBAAgB;YACtB,OAAO,EAAE,kDAAkD;YAC3D,SAAS,EAAE,KAAK;SACjB,CAAC,CAAA;IACJ,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,mBAAmB,CAAC,MAAM,CAAC,CAAA;IACjD,MAAM,SAAS,GAAG,eAAe,CAAC,OAAO,KAAK,GAAG,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,qBAAqB,EAAE;QAC7F,MAAM,EAAE,aAAa;KACtB,CAAC,CAAA;IAEF,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,SAAS,CAAC,OAAO,CAAC,CAAA;IAC5C,CAAC,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAA;IACzB,CAAC,CAAC,GAAG,CAAC,WAAW,EAAE,SAAS,CAAC,CAAA;IAC7B,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;IACvB,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;IACvB,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;IACrB,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;IACvB,MAAM,IAAI,EAAE,CAAA;IAEZ,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,CAAA;IACvB,MAAM,SAAS,CAAC,OAAO,CAAC,GAAG,EAAE,CAAA;AAC/B,CAAC,CAAC,CAAA"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import type { RepairSelect } from '../local-schema.ts';
|
|
2
|
+
import type { IndexerDatabase, LocalDatabase, WalletClient } from '../types.ts';
|
|
3
|
+
export type RunPullPiecesPhaseOptions = {
|
|
4
|
+
localDb: LocalDatabase;
|
|
5
|
+
indexerDb: IndexerDatabase;
|
|
6
|
+
repair: RepairSelect;
|
|
7
|
+
concurrency: number;
|
|
8
|
+
batchSize: number;
|
|
9
|
+
client: WalletClient;
|
|
10
|
+
};
|
|
11
|
+
export declare function runAddPieces({ localDb, indexerDb, repair, concurrency, batchSize, client, }: RunPullPiecesPhaseOptions): Promise<void>;
|
|
12
|
+
//# sourceMappingURL=add-pieces.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"add-pieces.d.ts","sourceRoot":"","sources":["../../../src/pipeline/add-pieces.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,EAAmB,YAAY,EAAE,MAAM,oBAAoB,CAAA;AACvE,OAAO,KAAK,EAAE,eAAe,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,aAAa,CAAA;AAG/E,MAAM,MAAM,yBAAyB,GAAG;IACtC,OAAO,EAAE,aAAa,CAAA;IACtB,SAAS,EAAE,eAAe,CAAA;IAC1B,MAAM,EAAE,YAAY,CAAA;IACpB,WAAW,EAAE,MAAM,CAAA;IACnB,SAAS,EAAE,MAAM,CAAA;IACjB,MAAM,EAAE,YAAY,CAAA;CACrB,CAAA;AAyHD,wBAAsB,YAAY,CAAC,EACjC,OAAO,EACP,SAAS,EACT,MAAM,EACN,WAAW,EACX,SAAS,EACT,MAAM,GACP,EAAE,yBAAyB,GAAG,OAAO,CAAC,IAAI,CAAC,CAiE3C"}
|
|
@@ -0,0 +1,142 @@
|
|
|
1
|
+
import { taskLog } from '@clack/prompts';
|
|
2
|
+
import * as Piece from '@filoz/synapse-core/piece';
|
|
3
|
+
import * as SP from '@filoz/synapse-core/sp';
|
|
4
|
+
import { and, asc, eq, gt, inArray } from 'drizzle-orm';
|
|
5
|
+
import PQueue from 'p-queue';
|
|
6
|
+
import { dedupeCids } from "../db/dedupe-cids.js";
|
|
7
|
+
import { getTargetDataset } from "../db/get-target-dataset.js";
|
|
8
|
+
import { repairUpdate } from "../db/repair-update.js";
|
|
9
|
+
import { upsertOperations } from "../db/upsert-operations.js";
|
|
10
|
+
import { excludeOperationsByCid, hashLink, operationsToPullPieces } from "../utils.js";
|
|
11
|
+
function createAddPiecesWorker({ localDb, indexerDb, repair, client, state, log }) {
|
|
12
|
+
return async (options) => {
|
|
13
|
+
let completedOps = 0;
|
|
14
|
+
let failedOps = 0;
|
|
15
|
+
let operations = options.operations;
|
|
16
|
+
const group = log.group(`Batch ${options.batchNumber}/${state.totalBatches}`);
|
|
17
|
+
try {
|
|
18
|
+
const dataset = await getTargetDataset({ localDb, repairId: repair.id, client });
|
|
19
|
+
if (repair.repairDataSetId == null) {
|
|
20
|
+
operations = await dedupeCids({ indexerDb, localDb, dataSetId: dataset.dataSetId, operations });
|
|
21
|
+
}
|
|
22
|
+
group.message(`Pulling ${operations.length} pieces...`);
|
|
23
|
+
if (operations.length > 0) {
|
|
24
|
+
const pullResult = await SP.waitForPullPieces(client, {
|
|
25
|
+
serviceURL: repair.targetProviderUrl,
|
|
26
|
+
dataSetId: dataset.dataSetId,
|
|
27
|
+
clientDataSetId: dataset.clientDataSetId,
|
|
28
|
+
pieces: operationsToPullPieces(operations),
|
|
29
|
+
timeout: 1000 * 60 * 30,
|
|
30
|
+
onStatus: (status) => {
|
|
31
|
+
const completed = status.pieces.filter((piece) => piece.status === 'complete').length;
|
|
32
|
+
const failed = status.pieces.filter((piece) => piece.status === 'failed').length;
|
|
33
|
+
group.message(`Pull ${completed} completed, ${failed} failed`);
|
|
34
|
+
},
|
|
35
|
+
});
|
|
36
|
+
for (const { pieceCid, status } of pullResult.pieces) {
|
|
37
|
+
const cid = pieceCid.toString();
|
|
38
|
+
if (status !== 'complete') {
|
|
39
|
+
state.failedOperations++;
|
|
40
|
+
failedOps++;
|
|
41
|
+
const { operationToFailed, operationToCommit } = excludeOperationsByCid(operations, cid);
|
|
42
|
+
operations = operationToCommit;
|
|
43
|
+
await upsertOperations({
|
|
44
|
+
localDb,
|
|
45
|
+
operations: operationToFailed.map((operation) => ({
|
|
46
|
+
...operation,
|
|
47
|
+
status: 'failed',
|
|
48
|
+
error: `pull failed with status ${status}`,
|
|
49
|
+
})),
|
|
50
|
+
});
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
if (operations.length > 0) {
|
|
55
|
+
group.message(`Adding ${operations.length} pieces...`);
|
|
56
|
+
const addPiecesResult = await SP.addPieces(client, {
|
|
57
|
+
serviceURL: repair.targetProviderUrl,
|
|
58
|
+
dataSetId: dataset.dataSetId,
|
|
59
|
+
clientDataSetId: dataset.clientDataSetId,
|
|
60
|
+
pieces: operations.map((operation) => ({
|
|
61
|
+
pieceCid: Piece.from(operation.cid),
|
|
62
|
+
metadata: operation.metadata,
|
|
63
|
+
})),
|
|
64
|
+
});
|
|
65
|
+
group.message(`Waiting for add pieces ${hashLink(addPiecesResult.txHash, client.chain)}...`);
|
|
66
|
+
const addPiecesResult2 = await SP.waitForAddPieces(addPiecesResult);
|
|
67
|
+
state.completedOperations += operations.length;
|
|
68
|
+
completedOps += operations.length;
|
|
69
|
+
await upsertOperations({
|
|
70
|
+
localDb,
|
|
71
|
+
operations: operations.map((operation) => ({
|
|
72
|
+
...operation,
|
|
73
|
+
status: 'completed',
|
|
74
|
+
error: null,
|
|
75
|
+
result: { dataSetId: addPiecesResult2.dataSetId, txHash: addPiecesResult2.txHash },
|
|
76
|
+
})),
|
|
77
|
+
});
|
|
78
|
+
}
|
|
79
|
+
group.message(`Done. ${completedOps} added, ${failedOps} failed`);
|
|
80
|
+
}
|
|
81
|
+
catch (error) {
|
|
82
|
+
state.failedOperations += operations.length;
|
|
83
|
+
const message = error instanceof Error ? error.message : 'Unknown error';
|
|
84
|
+
group.message(`${message.replace(/\n/g, ' ')}`);
|
|
85
|
+
await upsertOperations({
|
|
86
|
+
localDb,
|
|
87
|
+
operations: operations.map((operation) => ({
|
|
88
|
+
...operation,
|
|
89
|
+
status: 'failed',
|
|
90
|
+
error: message,
|
|
91
|
+
})),
|
|
92
|
+
});
|
|
93
|
+
}
|
|
94
|
+
};
|
|
95
|
+
}
|
|
96
|
+
export async function runAddPieces({ localDb, indexerDb, repair, concurrency, batchSize, client, }) {
|
|
97
|
+
const localSchema = localDb._.fullSchema;
|
|
98
|
+
let cursor = 0;
|
|
99
|
+
const totalOperations = await localDb.$count(localSchema.operations, and(eq(localSchema.operations.repairId, repair.id), eq(localSchema.operations.type, 'add_piece'), inArray(localSchema.operations.status, ['pending', 'failed'])));
|
|
100
|
+
let batchNumber = 0;
|
|
101
|
+
const state = {
|
|
102
|
+
totalBatches: Math.ceil(totalOperations / batchSize),
|
|
103
|
+
totalOperations,
|
|
104
|
+
completedOperations: 0,
|
|
105
|
+
failedOperations: 0,
|
|
106
|
+
};
|
|
107
|
+
const log = taskLog({
|
|
108
|
+
title: 'Adding pieces',
|
|
109
|
+
limit: 1,
|
|
110
|
+
});
|
|
111
|
+
async function getNextBatch() {
|
|
112
|
+
const operations = await localDb.query.operations.findMany({
|
|
113
|
+
where: and(eq(localSchema.operations.repairId, repair.id), eq(localSchema.operations.type, 'add_piece'), inArray(localSchema.operations.status, ['pending', 'failed']), gt(localSchema.operations.id, cursor)),
|
|
114
|
+
orderBy: [asc(localSchema.operations.id)],
|
|
115
|
+
limit: batchSize,
|
|
116
|
+
});
|
|
117
|
+
if (operations.length === 0) {
|
|
118
|
+
return null;
|
|
119
|
+
}
|
|
120
|
+
cursor = operations.at(-1)?.id ?? cursor;
|
|
121
|
+
return operations;
|
|
122
|
+
}
|
|
123
|
+
const addPiecesJob = createAddPiecesWorker({ localDb, indexerDb, repair, client, state, log });
|
|
124
|
+
const queue = new PQueue({ concurrency });
|
|
125
|
+
while (true) {
|
|
126
|
+
await queue.onSizeLessThan(concurrency);
|
|
127
|
+
const operations = await getNextBatch();
|
|
128
|
+
if (!operations)
|
|
129
|
+
break;
|
|
130
|
+
batchNumber++;
|
|
131
|
+
const currentBatchNumber = batchNumber;
|
|
132
|
+
queue.add(() => addPiecesJob({ operations, batchNumber: currentBatchNumber })).catch(console.error);
|
|
133
|
+
}
|
|
134
|
+
await queue.onIdle();
|
|
135
|
+
log.success(`Added ${state.completedOperations} pieces, ${state.failedOperations} failed`, { showLog: true });
|
|
136
|
+
await repairUpdate({
|
|
137
|
+
localDb,
|
|
138
|
+
repairId: repair.id,
|
|
139
|
+
status: state.failedOperations > 0 ? 'failed' : 'completed',
|
|
140
|
+
});
|
|
141
|
+
}
|
|
142
|
+
//# sourceMappingURL=add-pieces.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"add-pieces.js","sourceRoot":"","sources":["../../../src/pipeline/add-pieces.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAA;AACxC,OAAO,KAAK,KAAK,MAAM,2BAA2B,CAAA;AAClD,OAAO,KAAK,EAAE,MAAM,wBAAwB,CAAA;AAC5C,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,aAAa,CAAA;AACvD,OAAO,MAAM,MAAM,SAAS,CAAA;AAC5B,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAA;AACjD,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAA;AAC9D,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAA;AACrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,4BAA4B,CAAA;AAG7D,OAAO,EAAE,sBAAsB,EAAE,QAAQ,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAA;AAiCtF,SAAS,qBAAqB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAgC;IAC7G,OAAO,KAAK,EAAE,OAA4B,EAAE,EAAE;QAC5C,IAAI,YAAY,GAAG,CAAC,CAAA;QACpB,IAAI,SAAS,GAAG,CAAC,CAAA;QACjB,IAAI,UAAU,GAAsB,OAAO,CAAC,UAAU,CAAA;QAEtD,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,SAAS,OAAO,CAAC,WAAW,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC,CAAA;QAE7E,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,gBAAgB,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,CAAC,EAAE,EAAE,MAAM,EAAE,CAAC,CAAA;YAGhF,IAAI,MAAM,CAAC,eAAe,IAAI,IAAI,EAAE,CAAC;gBACnC,UAAU,GAAG,MAAM,UAAU,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,UAAU,EAAE,CAAC,CAAA;YACjG,CAAC;YACD,KAAK,CAAC,OAAO,CAAC,WAAW,UAAU,CAAC,MAAM,YAAY,CAAC,CAAA;YAEvD,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,MAAM,UAAU,GAAG,MAAM,EAAE,CAAC,iBAAiB,CAAC,MAAM,EAAE;oBACpD,UAAU,EAAE,MAAM,CAAC,iBAAiB;oBACpC,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,eAAe,EAAE,OAAO,CAAC,eAAe;oBACxC,MAAM,EAAE,sBAAsB,CAAC,UAAU,CAAC;oBAC1C,OAAO,EAAE,IAAI,GAAG,EAAE,GAAG,EAAE;oBACvB,QAAQ,EAAE,CAAC,MAAM,EAAE,EAAE;wBACnB,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,KAAK,UAAU,CAAC,CAAC,MAAM,CAAA;wBACrF,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,MAAM,CAAA;wBAChF,KAAK,CAAC,OAAO,CAAC,QAAQ,SAAS,eAAe,MAAM,SAAS,CAAC,CAAA;oBAChE,CAAC;iBACF,CAAC,CAAA;gBAEF,KAAK,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,UAAU,CAAC,MAAM,EAAE,CAAC;oBACrD,MAAM,GAAG,GAAG,QAAQ,CAAC,QAAQ,EAAE,CAAA;oBAC/B,IAAI,MAAM,KAAK,UAAU,EAAE,CAAC;wBAC1B,KAAK,CAAC,gBAAgB,EAAE,CAAA;wBACxB,SAAS,EAAE,CAAA;wBACX,MAAM,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,GAAG,sBAAsB,CAAC,UAAU,EAAE,GAAG,CAAC,CAAA;wBACxF,UAAU,GAAG,iBAAiB,CAAA;wBAC9B,MAAM,gBAAgB,CAAC;4BACrB,OAAO;4BACP,UAAU,EAAE,iBAAiB,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;gCAChD,GAAG,SAAS;gCACZ,MAAM,EAAE,QAAQ;gCAChB,KAAK,EAAE,2BAA2B,MAAM,EAAE;6BAC3C,CAAC,CAAC;yBACJ,CAAC,CAAA;oBACJ,CAAC;gBACH,CAAC;YACH,CAAC;YAGD,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC1B,KAAK,CAAC,OAAO,CAAC,UAAU,UAAU,CAAC,MAAM,YAAY,CAAC,CAAA;gBACtD,MAAM,eAAe,GAAG,MAAM,EAAE,CAAC,SAAS,CAAC,MAAM,EAAE;oBACjD,UAAU,EAAE,MAAM,CAAC,iBAAiB;oBACpC,SAAS,EAAE,OAAO,CAAC,SAAS;oBAC5B,eAAe,EAAE,OAAO,CAAC,eAAe;oBACxC,MAAM,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;wBACrC,QAAQ,EAAE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC;wBACnC,QAAQ,EAAE,SAAS,CAAC,QAAQ;qBAC7B,CAAC,CAAC;iBACJ,CAAC,CAAA;gBAEF,KAAK,CAAC,OAAO,CAAC,0BAA0B,QAAQ,CAAC,eAAe,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;gBAC5F,MAAM,gBAAgB,GAAG,MAAM,EAAE,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAA;gBACnE,KAAK,CAAC,mBAAmB,IAAI,UAAU,CAAC,MAAM,CAAA;gBAC9C,YAAY,IAAI,UAAU,CAAC,MAAM,CAAA;gBACjC,MAAM,gBAAgB,CAAC;oBACrB,OAAO;oBACP,UAAU,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;wBACzC,GAAG,SAAS;wBACZ,MAAM,EAAE,WAAW;wBACnB,KAAK,EAAE,IAAI;wBACX,MAAM,EAAE,EAAE,SAAS,EAAE,gBAAgB,CAAC,SAAS,EAAE,MAAM,EAAE,gBAAgB,CAAC,MAAM,EAAE;qBACnF,CAAC,CAAC;iBACJ,CAAC,CAAA;YACJ,CAAC;YACD,KAAK,CAAC,OAAO,CAAC,SAAS,YAAY,WAAW,SAAS,SAAS,CAAC,CAAA;QACnE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,KAAK,CAAC,gBAAgB,IAAI,UAAU,CAAC,MAAM,CAAA;YAC3C,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAA;YACxE,KAAK,CAAC,OAAO,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC,CAAA;YAC/C,MAAM,gBAAgB,CAAC;gBACrB,OAAO;gBACP,UAAU,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;oBACzC,GAAG,SAAS;oBACZ,MAAM,EAAE,QAAQ;oBAChB,KAAK,EAAE,OAAO;iBACf,CAAC,CAAC;aACJ,CAAC,CAAA;QACJ,CAAC;IACH,CAAC,CAAA;AACH,CAAC;AAKD,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,EACjC,OAAO,EACP,SAAS,EACT,MAAM,EACN,WAAW,EACX,SAAS,EACT,MAAM,GACoB;IAC1B,MAAM,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,UAAU,CAAA;IACxC,IAAI,MAAM,GAAG,CAAC,CAAA;IAEd,MAAM,eAAe,GAAG,MAAM,OAAO,CAAC,MAAM,CAC1C,WAAW,CAAC,UAAU,EACtB,GAAG,CACD,EAAE,CAAC,WAAW,CAAC,UAAU,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC,EAC9C,EAAE,CAAC,WAAW,CAAC,UAAU,CAAC,IAAI,EAAE,WAAW,CAAC,EAC5C,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC,CAC9D,CACF,CAAA;IACD,IAAI,WAAW,GAAG,CAAC,CAAA;IACnB,MAAM,KAAK,GAAG;QACZ,YAAY,EAAE,IAAI,CAAC,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QACpD,eAAe;QACf,mBAAmB,EAAE,CAAC;QACtB,gBAAgB,EAAE,CAAC;KACpB,CAAA;IAED,MAAM,GAAG,GAAG,OAAO,CAAC;QAClB,KAAK,EAAE,eAAe;QACtB,KAAK,EAAE,CAAC;KACT,CAAC,CAAA;IAEF,KAAK,UAAU,YAAY;QACzB,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC;YACzD,KAAK,EAAE,GAAG,CACR,EAAE,CAAC,WAAW,CAAC,UAAU,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,CAAC,EAC9C,EAAE,CAAC,WAAW,CAAC,UAAU,CAAC,IAAI,EAAE,WAAW,CAAC,EAC5C,OAAO,CAAC,WAAW,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC,EAC7D,EAAE,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE,EAAE,MAAM,CAAC,CACtC;YACD,OAAO,EAAE,CAAC,GAAG,CAAC,WAAW,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;YACzC,KAAK,EAAE,SAAS;SACjB,CAAC,CAAA;QACF,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,IAAI,CAAA;QACb,CAAC;QAED,MAAM,GAAG,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,IAAI,MAAM,CAAA;QACxC,OAAO,UAAU,CAAA;IACnB,CAAC;IAED,MAAM,YAAY,GAAG,qBAAqB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAA;IAC9F,MAAM,KAAK,GAAG,IAAI,MAAM,CAAC,EAAE,WAAW,EAAE,CAAC,CAAA;IAEzC,OAAO,IAAI,EAAE,CAAC;QACZ,MAAM,KAAK,CAAC,cAAc,CAAC,WAAW,CAAC,CAAA;QACvC,MAAM,UAAU,GAAG,MAAM,YAAY,EAAE,CAAA;QACvC,IAAI,CAAC,UAAU;YAAE,MAAK;QACtB,WAAW,EAAE,CAAA;QACb,MAAM,kBAAkB,GAAG,WAAW,CAAA;QACtC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,YAAY,CAAC,EAAE,UAAU,EAAE,WAAW,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;IACrG,CAAC;IAED,MAAM,KAAK,CAAC,MAAM,EAAE,CAAA;IAEpB,GAAG,CAAC,OAAO,CAAC,SAAS,KAAK,CAAC,mBAAmB,YAAY,KAAK,CAAC,gBAAgB,SAAS,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAA;IAE7G,MAAM,YAAY,CAAC;QACjB,OAAO;QACP,QAAQ,EAAE,MAAM,CAAC,EAAE;QACnB,MAAM,EAAE,KAAK,CAAC,gBAAgB,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW;KAC5D,CAAC,CAAA;AACJ,CAAC"}
|
|
@@ -1,10 +1,12 @@
|
|
|
1
1
|
import type { RepairSelect } from '../local-schema.ts';
|
|
2
2
|
import type { IndexerDatabase, LocalDatabase, WalletClient } from '../types.ts';
|
|
3
3
|
export type EnsureRepairDatasetOptions = {
|
|
4
|
+
source: string;
|
|
4
5
|
localDb: LocalDatabase;
|
|
5
6
|
indexerDb: IndexerDatabase;
|
|
6
7
|
client: WalletClient;
|
|
7
8
|
repair: RepairSelect;
|
|
8
9
|
};
|
|
9
|
-
export declare function ensureRepairDataset({ localDb, indexerDb, client, repair
|
|
10
|
+
export declare function ensureRepairDataset({ source, localDb, indexerDb, client, repair }: EnsureRepairDatasetOptions): Promise<bigint>;
|
|
11
|
+
export declare function ensureReplicateDataset({ localDb, indexerDb, client, repair }: EnsureRepairDatasetOptions): Promise<bigint>;
|
|
10
12
|
//# sourceMappingURL=create-datasets.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"create-datasets.d.ts","sourceRoot":"","sources":["../../../src/pipeline/create-datasets.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"create-datasets.d.ts","sourceRoot":"","sources":["../../../src/pipeline/create-datasets.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAA;AACtD,OAAO,KAAK,EAAE,eAAe,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,aAAa,CAAA;AAG/E,MAAM,MAAM,0BAA0B,GAAG;IACvC,MAAM,EAAE,MAAM,CAAA;IACd,OAAO,EAAE,aAAa,CAAA;IACtB,SAAS,EAAE,eAAe,CAAA;IAC1B,MAAM,EAAE,YAAY,CAAA;IACpB,MAAM,EAAE,YAAY,CAAA;CACrB,CAAA;AAOD,wBAAsB,mBAAmB,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,0BAA0B,mBA8CnH;AAOD,wBAAsB,sBAAsB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,0BAA0B,mBAqD9G"}
|
|
@@ -1,10 +1,11 @@
|
|
|
1
1
|
import * as p from '@clack/prompts';
|
|
2
2
|
import * as SP from '@filoz/synapse-core/sp';
|
|
3
3
|
import { getPDPProvider } from '@filoz/synapse-core/sp-registry';
|
|
4
|
-
import {
|
|
4
|
+
import { eq } from 'drizzle-orm';
|
|
5
|
+
import { findRepairDataset } from "../db/find-repair-dataset.js";
|
|
5
6
|
import { repairUpdate } from "../db/repair-update.js";
|
|
6
|
-
import {
|
|
7
|
-
export async function ensureRepairDataset({ localDb, indexerDb, client, repair
|
|
7
|
+
import { hashLink } from "../utils.js";
|
|
8
|
+
export async function ensureRepairDataset({ source, localDb, indexerDb, client, repair }) {
|
|
8
9
|
const log = p.taskLog({
|
|
9
10
|
title: 'Ensuring repair dataset',
|
|
10
11
|
});
|
|
@@ -14,10 +15,11 @@ export async function ensureRepairDataset({ localDb, indexerDb, client, repair,
|
|
|
14
15
|
if (!provider)
|
|
15
16
|
throw new Error(`Target provider ${repair.targetProviderId} not found or inactive`);
|
|
16
17
|
let datasetId = null;
|
|
17
|
-
const existingDatasetId = await
|
|
18
|
+
const existingDatasetId = await findRepairDataset({
|
|
18
19
|
indexerDb,
|
|
19
20
|
providerId: repair.targetProviderId,
|
|
20
21
|
payer: client.account.address,
|
|
22
|
+
source,
|
|
21
23
|
});
|
|
22
24
|
if (existingDatasetId) {
|
|
23
25
|
datasetId = existingDatasetId;
|
|
@@ -29,7 +31,10 @@ export async function ensureRepairDataset({ localDb, indexerDb, client, repair,
|
|
|
29
31
|
serviceURL: provider.pdp.serviceURL,
|
|
30
32
|
payer: client.account.address,
|
|
31
33
|
cdn: false,
|
|
32
|
-
metadata:
|
|
34
|
+
metadata: {
|
|
35
|
+
source,
|
|
36
|
+
withIPFSIndexing: '',
|
|
37
|
+
},
|
|
33
38
|
});
|
|
34
39
|
log.message(`Waiting for data to be created at ${provider.pdp.serviceURL} ${hashLink(txHash, client.chain)}...`);
|
|
35
40
|
const waitForResult = await SP.waitForCreateDataSet({
|
|
@@ -45,4 +50,51 @@ export async function ensureRepairDataset({ localDb, indexerDb, client, repair,
|
|
|
45
50
|
});
|
|
46
51
|
return datasetId;
|
|
47
52
|
}
|
|
53
|
+
export async function ensureReplicateDataset({ localDb, indexerDb, client, repair }) {
|
|
54
|
+
const log = p.taskLog({
|
|
55
|
+
title: 'Ensuring replication dataset',
|
|
56
|
+
});
|
|
57
|
+
if (repair.repairDataSetId == null) {
|
|
58
|
+
throw new Error('Missing source dataset ID');
|
|
59
|
+
}
|
|
60
|
+
const provider = await getPDPProvider(client, {
|
|
61
|
+
providerId: repair.targetProviderId,
|
|
62
|
+
});
|
|
63
|
+
if (!provider)
|
|
64
|
+
throw new Error(`Target provider ${repair.targetProviderId} not found or inactive`);
|
|
65
|
+
if (repair.targetDataSetId != null) {
|
|
66
|
+
log.success(`Data set #${repair.targetDataSetId} already exists at ${provider.pdp.serviceURL}`);
|
|
67
|
+
return repair.targetDataSetId;
|
|
68
|
+
}
|
|
69
|
+
const schema = indexerDb._.fullSchema;
|
|
70
|
+
const sourceDataSet = await indexerDb.query.dataSets.findFirst({
|
|
71
|
+
where: eq(schema.dataSets.dataSetId, repair.repairDataSetId),
|
|
72
|
+
columns: {
|
|
73
|
+
metadata: true,
|
|
74
|
+
withCdn: true,
|
|
75
|
+
},
|
|
76
|
+
});
|
|
77
|
+
if (!sourceDataSet) {
|
|
78
|
+
throw new Error(`Source dataset ${repair.repairDataSetId} not found`);
|
|
79
|
+
}
|
|
80
|
+
const { txHash, statusUrl } = await SP.createDataSet(client, {
|
|
81
|
+
payee: provider.payee,
|
|
82
|
+
serviceURL: provider.pdp.serviceURL,
|
|
83
|
+
payer: client.account.address,
|
|
84
|
+
cdn: sourceDataSet.withCdn,
|
|
85
|
+
metadata: sourceDataSet.metadata ?? undefined,
|
|
86
|
+
});
|
|
87
|
+
log.message(`Waiting for data to be created at ${provider.pdp.serviceURL} ${hashLink(txHash, client.chain)}...`);
|
|
88
|
+
const waitForResult = await SP.waitForCreateDataSet({
|
|
89
|
+
statusUrl,
|
|
90
|
+
});
|
|
91
|
+
const datasetId = waitForResult.dataSetId;
|
|
92
|
+
log.success(`Data set #${datasetId} created at ${provider.pdp.serviceURL}`);
|
|
93
|
+
await repairUpdate({
|
|
94
|
+
localDb,
|
|
95
|
+
repairId: repair.id,
|
|
96
|
+
targetDataSetId: datasetId,
|
|
97
|
+
});
|
|
98
|
+
return datasetId;
|
|
99
|
+
}
|
|
48
100
|
//# sourceMappingURL=create-datasets.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"create-datasets.js","sourceRoot":"","sources":["../../../src/pipeline/create-datasets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,CAAC,MAAM,gBAAgB,CAAA;AACnC,OAAO,KAAK,EAAE,MAAM,wBAAwB,CAAA;AAC5C,OAAO,EAAE,cAAc,EAAE,MAAM,iCAAiC,CAAA;AAChE,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"create-datasets.js","sourceRoot":"","sources":["../../../src/pipeline/create-datasets.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,CAAC,MAAM,gBAAgB,CAAA;AACnC,OAAO,KAAK,EAAE,MAAM,wBAAwB,CAAA;AAC5C,OAAO,EAAE,cAAc,EAAE,MAAM,iCAAiC,CAAA;AAChE,OAAO,EAAE,EAAE,EAAE,MAAM,aAAa,CAAA;AAChC,OAAO,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAA;AAChE,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAA;AAGrD,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAA;AAetC,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAA8B;IAClH,MAAM,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC;QACpB,KAAK,EAAE,yBAAyB;KACjC,CAAC,CAAA;IACF,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,MAAM,EAAE;QAC5C,UAAU,EAAE,MAAM,CAAC,gBAAgB;KACpC,CAAC,CAAA;IAEF,IAAI,CAAC,QAAQ;QAAE,MAAM,IAAI,KAAK,CAAC,mBAAmB,MAAM,CAAC,gBAAgB,wBAAwB,CAAC,CAAA;IAElG,IAAI,SAAS,GAAkB,IAAI,CAAA;IAEnC,MAAM,iBAAiB,GAAG,MAAM,iBAAiB,CAAC;QAChD,SAAS;QACT,UAAU,EAAE,MAAM,CAAC,gBAAgB;QACnC,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO;QAC7B,MAAM;KACP,CAAC,CAAA;IAEF,IAAI,iBAAiB,EAAE,CAAC;QACtB,SAAS,GAAG,iBAAiB,CAAA;QAC7B,GAAG,CAAC,OAAO,CAAC,aAAa,SAAS,sBAAsB,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC,CAAA;IACpF,CAAC;SAAM,CAAC;QACN,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,EAAE,CAAC,aAAa,CAAC,MAAM,EAAE;YAC3D,KAAK,EAAE,QAAQ,CAAC,KAAK;YACrB,UAAU,EAAE,QAAQ,CAAC,GAAG,CAAC,UAAU;YACnC,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO;YAC7B,GAAG,EAAE,KAAK;YACV,QAAQ,EAAE;gBACR,MAAM;gBACN,gBAAgB,EAAE,EAAE;aACrB;SACF,CAAC,CAAA;QACF,GAAG,CAAC,OAAO,CAAC,qCAAqC,QAAQ,CAAC,GAAG,CAAC,UAAU,IAAI,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;QAChH,MAAM,aAAa,GAAG,MAAM,EAAE,CAAC,oBAAoB,CAAC;YAClD,SAAS;SACV,CAAC,CAAA;QACF,SAAS,GAAG,aAAa,CAAC,SAAS,CAAA;QACnC,GAAG,CAAC,OAAO,CAAC,aAAa,SAAS,eAAe,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC,CAAA;IAC7E,CAAC;IACD,MAAM,YAAY,CAAC;QACjB,OAAO;QACP,QAAQ,EAAE,MAAM,CAAC,EAAE;QACnB,eAAe,EAAE,SAAS;KAC3B,CAAC,CAAA;IACF,OAAO,SAAS,CAAA;AAClB,CAAC;AAOD,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAAC,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,EAA8B;IAC7G,MAAM,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC;QACpB,KAAK,EAAE,8BAA8B;KACtC,CAAC,CAAA;IAEF,IAAI,MAAM,CAAC,eAAe,IAAI,IAAI,EAAE,CAAC;QACnC,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAA;IAC9C,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,MAAM,EAAE;QAC5C,UAAU,EAAE,MAAM,CAAC,gBAAgB;KACpC,CAAC,CAAA;IAEF,IAAI,CAAC,QAAQ;QAAE,MAAM,IAAI,KAAK,CAAC,mBAAmB,MAAM,CAAC,gBAAgB,wBAAwB,CAAC,CAAA;IAElG,IAAI,MAAM,CAAC,eAAe,IAAI,IAAI,EAAE,CAAC;QACnC,GAAG,CAAC,OAAO,CAAC,aAAa,MAAM,CAAC,eAAe,sBAAsB,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC,CAAA;QAC/F,OAAO,MAAM,CAAC,eAAe,CAAA;IAC/B,CAAC;IAED,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,UAAU,CAAA;IACrC,MAAM,aAAa,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC;QAC7D,KAAK,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC,eAAe,CAAC;QAC5D,OAAO,EAAE;YACP,QAAQ,EAAE,IAAI;YACd,OAAO,EAAE,IAAI;SACd;KACF,CAAC,CAAA;IAEF,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CAAC,kBAAkB,MAAM,CAAC,eAAe,YAAY,CAAC,CAAA;IACvE,CAAC;IAED,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,EAAE,CAAC,aAAa,CAAC,MAAM,EAAE;QAC3D,KAAK,EAAE,QAAQ,CAAC,KAAK;QACrB,UAAU,EAAE,QAAQ,CAAC,GAAG,CAAC,UAAU;QACnC,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC,OAAO;QAC7B,GAAG,EAAE,aAAa,CAAC,OAAO;QAC1B,QAAQ,EAAE,aAAa,CAAC,QAAQ,IAAI,SAAS;KAC9C,CAAC,CAAA;IACF,GAAG,CAAC,OAAO,CAAC,qCAAqC,QAAQ,CAAC,GAAG,CAAC,UAAU,IAAI,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA;IAChH,MAAM,aAAa,GAAG,MAAM,EAAE,CAAC,oBAAoB,CAAC;QAClD,SAAS;KACV,CAAC,CAAA;IACF,MAAM,SAAS,GAAG,aAAa,CAAC,SAAS,CAAA;IACzC,GAAG,CAAC,OAAO,CAAC,aAAa,SAAS,eAAe,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC,CAAA;IAE3E,MAAM,YAAY,CAAC;QACjB,OAAO;QACP,QAAQ,EAAE,MAAM,CAAC,EAAE;QACnB,eAAe,EAAE,SAAS;KAC3B,CAAC,CAAA;IACF,OAAO,SAAS,CAAA;AAClB,CAAC"}
|