Add deleted websocket message

This commit is contained in:
shamoon
2025-02-01 16:40:39 -08:00
parent bf2c6c4c25
commit eeb9347e99
5 changed files with 43 additions and 1 deletions

View File

@@ -241,6 +241,10 @@ export class DocumentListComponent
this.list.reload()
})
this.consumerStatusService.onDocumentDeleted().subscribe(() => {
this.list.reload()
})
this.route.paramMap
.pipe(
filter((params) => params.has('id')), // only on saved view e.g. /view/id

View File

@@ -323,4 +323,21 @@ describe('ConsumerStatusService', () => {
1
)
})
it('should trigger deleted subject on document deleted', () => {
let deleted = false
consumerStatusService.onDocumentDeleted().subscribe(() => {
deleted = true
})
consumerStatusService.connect()
server.send({
current_progress: 1,
max_progress: 1,
status: 'DELETED',
})
consumerStatusService.disconnect()
expect(deleted).toBeTruthy()
})
})

View File

@@ -95,6 +95,7 @@ export class ConsumerStatusService {
private documentDetectedSubject = new Subject<FileStatus>()
private documentConsumptionFinishedSubject = new Subject<FileStatus>()
private documentConsumptionFailedSubject = new Subject<FileStatus>()
private documentDeletedSubject = new Subject<boolean>()
private get(taskId: string, filename?: string) {
let status =
@@ -147,6 +148,11 @@ export class ConsumerStatusService {
)
this.statusWebSocket.onmessage = (ev) => {
let statusMessage: WebsocketConsumerStatusMessage = JSON.parse(ev['data'])
let isDelete = statusMessage.status === 'DELETED'
if (isDelete) {
this.documentDeletedSubject.next(true)
return
}
// fallback if backend didn't restrict message
if (
@@ -250,4 +256,8 @@ export class ConsumerStatusService {
onDocumentDetected() {
return this.documentDetectedSubject
}
onDocumentDeleted() {
return this.documentDeletedSubject
}
}