Skip to content

Commit 85eb2e7

Browse files
committed
tracer: allow host/port to be configured
1 parent faed10b commit 85eb2e7

File tree

5 files changed

+10
-11
lines changed

5 files changed

+10
-11
lines changed

ddtrace/api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11

22
# stdlib
3-
import httplib
3+
from .compat import httplib
44
import logging
55
import time
66

ddtrace/contrib/django/apps.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,8 @@ def ready(self):
3838
# AgentWriter, it breaks all tests. The configure() behavior must
3939
# be changed to use it in this integration
4040
tracer.enabled = settings.ENABLED
41-
tracer.writer._reporter.transport.hostname = settings.AGENT_HOSTNAME
42-
tracer.writer._reporter.transport.port = settings.AGENT_PORT
41+
tracer.writer.api.hostname = settings.AGENT_HOSTNAME
42+
tracer.writer.api.port = settings.AGENT_PORT
4343

4444
if settings.AUTO_INSTRUMENT:
4545
# trace Django internals

ddtrace/writer.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def __init__(self, hostname='localhost', port=7777):
2626
self._traces = None
2727
self._services = None
2828
self._worker = None
29-
self._api = api.API(hostname, port)
29+
self.api = api.API(hostname, port)
3030

3131
def write(self, spans=None, services=None):
3232
# if the worker needs to be reset, do it.
@@ -51,7 +51,7 @@ def _reset_worker(self):
5151

5252
# ensure we have an active thread working on this queue
5353
if not self._worker or not self._worker.is_alive():
54-
self._worker = AsyncWorker(self._api, self._traces, self._services)
54+
self._worker = AsyncWorker(self.api, self._traces, self._services)
5555

5656

5757
class AsyncWorker(object):
@@ -62,7 +62,7 @@ def __init__(self, api, trace_queue, service_queue, shutdown_timeout=DEFAULT_TIM
6262
self._lock = threading.Lock()
6363
self._thread = None
6464
self._shutdown_timeout = shutdown_timeout
65-
self._api = api
65+
self.api = api
6666
self.start()
6767

6868
def is_alive(self):
@@ -101,14 +101,14 @@ def _target(self):
101101
if traces:
102102
# If we have data, let's try to send it.
103103
try:
104-
self._api.send_traces(traces)
104+
self.api.send_traces(traces)
105105
except Exception:
106106
log.exception("error sending spans")
107107

108108
services = self._service_queue.pop()
109109
if services:
110110
try:
111-
self._api.send_services(services)
111+
self.api.send_services(services)
112112
except Exception:
113113
log.exception("error sending spans")
114114

tests/contrib/django/test_instrumentation.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,5 +14,5 @@ class DjangoInstrumentationTest(DjangoTraceTestCase):
1414
users settings
1515
"""
1616
def test_enabled_flag(self):
17-
eq_(self.tracer.writer._reporter.transport.hostname, 'agent.service.consul')
18-
eq_(self.tracer.writer._reporter.transport.port, '8777')
17+
eq_(self.tracer.writer.api.hostname, 'agent.service.consul')
18+
eq_(self.tracer.writer.api.port, '8777')

tests/test_tracer.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
from ddtrace import encoding
1111
from ddtrace.tracer import Tracer
1212
from ddtrace.writer import AgentWriter
13-
from ddtrace.transport import ThreadedHTTPTransport
1413

1514

1615
def test_tracer_vars():

0 commit comments

Comments
 (0)