add options of ActorConfig

wXwcoder · wXwcoder · commit 2cdafafd3a25 · 2022-11-05T14:34:15.000+08:00
diff --git a/actor/config/config.go b/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
+	IdleTimeout            string
+	ScanInterval           string
+	DrainOngingCallTimeout string
+	DrainBalancedActors    bool
 }
 
 // Option is option function of ActorConfig.
@@ -30,6 +34,34 @@ func WithSerializerName(serializerType string) Option {
 	}
 }
 
+// WithIdleTimeout set serializer type of the actor as @idleTimeout.
+func WithIdleTimeout(idleTimeout string) Option {
+	return func(config *ActorConfig) {
+		config.IdleTimeout = idleTimeout
+	}
+}
+
+// WithScanInterval set serializer type of the actor as @scanInterval.
+func WithScanInterval(scanInterval string) Option {
+	return func(config *ActorConfig) {
+		config.ScanInterval = scanInterval
+	}
+}
+
+// WithDrainOngingCallTimeout set serializer type of the actor as @drainOngingCallTimeout.
+func WithDrainOngingCallTimeout(drainOngingCallTimeout string) Option {
+	return func(config *ActorConfig) {
+		config.DrainOngingCallTimeout = drainOngingCallTimeout
+	}
+}
+
+// WithDrainBalancedActors set serializer 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{
diff --git a/actor/config/config_test.go b/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"),
+			WithIdleTimeout("1m"),
+			WithScanInterval("10s"),
+			WithDrainOngingCallTimeout("10s"),
+			WithDrainBalancedActors(true),
 		)
 		assert.NotNil(t, config)
 		assert.Equal(t, "mockSerializerType", config.SerializerType)
+		assert.Equal(t, "1m", config.IdleTimeout)
+		assert.Equal(t, "10s", config.ScanInterval)
+		assert.Equal(t, "10s", config.DrainOngingCallTimeout)
+		assert.Equal(t, true, config.DrainBalancedActors)
 	})
 }
