server-session-clientstreaming.swift 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. // client streaming
  2. public class Echo_EchoCollectSession {
  3. var handler : gRPC.Handler
  4. var provider : Echo_EchoProvider
  5. fileprivate init(handler:gRPC.Handler, provider: Echo_EchoProvider) {
  6. self.handler = handler
  7. self.provider = provider
  8. }
  9. public func Receive() throws -> Echo_EchoRequest {
  10. let done = NSCondition()
  11. var requestMessage : Echo_EchoRequest?
  12. try self.handler.receiveMessage() {(requestData) in
  13. if let requestData = requestData {
  14. requestMessage = try! Echo_EchoRequest(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 Echo_EchoServerError.endOfStream
  25. }
  26. return requestMessage!
  27. }
  28. public func SendAndClose(_ response: Echo_EchoResponse) 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. print("EchoCollectSession run")
  37. try self.handler.sendMetadata(initialMetadata:Metadata()) {
  38. queue.async {
  39. try! self.provider.collect(session:self)
  40. }
  41. }
  42. } catch (let callError) {
  43. print("grpc error: \(callError)")
  44. }
  45. }
  46. }