add support for actor runtime config, to make idle timeout scanning a…

…nd gc of runtime actor in effective dapr#334

Signed-off-by: wangxw <[email protected]>

actor/config/config.go

@@ -17,7 +17,11 @@ import "github.com/dapr/go-sdk/actor/codec/constant"
 
 // ActorConfig is Actor's configuration struct.
 type ActorConfig struct {
-	SerializerType string
+	SerializerType         string
+	ActorIdleTimeout       string
+	ActorScanInterval      string
+	DrainOngingCallTimeout string
+	DrainBalancedActors    bool
 }
 
 // Option is option function of ActorConfig.
@@ -30,6 +34,34 @@ func WithSerializerName(serializerType string) Option {
 	}
 }
 
+// WithActorIdleTimeout set actorIdleTimeout type of the actor as @actorIdleTimeout.
+func WithActorIdleTimeout(actorIdleTimeout string) Option {
+	return func(config *ActorConfig) {
+		config.ActorIdleTimeout = actorIdleTimeout
+	}
+}
+
+// WithActorScanInterval set actorScanInterval type of the actor as @actorScanInterval.
+func WithActorScanInterval(actorScanInterval string) Option {
+	return func(config *ActorConfig) {
+		config.ActorScanInterval = actorScanInterval
+	}
+}
+
+// WithDrainOngingCallTimeout set drainOngingCallTimeout type of the actor as @drainOngingCallTimeout.
+func WithDrainOngingCallTimeout(drainOngingCallTimeout string) Option {
+	return func(config *ActorConfig) {
+		config.DrainOngingCallTimeout = drainOngingCallTimeout
+	}
+}
+
+// WithDrainBalancedActors set drainBalancedActors type of the actor as @drainBalancedActors.
+func WithDrainBalancedActors(drainBalancedActors bool) Option {
+	return func(config *ActorConfig) {
+		config.DrainBalancedActors = drainBalancedActors
+	}
+}
+
 // GetConfigFromOptions get final ActorConfig set by @opts.
 func GetConfigFromOptions(opts ...Option) *ActorConfig {
 	conf := &ActorConfig{

actor/config/config_test.go

@@ -31,8 +31,16 @@ func TestRegisterActorTimer(t *testing.T) {
 	t.Run("get config with option", func(t *testing.T) {
 		config := GetConfigFromOptions(
 			WithSerializerName("mockSerializerType"),
+			WithActorIdleTimeout("1m"),
+			WithActorScanInterval("10s"),
+			WithDrainOngingCallTimeout("10s"),
+			WithDrainBalancedActors(true),
 		)
 		assert.NotNil(t, config)
 		assert.Equal(t, "mockSerializerType", config.SerializerType)
+		assert.Equal(t, "1m", config.ActorIdleTimeout)
+		assert.Equal(t, "10s", config.ActorScanInterval)
+		assert.Equal(t, "10s", config.DrainOngingCallTimeout)
+		assert.Equal(t, true, config.DrainBalancedActors)
 	})
 }

actor/runtime/actor_runtime.go

@@ -49,6 +49,10 @@ func (r *ActorRunTime) RegisterActorFactory(f actor.Factory, opt ...config.Optio
 	conf := config.GetConfigFromOptions(opt...)
 	actType := f().Type()
 	r.config.RegisteredActorTypes = append(r.config.RegisteredActorTypes, actType)
+	r.config.ActorIdleTimeout = conf.ActorIdleTimeout
+	r.config.ActorScanInterval = conf.ActorScanInterval
+	r.config.DrainOngingCallTimeout = conf.DrainOngingCallTimeout
+	r.config.DrainBalancedActors = conf.DrainBalancedActors
 	mng, ok := r.actorManagers.Load(actType)
 	if !ok {
 		newMng, err := manager.NewDefaultActorManager(conf.SerializerType)