fix: raw body upload (no base64), remove form parsing, bump 1.0.20
Deploy loadtest / validate (push) Waiting to run
Deploy loadtest / validate (push) Waiting to run
This commit is contained in:
+3
-27
@@ -1,13 +1,12 @@
|
|||||||
import time
|
import time
|
||||||
import json
|
import json
|
||||||
import base64
|
|
||||||
from flask import Flask, render_template, request
|
from flask import Flask, render_template, request
|
||||||
from flask_sock import Sock
|
from flask_sock import Sock
|
||||||
|
|
||||||
app = Flask(__name__)
|
app = Flask(__name__)
|
||||||
sock = Sock(app)
|
sock = Sock(app)
|
||||||
|
|
||||||
VERSION = '1.0.19'
|
VERSION = '1.0.20'
|
||||||
|
|
||||||
|
|
||||||
@app.route('/')
|
@app.route('/')
|
||||||
@@ -39,32 +38,9 @@ def ws_upload(ws):
|
|||||||
@app.route('/upload', methods=['POST'])
|
@app.route('/upload', methods=['POST'])
|
||||||
def upload():
|
def upload():
|
||||||
t0 = time.time()
|
t0 = time.time()
|
||||||
total = 0
|
# raw-режим: тело как есть (клиент шлёт файл напрямую через xhr.send(file))
|
||||||
# base64-режим: клиент отправляет поле "data"
|
raw = request.get_data()
|
||||||
data = request.form.get('data', '')
|
|
||||||
if data:
|
|
||||||
# убрать префикс data:...;base64,
|
|
||||||
if ',' in data:
|
|
||||||
data = data.split(',', 1)[1]
|
|
||||||
# form-encode превращает + в пробелы — восстановить
|
|
||||||
data = data.replace(' ', '+')
|
|
||||||
# добавить padding если потерялся
|
|
||||||
missing = len(data) % 4
|
|
||||||
if missing:
|
|
||||||
data += '=' * (4 - missing)
|
|
||||||
raw = base64.b64decode(data)
|
|
||||||
total = len(raw)
|
total = len(raw)
|
||||||
# файловый режим
|
|
||||||
elif request.files:
|
|
||||||
for f in request.files.values():
|
|
||||||
while True:
|
|
||||||
chunk = f.read(8192)
|
|
||||||
if not chunk:
|
|
||||||
break
|
|
||||||
total += len(chunk)
|
|
||||||
# raw-режим: тело как есть
|
|
||||||
else:
|
|
||||||
total = len(request.get_data())
|
|
||||||
elapsed = round((time.time() - t0) * 1000)
|
elapsed = round((time.time() - t0) * 1000)
|
||||||
return {'ok': True, 'bytes': total, 'elapsed_ms': elapsed}
|
return {'ok': True, 'bytes': total, 'elapsed_ms': elapsed}
|
||||||
|
|
||||||
|
|||||||
+4
-12
@@ -193,12 +193,9 @@ async function uploadAll() {
|
|||||||
fileSummary.textContent = summary;
|
fileSummary.textContent = summary;
|
||||||
}
|
}
|
||||||
|
|
||||||
/** uploadFileHTTP(f, idx) — загрузить файл через HTTP POST (base64, с retry) */
|
/** uploadFileHTTP(f, idx) — загрузить файл как raw body (без base64) */
|
||||||
function uploadFileHTTP(f, idx) {
|
function uploadFileHTTP(f, idx) {
|
||||||
return new Promise(function (resolve, reject) {
|
return new Promise(function (resolve, reject) {
|
||||||
var reader = new FileReader();
|
|
||||||
reader.onload = function () {
|
|
||||||
var dataUrl = reader.result;
|
|
||||||
var startTime = Date.now();
|
var startTime = Date.now();
|
||||||
var RETRY = 3;
|
var RETRY = 3;
|
||||||
|
|
||||||
@@ -206,6 +203,8 @@ function uploadFileHTTP(f, idx) {
|
|||||||
var xhr = new XMLHttpRequest();
|
var xhr = new XMLHttpRequest();
|
||||||
xhr.open('POST', '/upload');
|
xhr.open('POST', '/upload');
|
||||||
xhr.timeout = 300000;
|
xhr.timeout = 300000;
|
||||||
|
xhr.setRequestHeader('X-File-Name', encodeURIComponent(f.name));
|
||||||
|
xhr.setRequestHeader('X-File-Size', f.size);
|
||||||
|
|
||||||
xhr.upload.onprogress = function (e) {
|
xhr.upload.onprogress = function (e) {
|
||||||
if (e.lengthComputable) {
|
if (e.lengthComputable) {
|
||||||
@@ -234,17 +233,10 @@ function uploadFileHTTP(f, idx) {
|
|||||||
reject(new Error('Timeout'));
|
reject(new Error('Timeout'));
|
||||||
};
|
};
|
||||||
|
|
||||||
var form = new FormData();
|
xhr.send(f.file);
|
||||||
form.append('data', dataUrl);
|
|
||||||
form.append('name', f.name);
|
|
||||||
form.append('orig_size', f.size);
|
|
||||||
xhr.send(form);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
doUpload(1);
|
doUpload(1);
|
||||||
};
|
|
||||||
reader.onerror = function () { reject(new Error('Read error')); };
|
|
||||||
reader.readAsDataURL(f.file);
|
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user