Sanitize the statsd prefix

This commit is contained in:
Scott Sanders 2014-09-16 08:21:10 -04:00
parent c9c543dc82
commit 770c8219d2

View File

@ -5,6 +5,7 @@
"Bare-bones implementation of statsD's protocol, client-side" "Bare-bones implementation of statsD's protocol, client-side"
import re
import socket import socket
import logging import logging
from gunicorn.glogging import Logger from gunicorn.glogging import Logger
@ -25,7 +26,8 @@ class Statsd(Logger):
"""host, port: statsD server """host, port: statsD server
""" """
Logger.__init__(self, cfg) Logger.__init__(self, cfg)
self.prefix = cfg.statsd_prefix prefix_regex = re.compile(r"\.*$")
self.prefix = prefix_regex.sub(".", cfg.statsd_prefix)
try: try:
host, port = cfg.statsd_host host, port = cfg.statsd_host
self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)