- Provide grace timeout from the plugin

Signed-off-by: Valery Piashchynski <[email protected]>
temporalio · Mar 31, 2021 · 9662753 · 9662753
1 parent be27b2d
commit 9662753
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/activity/activity_pool.go b/activity/activity_pool.go
@@ -53,7 +53,7 @@ type activityPoolImpl struct {
 }
 
 // newActivityPool
-func newActivityPool(codec rrt.Codec, listener events.Listener, poolConfig pool.Config, server server.Server) (activityPool, error) {
+func newActivityPool(codec rrt.Codec, graceTimeout time.Duration, listener events.Listener, poolConfig pool.Config, server server.Server) (activityPool, error) {
 	const op = errors.Op("new_activity_pool")
 	// env variables
 	env := map[string]string{RR_MODE: roadrunner_temporal.RRMode, RR_CODEC: codec.GetName()}
@@ -63,9 +63,10 @@ func newActivityPool(codec rrt.Codec, listener events.Listener, poolConfig pool.
 	}
 
 	return &activityPoolImpl{
-		codec:   codec,
-		wp:      wp,
-		running: sync.Map{},
+		codec:        codec,
+		wp:           wp,
+		running:      sync.Map{},
+		graceTimeout: graceTimeout,
 	}, nil
 }
 

diff --git a/activity/plugin.go b/activity/plugin.go
@@ -169,6 +169,7 @@ func (p *Plugin) poolListener(event interface{}) {
 func (p *Plugin) startPool() (activityPool, error) {
 	pool, err := newActivityPool(
 		p.temporal.GetCodec().WithLogger(p.log),
+		p.graceTimeout,
 		p.poolListener,
 		*p.temporal.GetConfig().Activities,
 		p.server,