| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859 |
- // {{ 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 }} {
- try self.call.start(metadata: metadata, completion:{})
- 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()
- _ = 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()
- }
- }
|