def import_shares(input):
for i, line in enumerate(input):
data = json.loads(line)
data['time'] = datetime.datetime.utcfromtimestamp(data['time'])
slc = ShareSlice(algo="scrypt", **data)
floored = DeviceSlice.floor_time(data['time'], data['span'])
if data['time'] != floored:
current_app.logger.warn("{} != {}".format(data['time'], floored))
data['time'] = floored
db.session.add(slc)
if i % 100 == 0:
print "{} completed".format(i)
db.session.commit()
def collect_minutes():
""" Grabs all the pending minute shares out of redis and puts them in the
database """
unproc_mins = redis_conn.keys("min_*")
for key in unproc_mins:
current_app.logger.info("Processing key {}".format(key))
share_type, algo, stamp = key.split("_")[1:]
minute = datetime.datetime.utcfromtimestamp(float(stamp))
# To ensure invalid stampt don't get committed
minute = ShareSlice.floor_time(minute, 0)
if stamp < (time.time() - 30):
current_app.logger.info("Skipping timestamp {}, too young"
.format(minute))
continue
redis_conn.rename(key, "processing_shares")
for user, shares in redis_conn.hgetall("processing_shares").iteritems():
shares = float(shares)
# messily parse out the worker/address combo...
parts = user.split(".")
if len(parts) > 0:
worker = parts[1]
else:
worker = ''
address = parts[0]
if address != "pool":
try:
curr = currencies.lookup_payable_addr(address)
except InvalidAddressException:
curr = None
if not curr:
address = global_config.pool_payout_currency.pool_payout_addr
try:
slc = ShareSlice(user=address, time=minute, worker=worker, algo=algo,
share_type=share_type, value=shares, span=0)
db.session.add(slc)
db.session.commit()
except sqlalchemy.exc.IntegrityError:
db.session.rollback()
slc = ShareSlice.query.with_lockmode('update').filter_by(
user=address, time=minute, worker=worker, algo=algo,
share_type=share_type).one()
slc.value += shares
db.session.commit()
redis_conn.delete("processing_shares")
def test_span(self):
start = datetime.datetime.utcnow()
shares = list(xrange(300))
for x in shares:
now = start - datetime.timedelta(minutes=x)
v = ShareSlice(time=now, value=x, **self.slice_test_data)
self.db.session.add(v)
self.db.session.commit()
lower, upper = make_upper_lower()
res = ShareSlice.get_span(stamp=True, lower=lower, upper=upper)
self.assertEqual(len(res[0]['values']), 10)
self.assertEqual(sum(res[0]['values'].values()), 45)
res = ShareSlice.get_span(stamp=True)
self.assertEqual(sum(res[0]['values'].values()), 44850)
def test_compress(self):
start = datetime.datetime.utcnow().replace(second=0, microsecond=0)
for x in xrange(1500):
now = start - datetime.timedelta(minutes=x)
v = ShareSlice(time=now, value=x, **self.slice_test_data)
self.db.session.add(v)
self.db.session.commit()
ShareSlice.compress(0)
self.db.session.commit()
lower, upper = make_upper_lower()
res = ShareSlice.get_span(stamp=True, lower=lower, upper=upper)
self.assertEqual(len(res[0]['values']), 10)
self.assertEqual(sum(res[0]['values'].values()), 45)
res = ShareSlice.get_span(stamp=True)
print(res)
self.assertEqual(sum(res[0]['values'].values()), 1124250)
spans = {0: 0, 1: 0, 2: 0}
for slc in ShareSlice.get_span(ret_query=True):
spans[slc.span] += 1
print spans
assert spans[0] <= 65
assert spans[1] >= 48
ShareSlice.compress(1)
self.db.session.commit()
spans = {0: 0, 1: 0, 2: 0}
for slc in ShareSlice.get_span(ret_query=True):
spans[slc.span] += 1
print spans
assert spans[0] <= 65
assert spans[1] <= 288
assert spans[2] >= 1
res = ShareSlice.get_span(stamp=True)
self.assertEqual(sum(res[0]['values'].values()), 1124250)
