def run(self): self.set_wrapped({ 'elasticsearch.client.Elasticsearch.create': get_complex_wrapper('data.es.create.', state='data.elasticsearch', disable_if='model'), 'elasticsearch.client.Elasticsearch.get': get_complex_wrapper('data.es.get.', state='data.elasticsearch', disable_if='model'), 'elasticsearch.client.Elasticsearch.index': get_complex_wrapper('data.es.index.', state='data.elasticsearch', disable_if='model'), 'elasticsearch.client.Elasticsearch.search': get_complex_wrapper('data.es.search.', state='data.elasticsearch', disable_if='model'), 'elasticsearch.client.Elasticsearch.delete': get_complex_wrapper('data.es.delete.', state='data.elasticsearch', disable_if='model'), }) super(ElasticsearchInstrumentor, self).run()
def run(self): self.set_wrapped( { # Kafka (pykafka) 'pykafka.simpleconsumer.SimpleConsumer.consume': get_complex_wrapper('messaging.kafka.consume.', state='messaging.kafka', disable_if='model'), 'pykafka.Producer.produce': get_complex_wrapper('messaging.kafka.produce.', state='messaging.kafka', disable_if='model'), }) super(PykafkaInstrumentor, self).run()
def run(self): self.set_wrapped( { 'MySQLdb.cursors.Cursor.execute': get_complex_wrapper('data.mysql.execute.', state='data.mysql', disable_if='model'), 'MySQLdb.cursors.Cursor.callproc': get_complex_wrapper('data.mysql.callproc.', state='data.mysql', disable_if='model'), } ) super(MysqlInstrumentor, self).run()
def run(self): self.set_wrapped({ 'MySQLdb.cursors.Cursor.execute': get_complex_wrapper('data.mysql.execute.', state='data.mysql', disable_if='model'), 'MySQLdb.cursors.Cursor.callproc': get_complex_wrapper('data.mysql.callproc.', state='data.mysql', disable_if='model'), }) super(MysqlInstrumentor, self).run()
def run(self): self.set_wrapped({ # Kafka (pykafka) 'pykafka.simpleconsumer.SimpleConsumer.consume': get_complex_wrapper('messaging.kafka.consume.', state='messaging.kafka', disable_if='model'), 'pykafka.Producer.produce': get_complex_wrapper('messaging.kafka.produce.', state='messaging.kafka', disable_if='model'), }) super(PykafkaInstrumentor, self).run()
def run(self): wrappers = { "django.db.models.query.QuerySet.iterator": get_generator_wrapper( generate_record_telemetry("model.iterator."), state="model" ) } _query_set_methods = [ "aggregate", "count", "bulk_create", "create", "get", "get_or_create", "latest", "first", "last", "in_bulk", "iterator", "update_or_create", "delete", "update", "exists", ] for method in _query_set_methods: metric = "model." + method wrappers["django.db.models.query.QuerySet." + method] = get_complex_wrapper(metric, state="model") self.set_wrapped(wrappers) super(DjangoDbInstrumentor, self).run()
def run(self): self.set_wrapped( { 'elasticsearch.client.Elasticsearch.create': get_complex_wrapper('data.es.create.', state='data.elasticsearch', disable_if='model'), 'elasticsearch.client.Elasticsearch.get': get_complex_wrapper('data.es.get.', state='data.elasticsearch', disable_if='model'), 'elasticsearch.client.Elasticsearch.index': get_complex_wrapper('data.es.index.', state='data.elasticsearch', disable_if='model'), 'elasticsearch.client.Elasticsearch.search': get_complex_wrapper('data.es.search.', state='data.elasticsearch', disable_if='model'), 'elasticsearch.client.Elasticsearch.delete': get_complex_wrapper('data.es.delete.', state='data.elasticsearch', disable_if='model'), } ) super(ElasticsearchInstrumentor, self).run()
def run(self): # Instrument our wrapper connection class meths = { 'librato_python_web.instrumentor.data.sqlite.WrappedCursor.' + m: get_complex_wrapper('data.sqlite.%s.' % m, state='data.sqlite', disable_if='model') for m in ['execute', 'executemany', 'fetchone', 'fetchmany', 'fetchall'] } # Instrument connect method meths['sqlite3.connect'] = wrapped_connect self.set_wrapped(meths) super(SqliteInstrumentor, self).run()
def run(self): """ Instruments our cursor wrapper class and psycopg2.connect """ # Generate a list of methods in the cursor wrapper meth_names = [n for (n, _) in inspect.getmembers(WrappedCursor) if '_' not in n] meths = { 'librato_python_web.instrumentor.data.psycopg2.WrappedCursor.' + m: get_complex_wrapper('data.psycopg2.%s.' % m, state='data.psycopg2', disable_if='model') for m in meth_names } # Instrument connect method meths['psycopg2.connect'] = wrapped_connect self.set_wrapped(meths) super(Psycopg2Instrumentor, self).run()
def run(self): wrappers = { 'django.db.models.query.QuerySet.iterator': get_generator_wrapper(generate_record_telemetry('model.iterator.'), state='model'), } _query_set_methods = [ 'aggregate', 'count', 'bulk_create', 'create', 'get', 'get_or_create', 'latest', 'first', 'last', 'in_bulk', 'iterator', 'update_or_create', 'delete', 'update', 'exists' ] for method in _query_set_methods: metric = 'model.' + method wrappers['django.db.models.query.QuerySet.' + method] = get_complex_wrapper(metric, state='model') self.set_wrapped(wrappers) super(DjangoDbInstrumentor, self).run()
def run(self): """ Instruments our cursor wrapper class and psycopg2.connect """ # Generate a list of methods in the cursor wrapper meth_names = [ n for (n, _) in inspect.getmembers(WrappedCursor) if '_' not in n ] meths = { 'librato_python_web.instrumentor.data.psycopg2.WrappedCursor.' + m: get_complex_wrapper('data.psycopg2.%s.' % m, state='data.psycopg2', disable_if='model') for m in meth_names } # Instrument connect method meths['psycopg2.connect'] = wrapped_connect self.set_wrapped(meths) super(Psycopg2Instrumentor, self).run()