| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748 |
- // {{ method|methodDescriptorName }} (Client Streaming)
- {{ access }} final class {{ .|session:file,service,method }} : {{ .|service:file,service }}Session {
- private var provider : {{ .|provider:file,service }}
- /// Create a session.
- fileprivate init(handler:gRPC.Handler, provider: {{ .|provider:file,service }}) {
- self.provider = provider
- super.init(handler:handler)
- }
- /// Receive a message. Blocks until a message is received or the client closes the connection.
- {{ access }} func receive() throws -> {{ method|input }} {
- let sem = DispatchSemaphore(value: 0)
- var requestMessage : {{ method|input }}?
- try self.handler.receiveMessage() {(requestData) in
- if let requestData = requestData {
- requestMessage = try? {{ method|input }}(serializedData:requestData)
- }
- sem.signal()
- }
- _ = sem.wait(timeout: DispatchTime.distantFuture)
- if requestMessage == nil {
- throw {{ .|servererror:file,service }}.endOfStream
- }
- return requestMessage!
- }
- /// Send a response and close the connection.
- {{ access }} func sendAndClose(_ response: {{ method|output }}) throws {
- try self.handler.sendResponse(message:response.serializedData(),
- statusCode:self.statusCode,
- statusMessage:self.statusMessage,
- trailingMetadata:self.trailingMetadata)
- }
- /// Run the session. Internal.
- fileprivate func run(queue:DispatchQueue) throws {
- try self.handler.sendMetadata(initialMetadata:initialMetadata) {
- queue.async {
- do {
- try self.provider.{{ method|methodDescriptorName|lowercase }}(session:self)
- } catch (let error) {
- print("error \(error)")
- }
- }
- }
- }
- }
|