mirror of
https://github.com/google/flatbuffers.git
synced 2026-06-10 15:16:28 +00:00
34305c4ce4
* Adds the basic structure for required to add grpc support Added the message implementation Updated the code to confirm to the protocol flatbuffersobject Adds an example for Swift flatbuffers GRPC Started implementing the swift GRPC code Gen Generator generates protocols now Fixing ci issues Migrated the logic to use grpc_generator::File instead of string Refactored swift project Implemented GRPC in swift Finished implementing GRPC in swift Fixes issues Adds contiguousBytes initializer to swift Adds documentation + fixes buffer nameing in tables + structs Adds documentation + fixes buffer nameing in tables + structs Updated tests * Updated version
102 lines
3.5 KiB
Swift
102 lines
3.5 KiB
Swift
/*
|
|
* Copyright 2020, gRPC Authors All rights reserved.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
import GRPC
|
|
import Model
|
|
import NIO
|
|
import Logging
|
|
import FlatBuffers
|
|
|
|
// Quieten the logs.
|
|
LoggingSystem.bootstrap {
|
|
var handler = StreamLogHandler.standardOutput(label: $0)
|
|
handler.logLevel = .critical
|
|
return handler
|
|
}
|
|
|
|
func greet(name: String, client greeter: GreeterServiceClient) {
|
|
// Form the request with the name, if one was provided.
|
|
var builder = FlatBufferBuilder()
|
|
let name = builder.create(string: name)
|
|
let root = HelloRequest.createHelloRequest(builder, offsetOfName: name)
|
|
builder.finish(offset: root)
|
|
|
|
// Make the RPC call to the server.
|
|
let sayHello = greeter.SayHello(Message<HelloRequest>(builder: &builder))
|
|
// wait() on the response to stop the program from exiting before the response is received.
|
|
do {
|
|
let response = try sayHello.response.wait()
|
|
print("Greeter received: \(response.object.message)")
|
|
} catch {
|
|
print("Greeter failed: \(error)")
|
|
}
|
|
|
|
let surname = builder.create(string: "Name")
|
|
let manyRoot = ManyHellosRequest.createManyHellosRequest(builder, offsetOfName: surname, numGreetings: 2)
|
|
builder.finish(offset: manyRoot)
|
|
|
|
let call = greeter.SayManyHellos(Message(builder: &builder)) { message in
|
|
print(message.object.message)
|
|
}
|
|
|
|
let status = try! call.status.recover { _ in .processingError }.wait()
|
|
if status.code != .ok {
|
|
print("RPC failed: \(status)")
|
|
}
|
|
}
|
|
|
|
func main(args: [String]) {
|
|
// arg0 (dropped) is the program name. We expect arg1 to be the port, and arg2 (optional) to be
|
|
// the name sent in the request.
|
|
let arg1 = args.dropFirst(1).first
|
|
let arg2 = args.dropFirst(2).first
|
|
|
|
switch (arg1.flatMap(Int.init), arg2) {
|
|
case (.none, _):
|
|
print("Usage: PORT [NAME]")
|
|
exit(1)
|
|
|
|
case let (.some(port), name):
|
|
// Setup an `EventLoopGroup` for the connection to run on.
|
|
//
|
|
// See: https://github.com/apple/swift-nio#eventloops-and-eventloopgroups
|
|
let group = MultiThreadedEventLoopGroup(numberOfThreads: 1)
|
|
|
|
// Make sure the group is shutdown when we're done with it.
|
|
defer {
|
|
try! group.syncShutdownGracefully()
|
|
}
|
|
|
|
// Provide some basic configuration for the connection, in this case we connect to an endpoint on
|
|
// localhost at the given port.
|
|
let configuration = ClientConnection.Configuration(
|
|
target: .hostAndPort("localhost", port),
|
|
eventLoopGroup: group
|
|
)
|
|
|
|
// Create a connection using the configuration.
|
|
let connection = ClientConnection(configuration: configuration)
|
|
|
|
// Provide the connection to the generated client.
|
|
let greeter = GreeterServiceClient(connection: connection)
|
|
|
|
// Do the greeting.
|
|
greet(name: "Hello FlatBuffers!", client: greeter)
|
|
}
|
|
}
|
|
|
|
main(args: CommandLine.arguments)
|