unistack-org/micro
4
0
Fork 2
Code Issues 18 Pull Requests Actions Packages Projects Releases 331 Wiki Activity

Don't set stream unless its true

Browse Source
This commit is contained in:
Asim Aslam
2019-11-09 10:32:52 +00:00
parent 0b1e6d7eaf
commit 12faf3af2a
2 changed files with 19 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
server/extractor.go
View File

@@ -95,14 +95,20 @@ func extractEndpoint(method reflect.Method) *registry.Endpoint {
request := extractValue(reqType, 0) request := extractValue(reqType, 0)
response := extractValue(rspType, 0) response := extractValue(rspType, 0)
return &registry.Endpoint{ ep := &registry.Endpoint{
Name: method.Name, Name: method.Name,
Request: request, Request: request,
Response: response, Response: response,
Metadata: map[string]string{
"stream": fmt.Sprintf("%v", stream),
},
} }
// set endpoint metadata for stream
if stream {
ep.Metadata = map[string]string{
"stream": fmt.Sprintf("%v", stream),
}
}
return ep
} }
func extractSubValue(typ reflect.Type) *registry.Value { func extractSubValue(typ reflect.Type) *registry.Value {

13
server/grpc/extractor.go
View File

@@ -90,14 +90,19 @@ func extractEndpoint(method reflect.Method) *registry.Endpoint {
request := extractValue(reqType, 0) request := extractValue(reqType, 0)
response := extractValue(rspType, 0) response := extractValue(rspType, 0)
return &registry.Endpoint{ ep := &registry.Endpoint{
Name: method.Name, Name: method.Name,
Request: request, Request: request,
Response: response, Response: response,
Metadata: map[string]string{
"stream": fmt.Sprintf("%v", stream),
},
} }
if stream {
ep.Metadata = map[string]string{
"stream": fmt.Sprintf("%v", stream),
}
}
return ep
} }
func extractSubValue(typ reflect.Type) *registry.Value { func extractSubValue(typ reflect.Type) *registry.Value {