server-session-bidistreaming.swift 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. // {{ method|methodDescriptorName }} (Bidirectional Streaming)
  2. {{ access }} final class {{ .|session:file,service,method }} : {{ .|service:file,service }}Session {
  3. private var provider : {{ .|provider:file,service }}
  4. /// Create a session.
  5. fileprivate init(handler:gRPC.Handler, provider: {{ .|provider:file,service }}) {
  6. self.provider = provider
  7. super.init(handler:handler)
  8. }
  9. /// Receive a message. Blocks until a message is received or the client closes the connection.
  10. {{ access }} func receive() throws -> {{ method|input }} {
  11. let sem = DispatchSemaphore(value: 0)
  12. var requestMessage : {{ method|input }}?
  13. try self.handler.receiveMessage() {(requestData) in
  14. if let requestData = requestData {
  15. do {
  16. requestMessage = try {{ method|input }}(serializedData:requestData)
  17. } catch (let error) {
  18. print("error \(error)")
  19. }
  20. }
  21. sem.signal()
  22. }
  23. _ = sem.wait(timeout: DispatchTime.distantFuture)
  24. if let requestMessage = requestMessage {
  25. return requestMessage
  26. } else {
  27. throw {{ .|servererror:file,service }}.endOfStream
  28. }
  29. }
  30. /// Send a message. Nonblocking.
  31. {{ access }} func send(_ response: {{ method|output }}, completion: ((Bool)->())?) throws {
  32. try handler.sendResponse(message:response.serializedData(), completion: completion)
  33. }
  34. /// Close a connection. Blocks until the connection is closed.
  35. {{ access }} func close() throws {
  36. let sem = DispatchSemaphore(value: 0)
  37. try self.handler.sendStatus(statusCode:self.statusCode,
  38. statusMessage:self.statusMessage,
  39. trailingMetadata:self.trailingMetadata) { _ in sem.signal() }
  40. _ = sem.wait(timeout: DispatchTime.distantFuture)
  41. }
  42. /// Run the session. Internal.
  43. fileprivate func run(queue:DispatchQueue) throws {
  44. try self.handler.sendMetadata(initialMetadata:initialMetadata) { _ in
  45. queue.async {
  46. do {
  47. try self.provider.{{ method|methodDescriptorName|lowercase }}(session:self)
  48. } catch (let error) {
  49. print("error \(error)")
  50. }
  51. }
  52. }
  53. }
  54. }