server-session-serverstreaming.swift 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738
  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) {
  16. do {
  17. try self.handler.receiveMessage(initialMetadata:Metadata()) {(requestData) in
  18. if let requestData = requestData {
  19. let requestMessage = try! {{ method|input }}(protobuf:requestData)
  20. // to keep providers from blocking the server thread,
  21. // we dispatch them to another queue.
  22. queue.async {
  23. try! self.provider.expand(request:requestMessage, session: self)
  24. try! self.handler.sendStatus(statusCode:0,
  25. statusMessage:"OK",
  26. trailingMetadata:Metadata(),
  27. completion:{})
  28. }
  29. }
  30. }
  31. } catch (let callError) {
  32. print("grpc error: \(callError)")
  33. }
  34. }
  35. }