server-session-serverstreaming.swift 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  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) throws {
  16. try self.handler.receiveMessage(initialMetadata:Metadata()) {(requestData) in
  17. if let requestData = requestData {
  18. do {
  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. do {
  24. try self.provider.{{ method.name|lowercase }}(request:requestMessage, session: self)
  25. try self.handler.sendStatus(statusCode:0,
  26. statusMessage:"OK",
  27. trailingMetadata:Metadata(),
  28. completion:{})
  29. } catch (let error) {
  30. print("error: \(error)")
  31. }
  32. }
  33. } catch (let error) {
  34. print("error: \(error)")
  35. }
  36. }
  37. }
  38. }
  39. }