35 lines
734 B
Swift
35 lines
734 B
Swift
|
import Foundation
|
||
|
|
import Clairvoyant
|
||
|
|
import Vapor
|
||
|
|
import NIOCore
|
||
|
|
|
||
|
|
final class EventLoopScheduler {
|
||
|
|
|
||
|
|
private let backgroundGroup: EventLoopGroup
|
||
|
|
|
||
|
|
init(numberOfThreads: Int = 2) {
|
||
|
|
backgroundGroup = MultiThreadedEventLoopGroup(numberOfThreads: numberOfThreads)
|
||
|
|
}
|
||
|
|
|
||
|
|
func next() -> EventLoop {
|
||
|
|
backgroundGroup.next()
|
||
|
|
}
|
||
|
|
|
||
|
|
func provider() -> NIOEventLoopGroupProvider {
|
||
|
|
return .shared(backgroundGroup)
|
||
|
|
}
|
||
|
|
|
||
|
|
func shutdown() {
|
||
|
|
backgroundGroup.shutdownGracefully { _ in
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
extension EventLoopScheduler: AsyncScheduler {
|
||
|
|
|
||
|
|
func schedule(asyncJob: @escaping @Sendable () async throws -> Void) {
|
||
|
|
_ = backgroundGroup.any().makeFutureWithTask(asyncJob)
|
||
|
|
}
|
||
|
|
}
|