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:
+1
-3
@@ -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
@@ -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);
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user