diff options
| -rw-r--r-- | tools/feeder.py | 19 |
1 files changed, 12 insertions, 7 deletions
diff --git a/tools/feeder.py b/tools/feeder.py index de61ee2..bc3c37e 100644 --- a/tools/feeder.py +++ b/tools/feeder.py @@ -5,7 +5,7 @@ import ipaddress import datetime import itertools import json -import multiprocessing.dummy as mp +import threading data = { "ip": "", @@ -28,11 +28,13 @@ data = { def random_ipv4(): randint = random.randint(1, 0xffffffff) + return str(ipaddress.IPv4Address(randint)) def random_ipv6(): randint = random.randint(0, 2**128-1) + return str(ipaddress.IPv6Address(randint)) @@ -42,6 +44,7 @@ def random_port(): def random_asn(): randint = random.randint(1, 65536) + return f'AS{randint}' @@ -116,6 +119,7 @@ def random_tld(): def get_timestamp(): utc_datetime = datetime.datetime.utcnow() + return utc_datetime.strftime("%Y-%m-%dT%H:%M:%SUTC") @@ -159,17 +163,18 @@ def generate_blob(): return blob -def send_blob(url): +def send_blob(): + url = 'http://localhost:8000/sc/v0/add' + + print("Thread started") try: for i in range(1200): requests.post(url, json=generate_blob()) + print("Blob sent.") except Exception: print('Failed to send blob.') if __name__ == '__main__': - threads = mp.Pool(8) - - threads.map(send_blob, ['http://localhost:8000/sc/v0/add']) - threads.close() - threads.join() + for _ in range(0, 7): + threading.Thread(target=lambda: send_blob()).start() |