Add plumbing to send termination signal to handler

Summary:
TSIA. To be used downstream.

#build_rule_type[go_library,go_binary,go_test]

Reviewed By: siutsin

Differential Revision: D86181808

fbshipit-source-id: 9648a6933b5c0845e83c4617f45937418a5ff9e5

Author: echistyakov

third-party/thrift/src/thrift/compiler/generate/templates/go/svc/processor.mustache

@@ -99,3 +99,12 @@ func (p *{{service:go_name}}Processor) GetThriftMetadata() *{{program:metadata_q
 
 {{/function:go_server_supported?}}
 {{/service:functions}}
+{{#service:interaction?}}
+
+func (p *{{service:go_name}}Processor) OnTermination() {
+    // If the underlying handler implements OnTermination()
+    if terminable, ok := p.handler.(thrift.Terminable); ok {
+        terminable.OnTermination()
+    }
+}
+{{/service:interaction?}}

third-party/thrift/src/thrift/compiler/test/fixtures/basic-annotations/out/go/gen-go/svcs.go

@@ -141,6 +141,13 @@ func (p *procFuncBadInteractionFoo) RunContext(ctx context.Context, reqStruct th
     return result, nil
 }
 
+func (p *BadInteractionProcessor) OnTermination() {
+    // If the underlying handler implements OnTermination()
+    if terminable, ok := p.handler.(thrift.Terminable); ok {
+        terminable.OnTermination()
+    }
+}
+
 
 type MyService interface {
     Ping(ctx context.Context) (error)

third-party/thrift/src/thrift/compiler/test/fixtures/interactions/out/go/gen-go/svcs.go

@@ -299,6 +299,13 @@ func (p *procFuncMyInteractionTruthify) RunStreamContext(
     onStreamComplete()
 }
 
+func (p *MyInteractionProcessor) OnTermination() {
+    // If the underlying handler implements OnTermination()
+    if terminable, ok := p.handler.(thrift.Terminable); ok {
+        terminable.OnTermination()
+    }
+}
+
 type MyInteractionFast interface {
     Frobnicate(ctx context.Context) (int32, error)
     Ping(ctx context.Context) (error)
@@ -563,6 +570,13 @@ func (p *procFuncMyInteractionFastTruthify) RunStreamContext(
     onStreamComplete()
 }
 
+func (p *MyInteractionFastProcessor) OnTermination() {
+    // If the underlying handler implements OnTermination()
+    if terminable, ok := p.handler.(thrift.Terminable); ok {
+        terminable.OnTermination()
+    }
+}
+
 type SerialInteraction interface {
     Frobnicate(ctx context.Context) (error)
 }
@@ -677,6 +691,13 @@ func (p *procFuncSerialInteractionFrobnicate) RunContext(ctx context.Context, re
     return result, nil
 }
 
+func (p *SerialInteractionProcessor) OnTermination() {
+    // If the underlying handler implements OnTermination()
+    if terminable, ok := p.handler.(thrift.Terminable); ok {
+        terminable.OnTermination()
+    }
+}
+
 type BoxedInteraction interface {
     GetABox(ctx context.Context) (*ShouldBeBoxed, error)
 }
@@ -792,6 +813,13 @@ func (p *procFuncBoxedInteractionGetABox) RunContext(ctx context.Context, reqStr
     return result, nil
 }
 
+func (p *BoxedInteractionProcessor) OnTermination() {
+    // If the underlying handler implements OnTermination()
+    if terminable, ok := p.handler.(thrift.Terminable); ok {
+        terminable.OnTermination()
+    }
+}
+
 
 type MyService interface {
     Foo(ctx context.Context) (error)

third-party/thrift/src/thrift/lib/go/thrift/rocket_server.go

@@ -205,9 +205,15 @@ func (s *rocketServer) metadataPush(msg payload.Payload) {
 	if metadata.InteractionTerminate != nil {
 		interactionID := metadata.InteractionTerminate.InteractionId
 		s.interactionsMutex.Lock()
+		interaction, ok := s.interactions[interactionID]
 		delete(s.interactions, interactionID)
 		s.interactionsMutex.Unlock()
 		s.log("receive interaction terminate signal for ID %d", interactionID)
+		if ok {
+			if terminable, isTerminable := interaction.(types.Terminable); isTerminable {
+				terminable.OnTermination()
+			}
+		}
 	} else if metadata.StreamHeadersPush != nil {
 		s.log("unsupported StreamHeadersPush metadata type")
 	} else if metadata.TransportMetadataPush != nil {

third-party/thrift/src/thrift/lib/go/thrift/types/interaction.go

@@ -22,3 +22,8 @@ const (
 	interactionCreateKey interactionContextKey = 1
 	interactionIDKey     interactionContextKey = 2
 )
+
+// Terminable is implemented by an interaction processor.
+type Terminable interface {
+	OnTermination()
+}