yujunjun/strapi
1
0
Fork 0
mirror of https://github.com/strapi/strapi.git synced 2025-09-26 17:00:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #15912 from strapi/data-transfer/end-transfer

Browse Source
This commit is contained in:
Jean-Sébastien Herbaux 2023-02-24 15:27:43 +01:00 committed by GitHub
commit acbb752d95
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
packages/core/data-transfer/src/strapi/providers/remote-destination/index.ts
View File

@ -35,10 +35,13 @@ class RemoteStrapiDestinationProvider implements IDestinationProvider {
dispatcher: ReturnType<typeof createDispatcher> | null; dispatcher: ReturnType<typeof createDispatcher> | null;
transferID: string | null;
constructor(options: IRemoteStrapiDestinationProviderOptions) { constructor(options: IRemoteStrapiDestinationProviderOptions) {
this.options = options; this.options = options;
this.ws = null; this.ws = null;
this.dispatcher = null; this.dispatcher = null;
this.transferID = null;
} }
async initTransfer(): Promise<string> { async initTransfer(): Promise<string> {
@ -224,15 +227,23 @@ class RemoteStrapiDestinationProvider implements IDestinationProvider {
this.ws = ws; this.ws = ws;
this.dispatcher = createDispatcher(this.ws); this.dispatcher = createDispatcher(this.ws);
const transferID = await this.initTransfer(); this.transferID = await this.initTransfer();
this.dispatcher.setTransferProperties({ id: transferID, kind: 'push' }); this.dispatcher.setTransferProperties({ id: this.transferID, kind: 'push' });
await this.dispatcher.dispatchTransferAction('bootstrap'); await this.dispatcher.dispatchTransferAction('bootstrap');
} }
async close() { async close() {
await this.dispatcher?.dispatchTransferAction('close'); // Gracefully close the remote transfer process
if (this.transferID && this.dispatcher) {
await this.dispatcher.dispatchTransferAction('close');
await this.dispatcher.dispatchCommand({
command: 'end',
params: { transferID: this.transferID },
});
}
await new Promise<void>((resolve) => { await new Promise<void>((resolve) => {
const { ws } = this; const { ws } = this;