mirror of
https://github.com/google/flatbuffers.git
synced 2026-06-09 14:46:26 +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
94 lines
2.9 KiB
Swift
94 lines
2.9 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 NIO
|
|
import FlatBuffers
|
|
import Logging
|
|
import Model
|
|
|
|
class Greeter: GreeterProvider {
|
|
|
|
var hellos: [Message<HelloReply>] = []
|
|
|
|
init() {
|
|
let names = ["Stranger1", "Stranger2", "Stranger4", "Stranger3", "Stranger5", "Stranger6"]
|
|
for name in names {
|
|
var builder = FlatBufferBuilder()
|
|
let off = builder.create(string: name)
|
|
let root = HelloReply.createHelloReply(builder, offsetOfMessage: off)
|
|
builder.finish(offset: root)
|
|
hellos.append(Message(builder: &builder))
|
|
}
|
|
}
|
|
|
|
func SayHello(
|
|
_ request: Message<HelloRequest>,
|
|
context: StatusOnlyCallContext
|
|
) -> EventLoopFuture<Message<HelloReply>> {
|
|
let recipient = request.object.name ?? "Stranger"
|
|
|
|
var builder = FlatBufferBuilder()
|
|
let off = builder.create(string: recipient)
|
|
let root = HelloReply.createHelloReply(builder, offsetOfMessage: off)
|
|
builder.finish(offset: root)
|
|
return context.eventLoop.makeSucceededFuture(Message<HelloReply>(builder: &builder))
|
|
}
|
|
|
|
func SayManyHellos(
|
|
request: Message<ManyHellosRequest>,
|
|
context: StreamingResponseCallContext<Message<HelloReply>>
|
|
) -> EventLoopFuture<GRPCStatus> {
|
|
for _ in 0..<Int(request.object.numGreetings) {
|
|
let index = Int.random(in: 0..<hellos.count)
|
|
_ = context.sendResponse(hellos[index])
|
|
}
|
|
return context.eventLoop.makeSucceededFuture(.ok)
|
|
}
|
|
}
|
|
|
|
// Quieten the logs.
|
|
LoggingSystem.bootstrap {
|
|
var handler = StreamLogHandler.standardOutput(label: $0)
|
|
handler.logLevel = .critical
|
|
return handler
|
|
}
|
|
|
|
let group = MultiThreadedEventLoopGroup(numberOfThreads: 1)
|
|
defer {
|
|
try! group.syncShutdownGracefully()
|
|
}
|
|
|
|
// Create some configuration for the server:
|
|
let configuration = Server.Configuration(
|
|
target: .hostAndPort("localhost", 0),
|
|
eventLoopGroup: group,
|
|
serviceProviders: [Greeter()]
|
|
)
|
|
|
|
// Start the server and print its address once it has started.
|
|
let server = Server.start(configuration: configuration)
|
|
server.map {
|
|
$0.channel.localAddress
|
|
}.whenSuccess { address in
|
|
print("server started on port \(address!.port!)")
|
|
}
|
|
|
|
// Wait on the server's `onClose` future to stop the program from exiting.
|
|
_ = try server.flatMap {
|
|
$0.onClose
|
|
}.wait()
|