[~] Refactor
This commit is contained in:
parent
f3ca6933a9
commit
f928647c56
39
d1/f2.py
39
d1/f2.py
@ -463,9 +463,19 @@ class Application:
|
|||||||
returncode = None
|
returncode = None
|
||||||
|
|
||||||
sent_bytes = 0
|
sent_bytes = 0
|
||||||
|
self.op1(
|
||||||
|
json_data=dict(
|
||||||
|
input_content_length=input_content_length,
|
||||||
|
action='curl_stdin',
|
||||||
|
step='started',
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
started_at = datetime.datetime.now()
|
||||||
|
|
||||||
while sent_bytes < input_content_length and p.poll() is None:
|
while sent_bytes < input_content_length and p.poll() is None:
|
||||||
input_chunk = input_content.read(
|
input_chunk = input_content.read(
|
||||||
min(1024, input_content_length - sent_bytes)
|
min(512 * 1024, input_content_length - sent_bytes)
|
||||||
)
|
)
|
||||||
p.stdin.write(
|
p.stdin.write(
|
||||||
input_chunk,
|
input_chunk,
|
||||||
@ -473,6 +483,33 @@ class Application:
|
|||||||
p.stdin.flush()
|
p.stdin.flush()
|
||||||
|
|
||||||
sent_bytes += len(input_chunk)
|
sent_bytes += len(input_chunk)
|
||||||
|
|
||||||
|
elapsed = (datetime.datetime.now() - started_at).total_seconds()
|
||||||
|
|
||||||
|
if (
|
||||||
|
elapsed > Application.MAX_TIME or \
|
||||||
|
elapsed > 1 and sent_bytes == 0
|
||||||
|
):
|
||||||
|
self.op1(
|
||||||
|
json_data=dict(
|
||||||
|
input_content_length=input_content_length,
|
||||||
|
action='curl_stdin',
|
||||||
|
step='read_timeout',
|
||||||
|
chunk_len=len(input_chunk),
|
||||||
|
sent_byte=sent_bytes,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
break
|
||||||
|
|
||||||
|
self.op1(
|
||||||
|
json_data=dict(
|
||||||
|
input_content_length=input_content_length,
|
||||||
|
action='curl_stdin',
|
||||||
|
step='done',
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
p.stdin.close()
|
p.stdin.close()
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
Loading…
Reference in New Issue
Block a user