Skip to content

Commit 5b76f60

Browse files
committed
Fixup codegen
1 parent 632b9e5 commit 5b76f60

File tree

2 files changed

+13
-12
lines changed

2 files changed

+13
-12
lines changed

Sources/GRPCCodeGen/Internal/Translator/ClientCodeTranslator.swift

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -263,13 +263,13 @@ extension ClientCodeTranslator {
263263

264264
// All methods have a response handler.
265265
var responseHandler = ParameterDescription(label: "onResponse", name: "handleResponse")
266-
let responseKind = method.isOutputStreaming ? "Stream" : "Single"
266+
let responseKind = method.isOutputStreaming ? "Streaming" : ""
267267
responseHandler.type = .closure(
268268
ClosureSignatureDescription(
269269
parameters: [
270270
ParameterDescription(
271271
type: .generic(
272-
wrapper: .member(["GRPCCore", "ClientResponse", responseKind]),
272+
wrapper: .member(["GRPCCore", "\(responseKind)ClientResponse"]),
273273
wrapped: .member(method.outputType)
274274
)
275275
)
@@ -308,12 +308,12 @@ extension ClientCodeTranslator {
308308
// return try await method(request: request, options: options, responseHandler: responseHandler)
309309

310310
// First, make the init for the ClientRequest
311-
let requestType = method.isInputStreaming ? "Stream" : "Single"
311+
let requestType = method.isInputStreaming ? "Streaming" : ""
312312
var requestInit = FunctionCallDescription(
313313
calledExpression: .identifier(
314314
.type(
315315
.generic(
316-
wrapper: .member(["GRPCCore", "ClientRequest", requestType]),
316+
wrapper: .member(["GRPCCore", "\(requestType)ClientRequest"]),
317317
wrapped: .member(method.inputType)
318318
)
319319
)
@@ -490,9 +490,10 @@ extension ClientCodeTranslator {
490490
for method: CodeGenerationRequest.ServiceDescriptor.MethodDescriptor,
491491
in service: CodeGenerationRequest.ServiceDescriptor
492492
) -> ParameterDescription {
493-
let requestType = method.isInputStreaming ? "Stream" : "Single"
493+
let requestType = method.isInputStreaming ? "Streaming" : ""
494494
let clientRequestType = ExistingTypeDescription.member([
495-
"GRPCCore", "ClientRequest", requestType,
495+
"GRPCCore",
496+
"\(requestType)ClientRequest"
496497
])
497498
return ParameterDescription(
498499
label: "request",
@@ -538,9 +539,9 @@ extension ClientCodeTranslator {
538539
in service: CodeGenerationRequest.ServiceDescriptor,
539540
includeDefaultResponseHandler: Bool
540541
) -> ParameterDescription {
541-
let clientStreaming = method.isOutputStreaming ? "Stream" : "Single"
542+
let clientStreaming = method.isOutputStreaming ? "Streaming" : ""
542543
let closureParameterType = ExistingTypeDescription.generic(
543-
wrapper: .member(["GRPCCore", "ClientResponse", clientStreaming]),
544+
wrapper: .member(["GRPCCore", "\(clientStreaming)ClientResponse"]),
544545
wrapped: .member(method.outputType)
545546
)
546547

Sources/GRPCCodeGen/Internal/Translator/ServerCodeTranslator.swift

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -313,8 +313,8 @@ extension ServerCodeTranslator {
313313
in service: CodeGenerationRequest.ServiceDescriptor,
314314
accessModifier: AccessModifier? = nil
315315
) -> Declaration {
316-
let inputStreaming = method.isInputStreaming ? "Stream" : "Single"
317-
let outputStreaming = method.isOutputStreaming ? "Stream" : "Single"
316+
let inputStreaming = method.isInputStreaming ? "Streaming" : ""
317+
let outputStreaming = method.isOutputStreaming ? "Streaming" : ""
318318

319319
let functionSignature = FunctionSignatureDescription(
320320
accessModifier: accessModifier,
@@ -324,7 +324,7 @@ extension ServerCodeTranslator {
324324
label: "request",
325325
type:
326326
.generic(
327-
wrapper: .member(["GRPCCore", "ServerRequest", inputStreaming]),
327+
wrapper: .member(["GRPCCore", "\(inputStreaming)ServerRequest"]),
328328
wrapped: .member(method.inputType)
329329
)
330330
),
@@ -333,7 +333,7 @@ extension ServerCodeTranslator {
333333
keywords: [.async, .throws],
334334
returnType: .identifierType(
335335
.generic(
336-
wrapper: .member(["GRPCCore", "ServerResponse", outputStreaming]),
336+
wrapper: .member(["GRPCCore", "\(outputStreaming)ServerResponse"]),
337337
wrapped: .member(method.outputType)
338338
)
339339
)

0 commit comments

Comments
 (0)