server-session-serverstreaming.swift 1.3 KB

12345678910111213141516171819202122232425262728293031323334
  1. // {{ method.name }} (Server Streaming)
  2. public class {{ .|session:protoFile,service,method }} {
  3. private var handler : gRPC.Handler
  4. private var provider : {{ .|provider:protoFile,service }}
  5. /// Create a session.
  6. fileprivate init(handler:gRPC.Handler, provider: {{ .|provider:protoFile,service }}) {
  7. self.handler = handler
  8. self.provider = provider
  9. }
  10. /// Send a message. Nonblocking.
  11. public func Send(_ response: {{ method|output }}) throws {
  12. try! handler.sendResponse(message:response.serializeProtobuf()) {}
  13. }
  14. /// Run the session. Internal.
  15. fileprivate func run(queue:DispatchQueue) throws {
  16. try self.handler.receiveMessage(initialMetadata:Metadata()) {(requestData) in
  17. if let requestData = requestData {
  18. let requestMessage = try! {{ method|input }}(protobuf:requestData)
  19. // to keep providers from blocking the server thread,
  20. // we dispatch them to another queue.
  21. queue.async {
  22. try! self.provider.expand(request:requestMessage, session: self)
  23. try! self.handler.sendStatus(statusCode:0,
  24. statusMessage:"OK",
  25. trailingMetadata:Metadata(),
  26. completion:{})
  27. }
  28. }
  29. }
  30. }
  31. }