File size: 1,794 Bytes
3e05655 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | export function createLatestWorkerQueue<T extends { key: string }>(input: {
run: (request: T) => Promise<void>
supersede: (request: T) => void
dispose: (key: string) => void
}) {
type Slot = { type: "highlight"; key: string; request?: T }
const jobs: Array<Slot | { type: "dispose"; key: string }> = []
const slots = new Map<string, Slot>()
let running: Promise<void> | undefined
let cursor = 0
const schedule = () => {
if (running) return
running = Promise.resolve()
.then(async () => {
while (cursor < jobs.length) {
const job = jobs[cursor++]!
if (job.type === "dispose") {
input.dispose(job.key)
continue
}
if (slots.get(job.key) === job) slots.delete(job.key)
const request = job.request
job.request = undefined
if (request) await input.run(request)
}
})
.finally(() => {
jobs.splice(0, cursor)
cursor = 0
running = undefined
if (jobs.length > 0) schedule()
})
}
return {
highlight(request: T) {
const slot = slots.get(request.key)
if (slot) {
if (slot.request) input.supersede(slot.request)
slot.request = request
return
}
const next: Slot = { type: "highlight", key: request.key, request }
slots.set(request.key, next)
jobs.push(next)
schedule()
},
dispose(key: string) {
const slot = slots.get(key)
if (slot?.request) input.supersede(slot.request)
if (slot) {
slot.request = undefined
slots.delete(key)
}
jobs.push({ type: "dispose", key })
schedule()
},
pending: () => slots.size,
async idle() {
while (running) await running
},
}
}
|