server-session-serverstreaming.swift 1.2 KB

1234567891011121314151617181920212223242526272829303132333435
  1. // server 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 Send(_ response: {{ method|output }}) throws {
  10. try! handler.sendResponse(message:response.serializeProtobuf()) {}
  11. }
  12. fileprivate func run(queue:DispatchQueue) {
  13. do {
  14. try self.handler.receiveMessage(initialMetadata:Metadata()) {(requestData) in
  15. if let requestData = requestData {
  16. let requestMessage = try! {{ method|input }}(protobuf:requestData)
  17. // to keep providers from blocking the server thread,
  18. // we dispatch them to another queue.
  19. queue.async {
  20. try! self.provider.expand(request:requestMessage, session: self)
  21. try! self.handler.sendStatus(statusCode:0,
  22. statusMessage:"OK",
  23. trailingMetadata:Metadata(),
  24. completion:{})
  25. }
  26. }
  27. }
  28. } catch (let callError) {
  29. print("grpc error: \(callError)")
  30. }
  31. }
  32. }