Skip to content

Commit 8bcadcd

Browse files
committed
fix: typo and run tox
1 parent 51b932e commit 8bcadcd

File tree

7 files changed

+64
-34
lines changed

7 files changed

+64
-34
lines changed

instrumentation/README.md

+1
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
| [opentelemetry-instrumentation-pymysql](./opentelemetry-instrumentation-pymysql) | PyMySQL < 2 |
2929
| [opentelemetry-instrumentation-pyramid](./opentelemetry-instrumentation-pyramid) | pyramid >= 1.7 |
3030
| [opentelemetry-instrumentation-redis](./opentelemetry-instrumentation-redis) | redis >= 2.6 |
31+
| [opentelemetry-instrumentation-remoulade](./opentelemetry-instrumentation-remoulade) | remoulade >= 0.49 |
3132
| [opentelemetry-instrumentation-requests](./opentelemetry-instrumentation-requests) | requests ~= 2.0 |
3233
| [opentelemetry-instrumentation-sklearn](./opentelemetry-instrumentation-sklearn) | scikit-learn ~= 0.24.0 |
3334
| [opentelemetry-instrumentation-sqlalchemy](./opentelemetry-instrumentation-sqlalchemy) | sqlalchemy |

instrumentation/opentelemetry-instrumentation-remoulade/setup.py

+12-2
Original file line numberDiff line numberDiff line change
@@ -39,13 +39,23 @@
3939
PACKAGE_INFO = {}
4040

4141
VERSION_FILENAME = os.path.join(
42-
BASE_DIR, "src", "opentelemetry", "instrumentation", "remoulade", "version.py"
42+
BASE_DIR,
43+
"src",
44+
"opentelemetry",
45+
"instrumentation",
46+
"remoulade",
47+
"version.py",
4348
)
4449
with open(VERSION_FILENAME, encoding="utf-8") as f:
4550
exec(f.read(), PACKAGE_INFO)
4651

4752
PACKAGE_FILENAME = os.path.join(
48-
BASE_DIR, "src", "opentelemetry", "instrumentation", "remoulade", "package.py"
53+
BASE_DIR,
54+
"src",
55+
"opentelemetry",
56+
"instrumentation",
57+
"remoulade",
58+
"package.py",
4959
)
5060
with open(PACKAGE_FILENAME, encoding="utf-8") as f:
5161
exec(f.read(), PACKAGE_INFO)

instrumentation/opentelemetry-instrumentation-remoulade/src/opentelemetry/instrumentation/remoulade/__init__.py

+42-14
Original file line numberDiff line numberDiff line change
@@ -4,14 +4,13 @@
44
from remoulade import Middleware, broker
55

66
from opentelemetry import trace
7+
from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
78
from opentelemetry.instrumentation.remoulade import utils
8-
from opentelemetry.instrumentation.remoulade.version import __version__
99
from opentelemetry.instrumentation.remoulade.package import _instruments
10-
from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
10+
from opentelemetry.instrumentation.remoulade.version import __version__
1111
from opentelemetry.propagate import extract, inject
1212
from opentelemetry.semconv.trace import SpanAttributes
1313

14-
1514
_MESSAGE_TAG_KEY = "remoulade.action"
1615
_MESSAGE_SEND = "send"
1716
_MESSAGE_RUN = "run"
@@ -28,20 +27,32 @@ def before_process_message(self, _broker, message):
2827
trace_ctx = extract(message.options["trace_ctx"])
2928
retry_count = message.options.get("retries")
3029

31-
operation_name = "remoulade/process" if retry_count is None else f"remoulade/process(retry-{retry_count})"
30+
operation_name = (
31+
"remoulade/process"
32+
if retry_count is None
33+
else f"remoulade/process(retry-{retry_count})"
34+
)
3235

33-
span = self._tracer.start_span(operation_name, kind=trace.SpanKind.CONSUMER, context=trace_ctx)
36+
span = self._tracer.start_span(
37+
operation_name, kind=trace.SpanKind.CONSUMER, context=trace_ctx
38+
)
3439

3540
if retry_count is not None:
3641
span.set_attribute("retry_count", retry_count)
3742

3843
activation = trace.use_span(span, end_on_exit=True)
3944
activation.__enter__()
4045

41-
utils.attach_span(self._span_registry, message.message_id, (span, activation))
46+
utils.attach_span(
47+
self._span_registry, message.message_id, (span, activation)
48+
)
4249

43-
def after_process_message(self, _broker, message, *, result=None, exception=None):
44-
span, activation = utils.retrieve_span(self._span_registry, message.message_id)
50+
def after_process_message(
51+
self, _broker, message, *, result=None, exception=None
52+
):
53+
span, activation = utils.retrieve_span(
54+
self._span_registry, message.message_id
55+
)
4556

4657
if span is None:
4758
# no existing span found for message_id
@@ -58,37 +69,54 @@ def after_process_message(self, _broker, message, *, result=None, exception=None
5869
def before_enqueue(self, _broker, message, delay):
5970
retry_count = message.options.get("retries")
6071

61-
operation_name = "remoulade/send" if retry_count is None else f"remoulade/send(retry-{retry_count})"
72+
operation_name = (
73+
"remoulade/send"
74+
if retry_count is None
75+
else f"remoulade/send(retry-{retry_count})"
76+
)
6277

63-
span = self._tracer.start_span(operation_name, kind=trace.SpanKind.PRODUCER)
78+
span = self._tracer.start_span(
79+
operation_name, kind=trace.SpanKind.PRODUCER
80+
)
6481

6582
if retry_count is not None:
6683
span.set_attribute("retry_count", retry_count)
6784

6885
if span.is_recording():
6986
span.set_attribute(_MESSAGE_TAG_KEY, _MESSAGE_SEND)
70-
span.set_attribute(SpanAttributes.MESSAGING_MESSAGE_ID, message.message_id)
87+
span.set_attribute(
88+
SpanAttributes.MESSAGING_MESSAGE_ID, message.message_id
89+
)
7190
span.set_attribute(_MESSAGE_NAME_KEY, message.actor_name)
7291
pass
7392

7493
activation = trace.use_span(span, end_on_exit=True)
7594
activation.__enter__()
7695

77-
utils.attach_span(self._span_registry, message.message_id, (span, activation), is_publish=True)
96+
utils.attach_span(
97+
self._span_registry,
98+
message.message_id,
99+
(span, activation),
100+
is_publish=True,
101+
)
78102

79103
if "trace_ctx" not in message.options:
80104
message.options["trace_ctx"] = {}
81105
inject(message.options["trace_ctx"])
82106

83107
def after_enqueue(self, _broker, message, delay, exception=None):
84-
_, activation = utils.retrieve_span(self._span_registry, message.message_id, is_publish=True)
108+
_, activation = utils.retrieve_span(
109+
self._span_registry, message.message_id, is_publish=True
110+
)
85111

86112
if activation is None:
87113
# no existing span found for message_id
88114
return
89115

90116
activation.__exit__(None, None, None)
91-
utils.detach_span(self._span_registry, message.message_id, is_publish=True)
117+
utils.detach_span(
118+
self._span_registry, message.message_id, is_publish=True
119+
)
92120

93121

94122
class RemouladeInstrumentor(BaseInstrumentor):

instrumentation/opentelemetry-instrumentation-remoulade/tests/test_messages.py

+5-16
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import remoulade
22
from remoulade.brokers.local import LocalBroker
3+
34
from opentelemetry.instrumentation.remoulade import RemouladeInstrumentor
45
from opentelemetry.test.test_base import TestBase
56
from opentelemetry.trace import SpanKind
@@ -66,25 +67,13 @@ def test_retries(self):
6667
producer_spans = spans[1::2]
6768

6869
self.assertEqual(consumer_spans[0].name, "remoulade/process(retry-3)")
69-
self.assertSpanHasAttributes(
70-
consumer_spans[0],
71-
{"retry_count": 3}
72-
)
70+
self.assertSpanHasAttributes(consumer_spans[0], {"retry_count": 3})
7371
self.assertEqual(consumer_spans[1].name, "remoulade/process(retry-2)")
74-
self.assertSpanHasAttributes(
75-
consumer_spans[1],
76-
{"retry_count": 2}
77-
)
72+
self.assertSpanHasAttributes(consumer_spans[1], {"retry_count": 2})
7873
self.assertEqual(consumer_spans[3].name, "remoulade/process")
7974

8075
self.assertEqual(producer_spans[0].name, "remoulade/send(retry-3)")
81-
self.assertSpanHasAttributes(
82-
producer_spans[0],
83-
{"retry_count": 3}
84-
)
76+
self.assertSpanHasAttributes(producer_spans[0], {"retry_count": 3})
8577
self.assertEqual(producer_spans[1].name, "remoulade/send(retry-2)")
86-
self.assertSpanHasAttributes(
87-
producer_spans[1],
88-
{"retry_count": 2}
89-
)
78+
self.assertSpanHasAttributes(producer_spans[1], {"retry_count": 2})
9079
self.assertEqual(producer_spans[3].name, "remoulade/send")

opentelemetry-contrib-instrumentations/setup.cfg

+1
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ install_requires =
5555
opentelemetry-instrumentation-pymysql==0.30b0
5656
opentelemetry-instrumentation-pyramid==0.30b0
5757
opentelemetry-instrumentation-redis==0.30b0
58+
opentelemetry-instrumentation-remoulade==0.30b0
5859
opentelemetry-instrumentation-requests==0.30b0
5960
opentelemetry-instrumentation-sklearn==0.30b0
6061
opentelemetry-instrumentation-sqlalchemy==0.30b0

opentelemetry-instrumentation/src/opentelemetry/instrumentation/bootstrap_gen.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@
113113
"instrumentation": "opentelemetry-instrumentation-redis==0.30b0",
114114
},
115115
"remoulade": {
116-
"library": "remoulade >= 0.48",
116+
"library": "remoulade >= 0.49",
117117
"instrumentation": "opentelemetry-instrumentation-remoulade==0.30b0",
118118
},
119119
"requests": {

tox.ini

+2-1
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@ envlist =
157157
pypy3-test-instrumentation-redis
158158

159159
; opentelemetry-instrumentation-remoulade
160-
py3{6,7,8,9,10}-test-instrumentation-remoualde
160+
py3{6,7,8,9,10}-test-instrumentation-remoulade
161161
pypy3-test-instrumentation-remoulade
162162

163163
; opentelemetry-instrumentation-celery
@@ -282,6 +282,7 @@ changedir =
282282
test-instrumentation-pymysql: instrumentation/opentelemetry-instrumentation-pymysql/tests
283283
test-instrumentation-pyramid: instrumentation/opentelemetry-instrumentation-pyramid/tests
284284
test-instrumentation-redis: instrumentation/opentelemetry-instrumentation-redis/tests
285+
test-instrumentation-remoulade: instrumentation/opentelemetry-instrumentation-remoulade/tests
285286
test-instrumentation-requests: instrumentation/opentelemetry-instrumentation-requests/tests
286287
test-instrumentation-sklearn: instrumentation/opentelemetry-instrumentation-sklearn/tests
287288
test-instrumentation-sqlalchemy{11,14}: instrumentation/opentelemetry-instrumentation-sqlalchemy/tests

0 commit comments

Comments
 (0)