fix: close WS on client after last chunk, no DONE needed, bump 1.0.13→1.0.14
Deploy loadtest / validate (push) Waiting to run
Deploy loadtest / validate (push) Waiting to run
This commit is contained in:
+19
-19
@@ -200,25 +200,20 @@ async function uploadAll() {
|
||||
function uploadFileWS(url, f, idx) {
|
||||
return new Promise(function (resolve, reject) {
|
||||
var ws = new WebSocket(url);
|
||||
var CHUNK = 64 * 1024; // 64KB chunks
|
||||
var CHUNK = 64 * 1024;
|
||||
var offset = 0;
|
||||
var startTime = Date.now();
|
||||
var lastProgress = 0;
|
||||
var timer;
|
||||
|
||||
ws.binaryType = 'arraybuffer';
|
||||
|
||||
ws.onopen = function () {
|
||||
// Шлём заголовок: имя файла + размер
|
||||
ws.send(JSON.stringify({ name: f.name, size: f.size }));
|
||||
|
||||
// Читаем и шлём чанки
|
||||
var reader = new FileReader();
|
||||
reader.onload = function (e) {
|
||||
ws.send(e.target.result);
|
||||
offset += CHUNK;
|
||||
|
||||
if (offset < f.size) {
|
||||
// Прогресс каждые 10%
|
||||
var pct = Math.round(offset / f.size * 100);
|
||||
if (pct - lastProgress >= 10) {
|
||||
lastProgress = pct;
|
||||
@@ -227,41 +222,46 @@ function uploadFileWS(url, f, idx) {
|
||||
}
|
||||
readNext();
|
||||
} else {
|
||||
// Все чанки отправлены — шлём сигнал завершения
|
||||
ws.send('DONE');
|
||||
// Все чанки ушли — закрываем, сервер поймёт что конец
|
||||
f.progress = 100;
|
||||
renderTable();
|
||||
ws.close();
|
||||
}
|
||||
};
|
||||
|
||||
function readNext() {
|
||||
var end = Math.min(offset + CHUNK, f.size);
|
||||
reader.readAsArrayBuffer(f.file.slice(offset, end));
|
||||
}
|
||||
|
||||
readNext();
|
||||
};
|
||||
|
||||
ws.onmessage = function (e) {
|
||||
try {
|
||||
var r = JSON.parse(e.data);
|
||||
var elapsed = Date.now() - startTime;
|
||||
clearTimeout(timer);
|
||||
f.progress = 100;
|
||||
renderTable();
|
||||
ws.close();
|
||||
resolve({ bytes: r.bytes, elapsed_ms: elapsed });
|
||||
resolve({ bytes: r.bytes, elapsed_ms: Date.now() - startTime });
|
||||
} catch (err) {
|
||||
reject(new Error('Bad response'));
|
||||
}
|
||||
};
|
||||
|
||||
ws.onerror = function () {
|
||||
reject(new Error('WebSocket error'));
|
||||
};
|
||||
ws.onerror = function () { reject(new Error('WebSocket error')); };
|
||||
|
||||
ws.onclose = function (e) {
|
||||
if (e.code !== 1000 && offset < f.size) {
|
||||
reject(new Error('WS closed (' + e.code + ')'));
|
||||
ws.onclose = function () {
|
||||
// Если файл не до конца — ошибка
|
||||
if (offset < f.size) {
|
||||
reject(new Error('WS closed early'));
|
||||
}
|
||||
};
|
||||
|
||||
// Таймаут на случай зависания
|
||||
timer = setTimeout(function () {
|
||||
ws.close();
|
||||
reject(new Error('Timeout'));
|
||||
}, 300000);
|
||||
});
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user