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)
