summaryrefslogtreecommitdiffstats
path: root/storperf/carbon/emitter.py
blob: 6104fd484bbf952682e41ba0d354c6fcd2d1ff11 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
##############################################################################
# Copyright (c) 2015 EMC and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
import calendar
import logging
import socket
import time


class CarbonMetricTransmitter():

    carbon_host = '127.0.0.1'
    carbon_port = 2003

    def __init__(self):
        self.logger = logging.getLogger(__name__)

    def transmit_metrics(self, metrics):
        if 'timestamp' in metrics:
            timestamp = metrics.pop('timestamp')
        else:
            timestamp = str(calendar.timegm(time.gmtime()))

        carbon_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        carbon_socket.connect((self.carbon_host, self.carbon_port))

        for key, metric in metrics.items():
            message = key + " " + metric + " " + timestamp
            self.logger.debug("Metric: " + message)
            carbon_socket.send(message + '\n')

        carbon_socket.close()