From 5306f88b5739322209e4014aae11327c8430dcdd Mon Sep 17 00:00:00 2001
From: Andy Stark <andrew.stark@redis.com>
Date: Fri, 13 Sep 2024 14:37:55 +0100
Subject: [PATCH] DOC-4239 added Count-min sketch examples

---
 doctests/cms_tutorial_test.go | 84 +++++++++++++++++++++++++++++++++++
 1 file changed, 84 insertions(+)
 create mode 100644 doctests/cms_tutorial_test.go

diff --git a/doctests/cms_tutorial_test.go b/doctests/cms_tutorial_test.go
new file mode 100644
index 000000000..ade1fa93d
--- /dev/null
+++ b/doctests/cms_tutorial_test.go
@@ -0,0 +1,84 @@
+// EXAMPLE: cms_tutorial
+// HIDE_START
+package example_commands_test
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/redis/go-redis/v9"
+)
+
+// HIDE_END
+
+func ExampleClient_cms() {
+	ctx := context.Background()
+
+	rdb := redis.NewClient(&redis.Options{
+		Addr:     "localhost:6379",
+		Password: "", // no password docs
+		DB:       0,  // use default DB
+	})
+
+	// REMOVE_START
+	rdb.Del(ctx, "bikes:profit")
+	// REMOVE_END
+
+	// STEP_START cms
+	res1, err := rdb.CMSInitByProb(ctx, "bikes:profit", 0.001, 0.002).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(res1) // >>> OK
+
+	res2, err := rdb.CMSIncrBy(ctx, "bikes:profit",
+		"Smoky Mountain Striker", 100,
+	).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(res2) // >>> [100]
+
+	res3, err := rdb.CMSIncrBy(ctx, "bikes:profit",
+		"Rocky Mountain Racer", 200,
+		"Cloudy City Cruiser", 150,
+	).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(res3) // >>> [200 150]
+
+	res4, err := rdb.CMSQuery(ctx, "bikes:profit",
+		"Smoky Mountain Striker",
+	).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(res4) // >>> [100]
+
+	res5, err := rdb.CMSInfo(ctx, "bikes:profit").Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Printf("Width: %v, Depth: %v, Count: %v",
+		res5.Width, res5.Depth, res5.Count)
+	// >>> Width: 2000, Depth: 9, Count: 450
+	// STEP_END
+
+	// Output:
+	// OK
+	// [100]
+	// [200 150]
+	// [100]
+	// Width: 2000, Depth: 9, Count: 450
+}