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()
|