Madiator2011/lbry-sdk
1
0
Fork 0
forked from LBRYCommunity/lbry-sdk
Code Pull requests Activity

add head_blob_only to ttfb

Browse source
This commit is contained in:
Victor Shyba 2019-05-01 13:59:45 -03:00
parent 895b3992c9
commit fc88261cc6
1 changed files with 14 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
scripts/time_to_first_byte.py
View file

@ -118,23 +118,24 @@ async def main(uris=None, cmd_args=None):
first_byte = time.time() first_byte = time.time()
first_byte_times.append(first_byte - start) first_byte_times.append(first_byte - start)
print(f"{i + 1}/{len(resolvable)} - {first_byte - start} {uri}") print(f"{i + 1}/{len(resolvable)} - {first_byte - start} {uri}")
downloaded, amount_downloaded, blobs_in_stream = await wait_for_done( if not cmd_args.head_blob_only:
conf, uri, cmd_args.stall_download_timeout downloaded, amount_downloaded, blobs_in_stream = await wait_for_done(
) conf, uri, cmd_args.stall_download_timeout
if downloaded: )
download_successes.append(uri) if downloaded:
else: download_successes.append(uri)
download_failures.append(uri) else:
mbs = round((blobs_in_stream * (MAX_BLOB_SIZE - 1)) / (time.time() - start) / 1000000, 2) download_failures.append(uri)
download_speeds.append(mbs) mbs = round((blobs_in_stream * (MAX_BLOB_SIZE - 1)) / (time.time() - start) / 1000000, 2)
print(f"downloaded {amount_downloaded}/{blobs_in_stream} blobs for {uri} at " download_speeds.append(mbs)
f"{mbs}mb/s") print(f"downloaded {amount_downloaded}/{blobs_in_stream} blobs for {uri} at "
f"{mbs}mb/s")
except Exception as e: except Exception as e:
print(f"{i + 1}/{len(uris)} - failed to start {uri}: {e}") print(f"{i + 1}/{len(uris)} - failed to start {uri}: {e}")
failed_to_start.append(uri) failed_to_start.append(uri)
if cmd_args.exit_on_error: if cmd_args.exit_on_error:
return return
if cmd_args.delete_after_download: if cmd_args.delete_after_download or cmd_args.head_blob_only:
await daemon_rpc(conf, 'file_delete', delete_from_download_dir=True, claim_name=parse_lbry_uri(uri).name) await daemon_rpc(conf, 'file_delete', delete_from_download_dir=True, claim_name=parse_lbry_uri(uri).name)
await asyncio.sleep(0.1) await asyncio.sleep(0.1)
@ -171,4 +172,5 @@ if __name__ == "__main__":
parser.add_argument("--exit_on_error", action='store_true') parser.add_argument("--exit_on_error", action='store_true')
parser.add_argument("--stall_download_timeout", default=10, type=int) parser.add_argument("--stall_download_timeout", default=10, type=int)
parser.add_argument("--delete_after_download", action='store_true') parser.add_argument("--delete_after_download", action='store_true')
parser.add_argument("--head_blob_only", action='store_true')
asyncio.run(main(cmd_args=parser.parse_args())) asyncio.run(main(cmd_args=parser.parse_args()))