Extracted redis connection and quota check to be reusable

server/extension/sql/0.0.1/11_redis_helper.sql

@@ -0,0 +1,11 @@
+-- Get the connection to redis from cache or create a new one
+CREATE OR REPLACE FUNCTION cdb_geocoder_server._connect_to_redis(user_id name)
+RETURNS boolean AS $$
+  if user_id in GD and 'redis_connection' in GD[user_id]:
+    return False
+  else:
+    from cartodb_geocoder import redis_helper
+    redis_connection = redis_helper.RedisHelper('localhost', 6379, 5).redis_connection()
+    GD[user_id] = {'redis_connection': redis_connection}
+    return True
+$$ LANGUAGE plpythonu;

server/extension/sql/0.0.1/30_admin0.sql

@@ -11,14 +11,23 @@ RETURNS Geometry AS $$
         plpy.error('The api_key must be provided')
 
     #--TODO: rate limiting check
-    #--TODO: quota check
+    #--This will create and cache a redis connection, if needed, in the GD object for the current user
+    redis_conn_plan = plpy.prepare("SELECT cdb_geocoder_server._connect_to_redis($1)", ["name"])
+    redis_conn_result = plpy.execute(redis_conn_plan, [user_id], 1)
+    qs = quota_service.QuotaService(user_id, tx_id, GD[user_id]['redis_connection'])
+
+    if not qs.check_user_quota():
+      plpy.error("Not enough quota for this user")
 
     #-- Copied from the doc, see http://www.postgresql.org/docs/9.4/static/plpython-database.html
     plan = plpy.prepare("SELECT cdb_geocoder_server._geocode_admin0_polygon($1) AS mypolygon", ["text"])
-    rv = plpy.execute(plan, [country_name], 1)
-
-    plpy.debug('Returning from Returning from geocode_admin0_polygons')
-    return rv[0]["mypolygon"]
+    result = plpy.execute(plan, [country_name], 1)
+    if result.status() == 5 and result.nrows() == 1:
+      qs.increment_geocoder_use()
+      plpy.debug('Returning from geocode_admin0_polygons')
+      return result[0]["mypolygon"]
+    else:
+      plpy.error('Something wrong with the georefence operation')
 $$ LANGUAGE plpythonu;