add support for sets

cyberdelia · Nov 10, 2012 · 8c2eebe · 8c2eebe
1 parent 3ba8369
commit 8c2eebe
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/statsd.go b/statsd.go
@@ -73,6 +73,11 @@ func (c *Client) Gauge(stat string, value int, rate float64) error {
 	return c.send(stat, rate, "%d|g", value)
 }
 
+// Record unique occurences of events
+func (c *Client) Unique(stat string, value int, rate float64) error {
+	return c.send(stat, rate, "%d|s", value)
+}
+
 func (c *Client) Close() error {
 	err := c.buf.Flush()
 	if err != nil {

diff --git a/statsd_test.go b/statsd_test.go
@@ -52,6 +52,14 @@ func TestGauge(t *testing.T) {
 	assert.Equal(t, data, "gauge:300|g")
 }
 
+func TestUnique(t *testing.T) {
+	c := fakeClient()
+	err := c.Unique("unique", 765, 1)
+	assert.Equal(t, err, nil)
+	data := readData(c.buf)
+	assert.Equal(t, data, "unique:765|s")
+}
+
 func TestMilliseconds(t *testing.T) {
 	msec, _ := time.ParseDuration("350ms")
 	assert.Equal(t, 350, millisecond(msec))