Activity Timeout Engine changes.

Reviewers: maxim, tamer, samar Reviewed By: samar Subscribers: jenkins Differential Revision: https://code.uberinternal.com/D698711
zshni · Jan 23, 2017 · 2700912 · 2700912
1 parent 1f7f0f3
commit 2700912
Show file tree

Hide file tree

Showing 27 changed files with 1,539 additions and 85 deletions.
diff --git a/.gen/go/shared/ttypes.go b/.gen/go/shared/ttypes.go
diff --git a/.gen/js/shared_types.js b/.gen/js/shared_types.js
diff --git a/.gen/py/shared/ttypes.py b/.gen/py/shared/ttypes.py
diff --git a/client/history/client.go b/client/history/client.go
@@ -123,3 +123,13 @@ func (c *clientImpl) RespondActivityTaskFailed(request *workflow.RespondActivity
 	defer cancel()
 	return client.RespondActivityTaskFailed(ctx, request)
 }
+
+func (c *clientImpl) RecordActivityTaskHeartbeat(request *workflow.RecordActivityTaskHeartbeatRequest) (*workflow.RecordActivityTaskHeartbeatResponse, error) {
+	client, err := c.getHostForRequest(shardID)
+	if err != nil {
+		return nil, err
+	}
+	ctx, cancel := c.createContext()
+	defer cancel()
+	return client.RecordActivityTaskHeartbeat(ctx, request)
+}
diff --git a/client/history/interface.go b/client/history/interface.go
@@ -15,4 +15,5 @@ type Client interface {
 	RespondDecisionTaskCompleted(request *workflow.RespondDecisionTaskCompletedRequest) error
 	RespondActivityTaskCompleted(request *workflow.RespondActivityTaskCompletedRequest) error
 	RespondActivityTaskFailed(request *workflow.RespondActivityTaskFailedRequest) error
+	RecordActivityTaskHeartbeat(request *workflow.RecordActivityTaskHeartbeatRequest) (*workflow.RecordActivityTaskHeartbeatResponse, error)
 }
diff --git a/common/metrics/constants.go b/common/metrics/constants.go
@@ -85,6 +85,7 @@ const (
 	RespondActivityTaskFailedScope
 	GetWorkflowExecutionHistoryScope
 	GetTimerIndexTasksScope
+	GetWorkflowMutableStateScope
 )
 
 // ScopeToTags record the scope name for all services
@@ -111,6 +112,7 @@ var ScopeToTags = [NumServices][]map[string]string{
 		{OperationTagName: RespondActivityTaskFailedOperationTagValue},
 		{OperationTagName: GetWorkflowExecutionHistoryOperationTagValue},
 		{OperationTagName: GetTimerIndexTasksOperationTagValue},
+		{OperationTagName: GetWorkflowMutableStateOperationTagValue},
 	},
 }
 
@@ -169,6 +171,7 @@ const (
 	RespondActivityTaskFailedOperationTagValue    = "RespondActivityTaskFailed"
 	GetWorkflowExecutionHistoryOperationTagValue  = "GetWorkflowExecutionHistory"
 	GetTimerIndexTasksOperationTagValue           = "GetTimerIndexTasks"
+	GetWorkflowMutableStateOperationTagValue      = "GetWorkflowMutableState"
 )
 
 // ErrorClass is an enum to help with classifying SLA vs. non-SLA errors (SLA = "service level agreement")