Fix pushes for new clients in calendar (#8115)

Signed-off-by: Denis Bykhov <bykhov.denis@gmail.com>
This commit is contained in:
Denis Bykhov 2025-02-28 12:46:46 +05:00 committed by GitHub
parent b858dd7696
commit 143cf905b5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 91 additions and 79 deletions

View File

@ -241,6 +241,7 @@ export class CalendarClient {
} }
close (): void { close (): void {
clearTimeout(this.inactiveTimer)
this.googleClient.close() this.googleClient.close()
for (const watch of this.dummyWatches) { for (const watch of this.dummyWatches) {
clearTimeout(watch.timer) clearTimeout(watch.timer)

View File

@ -108,6 +108,12 @@ export class CalendarController {
} }
} }
pushWorkspaceByEmail (email: string, workspace: string): void {
const arr = this.workspacesByEmail.get(email) ?? []
arr.push(workspace)
this.workspacesByEmail.set(email, arr)
}
async startWorkspace (workspace: string, tokens: Token[]): Promise<WorkspaceClient> { async startWorkspace (workspace: string, tokens: Token[]): Promise<WorkspaceClient> {
const workspaceClient = await this.getWorkspaceClient(workspace) const workspaceClient = await this.getWorkspaceClient(workspace)
for (const token of tokens) { for (const token of tokens) {
@ -117,9 +123,7 @@ export class CalendarController {
}, 60000) }, 60000)
console.log('init client', token.workspace, token.userId) console.log('init client', token.workspace, token.userId)
await workspaceClient.createCalendarClient(token, true) await workspaceClient.createCalendarClient(token, true)
const arr = this.workspacesByEmail.get(token.email) ?? [] this.pushWorkspaceByEmail(token.email, token.workspace)
arr.push(token.workspace)
this.workspacesByEmail.set(token.email, arr)
clearTimeout(timeout) clearTimeout(timeout)
} catch (err) { } catch (err) {
console.error(`Couldn't create client for ${workspace} ${token.userId} ${token.email}`) console.error(`Couldn't create client for ${workspace} ${token.userId} ${token.email}`)

View File

@ -212,10 +212,13 @@ export class GoogleClient {
workspace: this.user.workspace, workspace: this.user.workspace,
calendarId: null calendarId: null
}) })
if (current == null || current.expired < Date.now() + 24 * 60 * 60 * 1000) {
if (current != null) { if (current != null) {
await this.rateLimiter.take(1) await this.rateLimiter.take(1)
try { try {
await this.calendar.channels.stop({ requestBody: { id: current.channelId, resourceId: current.resourceId } }) await this.calendar.channels.stop({
requestBody: { id: current.channelId, resourceId: current.resourceId }
})
} catch {} } catch {}
} }
const channelId = generateId() const channelId = generateId()
@ -250,6 +253,7 @@ export class GoogleClient {
}) })
} }
} }
}
} catch (err: any) { } catch (err: any) {
console.error('Calendar watch error', err.message) console.error('Calendar watch error', err.message)
} }
@ -262,6 +266,7 @@ export class GoogleClient {
workspace: this.user.workspace, workspace: this.user.workspace,
calendarId calendarId
}) })
if (current == null || current.expired < Date.now() + 24 * 60 * 60 * 1000) {
if (current != null) { if (current != null) {
await this.rateLimiter.take(1) await this.rateLimiter.take(1)
try { try {
@ -307,12 +312,13 @@ export class GoogleClient {
}) })
} }
} }
}
return true return true
} catch (err: any) { } catch (err: any) {
if (err?.errors?.[0]?.reason === 'pushNotSupportedForRequestedResource') { if (err?.errors?.[0]?.reason === 'pushNotSupportedForRequestedResource') {
return false return false
} else { } else {
console.error('Watch error', err.message) console.error('Watch error', err)
await this.checkError(err) await this.checkError(err)
return false return false
} }

View File

@ -116,6 +116,7 @@ export class WorkspaceClient {
throw new Error('Client already exist') throw new Error('Client already exist')
} }
this.clients.set(email, newClient) this.clients.set(email, newClient)
this.serviceController.pushWorkspaceByEmail(email, user.workspace)
return newClient return newClient
} }