Adding the 'set' data type to the client

2 files changed, 18 insertions, 0 deletions

diff --git a/statsd/client.py b/statsd/client.py
index e88d23a..9d34bed 100644
--- a/statsd/client.py
+++ b/statsd/client.py
@@ -78,6 +78,12 @@ class StatsClient(object):
         if data is not None:
             self._after(data)
 
+    def set(self, stat, value, rate=1):
+        """Set a set value."""
+        data = self._prepare(stat, '%s|s' % value, rate)
+        if data is not None:
+            self._after(data)
+
     def _prepare(self, stat, value, rate=1):
         if rate < 1:
             if random.random() < rate:
diff --git a/statsd/tests.py b/statsd/tests.py
index 59e8f24..fb9a882 100644
--- a/statsd/tests.py
+++ b/statsd/tests.py
@@ -146,6 +146,18 @@ def test_gauge_delta():
     _sock_check(sc, 4, 'foo:-1.3|g')
 
 
+def test_set():
+    sc = _client()
+    sc.set('foo', 10)
+    _sock_check(sc, 1, 'foo:10|s')
+
+    sc.set('foo', 2.3)
+    _sock_check(sc, 2, 'foo:2.3|s')
+
+    sc.set('foo', 2.3, 0.5)
+    _sock_check(sc, 3, 'foo:2.3|s|@0.5')
+
+
 @mock.patch.object(random, 'random', lambda: -1)
 def test_timing():
     sc = _client()