mirror of
https://github.com/GayPizzaSpecifications/stable-diffusion-rpc.git
synced 2025-08-05 22:41:30 +00:00
Split out worker related things to a separate service definition.
This commit is contained in:
37
Sources/StableDiffusionNode/JobService.swift
Normal file
37
Sources/StableDiffusionNode/JobService.swift
Normal file
@ -0,0 +1,37 @@
|
||||
import Foundation
|
||||
import GRPC
|
||||
import StableDiffusionCore
|
||||
import StableDiffusionProtos
|
||||
|
||||
class JobServiceProvider: SdJobServiceAsyncProvider {
|
||||
private let jobManager: JobManager
|
||||
|
||||
init(jobManager: JobManager) {
|
||||
self.jobManager = jobManager
|
||||
}
|
||||
|
||||
func getJob(request: SdGetJobRequest, context _: GRPCAsyncServerCallContext) async throws -> SdGetJobResponse {
|
||||
var response = SdGetJobResponse()
|
||||
guard let job = await jobManager.job(id: request.id) else {
|
||||
throw SdCoreError.jobNotFound
|
||||
}
|
||||
response.job = job
|
||||
return response
|
||||
}
|
||||
|
||||
func cancelJob(request _: SdCancelJobRequest, context _: GRPCAsyncServerCallContext) async throws -> SdCancelJobResponse {
|
||||
throw SdCoreError.notImplemented
|
||||
}
|
||||
|
||||
func streamJobUpdates(request: SdStreamJobUpdatesRequest, responseStream: GRPCAsyncResponseStreamWriter<SdJobUpdate>, context _: GRPCAsyncServerCallContext) async throws {
|
||||
let isFilteredById = request.id != 0
|
||||
for await job in await jobManager.jobUpdatePublisher {
|
||||
if isFilteredById, job.id != request.id {
|
||||
continue
|
||||
}
|
||||
var update = SdJobUpdate()
|
||||
update.job = job
|
||||
try await responseStream.send(update)
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user