voussoir/pixelcanvas
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add argument as_of to get chunk at or before a particular time.

Browse source
This commit is contained in:
voussoir 2020-02-11 15:38:04 -08:00
parent d649be2de5
commit f22746d389
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
pixelcanvas.py
View file

@ -88,17 +88,20 @@ def now():
# DB FUNCTIONS # DB FUNCTIONS
################################################################################ ################################################################################
def get_chunk_from_db(chunk_x, chunk_y): def get_chunk_from_db(chunk_x, chunk_y, as_of=None):
''' '''
Get the chunk from the database, and raise IndexError if it doesn't exist. Get the chunk from the database, and raise IndexError if it doesn't exist.
''' '''
query = ''' query = f'''
SELECT x, y, data FROM chunks SELECT x, y, data FROM chunks
WHERE x == ? AND y == ? WHERE x == ? AND y == ?
{'AND updated_at <= ?' if as_of is not None else ''}
ORDER BY updated_at DESC ORDER BY updated_at DESC
LIMIT 1 LIMIT 1
''' '''
bindings = [chunk_x, chunk_y] bindings = [chunk_x, chunk_y]
if as_of is not None:
bindings.append(as_of)
cur.execute(query, bindings) cur.execute(query, bindings)
fetch = cur.fetchone() fetch = cur.fetchone()
if fetch is None: if fetch is None:
@ -107,16 +110,16 @@ def get_chunk_from_db(chunk_x, chunk_y):
data = gzip.decompress(data) data = gzip.decompress(data)
return (x, y, data) return (x, y, data)
def get_chunk(chunk_x, chunk_y): def get_chunk(chunk_x, chunk_y, *args, **kwargs):
''' '''
Get the chunk from the database if it exists, or else download it. Get the chunk from the database if it exists, or else download it.
''' '''
try: try:
return get_chunk_from_db(chunk_x, chunk_y) return get_chunk_from_db(chunk_x, chunk_y, *args, **kwargs)
except IndexError: except IndexError:
(bigchunk_x, bigchunk_y) = chunk_to_bigchunk(chunk_x, chunk_y) (bigchunk_x, bigchunk_y) = chunk_to_bigchunk(chunk_x, chunk_y)
download_bigchunk(bigchunk_x, bigchunk_y) download_bigchunk(bigchunk_x, bigchunk_y)
return get_chunk_from_db(chunk_x, chunk_y) return get_chunk_from_db(chunk_x, chunk_y, *args, **kwargs)
def insert_chunk(chunk_x, chunk_y, data, commit=True): def insert_chunk(chunk_x, chunk_y, data, commit=True):
try: try: