async def setup_graphite_exporter(): graphite_server = os.environ.get("GRAPHITE_SERVER", None) graphite_key = os.environ.get("GRAPHITE_KEY", None) graphite_port = os.environ.get("GRAPHITE_PORT", 2003) if graphite_server is None or graphite_key is None: logger.warn( "Graphite server (" + str(graphite_server) + ") or Graphite key (" + str(graphite_key) + ") is not defined. Not setting up Monitoring graphite server!" ) return logger.info( "Monitoring graphite server setup! Reporting to server: " + graphite_server + ":" + str(graphite_port) + " with prefix: " + str(graphite_key) ) gb = GraphiteBridge((graphite_server, graphite_port)) gb.start(60.0, prefix=graphite_key)
class TestGraphiteBridge(unittest.TestCase): def setUp(self): self.registry = CollectorRegistry() self.data = '' class TCPHandler(SocketServer.BaseRequestHandler): def handle(s): self.data = s.request.recv(1024) server = SocketServer.TCPServer(('', 0), TCPHandler) class ServingThread(threading.Thread): def run(self): server.handle_request() server.socket.close() self.t = ServingThread() self.t.start() # Explicitly use localhost as the target host, since connecting to 0.0.0.0 fails on Windows address = ('localhost', server.server_address[1]) self.gb = GraphiteBridge(address, self.registry, _timer=fake_timer) def test_nolabels(self): gauge = Gauge('g', 'help', registry=self.registry) gauge.inc() self.gb.push() self.t.join() self.assertEqual(b'g 1.0 1434898897\n', self.data) def test_labels(self): labels = Gauge('labels', 'help', ['a', 'b'], registry=self.registry) labels.labels('c', 'd').inc() self.gb.push() self.t.join() self.assertEqual(b'labels.a.c.b.d 1.0 1434898897\n', self.data) def test_prefix(self): labels = Gauge('labels', 'help', ['a', 'b'], registry=self.registry) labels.labels('c', 'd').inc() self.gb.push(prefix='pre.fix') self.t.join() self.assertEqual(b'pre.fix.labels.a.c.b.d 1.0 1434898897\n', self.data) def test_sanitizing(self): labels = Gauge('labels', 'help', ['a'], registry=self.registry) labels.labels('c.:8').inc() self.gb.push() self.t.join() self.assertEqual(b'labels.a.c__8 1.0 1434898897\n', self.data)
class TestGraphiteBridge(unittest.TestCase): def setUp(self): self.registry = CollectorRegistry() self.data = '' class TCPHandler(SocketServer.BaseRequestHandler): def handle(s): self.data = s.request.recv(1024) server = SocketServer.TCPServer(('', 0), TCPHandler) class ServingThread(threading.Thread): def run(self): server.handle_request() server.socket.close() self.t = ServingThread() self.t.start() # Explicitly use localhost as the target host, since connecting to 0.0.0.0 fails on Windows address = ('localhost', server.server_address[1]) self.gb = GraphiteBridge(address, self.registry, _time=FakeTime()) def test_nolabels(self): counter = Counter('c', 'help', registry=self.registry) counter.inc() self.gb.push() self.t.join() self.assertEqual(b'c 1.0 1434898897\n', self.data) def test_labels(self): labels = Counter('labels', 'help', ['a', 'b'], registry=self.registry) labels.labels('c', 'd').inc() self.gb.push() self.t.join() self.assertEqual(b'labels.a.c.b.d 1.0 1434898897\n', self.data) def test_prefix(self): labels = Counter('labels', 'help', ['a', 'b'], registry=self.registry) labels.labels('c', 'd').inc() self.gb.push(prefix = 'pre.fix') self.t.join() self.assertEqual(b'pre.fix.labels.a.c.b.d 1.0 1434898897\n', self.data) def test_sanitizing(self): labels = Counter('labels', 'help', ['a'], registry=self.registry) labels.labels('c.:8').inc() self.gb.push() self.t.join() self.assertEqual(b'labels.a.c__8 1.0 1434898897\n', self.data)
def setup_bridge(prom, retry, graphite): registry = CollectorRegistry() registry.register(ScrapeCollector(prom, retry)) logging.info('Scrape target: %s', prom) logging.info('Graphite server: %s:%i', *graphite) bridge = GraphiteBridge(graphite, registry=registry) return bridge
def set_config(self, config: ValueExtractor) -> None: super().set_config(config) registry = get_registry(self.config.get('registry', REGISTRY)) if MULTIPROC_DIR: registry = MultiProcessCollector(registry) self._registry = registry port: int = self.config.get_int('port', default=0) if port: start_http_server(port=port, registry=registry) graphite: Optional[ValueExtractor] = self.config.get('graphite') if graphite: address = graphite.get('address') if isinstance(address, str): addr = address.split(':') address = (addr[0], int(addr[-1])) gb = GraphiteBridge(address, registry=registry) interval: float = graphite.get_duration('interval', 60) prefix: str = graphite.get('prefix', '') gb.start(interval, prefix=prefix)
def setUp(self): self.registry = CollectorRegistry() self.data = '' class TCPHandler(SocketServer.BaseRequestHandler): def handle(s): self.data = s.request.recv(1024) server = SocketServer.TCPServer(('', 0), TCPHandler) class ServingThread(threading.Thread): def run(self): server.handle_request() server.socket.close() self.t = ServingThread() self.t.start() self.gb = GraphiteBridge(server.server_address, self.registry, _time=FakeTime())
def setUp(self): self.registry = CollectorRegistry() self.data = '' class TCPHandler(SocketServer.BaseRequestHandler): def handle(s): self.data = s.request.recv(1024) server = SocketServer.TCPServer(('', 0), TCPHandler) class ServingThread(threading.Thread): def run(self): server.handle_request() server.socket.close() self.t = ServingThread() self.t.start() # Explicitly use localhost as the target host, since connecting to 0.0.0.0 fails on Windows address = ('localhost', server.server_address[1]) self.gb = GraphiteBridge(address, self.registry, _timer=fake_timer)
def setUp(self): self.registry = CollectorRegistry() self.data = '' class TCPHandler(SocketServer.BaseRequestHandler): def handle(s): self.data = s.request.recv(1024) server = SocketServer.TCPServer(('', 0), TCPHandler) class ServingThread(threading.Thread): def run(self): server.handle_request() server.socket.close() self.t = ServingThread() self.t.start() # Explicitly use localhost as the target host, since connecting to 0.0.0.0 fails on Windows address = ('localhost', server.server_address[1]) self.gb = GraphiteBridge(address, self.registry, _time=FakeTime())
class TestGraphiteBridge(unittest.TestCase): def setUp(self): self.registry = CollectorRegistry() self.data = '' class TCPHandler(SocketServer.BaseRequestHandler): def handle(s): self.data = s.request.recv(1024) server = SocketServer.TCPServer(('', 0), TCPHandler) class ServingThread(threading.Thread): def run(self): server.handle_request() server.socket.close() self.t = ServingThread() self.t.start() self.gb = GraphiteBridge(server.server_address, self.registry, _time=FakeTime()) def test_nolabels(self): counter = Counter('c', 'help', registry=self.registry) counter.inc() self.gb.push() self.t.join() self.assertEqual(b'c 1.0 1434898897\n', self.data) def test_labels(self): labels = Counter('labels', 'help', ['a', 'b'], registry=self.registry) labels.labels('c', 'd').inc() self.gb.push() self.t.join() self.assertEqual(b'labels.a.c.b.d 1.0 1434898897\n', self.data) def test_sanitizing(self): labels = Counter('labels', 'help', ['a'], registry=self.registry) labels.labels('c.:8').inc() self.gb.push() self.t.join() self.assertEqual(b'labels.a.c__8 1.0 1434898897\n', self.data)
def start(self): print("Starting energy monitor:") print(" GPIO: {:d}".format(self.sensor_gpio)) print(" graphite:") print(" server : {}:{}".format(self.graphite_host, self.graphite_port)) print(" metric prefix : {}".format(self.graphite_metric_prefix)) print(" report interval: {:d}".format( self.graphite_report_interval)) print(" influxdb (graphite protocol):") print(" server : {}:{}".format(self.influxdb_host, self.influxdb_port)) print(" metric prefix : {}".format(self.graphite_metric_prefix)) print(" report interval: {:d}".format( self.graphite_report_interval)) print(" prometheus:") print(" port: {}".format(self.prometheus_port)) print(" path: /metrics") registry = CollectorRegistry() self.counter = Counter('meter_power', 'Total watt-hours consumed', registry=registry) gb = GraphiteBridge((self.graphite_host, self.graphite_port), registry=registry) gb.start(self.graphite_report_interval, prefix=self.graphite_metric_prefix) influxbridge = GraphiteBridge((self.influxdb_host, self.influxdb_port), registry=registry) influxbridge.start(self.graphite_report_interval, prefix=self.graphite_metric_prefix) start_http_server(self.prometheus_port, registry=registry) GPIO.setmode(GPIO.BCM) GPIO.setup(self.sensor_gpio, GPIO.IN, pull_up_down=GPIO.PUD_DOWN) GPIO.add_event_detect(self.sensor_gpio, GPIO.RISING, callback=self.increment, bouncetime=100)
def setUp(self): self.registry = CollectorRegistry() self.data = "" class TCPHandler(SocketServer.BaseRequestHandler): def handle(s): self.data = s.request.recv(1024) server = SocketServer.TCPServer(("", 0), TCPHandler) class ServingThread(threading.Thread): def run(self): server.handle_request() server.socket.close() self.t = ServingThread() self.t.start() self.gb = GraphiteBridge(server.server_address, self.registry, _time=FakeTime())
class TestGraphiteBridge(unittest.TestCase): def setUp(self): self.registry = CollectorRegistry() self.data = "" class TCPHandler(SocketServer.BaseRequestHandler): def handle(s): self.data = s.request.recv(1024) server = SocketServer.TCPServer(("", 0), TCPHandler) class ServingThread(threading.Thread): def run(self): server.handle_request() server.socket.close() self.t = ServingThread() self.t.start() self.gb = GraphiteBridge(server.server_address, self.registry, _time=FakeTime()) def test_nolabels(self): counter = Counter("c", "help", registry=self.registry) counter.inc() self.gb.push() self.t.join() self.assertEqual(b"c 1.0 1434898897\n", self.data) def test_labels(self): labels = Counter("labels", "help", ["a", "b"], registry=self.registry) labels.labels("c", "d").inc() self.gb.push() self.t.join() self.assertEqual(b"labels.a.c.b.d 1.0 1434898897\n", self.data) def test_sanitizing(self): labels = Counter("labels", "help", ["a"], registry=self.registry) labels.labels("c.:8").inc() self.gb.push() self.t.join() self.assertEqual(b"labels.a.c__8 1.0 1434898897\n", self.data)
def _use_tags(self): self.gb = GraphiteBridge(self.address, self.registry, tags=True, _timer=fake_timer)
import time from prometheus_client.bridge.graphite import GraphiteBridge gb = GraphiteBridge(['graphite.your.org', 2003]) gb.start(10) while True: time.sleep(1)