| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869 |
- // {{ method.name }} (Bidirectional Streaming)
- public class {{ .|call:protoFile,service,method }} {
- private var call : Call
- /// Create a call.
- fileprivate init(_ channel: Channel) {
- self.call = channel.makeCall("{{ .|path:protoFile,service,method }}")
- }
- fileprivate func run(metadata:Metadata) throws -> {{ .|call:protoFile,service,method }} {
- let done = NSCondition()
- try self.call.start(.bidiStreaming,
- metadata:metadata)
- {callResult in
- done.lock()
- done.signal()
- done.unlock()
- }
- done.lock()
- done.wait()
- done.unlock()
- return self
- }
- public func Receive() throws -> {{ method|output }} {
- var returnError : {{ .|clienterror:protoFile,service }}?
- var returnMessage : {{ method|output }}!
- let done = NSCondition()
- do {
- try call.receiveMessage() {(data) in
- if let data = data {
- returnMessage = try? {{ method|output }}(protobuf:data)
- if returnMessage == nil {
- returnError = {{ .|clienterror:protoFile,service }}.invalidMessageReceived
- }
- } else {
- returnError = {{ .|clienterror:protoFile,service }}.endOfStream
- }
- done.lock()
- done.signal()
- done.unlock()
- }
- done.lock()
- done.wait()
- done.unlock()
- }
- if let returnError = returnError {
- throw returnError
- }
- return returnMessage
- }
- public func Send(_ message:{{ method|input }}) throws {
- let messageData = try message.serializeProtobuf()
- try call.sendMessage(data:messageData)
- }
- public func CloseSend() throws {
- let done = NSCondition()
- try call.close() {
- done.lock()
- done.signal()
- done.unlock()
- }
- done.lock()
- done.wait()
- done.unlock()
- }
- }
|