Implement a Java/Kotlin client.

This commit is contained in:
2023-04-22 23:24:36 -07:00
parent 0b5f5dae57
commit 9b0c174df4
10 changed files with 195 additions and 11 deletions

View File

@ -0,0 +1,38 @@
package gay.pizza.stable.diffusion
import io.grpc.Channel
@Suppress("MemberVisibilityCanBePrivate")
class StableDiffusionRpcClient(val channel: Channel) {
val modelService: ModelServiceGrpc.ModelServiceStub by lazy {
ModelServiceGrpc.newStub(channel)
}
val modelServiceBlocking: ModelServiceGrpc.ModelServiceBlockingStub by lazy {
ModelServiceGrpc.newBlockingStub(channel)
}
val modelServiceFuture: ModelServiceGrpc.ModelServiceFutureStub by lazy {
ModelServiceGrpc.newFutureStub(channel)
}
val modelServiceCoroutine: ModelServiceGrpcKt.ModelServiceCoroutineStub by lazy {
ModelServiceGrpcKt.ModelServiceCoroutineStub(channel)
}
val imageGenerationService: ImageGenerationServiceGrpc.ImageGenerationServiceStub by lazy {
ImageGenerationServiceGrpc.newStub(channel)
}
val imageGenerationServiceBlocking: ImageGenerationServiceGrpc.ImageGenerationServiceBlockingStub by lazy {
ImageGenerationServiceGrpc.newBlockingStub(channel)
}
val imageGenerationServiceFuture: ImageGenerationServiceGrpc.ImageGenerationServiceFutureStub by lazy {
ImageGenerationServiceGrpc.newFutureStub(channel)
}
val imageGenerationServiceCoroutine: ImageGenerationServiceGrpcKt.ImageGenerationServiceCoroutineStub by lazy {
ImageGenerationServiceGrpcKt.ImageGenerationServiceCoroutineStub(channel)
}
}