| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| |
| import os |
| import time |
| |
| from tashi import scrubString |
| |
| class GangliaPublisher(object): |
| def __init__(self, config): |
| self.disable = False |
| self.disableAt = 0.0 |
| self.retry = float(config.get("GangliaPublisher", "retry")) |
| self.dmax = float(config.get("GangliaPublisher", "dmax")) |
| |
| def publish(self, message): |
| for (key, val) in message.iteritems(): |
| if (self.disable): |
| if (time.time() - self.disableAt > self.retry): |
| disable = False |
| else: |
| return |
| key = scrubString(str(key)) |
| val = str(val) |
| metricName = "tashi_%s" % (key) |
| val = val.replace('"', "'") |
| val = val.replace('<', '<') |
| val = val.replace('>', '>') |
| metricValue = val |
| metricType = "string" |
| try: |
| metricValue = float(metricValue) |
| metricType = "float" |
| metricValue = "%3.3f" % (metricValue) |
| except: |
| pass |
| cmd = 'gmetric -n "%s" -v "%s" -t "%s" -d "%d"' % (metricName, metricValue, metricType, self.dmax) |
| # XXXpipe: send a datum to ganglia |
| (stdin, stdout) = os.popen4(cmd) |
| stdin.close() |
| res = stdout.read() |
| stdout.close() |
| if (res != ""): |
| self.disable = True |
| self.disableAt = time.time() |
| print "Failed to exec gmetric, disabling: %s" % (res) |
| |
| def publishList(self, messages): |
| for message in messages: |
| self.publish(message) |
