[~] 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
|
||||
|
||||
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:
|
||||
input_chunk = input_content.read(
|
||||
min(1024, input_content_length - sent_bytes)
|
||||
min(512 * 1024, input_content_length - sent_bytes)
|
||||
)
|
||||
p.stdin.write(
|
||||
input_chunk,
|
||||
@ -473,6 +483,33 @@ class Application:
|
||||
p.stdin.flush()
|
||||
|
||||
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()
|
||||
|
||||
try:
|
||||
|
Loading…
Reference in New Issue
Block a user