From 621dadefc0c1d41bf78ae687fbe1732a03f78028 Mon Sep 17 00:00:00 2001
From: sadath-12 <sadathsadu2002@gmail.com>
Date: Fri, 15 Dec 2023 00:19:28 +0530
Subject: [PATCH] perf: enhanced configuration example

Signed-off-by: sadath-12 <sadathsadu2002@gmail.com>
---
 examples/configuration/README.md | 28 ++++++++++++++++++----------
 examples/configuration/main.go   | 16 +++++++++++-----
 2 files changed, 29 insertions(+), 15 deletions(-)

diff --git a/examples/configuration/README.md b/examples/configuration/README.md
index db8bb37c..eb71cc54 100644
--- a/examples/configuration/README.md
+++ b/examples/configuration/README.md
@@ -47,17 +47,25 @@ dapr run --app-id configuration-api\
 The subscription event order may out of order.
 
 ```
-get config = myConfigValue
-get updated config key = mySubscribeKey1, value = mySubscribeValue1 
-get updated config key = mySubscribeKey2, value = mySubscribeValue1 
-get updated config key = mySubscribeKey3, value = mySubscribeValue1 
-get updated config key = mySubscribeKey1, value = mySubscribeValue2 
-get updated config key = mySubscribeKey2, value = mySubscribeValue2 
-get updated config key = mySubscribeKey3, value = mySubscribeValue2 
-get updated config key = mySubscribeKey1, value = mySubscribeValue3 
-get updated config key = mySubscribeKey2, value = mySubscribeValue3 
-get updated config key = mySubscribeKey3, value = mySubscribeValue3 
+got config key = mykey, with value = myConfigValue
+
+got config key = mySubscribeKey1, with value = mySubscribeValue1 
+got config key = mySubscribeKey2, with value = mySubscribeValue1 
+got config key = mySubscribeKey3, with value = mySubscribeValue1 
+got config key = mySubscribeKey1, with value = mySubscribeValue2 
+got config key = mySubscribeKey2, with value = mySubscribeValue2 
+got config key = mySubscribeKey3, with value = mySubscribeValue2 
+got config key = mySubscribeKey1, with value = mySubscribeValue3 
+got config key = mySubscribeKey2, with value = mySubscribeValue3 
+got config key = mySubscribeKey3, with value = mySubscribeValue3 
+got config key = mySubscribeKey1, with value = mySubscribeValue4 
+got config key = mySubscribeKey2, with value = mySubscribeValue4 
+got config key = mySubscribeKey3, with value = mySubscribeValue4 
+got config key = mySubscribeKey1, with value = mySubscribeValue5 
+got config key = mySubscribeKey2, with value = mySubscribeValue5 
+got config key = mySubscribeKey3, with value = mySubscribeValue5 
 dapr configuration subscribe finished.
+dapr configuration unsubscribed
 ✅  Exited App successfully
 
 ```
diff --git a/examples/configuration/main.go b/examples/configuration/main.go
index 66a4cb41..687c9a4c 100644
--- a/examples/configuration/main.go
+++ b/examples/configuration/main.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"strconv"
+	"sync"
 	"time"
 
 	dapr "github.com/dapr/go-sdk/client"
@@ -11,7 +12,7 @@ import (
 	"google.golang.org/grpc/metadata"
 )
 
-func init() {
+func addItems(wg *sync.WaitGroup) {
 	opts := &redis.Options{
 		Addr: "127.0.0.1:6379",
 	}
@@ -19,6 +20,7 @@ func init() {
 	// set config value
 	client.Set(context.Background(), "mykey", "myConfigValue", -1)
 	ticker := time.NewTicker(time.Second)
+	wg.Add(3*5)
 	go func() {
 		for i := 0; i < 5; i++ {
 			<-ticker.C
@@ -32,6 +34,8 @@ func init() {
 }
 
 func main() {
+	var wg sync.WaitGroup
+	addItems(&wg)
 	ctx := context.Background()
 	client, err := dapr.NewClient()
 	if err != nil {
@@ -42,25 +46,27 @@ func main() {
 	if err != nil {
 		panic(err)
 	}
-	fmt.Printf("get config = %s\n", (*items).Value)
+	fmt.Printf("got config key = mykey, with value = %s \n", (*items).Value)
 
 	ctx, f := context.WithTimeout(ctx, 60*time.Second)
 	md := metadata.Pairs("dapr-app-id", "configuration-api")
 	ctx = metadata.NewOutgoingContext(ctx, md)
 	defer f()
 	subscribeID, err := client.SubscribeConfigurationItems(ctx, "example-config", []string{"mySubscribeKey1", "mySubscribeKey2", "mySubscribeKey3"}, func(id string, items map[string]*dapr.ConfigurationItem) {
+		wg.Done()
 		for k, v := range items {
-			fmt.Printf("get updated config key = %s, value = %s \n", k, v.Value)
+			fmt.Printf("got config key = %s, with value = %s \n", k, v.Value)
 		}
 	})
 	if err != nil {
 		panic(err)
 	}
-	time.Sleep(time.Second*3 + time.Millisecond*500)
+	wg.Wait()
 
 	// dapr configuration unsubscribe called.
 	if err := client.UnsubscribeConfigurationItems(ctx, "example-config", subscribeID); err != nil {
 		panic(err)
 	}
-	time.Sleep(time.Second * 5)
+	fmt.Println("dapr configuration unsubscribed")
+	time.Sleep(time.Second)
 }