mirror of
https://github.com/jpetazzo/container.training.git
synced 2026-03-03 01:40:19 +00:00
85 lines
2.4 KiB
Python
85 lines
2.4 KiB
Python
import http.client
|
|
import logging
|
|
import os
|
|
from redis import Redis
|
|
import time
|
|
|
|
DEBUG = os.environ.get("DEBUG", "").lower().startswith("y")
|
|
|
|
log = logging.getLogger(__name__)
|
|
level = logging.DEBUG if DEBUG else logging.INFO
|
|
logging.basicConfig(level=level)
|
|
|
|
|
|
redis = Redis("redis")
|
|
|
|
|
|
def get_random_bytes():
|
|
log.debug("Connecting to RNG")
|
|
connection = http.client.HTTPConnection("rng")
|
|
log.debug("Sending request")
|
|
connection.request("GET", "/1000")
|
|
response = connection.getresponse()
|
|
random_bytes = response.read()
|
|
log.debug("Got {} bytes of random data".format(len(random_bytes)))
|
|
return random_bytes
|
|
|
|
|
|
def hash_bytes(data):
|
|
log.debug("Connecting to HASHER")
|
|
connection = http.client.HTTPConnection("hasher")
|
|
log.debug("Sending request")
|
|
connection.request(
|
|
"POST", "/", data, {"Content-Type": "application/octet-stream"})
|
|
response = connection.getresponse()
|
|
hex_hash = response.read()
|
|
log.debug("Got hash: {}...".format(hex_hash[:8]))
|
|
return hex_hash
|
|
|
|
|
|
def work_loop(interval=1):
|
|
deadline = 0
|
|
loops_done = 0
|
|
old_total = 0
|
|
while True:
|
|
if time.time() > deadline:
|
|
new_total = redis.incrby("hashes", loops_done)
|
|
deadline = time.time() + interval
|
|
loops_done = 0
|
|
log.info("Current apparent speed: {}h/s"
|
|
.format((new_total-old_total)/interval))
|
|
old_total = new_total
|
|
work_once()
|
|
loops_done += 1
|
|
|
|
|
|
def work_once():
|
|
log.debug("Doing one unit of work")
|
|
random_bytes = get_random_bytes()
|
|
hex_hash = hash_bytes(random_bytes)
|
|
if not hex_hash.startswith(b'0'):
|
|
log.debug("No coin found")
|
|
return
|
|
log.info("Coin found: {}...".format(hex_hash[:8]))
|
|
created = redis.hset("wallet", hex_hash, random_bytes)
|
|
if not created:
|
|
log.info("We already had that coin")
|
|
return
|
|
log.debug("Storing timing information")
|
|
now = time.time()
|
|
redis.rpush("timing", now)
|
|
redis.ltrim("timing", -1000, -1)
|
|
log.debug("Getting timing information")
|
|
oldest = float(redis.lrange("timing", 0, 0)[0])
|
|
total = redis.llen("timing")
|
|
if oldest == now:
|
|
log.debug("oldest == now, can't compute timing")
|
|
return
|
|
speed = total / (now - oldest)
|
|
log.info("Speed over the last {} coins: {} coins/s".format(total, speed))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
work_loop()
|
|
|