diff --git a/packages/data-center/src/datacenter.ts b/packages/data-center/src/datacenter.ts index 8cc8f80fa3..79d8c81eca 100644 --- a/packages/data-center/src/datacenter.ts +++ b/packages/data-center/src/datacenter.ts @@ -127,16 +127,16 @@ export class DataCenter { public async login(providerId = 'affine') { const provider = this.providerMap.get(providerId); assert(provider, `provide '${providerId}' is not registered`); - provider.auth(); + await provider.auth(); } /** * logout from all providers */ - public async logout() { - this.providers.forEach(p => { - p.logout(); - }); + public async logout(providerId = 'affine') { + const provider = this.providerMap.get(providerId); + assert(provider, `provide '${providerId}' is not registered`); + await provider.logout(); } /**