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__) app = Flask(__name__)
sock = Sock(app) sock = Sock(app)
VERSION = '1.0.13' VERSION = '1.0.14'
@app.route('/') @app.route('/')
@@ -29,8 +29,6 @@ def ws_upload(ws):
chunk = ws.receive() chunk = ws.receive()
if chunk is None: if chunk is None:
break break
if isinstance(chunk, str) and chunk == 'DONE':
break
total += len(chunk) total += len(chunk)
elapsed = round((time.time() - t0) * 1000) elapsed = round((time.time() - t0) * 1000)
ws.send(json.dumps({'ok': True, 'bytes': total, 'elapsed_ms': elapsed})) 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) { function uploadFileWS(url, f, idx) {
return new Promise(function (resolve, reject) { return new Promise(function (resolve, reject) {
var ws = new WebSocket(url); var ws = new WebSocket(url);
var CHUNK = 64 * 1024; // 64KB chunks var CHUNK = 64 * 1024;
var offset = 0; var offset = 0;
var startTime = Date.now(); var startTime = Date.now();
var lastProgress = 0; var lastProgress = 0;
var timer;
ws.binaryType = 'arraybuffer'; ws.binaryType = 'arraybuffer';
ws.onopen = function () { ws.onopen = function () {
// Шлём заголовок: имя файла + размер
ws.send(JSON.stringify({ name: f.name, size: f.size }));
// Читаем и шлём чанки
var reader = new FileReader(); var reader = new FileReader();
reader.onload = function (e) { reader.onload = function (e) {
ws.send(e.target.result); ws.send(e.target.result);
offset += CHUNK; offset += CHUNK;
if (offset < f.size) { if (offset < f.size) {
// Прогресс каждые 10%
var pct = Math.round(offset / f.size * 100); var pct = Math.round(offset / f.size * 100);
if (pct - lastProgress >= 10) { if (pct - lastProgress >= 10) {
lastProgress = pct; lastProgress = pct;
@@ -227,41 +222,46 @@ function uploadFileWS(url, f, idx) {
} }
readNext(); readNext();
} else { } else {
// Все чанки отправлены — шлём сигнал завершения // Все чанки ушли — закрываем, сервер поймёт что конец
ws.send('DONE'); f.progress = 100;
renderTable();
ws.close();
} }
}; };
function readNext() { function readNext() {
var end = Math.min(offset + CHUNK, f.size); var end = Math.min(offset + CHUNK, f.size);
reader.readAsArrayBuffer(f.file.slice(offset, end)); reader.readAsArrayBuffer(f.file.slice(offset, end));
} }
readNext(); readNext();
}; };
ws.onmessage = function (e) { ws.onmessage = function (e) {
try { try {
var r = JSON.parse(e.data); var r = JSON.parse(e.data);
var elapsed = Date.now() - startTime; clearTimeout(timer);
f.progress = 100; f.progress = 100;
renderTable(); renderTable();
ws.close(); ws.close();
resolve({ bytes: r.bytes, elapsed_ms: elapsed }); resolve({ bytes: r.bytes, elapsed_ms: Date.now() - startTime });
} catch (err) { } catch (err) {
reject(new Error('Bad response')); reject(new Error('Bad response'));
} }
}; };
ws.onerror = function () { ws.onerror = function () { reject(new Error('WebSocket error')); };
reject(new Error('WebSocket error'));
};
ws.onclose = function (e) { ws.onclose = function () {
if (e.code !== 1000 && offset < f.size) { // Если файл не до конца — ошибка
reject(new Error('WS closed (' + e.code + ')')); if (offset < f.size) {
reject(new Error('WS closed early'));
} }
}; };
// Таймаут на случай зависания
timer = setTimeout(function () {
ws.close();
reject(new Error('Timeout'));
}, 300000);
}); });
} }