server-session-clientstreaming.swift 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. // client streaming
  2. public class {{ .|session:protoFile,service,method }} {
  3. var handler : gRPC.Handler
  4. var provider : {{ .|provider:protoFile,service }}
  5. fileprivate init(handler:gRPC.Handler, provider: {{ .|provider:protoFile,service }}) {
  6. self.handler = handler
  7. self.provider = provider
  8. }
  9. public func Receive() throws -> {{ method|input }} {
  10. let done = NSCondition()
  11. var requestMessage : {{ method|input }}?
  12. try self.handler.receiveMessage() {(requestData) in
  13. if let requestData = requestData {
  14. requestMessage = try! {{ method|input }}(protobuf:requestData)
  15. }
  16. done.lock()
  17. done.signal()
  18. done.unlock()
  19. }
  20. done.lock()
  21. done.wait()
  22. done.unlock()
  23. if requestMessage == nil {
  24. throw {{ .|servererror:protoFile,service }}.endOfStream
  25. }
  26. return requestMessage!
  27. }
  28. public func SendAndClose(_ response: {{ method|output }}) throws {
  29. try! self.handler.sendResponse(message:response.serializeProtobuf(),
  30. statusCode: 0,
  31. statusMessage: "OK",
  32. trailingMetadata: Metadata())
  33. }
  34. fileprivate func run(queue:DispatchQueue) {
  35. do {
  36. try self.handler.sendMetadata(initialMetadata:Metadata()) {
  37. queue.async {
  38. try! self.provider.collect(session:self)
  39. }
  40. }
  41. } catch (let callError) {
  42. print("grpc error: \(callError)")
  43. }
  44. }
  45. }