Add GetThriftMetadata to thrift.Processor

Summary: Add GetThriftMetadata to thrift.Processor

Reviewed By: codyohl, podtserkovskiy

Differential Revision: D64175681

fbshipit-source-id: f9a64d7a6e1e9f2dfef69dafae2283acc37fbff2

thrift/lib/go/thrift/header_server_test.go

@@ -23,6 +23,7 @@ import (
 	"time"
 
 	"github.com/facebook/fbthrift/thrift/lib/go/thrift/types"
+	"github.com/facebook/fbthrift/thrift/lib/thrift/metadata"
 )
 
 type headerServerTestProcessor struct {
@@ -33,6 +34,10 @@ func (t *headerServerTestProcessor) ProcessorFunctionMap() map[string]types.Proc
 	return map[string]types.ProcessorFunction{"test": &headerServerTestProcessorFunction{&testProcessorFunction{}, t.requests}}
 }
 
+func (t *headerServerTestProcessor) GetThriftMetadata() *metadata.ThriftMetadata {
+	return nil
+}
+
 type headerServerTestProcessorFunction struct {
 	types.ProcessorFunction
 	requests chan<- *MyTestStruct

thrift/lib/go/thrift/interceptor.go

@@ -20,6 +20,7 @@ import (
 	"context"
 
 	"github.com/facebook/fbthrift/thrift/lib/go/thrift/types"
+	"github.com/facebook/fbthrift/thrift/lib/thrift/metadata"
 )
 
 // Interceptor is a function that runs before the actual method. It is passed
@@ -60,6 +61,10 @@ func (p *interceptorProcessor) ProcessorFunctionMap() map[string]types.Processor
 	return mi
 }
 
+func (p *interceptorProcessor) GetThriftMetadata() *metadata.ThriftMetadata {
+	return p.Processor.GetThriftMetadata()
+}
+
 type interceptorProcessorFunction struct {
 	interceptor Interceptor
 	methodName  string

thrift/lib/go/thrift/processor.go

@@ -22,6 +22,7 @@ import (
 	"strings"
 
 	"github.com/facebook/fbthrift/thrift/lib/go/thrift/types"
+	"github.com/facebook/fbthrift/thrift/lib/thrift/metadata"
 )
 
 // Processor exposes access to processor functions which
@@ -37,6 +38,7 @@ type Processor interface {
 	// sent which explains that no processor function exists with the specified
 	// name on this server.
 	ProcessorFunctionMap() map[string]types.ProcessorFunction
+	GetThriftMetadata() *metadata.ThriftMetadata
 }
 
 func errorType(err error) string {

thrift/lib/go/thrift/processor_composite.go

@@ -18,6 +18,7 @@ package thrift
 
 import (
 	"github.com/facebook/fbthrift/thrift/lib/go/thrift/types"
+	"github.com/facebook/fbthrift/thrift/lib/thrift/metadata"
 )
 
 // CompositeProcessor is a serial Processor can report what functions it has
@@ -31,12 +32,14 @@ type CompositeProcessor interface {
 // server as long as their functions carry distinct names
 type compositeProcessor struct {
 	serviceProcessorMap map[string]types.ProcessorFunction
+	metadata            *metadata.ThriftMetadata
 }
 
 // NewCompositeProcessor creates a new CompositeProcessor
 func NewCompositeProcessor() CompositeProcessor {
 	return &compositeProcessor{
 		serviceProcessorMap: make(map[string]types.ProcessorFunction),
+		metadata:            metadata.NewThriftMetadata(),
 	}
 }
 
@@ -49,6 +52,18 @@ func (p *compositeProcessor) Include(processor Processor) {
 	for name, tfunc := range processor.ProcessorFunctionMap() {
 		p.serviceProcessorMap[name] = tfunc
 	}
+	for name, v := range processor.GetThriftMetadata().GetEnums() {
+		p.metadata.Enums[name] = v
+	}
+	for name, v := range processor.GetThriftMetadata().GetStructs() {
+		p.metadata.Structs[name] = v
+	}
+	for name, v := range processor.GetThriftMetadata().GetExceptions() {
+		p.metadata.Exceptions[name] = v
+	}
+	for name, v := range processor.GetThriftMetadata().GetServices() {
+		p.metadata.Services[name] = v
+	}
 }
 
 // GetProcessorFunction multiplexes redirects to the appropriate Processor
@@ -61,3 +76,7 @@ func (p *compositeProcessor) GetProcessorFunction(name string) types.ProcessorFu
 func (p *compositeProcessor) ProcessorFunctionMap() map[string]types.ProcessorFunction {
 	return p.serviceProcessorMap
 }
+
+func (p *compositeProcessor) GetThriftMetadata() *metadata.ThriftMetadata {
+	return p.metadata
+}

thrift/lib/go/thrift/rocket_server_test.go

@@ -24,6 +24,7 @@ import (
 	"time"
 
 	"github.com/facebook/fbthrift/thrift/lib/go/thrift/types"
+	"github.com/facebook/fbthrift/thrift/lib/thrift/metadata"
 )
 
 type rocketServerTestProcessor struct {
@@ -34,6 +35,10 @@ func (t *rocketServerTestProcessor) ProcessorFunctionMap() map[string]types.Proc
 	return map[string]types.ProcessorFunction{"test": &rocketServerTestProcessorFunction{&testProcessorFunction{}, t.requests}}
 }
 
+func (t *rocketServerTestProcessor) GetThriftMetadata() *metadata.ThriftMetadata {
+	return nil
+}
+
 type rocketServerTestProcessorFunction struct {
 	types.ProcessorFunction
 	requests chan<- *MyTestStruct

thrift/lib/go/thrift/rocket_upgrade_processor.go

@@ -21,6 +21,7 @@ import (
 	"maps"
 
 	"github.com/facebook/fbthrift/thrift/lib/go/thrift/types"
+	"github.com/facebook/fbthrift/thrift/lib/thrift/metadata"
 )
 
 type rocketUpgradeProcessor struct {
@@ -43,6 +44,10 @@ func (r *rocketUpgradeProcessor) ProcessorFunctionMap() map[string]types.Process
 	return mr
 }
 
+func (r *rocketUpgradeProcessor) GetThriftMetadata() *metadata.ThriftMetadata {
+	return r.processor.GetThriftMetadata()
+}
+
 type rocketUpgradeProcessorFunction struct {
 	upgraded *bool
 }

thrift/lib/go/thrift/server_test.go

@@ -22,6 +22,7 @@ import (
 	"testing"
 
 	"github.com/facebook/fbthrift/thrift/lib/go/thrift/types"
+	"github.com/facebook/fbthrift/thrift/lib/thrift/metadata"
 )
 
 // TestSimpleServer is a simple tests that simple sends an empty message to a server and receives an empty result.
@@ -66,6 +67,10 @@ func (t *testProcessor) ProcessorFunctionMap() map[string]types.ProcessorFunctio
 	return map[string]types.ProcessorFunction{"test": &testProcessorFunction{}}
 }
 
+func (t *testProcessor) GetThriftMetadata() *metadata.ThriftMetadata {
+	return nil
+}
+
 type testProcessorFunction struct{}
 
 func (p *testProcessorFunction) Read(prot types.Decoder) (types.Struct, types.Exception) {