server-session-clientstreaming.swift 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. // {{ method.name }} (Client Streaming)
  2. public class {{ .|session:protoFile,service,method }} : {{ .|service:protoFile,service }}Session {
  3. private var provider : {{ .|provider:protoFile,service }}
  4. /// Create a session.
  5. fileprivate init(handler:gRPC.Handler, provider: {{ .|provider:protoFile,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. public func Receive() throws -> {{ method|input }} {
  11. let done = NSCondition()
  12. var requestMessage : {{ method|input }}?
  13. try self.handler.receiveMessage() {(requestData) in
  14. if let requestData = requestData {
  15. requestMessage = try? {{ method|input }}(protobuf:requestData)
  16. }
  17. done.lock()
  18. done.signal()
  19. done.unlock()
  20. }
  21. done.lock()
  22. done.wait()
  23. done.unlock()
  24. if requestMessage == nil {
  25. throw {{ .|servererror:protoFile,service }}.endOfStream
  26. }
  27. return requestMessage!
  28. }
  29. /// Send a response and close the connection.
  30. public func SendAndClose(_ response: {{ method|output }}) throws {
  31. try self.handler.sendResponse(message:response.serializeProtobuf(),
  32. statusCode:self.statusCode,
  33. statusMessage:self.statusMessage,
  34. trailingMetadata:self.trailingMetadata)
  35. }
  36. /// Run the session. Internal.
  37. fileprivate func run(queue:DispatchQueue) throws {
  38. try self.handler.sendMetadata(initialMetadata:initialMetadata) {
  39. queue.async {
  40. do {
  41. try self.provider.{{ method.name|lowercase }}(session:self)
  42. } catch (let error) {
  43. print("error \(error)")
  44. }
  45. }
  46. }
  47. }
  48. }