From e69895b94d276e03d8e0e88a0e7514060b9a0639 Mon Sep 17 00:00:00 2001
From: andy-stark-redis <164213578+andy-stark-redis@users.noreply.github.com>
Date: Wed, 9 Oct 2024 11:42:24 +0100
Subject: [PATCH] DOC-4324 added ZADD and ZRANGE command examples (#3142)

Co-authored-by: ofekshenawa <104765379+ofekshenawa@users.noreply.github.com>
---
 doctests/cmds_sorted_set_test.go | 220 +++++++++++++++++++++++++++++++
 1 file changed, 220 insertions(+)
 create mode 100644 doctests/cmds_sorted_set_test.go

diff --git a/doctests/cmds_sorted_set_test.go b/doctests/cmds_sorted_set_test.go
new file mode 100644
index 00000000..8704fc20
--- /dev/null
+++ b/doctests/cmds_sorted_set_test.go
@@ -0,0 +1,220 @@
+// EXAMPLE: cmds_sorted_set
+// HIDE_START
+package example_commands_test
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/redis/go-redis/v9"
+)
+
+// HIDE_END
+
+func ExampleClient_zadd_cmd() {
+	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, "myzset")
+	// REMOVE_END
+
+	// STEP_START zadd
+	zAddResult1, err := rdb.ZAdd(ctx, "myzset",
+		redis.Z{Member: "one", Score: 1},
+	).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(zAddResult1) // >>> 1
+
+	zAddResult2, err := rdb.ZAdd(ctx, "myzset",
+		redis.Z{Member: "uno", Score: 1},
+	).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(zAddResult2)
+
+	zAddResult3, err := rdb.ZAdd(ctx, "myzset",
+		redis.Z{Member: "two", Score: 2},
+		redis.Z{Member: "three", Score: 3},
+	).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(zAddResult3) // >>> 2
+
+	zAddResult4, err := rdb.ZRangeWithScores(ctx, "myzset", 0, -1).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(zAddResult4) // >>> [{1 one} {1 uno} {2 two} {3 three}]
+	// STEP_END
+
+	// Output:
+	// 1
+	// 1
+	// 2
+	// [{1 one} {1 uno} {2 two} {3 three}]
+}
+
+func ExampleClient_zrange1() {
+	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, "myzset")
+	// REMOVE_END
+
+	// STEP_START zrange1
+	zrangeResult1, err := rdb.ZAdd(ctx, "myzset",
+		redis.Z{Member: "one", Score: 1},
+		redis.Z{Member: "two", Score: 2},
+		redis.Z{Member: "three", Score: 3},
+	).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(zrangeResult1) // >>> 3
+
+	zrangeResult2, err := rdb.ZRange(ctx, "myzset", 0, -1).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(zrangeResult2) // >>> [one two three]
+
+	zrangeResult3, err := rdb.ZRange(ctx, "myzset", 2, 3).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(zrangeResult3) // >>> [three]
+
+	zrangeResult4, err := rdb.ZRange(ctx, "myzset", -2, -1).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(zrangeResult4) // >>> [two three]
+	// STEP_END
+
+	// Output:
+	// 3
+	// [one two three]
+	// [three]
+	// [two three]
+}
+
+func ExampleClient_zrange2() {
+	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, "myzset")
+	// REMOVE_END
+
+	// STEP_START zrange2
+	zRangeResult5, err := rdb.ZAdd(ctx, "myzset",
+		redis.Z{Member: "one", Score: 1},
+		redis.Z{Member: "two", Score: 2},
+		redis.Z{Member: "three", Score: 3},
+	).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(zRangeResult5) // >>> 3
+
+	zRangeResult6, err := rdb.ZRangeWithScores(ctx, "myzset", 0, 1).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(zRangeResult6) // >>> [{1 one} {2 two}]
+	// STEP_END
+
+	// Output:
+	// 3
+	// [{1 one} {2 two}]
+}
+
+func ExampleClient_zrange3() {
+	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, "myzset")
+	// REMOVE_END
+
+	// STEP_START zrange3
+	zRangeResult7, err := rdb.ZAdd(ctx, "myzset",
+		redis.Z{Member: "one", Score: 1},
+		redis.Z{Member: "two", Score: 2},
+		redis.Z{Member: "three", Score: 3},
+	).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(zRangeResult7) // >>> 3
+
+	zRangeResult8, err := rdb.ZRangeArgs(ctx,
+		redis.ZRangeArgs{
+			Key:     "myzset",
+			ByScore: true,
+			Start:   "(1",
+			Stop:    "+inf",
+			Offset:  1,
+			Count:   1,
+		},
+	).Result()
+
+	if err != nil {
+		panic(err)
+	}
+
+	fmt.Println(zRangeResult8) // >>> [three]
+	// STEP_END
+
+	// Output:
+	// 3
+	// [three]
+}