Added QPS decorator

This commit is contained in:
Antonio
2017-12-27 14:55:50 +01:00
parent 8c0af7d51d
commit dabd596271
5 changed files with 78 additions and 0 deletions
@@ -7,6 +7,7 @@ import requests
from mapbox import Geocoder
from cartodb_services.metrics import Traceable
from cartodb_services.mapbox.exceptions import ServiceException
from cartodb_services.tools.qps import qps_retry
ACCESS_TOKEN = 'pk.eyJ1IjoiYWNhcmxvbiIsImEiOiJjamJuZjQ1Zjc0Ymt4Mnh0YmFrMmhtYnY4In0.gt9cw0VeKc3rM2mV5pcEmg'
@@ -48,6 +49,7 @@ class MapboxGeocoder(Traceable):
latitude = location[1]
return [longitude, latitude]
@qps_retry(qps=10)
def geocode(self, address, country=None):
response = self._geocoder.forward(address=address,
country=country,
@@ -7,6 +7,7 @@ from cartodb_services.metrics import Traceable
from cartodb_services.tools.coordinates import (validate_coordinates,
marshall_coordinates)
from exceptions import ServiceException
from cartodb_services.tools.qps import qps_retry
ACCESS_TOKEN = 'pk.eyJ1IjoiYWNhcmxvbiIsImEiOiJjamJuZjQ1Zjc0Ymt4Mnh0YmFrMmhtYnY4In0.gt9cw0VeKc3rM2mV5pcEmg'
@@ -57,6 +58,7 @@ class MapboxMatrixClient(Traceable):
def _parse_matrix_response(self, response):
return response
@qps_retry(qps=1)
def matrix(self, coordinates, profile=DEFAULT_PROFILE):
validate_profile(profile)
validate_coordinates(coordinates,
@@ -9,6 +9,7 @@ from cartodb_services.tools import PolyLine
from cartodb_services.tools.coordinates import (validate_coordinates,
marshall_coordinates)
from cartodb_services.mapbox.exceptions import ServiceException
from cartodb_services.tools.qps import qps_retry
ACCESS_TOKEN = 'pk.eyJ1IjoiYWNhcmxvbiIsImEiOiJjamJuZjQ1Zjc0Ymt4Mnh0YmFrMmhtYnY4In0.gt9cw0VeKc3rM2mV5pcEmg'
@@ -70,6 +71,7 @@ class MapboxRouting(Traceable):
return MapboxRoutingResponse(geometry, distance, duration)
@qps_retry(qps=1)
def directions(self, waypoints, profile=DEFAULT_PROFILE):
self._validate_profile(profile)
validate_coordinates(waypoints, NUM_WAYPOINTS_MIN, NUM_WAYPOINTS_MAX)