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

This commit is contained in:
2026-07-11 00:20:13 +04:00
parent ada363b9ba
commit a2cd7bb88f
2 changed files with 20 additions and 22 deletions
+1 -3
View File
@@ -7,7 +7,7 @@ from flask_sock import Sock
app = Flask(__name__)
sock = Sock(app)
VERSION = '1.0.13'
VERSION = '1.0.14'
@app.route('/')
@@ -29,8 +29,6 @@ def ws_upload(ws):
chunk = ws.receive()
if chunk is None:
break
if isinstance(chunk, str) and chunk == 'DONE':
break
total += len(chunk)
elapsed = round((time.time() - t0) * 1000)
ws.send(json.dumps({'ok': True, 'bytes': total, 'elapsed_ms': elapsed}))
+19 -19
View File
@@ -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);
});
}