Updated the code to GRPCPayload and added GRPCProtobufPayload

Author: mustii

Sources/Examples/Echo/Model/echo.grpc.swift

@@ -149,26 +149,5 @@ extension Echo_EchoProvider {
   }
 }
 
-extension Echo_EchoRequest: Payload {
-    public init(serializedByteBuffer: inout ByteBuffer) throws {
-        try self.init(serializedData: serializedByteBuffer.readData(length: serializedByteBuffer.readableBytes)!,
-                  extensions: nil,
-                  partial: false,
-                  options: BinaryDecodingOptions())
-    }
-    public func serialize(into buffer: inout ByteBuffer) throws {
-        let data = try serializedData()
-        buffer.writeBytes(data)
-    }
-}
-
-extension Echo_EchoResponse: Payload {
-    public init(serializedByteBuffer: inout ByteBuffer) throws {
-        try self.init(serializedData: serializedByteBuffer.readData(length: serializedByteBuffer.readableBytes)!,
-                  extensions: nil, partial: false, options: BinaryDecodingOptions())
-    }
-    public func serialize(into buffer: inout ByteBuffer) throws {
-        let data = try serializedData()
-        buffer.writeBytes(data)
-    }
-}
+extension Echo_EchoRequest: GRPCProtobufPayload {}
+extension Echo_EchoResponse: GRPCProtobufPayload {}

Sources/Examples/HelloWorld/Model/helloworld.grpc.swift

@@ -84,31 +84,5 @@ extension Helloworld_GreeterProvider {
   }
 }
 
-extension Helloworld_HelloRequest: Payload {
-    public init(serializedByteBuffer: inout ByteBuffer) throws {
-        try self.init(serializedData: serializedByteBuffer.readData(length: serializedByteBuffer.readableBytes)!,
-                  extensions: nil,
-                  partial: false,
-                  options: BinaryDecodingOptions())
-    }
-    
-    public func serialize(into buffer: inout ByteBuffer) throws {
-        let data = try serializedData()
-        buffer.writeBytes(data)
-    }
-}
-
-extension Helloworld_HelloReply: Payload {
-    public init(serializedByteBuffer: inout ByteBuffer) throws {
-        try self.init(serializedData: serializedByteBuffer.readData(length: serializedByteBuffer.readableBytes)!,
-                  extensions: nil,
-                  partial: false,
-                  options: BinaryDecodingOptions())
-    }
-    
-    public func serialize(into buffer: inout ByteBuffer) throws {
-        let data = try serializedData()
-        buffer.writeBytes(data)
-    }
-}
-
+extension Helloworld_HelloRequest: GRPCProtobufPayload {}
+extension Helloworld_HelloReply: GRPCProtobufPayload {}

Sources/Examples/RouteGuide/Model/route_guide.grpc.swift

@@ -149,28 +149,8 @@ extension Routeguide_RouteGuideProvider {
   }
 }
 
-extension Routeguide_RouteNote: Payload {
-    public init(serializedByteBuffer: inout ByteBuffer) throws {}
-    public func serialize(into buffer: inout ByteBuffer) throws {}
-}
-
-extension Routeguide_Feature: Payload {
-    public init(serializedByteBuffer: inout ByteBuffer) throws {}
-    public func serialize(into buffer: inout ByteBuffer) throws {}
-}
-
-extension Routeguide_RouteSummary: Payload {
-    public init(serializedByteBuffer: inout ByteBuffer) throws {}
-    public func serialize(into buffer: inout ByteBuffer) throws {}
-}
-
-extension Routeguide_Point: Payload {
-    public init(serializedByteBuffer: inout ByteBuffer) throws {}
-    public func serialize(into buffer: inout ByteBuffer) throws {}
-}
-
-extension Routeguide_Rectangle: Payload {
-    public init(serializedByteBuffer: inout ByteBuffer) throws {}
-    public func serialize(into buffer: inout ByteBuffer) throws {}
-}
-
+extension Routeguide_RouteNote: GRPCProtobufPayload {}
+extension Routeguide_Feature: GRPCProtobufPayload {}
+extension Routeguide_RouteSummary: GRPCProtobufPayload {}
+extension Routeguide_Point: GRPCProtobufPayload {}
+extension Routeguide_Rectangle: GRPCProtobufPayload {}

Sources/GRPC/CallHandlers/BidirectionalStreamingCallHandler.swift

@@ -26,8 +26,8 @@ import Logging
 ///   they can fail the observer block future.
 /// - To close the call and send the status, complete `context.statusPromise`.
 public class BidirectionalStreamingCallHandler<
-  RequestMessage: Payload,
-  ResponseMessage: Payload
+  RequestMessage: GRPCPayload,
+  ResponseMessage: GRPCPayload
 >: _BaseCallHandler<RequestMessage, ResponseMessage> {
   public typealias Context = StreamingResponseCallContext<ResponseMessage>
   public typealias EventObserver = (StreamEvent<RequestMessage>) -> Void

Sources/GRPC/CallHandlers/ClientStreamingCallHandler.swift

@@ -34,8 +34,8 @@ enum ClientStreamingHandlerObserverState<Factory, Observer> {
 ///   they can fail the observer block future.
 /// - To close the call and send the response, complete `context.responsePromise`.
 public final class ClientStreamingCallHandler<
-  RequestMessage: Payload,
-  ResponseMessage: Payload
+  RequestMessage: GRPCPayload,
+  ResponseMessage: GRPCPayload
 >: _BaseCallHandler<RequestMessage, ResponseMessage> {
   public typealias Context = UnaryResponseCallContext<ResponseMessage>
   public typealias EventObserver = (StreamEvent<RequestMessage>) -> Void

Sources/GRPC/CallHandlers/ServerStreamingCallHandler.swift

@@ -24,8 +24,8 @@ import Logging
 /// - The observer block is implemented by the framework user and calls `context.sendResponse` as needed.
 /// - To close the call and send the status, complete the status future returned by the observer block.
 public final class ServerStreamingCallHandler<
-  RequestMessage: Payload,
-  ResponseMessage: Payload
+  RequestMessage: GRPCPayload,
+  ResponseMessage: GRPCPayload
 >: _BaseCallHandler<RequestMessage, ResponseMessage> {
   public typealias EventObserver = (RequestMessage) -> EventLoopFuture<GRPCStatus>
 

Sources/GRPC/CallHandlers/UnaryCallHandler.swift

@@ -25,8 +25,8 @@ import Logging
 /// - To return a response to the client, the framework user should complete that future
 ///   (similar to e.g. serving regular HTTP requests in frameworks such as Vapor).
 public final class UnaryCallHandler<
-  RequestMessage: Payload,
-  ResponseMessage: Payload
+  RequestMessage: GRPCPayload,
+  ResponseMessage: GRPCPayload
 >: _BaseCallHandler<RequestMessage, ResponseMessage> {
   public typealias EventObserver = (RequestMessage) -> EventLoopFuture<ResponseMessage>
   private var eventObserver: EventObserver?

Sources/GRPC/CallHandlers/_BaseCallHandler.swift

@@ -23,7 +23,7 @@ import Logging
 ///
 /// Calls through to `processMessage` for individual messages it receives, which needs to be implemented by subclasses.
 /// - Important: This is **NOT** part of the public API.
-public class _BaseCallHandler<RequestMessage: Payload, ResponseMessage: Payload>: GRPCCallHandler {
+public class _BaseCallHandler<RequestMessage: GRPCPayload, ResponseMessage: GRPCPayload>: GRPCCallHandler {
   public func makeGRPCServerCodec() -> ChannelHandler {
     return GRPCServerCodec<RequestMessage, ResponseMessage>()
   }

Sources/GRPC/ClientCalls/BaseClientCall.swift

@@ -48,7 +48,7 @@ import Logging
 ///
 /// This class also provides much of the framework user facing functionality via conformance to
 /// `ClientCall`.
-public class BaseClientCall<Request: Payload, Response: Payload>: ClientCall {
+public class BaseClientCall<Request: GRPCPayload, Response: GRPCPayload>: ClientCall {
   public typealias RequestMessage = Request
   public typealias ResponseMessage = Response
 

Sources/GRPC/ClientCalls/BidirectionalStreamingCall.swift

@@ -27,7 +27,7 @@ import Logging
 /// - `initialMetadata`: the initial metadata returned from the server,
 /// - `status`: the status of the gRPC call after it has ended,
 /// - `trailingMetadata`: any metadata returned from the server alongside the `status`.
-public final class BidirectionalStreamingCall<RequestMessage: Payload, ResponseMessage: Payload>
+public final class BidirectionalStreamingCall<RequestMessage: GRPCPayload, ResponseMessage: GRPCPayload>
   : BaseClientCall<RequestMessage, ResponseMessage>,
     StreamingRequestClientCall {
   private var messageQueue: EventLoopFuture<Void>

Sources/GRPC/ClientCalls/ClientCall.swift

@@ -23,9 +23,9 @@ import SwiftProtobuf
 /// Base protocol for a client call to a gRPC service.
 public protocol ClientCall {
   /// The type of the request message for the call.
-  associatedtype RequestMessage: Payload
+  associatedtype RequestMessage: GRPCPayload
   /// The type of the response message for the call.
-  associatedtype ResponseMessage: Payload
+  associatedtype ResponseMessage: GRPCPayload
 
   /// HTTP/2 stream that requests and responses are sent and received on.
   var subchannel: EventLoopFuture<Channel> { get }

Sources/GRPC/ClientCalls/ClientStreamingCall.swift

@@ -28,7 +28,7 @@ import Logging
 /// - `response`: the response from the call,
 /// - `status`: the status of the gRPC call after it has ended,
 /// - `trailingMetadata`: any metadata returned from the server alongside the `status`.
-public final class ClientStreamingCall<RequestMessage: Payload, ResponseMessage: Payload>
+public final class ClientStreamingCall<RequestMessage: GRPCPayload, ResponseMessage: GRPCPayload>
   : BaseClientCall<RequestMessage, ResponseMessage>,
     StreamingRequestClientCall,
     UnaryResponseClientCall {

Sources/GRPC/ClientCalls/ServerStreamingCall.swift

@@ -24,7 +24,7 @@ import Logging
 /// - `initialMetadata`: the initial metadata returned from the server,
 /// - `status`: the status of the gRPC call after it has ended,
 /// - `trailingMetadata`: any metadata returned from the server alongside the `status`.
-public final class ServerStreamingCall<RequestMessage: Payload, ResponseMessage: Payload>: BaseClientCall<RequestMessage, ResponseMessage> {
+public final class ServerStreamingCall<RequestMessage: GRPCPayload, ResponseMessage: GRPCPayload>: BaseClientCall<RequestMessage, ResponseMessage> {
   public init(
     connection: ClientConnection,
     path: String,

Sources/GRPC/ClientCalls/UnaryCall.swift

@@ -27,7 +27,7 @@ import Logging
 /// - `response`: the response from the unary call,
 /// - `status`: the status of the gRPC call after it has ended,
 /// - `trailingMetadata`: any metadata returned from the server alongside the `status`.
-public final class UnaryCall<RequestMessage: Payload, ResponseMessage: Payload>
+public final class UnaryCall<RequestMessage: GRPCPayload, ResponseMessage: GRPCPayload>
   : BaseClientCall<RequestMessage, ResponseMessage>,
     UnaryResponseClientCall {
   public let response: EventLoopFuture<ResponseMessage>

Sources/GRPC/GRPCClientResponseChannelHandler.swift

@@ -24,7 +24,7 @@ import Logging
 /// This includes holding promises for the initial metadata and status of the gRPC call. This handler
 /// is also responsible for error handling, via an error delegate and by appropriately failing the
 /// aforementioned promises.
-internal class GRPCClientResponseChannelHandler<ResponseMessage: Payload>: ChannelInboundHandler {
+internal class GRPCClientResponseChannelHandler<ResponseMessage: GRPCPayload>: ChannelInboundHandler {
   public typealias InboundIn = _GRPCClientResponsePart<ResponseMessage>
   internal let logger: Logger
   internal var stopwatch: Stopwatch?
@@ -186,7 +186,7 @@ internal class GRPCClientResponseChannelHandler<ResponseMessage: Payload>: Chann
 }
 
 /// A channel handler for client calls which receive a single response.
-final class GRPCClientUnaryResponseChannelHandler<ResponseMessage: Payload>: GRPCClientResponseChannelHandler<ResponseMessage> {
+final class GRPCClientUnaryResponseChannelHandler<ResponseMessage: GRPCPayload>: GRPCClientResponseChannelHandler<ResponseMessage> {
   let responsePromise: EventLoopPromise<ResponseMessage>
 
   internal init(
@@ -236,7 +236,7 @@ final class GRPCClientUnaryResponseChannelHandler<ResponseMessage: Payload>: GRP
 }
 
 /// A channel handler for client calls which receive a stream of responses.
-final class GRPCClientStreamingResponseChannelHandler<ResponseMessage: Payload>: GRPCClientResponseChannelHandler<ResponseMessage> {
+final class GRPCClientStreamingResponseChannelHandler<ResponseMessage: GRPCPayload>: GRPCClientResponseChannelHandler<ResponseMessage> {
   typealias ResponseHandler = (ResponseMessage) -> Void
 
   let responseHandler: ResponseHandler

Sources/GRPC/GRPCClientStateMachine.swift

@@ -66,7 +66,7 @@ enum SendEndOfRequestStreamError: Error {
 /// A state machine for a single gRPC call from the perspective of a client.
 ///
 /// See: https://github.com/grpc/grpc/blob/master/doc/PROTOCOL-HTTP2.md
-struct GRPCClientStateMachine<Request: Payload, Response: Payload> {
+struct GRPCClientStateMachine<Request: GRPCPayload, Response: GRPCPayload> {
   /// The combined state of the request (client) and response (server) streams for an RPC call.
   ///
   /// The following states are not possible:

Sources/GRPC/Payload.swift renamed to Sources/GRPC/GRPCPayload.swift

@@ -1,5 +1,5 @@
 /*
- * Copyright YEARS, gRPC Authors All rights reserved.
+ * Copyright 2020, gRPC Authors All rights reserved.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -14,9 +14,15 @@
  * limitations under the License.
  */
 import NIO
-import SwiftProtobuf
 
-public protocol Payload {
-    init(serializedByteBuffer: inout NIO.ByteBuffer) throws
-    func serialize(into buffer: inout NIO.ByteBuffer) throws
+public protocol GRPCPayload {
+    init()
+}
+
+extension GRPCPayload {
+    init(serializedByteBuffer: inout NIO.ByteBuffer) throws {
+        self.init()
+    }
+    
+    func serialize(into buffer: inout NIO.ByteBuffer) throws {}
 }

Sources/GRPC/GRPCProtobufPayload.swift

@@ -0,0 +1,32 @@
+/*
+ * Copyright 2020, gRPC Authors All rights reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+import NIO
+import SwiftProtobuf
+
+public protocol GRPCProtobufPayload: GRPCPayload {}
+
+extension GRPCProtobufPayload where Self: SwiftProtobuf.Message {
+    init(serializedByteBuffer: inout NIO.ByteBuffer) throws {
+        try self.init(serializedData: serializedByteBuffer.readData(length: serializedByteBuffer.readableBytes)!,
+                  extensions: nil,
+                  partial: false,
+                  options: BinaryDecodingOptions())
+    }
+    func serialize(into buffer: inout NIO.ByteBuffer) throws {
+        let data = try self.serializedData()
+        buffer.writeBytes(data)
+    }
+}

Sources/GRPC/GRPCServerCodec.swift

@@ -22,7 +22,7 @@ import NIOHTTP1
 /// Incoming gRPC package with a fixed message type.
 ///
 /// - Important: This is **NOT** part of the public API.
-public enum _GRPCServerRequestPart<RequestMessage: Payload> {
+public enum _GRPCServerRequestPart<RequestMessage: GRPCPayload> {
   case head(HTTPRequestHead)
   case message(RequestMessage)
   case end
@@ -31,14 +31,14 @@ public enum _GRPCServerRequestPart<RequestMessage: Payload> {
 /// Outgoing gRPC package with a fixed message type.
 ///
 /// - Important: This is **NOT** part of the public API.
-public enum _GRPCServerResponsePart<ResponseMessage: Payload> {
+public enum _GRPCServerResponsePart<ResponseMessage: GRPCPayload> {
   case headers(HTTPHeaders)
   case message(ResponseMessage)
   case statusAndTrailers(GRPCStatus, HTTPHeaders)
 }
 
 /// A simple channel handler that translates raw gRPC packets into decoded protobuf messages, and vice versa.
-internal final class GRPCServerCodec<RequestMessage: Payload, ResponseMessage: Payload> {}
+internal final class GRPCServerCodec<RequestMessage: GRPCPayload, ResponseMessage: GRPCPayload> {}
 
 extension GRPCServerCodec: ChannelInboundHandler {
   typealias InboundIn = _RawGRPCServerRequestPart

Sources/GRPC/ReadWriteStates.swift

@@ -57,7 +57,7 @@ enum WriteState {
   /// - Parameter allocator: An allocator to provide a `ByteBuffer` into which the message will be
   ///     written.
   mutating func write(
-    _ message: Payload,
+    _ message: GRPCPayload,
     allocator: ByteBufferAllocator
   ) -> Result<ByteBuffer, MessageWriteError> {
     switch self {
@@ -116,7 +116,7 @@ enum ReadState {
   /// a message has been produced then subsequent calls will result in an error.
   ///
   /// - Parameter buffer: The buffer to read from.
-  mutating func readMessages<MessageType: Payload>(
+  mutating func readMessages<MessageType: GRPCPayload>(
     _ buffer: inout ByteBuffer,
     as: MessageType.Type = MessageType.self
   ) -> Result<[MessageType], MessageReadError> {

Sources/GRPC/ServerCallContexts/StreamingResponseCallContext.swift

@@ -25,7 +25,7 @@ import Logging
 /// - If `statusPromise` is failed and the error is of type `GRPCStatusTransformable`,
 ///   the result of `error.asGRPCStatus()` will be returned to the client.
 /// - If `error.asGRPCStatus()` is not available, `GRPCStatus.processingError` is returned to the client.
-open class StreamingResponseCallContext<ResponseMessage: Payload>: ServerCallContextBase {
+open class StreamingResponseCallContext<ResponseMessage: GRPCPayload>: ServerCallContextBase {
   typealias WrappedResponse = _GRPCServerResponsePart<ResponseMessage>
 
   public let statusPromise: EventLoopPromise<GRPCStatus>
@@ -41,7 +41,7 @@ open class StreamingResponseCallContext<ResponseMessage: Payload>: ServerCallCon
 }
 
 /// Concrete implementation of `StreamingResponseCallContext` used by our generated code.
-open class StreamingResponseCallContextImpl<ResponseMessage: Payload>: StreamingResponseCallContext<ResponseMessage> {
+open class StreamingResponseCallContextImpl<ResponseMessage: GRPCPayload>: StreamingResponseCallContext<ResponseMessage> {
   public let channel: Channel
 
   /// - Parameters:
@@ -80,7 +80,7 @@ open class StreamingResponseCallContextImpl<ResponseMessage: Payload>: Streaming
 /// Concrete implementation of `StreamingResponseCallContext` used for testing.
 ///
 /// Simply records all sent messages.
-open class StreamingResponseCallContextTestStub<ResponseMessage: Payload>: StreamingResponseCallContext<ResponseMessage> {
+open class StreamingResponseCallContextTestStub<ResponseMessage: GRPCPayload>: StreamingResponseCallContext<ResponseMessage> {
   open var recordedResponses: [ResponseMessage] = []
 
   open override func sendResponse(_ message: ResponseMessage) -> EventLoopFuture<Void> {

Sources/GRPC/ServerCallContexts/UnaryResponseCallContext.swift

@@ -28,7 +28,7 @@ import Logging
 ///
 /// For unary calls, the response is not actually provided by fulfilling `responsePromise`, but instead by completing
 /// the future returned by `UnaryCallHandler.EventObserver`.
-open class UnaryResponseCallContext<ResponseMessage: Payload>: ServerCallContextBase, StatusOnlyCallContext {
+open class UnaryResponseCallContext<ResponseMessage: GRPCPayload>: ServerCallContextBase, StatusOnlyCallContext {
   typealias WrappedResponse = _GRPCServerResponsePart<ResponseMessage>
 
   public let responsePromise: EventLoopPromise<ResponseMessage>
@@ -55,7 +55,7 @@ public protocol StatusOnlyCallContext: ServerCallContext {
 }
 
 /// Concrete implementation of `UnaryResponseCallContext` used by our generated code.
-open class UnaryResponseCallContextImpl<ResponseMessage: Payload>: UnaryResponseCallContext<ResponseMessage> {
+open class UnaryResponseCallContextImpl<ResponseMessage: GRPCPayload>: UnaryResponseCallContext<ResponseMessage> {
   public let channel: Channel
 
   /// - Parameters:
@@ -93,4 +93,4 @@ open class UnaryResponseCallContextImpl<ResponseMessage: Payload>: UnaryResponse
 /// Concrete implementation of `UnaryResponseCallContext` used for testing.
 ///
 /// Only provided to make it clear in tests that no "real" implementation is used.
-open class UnaryResponseCallContextTestStub<ResponseMessage: Payload>: UnaryResponseCallContext<ResponseMessage> { }
+open class UnaryResponseCallContextTestStub<ResponseMessage: GRPCPayload>: UnaryResponseCallContext<ResponseMessage> { }