| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960 |
- // fully streaming
- public class {{ .|session:protoFile,service,method }} {
- var handler : gRPC.Handler
- var provider : {{ .|provider:protoFile,service }}
- fileprivate init(handler:gRPC.Handler, provider: {{ .|provider:protoFile,service }}) {
- self.handler = handler
- self.provider = provider
- }
- public func Receive() throws -> {{ method|input }} {
- let done = NSCondition()
- var requestMessage : {{ method|input }}?
- try self.handler.receiveMessage() {(requestData) in
- if let requestData = requestData {
- requestMessage = try! {{ method|input }}(protobuf:requestData)
- }
- done.lock()
- done.signal()
- done.unlock()
- }
- done.lock()
- done.wait()
- done.unlock()
- if requestMessage == nil {
- throw {{ .|servererror:protoFile,service }}.endOfStream
- }
- return requestMessage!
- }
- public func Send(_ response: {{ method|output }}) throws {
- try handler.sendResponse(message:response.serializeProtobuf()) {}
- }
- public func Close() {
- let done = NSCondition()
- try! self.handler.sendStatus(statusCode: 0,
- statusMessage: "OK",
- trailingMetadata: Metadata()) {
- done.lock()
- done.signal()
- done.unlock()
- }
- done.lock()
- done.wait()
- done.unlock()
- }
- fileprivate func run(queue:DispatchQueue) {
- do {
- try self.handler.sendMetadata(initialMetadata:Metadata()) {
- queue.async {
- try! self.provider.update(session:self)
- }
- }
- } catch (let callError) {
- print("grpc error: \(callError)")
- }
- }
- }
|