135 lines
3.1 KiB
Plaintext
135 lines
3.1 KiB
Plaintext
import * as cornerstone from 'cornerstone-core'
|
|
var taskPool = [] // 请求池
|
|
let numRequest = 0 // 正在执行数量
|
|
const maxRequest = 36 // 可配置
|
|
let taskTimer // 轮询的定时器
|
|
var cachedTask = {} // 存放的任务数据
|
|
|
|
// 执行下载
|
|
async function executeTask() {
|
|
// stopTaskTimer()
|
|
if (taskPool.length > 0) {
|
|
|
|
sortTaskPool()
|
|
var requestNum = Math.min(taskPool.length, maxRequest)
|
|
const tasks = taskPool.splice(0, requestNum);
|
|
|
|
const taskPromises = tasks.map((task) => {
|
|
return task.execute().then((result) => {
|
|
task.callback({ success: true, res: result })
|
|
|
|
}).catch((error) => {
|
|
task.callback({ success: true, err: error })
|
|
})
|
|
})
|
|
await Promise.all(taskPromises)
|
|
executeTask()
|
|
}else{
|
|
startTaskTimer()
|
|
}
|
|
|
|
}
|
|
|
|
function sortTaskPool() {
|
|
if (taskPool.length > 0) {
|
|
taskPool.sort((a, b) => b.priority-a.priority )
|
|
|
|
}
|
|
}
|
|
|
|
// 轮询检查请求池中是否有请求需要执行
|
|
function startTaskTimer() {
|
|
if (taskTimer) return; // 如果已存在定时器，则不重复启动
|
|
|
|
taskTimer = setInterval(() => {
|
|
if (taskPool.length > 0) {
|
|
stopTaskTimer();
|
|
executeTask();
|
|
}
|
|
}, 50);
|
|
}
|
|
|
|
// 停止轮询
|
|
function stopTaskTimer() {
|
|
clearInterval(taskTimer)
|
|
taskTimer = null
|
|
}
|
|
function loadAndCacheImagePlus(imageId,seriesId, priority) {
|
|
return new Promise((resolve, reject) => {
|
|
const imageLoadObject = cornerstone.imageCache.getImageLoadObject(imageId)
|
|
if (imageLoadObject) {
|
|
imageLoadObject.promise.then((image) => {
|
|
resolve(image)
|
|
}, (err) => {
|
|
reject(err)
|
|
})
|
|
} else {
|
|
const imageTask = buildImageRequestTask(imageId,seriesId, { priority })
|
|
addTaskIntoPool(imageTask).then((res) => {
|
|
resolve(res)
|
|
}).catch(e => {
|
|
reject(e)
|
|
})
|
|
}
|
|
})
|
|
}
|
|
// 预加载池的添加
|
|
function addTaskIntoPool(task) {
|
|
return new Promise((resolve, reject) => {
|
|
const callback = (executeRes) => {
|
|
if (executeRes.success) {
|
|
resolve(executeRes.res)
|
|
} else {
|
|
reject(executeRes.err)
|
|
}
|
|
}
|
|
task.callback =callback
|
|
taskPool.push(task)
|
|
})
|
|
}
|
|
function removeTask(seriesId){
|
|
stopTaskTimer()
|
|
if (taskPool.length > 0) {
|
|
for(var i =taskPool.length-1; i>=0;i--){
|
|
if(taskPool[i] && taskPool[i].seriesId === seriesId){
|
|
delete cachedTask[taskPool[i].key]
|
|
taskPool.splice(i,1)
|
|
|
|
}
|
|
}
|
|
|
|
}
|
|
startTaskTimer()
|
|
}
|
|
function resetRequestPool(){
|
|
taskPool = []
|
|
pendingList =[]
|
|
cachedTask = {}
|
|
}
|
|
function resetCachedTask(){
|
|
taskPool = []
|
|
pendingList =[]
|
|
cachedTask = {}
|
|
}
|
|
function buildImageRequestTask(imageId,seriesId, config = {}) {
|
|
return {
|
|
key: imageId,
|
|
seriesId:seriesId,
|
|
...config,
|
|
execute: () => {
|
|
return cornerstone.loadAndCacheImage(imageId)
|
|
}
|
|
};
|
|
}
|
|
|
|
export default {
|
|
addTaskIntoPool,
|
|
executeTask,
|
|
startTaskTimer,
|
|
stopTaskTimer,
|
|
loadAndCacheImagePlus,
|
|
removeTask,
|
|
resetRequestPool,
|
|
resetCachedTask
|
|
}
|