Merge pull request #38 from tpiperatgod/dev

Keep retrying when dapr client initialization fails instead of just panic

Author: benjaminhuo

context/context.go

@@ -10,6 +10,7 @@ import (
 	"strconv"
 	"strings"
 	"sync"
+	"time"
 
 	cloudevents "github.com/cloudevents/sdk-go/v2"
 	"github.com/dapr/go-sdk/service/common"
@@ -418,13 +419,23 @@ func (ctx *FunctionContext) InitDaprClientIfNil() {
 	}
 
 	if ctx.daprClient == nil {
+		var err error
 		ctx.mu.Lock()
 		defer ctx.mu.Unlock()
-		c, e := dapr.NewClientWithPort(clientGRPCPort)
-		if e != nil {
-			panic(e)
+
+		for attempts := 120; attempts > 0; attempts-- {
+			c, e := dapr.NewClientWithPort(clientGRPCPort)
+			if e == nil {
+				ctx.daprClient = c
+				break
+			}
+			err = e
+			time.Sleep(500 * time.Millisecond)
+		}
+
+		if ctx.daprClient == nil {
+			klog.Errorf("failed to init dapr client: %v", err)
 		}
-		ctx.daprClient = c
 	}
 }
 

runtime/async/async.go

@@ -107,6 +107,9 @@ func (r *Runtime) RegisterOpenFunction(
 							return nil, nil
 						}
 					})
+					if funcErr == nil {
+						klog.Infof("registered bindings handler: %s", input.Uri)
+					}
 				case ofctx.OpenFuncTopic:
 					sub := &dapr.Subscription{
 						PubsubName: input.ComponentName,
@@ -136,6 +139,9 @@ func (r *Runtime) RegisterOpenFunction(
 							return false, nil
 						}
 					})
+					if funcErr == nil {
+						klog.Infof("registered pubsub handler: %s, topic: %s", input.ComponentName, input.Uri)
+					}
 				default:
 					return fmt.Errorf("invalid input type: %s", input.GetType())
 				}