1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148
|
import json
try:
# Python 2.x
import urllib2 as HTTPClient
from urllib2 import HTTPError
except ImportError:
# Python 3.x
import urllib.request as HTTPClient
from urllib.error import HTTPError
DEFAULT_ENDPOINT_URL = 'https://dc.services.visualstudio.com/v2/track'
class SenderBase(object):
"""The base class for all types of senders for use in conjunction with an implementation of :class:`QueueBase`.
The queue will notify the sender that it needs to pick up items. The concrete sender implementation will
listen to these notifications and will pull items from the queue getting at most :func:`send_buffer_size` items.
It will then call :func:`send` using the list of items pulled from the queue.
"""
def __init__(self, service_endpoint_uri):
"""Initializes a new instance of the class.
Args:
service_endpoint_uri (str) the address of the service to send telemetry data to.
"""
self._service_endpoint_uri = service_endpoint_uri
self._queue = None
self._send_buffer_size = 100
self._timeout = 10
@property
def service_endpoint_uri(self):
"""The HTTP or HTTPS endpoint that this sender will send data to.
Args:
value (str). the service endpoint URI.
Returns:
str. the service endpoint URI.
"""
return self._service_endpoint_uri
@service_endpoint_uri.setter
def service_endpoint_uri(self, value):
"""The service endpoint URI where this sender will send data to.
Args:
value (str). the service endpoint URI.
Returns:
str. the service endpoint URI.
"""
self._service_endpoint_uri = value
@property
def queue(self):
"""The queue that this sender is draining. While :class:`SenderBase` doesn't implement any means of doing
so, derivations of this class do.
Args:
value (:class:`QueueBase`). the queue instance that this sender is draining.
Returns:
:class:`QueueBase`. the queue instance that this sender is draining.
"""
return self._queue
@property
def send_timeout(self):
"""Time in seconds that the sender should wait before giving up."""
return self._timeout
@send_timeout.setter
def send_timeout(self, seconds):
"""Configures the timeout in seconds the sender waits for a response for the server.
Args:
seconds(float). Timeout in seconds.
"""
self._timeout = seconds
@queue.setter
def queue(self, value):
"""The queue that this sender is draining. While :class:`SenderBase` doesn't implement any means of doing
so, derivations of this class do.
Args:
value (:class:`QueueBase`). the queue instance that this sender is draining.
Returns:
:class:`QueueBase`. the queue instance that this sender is draining.
"""
self._queue = value
@property
def send_buffer_size(self):
"""The buffer size for a single batch of telemetry. This is the maximum number of items in a single service
request that this sender is going to send.
Args:
value (int). the maximum number of items in a telemetry batch.
Returns:
int. the maximum number of items in a telemetry batch.
"""
return self._send_buffer_size
@send_buffer_size.setter
def send_buffer_size(self, value):
"""The buffer size for a single batch of telemetry. This is the maximum number of items in a single service
request that this sender is going to send.
Args:
value (int). the maximum number of items in a telemetry batch.
Returns:
int. the maximum number of items in a telemetry batch.
"""
if value < 1:
value = 1
self._send_buffer_size = value
def send(self, data_to_send):
""" Immediately sends the data passed in to :func:`service_endpoint_uri`. If the service request fails, the
passed in items are pushed back to the :func:`queue`.
Args:
data_to_send (Array): an array of :class:`contracts.Envelope` objects to send to the service.
"""
request_payload = json.dumps([ a.write() for a in data_to_send ])
request = HTTPClient.Request(self._service_endpoint_uri, bytearray(request_payload, 'utf-8'), { 'Accept': 'application/json', 'Content-Type' : 'application/json; charset=utf-8' })
try:
response = HTTPClient.urlopen(request, timeout=self._timeout)
status_code = response.getcode()
if 200 <= status_code < 300:
return
except HTTPError as e:
if e.getcode() == 400:
return
except Exception as e:
pass
# Add our unsent data back on to the queue
for data in data_to_send:
self._queue.put(data)
|