def test_bad_construction(self):
mm = MetricManager()
class NotMetricReporter(object):
pass
mr = [NotMetricReporter()]
with self.assertRaises(MogwaiMetricsException):
mm.setup_reporters(mr)
def test_multiple_reporter_constuction(self):
mm = MetricManager()
r1 = [MetricsRegistry(), MetricsRegistry()]
mr1 = BaseMetricsReporter(registry=r1)
r2 = [MetricsRegistry(), MetricsRegistry()]
mr2 = BaseMetricsReporter(registry=r2)
mm.setup_reporters([mr1, mr2])
self.assertIsInstance(mm.metric_reporters, (tuple, list))
self.assertEqual(len(mm.metric_reporters), 2)
self.assertIsInstance(mm.metric_reporters[0], BaseMetricsReporter)
self.assertIsInstance(mm.metric_reporters[1], BaseMetricsReporter)
self.assertEqual(mm.metric_reporters, [mr1, mr2])
def test_multiple_decorators(self):
mm = MetricManager()
mr = BaseMetricsReporter()
mm.setup_reporters(mr)
mm.start()
@mm.meter_calls
@mm.count_calls
@mm.hist_calls
@mm.time_calls
def somefunc(i):
return i
self.assertEqual(somefunc(1), 1)
self.assertEqual(somefunc(1, context='test'), 1)
# check metrics collected
for reporter in mm.metric_reporters:
timestamp, metrics = reporter.get_metrics()
self.assertIsInstance(timestamp, integer_types)
self.assertIsInstance(metrics, dict)
self.assertEqual(len(metrics), 3)
# context aware
self.assertIn('test.meter', metrics)
self.assertIsInstance(metrics['test.meter'], dict)
self.assertGreater(metrics['test.meter']['mean_rate'], 100)
# normal one
self.assertIn('somefunc.calls', metrics)
self.assertIsInstance(metrics['somefunc.calls'], dict)
self.assertEqual(metrics['somefunc.calls']['count'], 2)
mm.stop()
def test_start_stop_mechanism(self):
mm = MetricManager()
mr = BaseMetricsReporter()
mm.setup_reporters(mr)
mm.start()
for mr in mm.metric_reporters:
self.assertTrue(mr.task.running)
mm.stop()
for mr in mm.metric_reporters:
self.assertFalse(mr.task.running)
def test_metric_getter(self):
mm = MetricManager()
mr = BaseMetricsReporter()
mm.setup_reporters(mr)
mm.start()
# increment counters
for counter in mm.counters('test'):
counter.inc()
timestamp, metrics = mr.get_metrics()
self.assertIsInstance(timestamp, integer_types)
self.assertIsInstance(metrics, dict)
self.assertEqual(len(metrics), 1)
self.assertIn('test', metrics)
self.assertIsInstance(metrics['test'], dict)
self.assertEqual(metrics['test']['count'], 1)
mm.stop()
def test_default_construction(self):
mm = MetricManager()
self.assertEqual(mm.metric_reporters, [])
with self.assertRaises(MogwaiMetricsException):
mm.setup_reporters(None)
mr = BaseMetricsReporter()
mm.setup_reporters(mr)
self.assertIsInstance(mm.metric_reporters, (tuple, list))
self.assertEqual(len(mm.metric_reporters), 1)
self.assertIsInstance(mm.metric_reporters[0], BaseMetricsReporter)
self.assertEqual(mm.metric_reporters, [mr])
def test_get_all_counters(self):
mm = MetricManager()
mr = BaseMetricsReporter()
mm.setup_reporters(mr)
# test default implementation
for counter in mm.counters():
self.assertIsNotNone(get_method_self(counter))
self.assertIsInstance(get_method_self(counter), MetricsRegistry)
# test context aware
for counter in mm.counters('test'):
from pyformance.meters.counter import Counter
self.assertIsInstance(counter, Counter)
def test_get_all_histograms(self):
mm = MetricManager()
mr = BaseMetricsReporter()
mm.setup_reporters(mr)
# test default implementation
for hist in mm.histograms():
self.assertIsNotNone(get_method_self(hist))
self.assertIsInstance(get_method_self(hist), MetricsRegistry)
# test context aware
for hist in mm.histograms('test'):
from pyformance.meters.histogram import Histogram
self.assertIsInstance(hist, Histogram)
def test_timer_decorator(self):
mm = MetricManager()
mr = BaseMetricsReporter()
mm.setup_reporters(mr)
mm.start()
@mm.time_calls
def somefunc(i):
return i
@mm.time_calls
def badfunc(i):
raise Exception("test exception")
self.assertEqual(somefunc(1), 1)
self.assertEqual(somefunc(1, context='test'), 1)
with self.assertRaises(Exception):
badfunc(1)
# check metrics collected
for reporter in mm.metric_reporters:
timestamp, metrics = reporter.get_metrics()
self.assertIsInstance(timestamp, integer_types)
self.assertIsInstance(metrics, dict)
self.assertEqual(len(metrics), 5)
# context aware
self.assertIn('test.timer', metrics)
self.assertIsInstance(metrics['test.timer'], dict)
self.assertLess(metrics['test.timer']['max'], 0.01)
# normal one
self.assertIn('somefunc.timer', metrics)
self.assertIsInstance(metrics['somefunc.timer'], dict)
self.assertLess(metrics['somefunc.timer']['max'], 0.01)
mm.stop()
from re import compile
from rexpro.connection import RexProConnection
from rexpro.utils import get_rexpro
from rexpro.exceptions import RexProConnectionException, RexProScriptException
from mogwai.exceptions import MogwaiConnectionError, MogwaiQueryError
from mogwai.metrics.manager import MetricManager
logger = logging.getLogger(__name__)
SOCKET_TYPE = None
CONNECTION_TYPE = None
CONNECTION_POOL_TYPE = None
HOST_PARAMS = None
_connection_pool = None
_graph_name = None
metric_manager = MetricManager()
_loaded_models = []
__cached_spec = None
@metric_manager.time_calls
def execute_query(query,
params={},
transaction=True,
isolate=True,
pool=None,
*args,
**kwargs):
"""
Execute a raw Gremlin query with the given parameters passed in.
