cartodb4/dataservices-api
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Cached redis connections

Browse Source
This commit is contained in:
Mario de Frutos
2015-11-09 16:43:10 +01:00
parent 904336a298
commit a375e9dcfb
2 changed files with 57 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
server/lib/python/cartodb_geocoder/example/server_func_example.sql
View File

@@ -10,15 +10,21 @@ $$
LOG_FILENAME = '/tmp/plpython.log'
logging.basicConfig(filename=LOG_FILENAME,level=logging.DEBUG)
qs = quota_service.QuotaService(logging, user_id, tx_id)
if user_id in SD and tx_id in SD[user_id] and 'redis_connection' in SD[user_id][tx_id]:
logging.debug("Using redis cached connection...")
qs = quota_service.QuotaService(logging, user_id, tx_id, redis_connection=SD[user_id][tx_id]['redis_connection'])
else:
qs = quota_service.QuotaService(logging, user_id, tx_id, redis_host='localhost', redis_port=6379, redis_db=5)
if qs.check_user_quota():
result = plpy.execute("SELECT geom FROM geocode_admin0_polygons(Array[\'{0}\']::text[])".format(search))
logging.debug("Number of rows: {0} --- Status: {1}".format(result.nrows(), result.status()))
if result.status() == 5 and result.nrows() == 1:
qs.increment_georeference_use()
SD[user_id] = {tx_id: {'redis_connection': qs.get_redis_connection()}}
return result[0]["geom"]
else:
raise Exception('Something wrong with the georefence operation')
else:
raise Exception('Not enough quota for this user')
$$ LANGUAGE plpythonu;