Working C++ Sample

This commit is contained in:
2023-05-01 18:05:47 -07:00
parent 28ebb29a80
commit 7325f96b67
5 changed files with 26 additions and 14 deletions

View File

@ -6,7 +6,7 @@ find_package(gRPC CONFIG REQUIRED)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
add_library(sdrpc src/StableDiffusion.proto)
add_library(sdrpc StableDiffusion.proto)
get_target_property(grpc_cpp_plugin_location gRPC::grpc_cpp_plugin LOCATION)
@ -17,6 +17,6 @@ protobuf_generate(TARGET sdrpc LANGUAGE grpc
target_include_directories(sdrpc PUBLIC ${CMAKE_CURRENT_BINARY_DIR})
target_link_libraries(sdrpc PUBLIC protobuf::libprotobuf gRPC::grpc gRPC::grpc++)
add_executable(sdrpc_sample src/sample.cpp)
add_executable(sdrpc_sample sample.cpp)
target_include_directories(sdrpc_sample PRIVATE ${CMAKE_CURRENT_BINARY_DIR})
target_link_libraries(sdrpc_sample PRIVATE sdrpc)

View File

@ -0,0 +1 @@
../../Common/StableDiffusion.proto

23
Clients/Cpp/sample.cpp Normal file
View File

@ -0,0 +1,23 @@
#include "StableDiffusion.pb.h"
#include "StableDiffusion.grpc.pb.h"
#include <grpc++/grpc++.h>
using namespace gay::pizza::stable::diffusion;
int main() {
auto channel = grpc::CreateChannel("localhost:4546", grpc::InsecureChannelCredentials());
auto modelService = ModelService::NewStub(channel);
auto imageGenerationService = ImageGenerationService::NewStub(channel);
grpc::ClientContext context;
ListModelsRequest listModelsRequest;
ListModelsResponse response;
modelService->ListModels(&context, listModelsRequest, &response);
auto models = response.available_models();
for (const auto &item: models) {
std::cout << "Model Name: " << item.name() << std::endl;
}
return 0;
}

View File

@ -1 +0,0 @@
../../../Common/StableDiffusion.proto

View File

@ -1,11 +0,0 @@
#include <StableDiffusion.pb.h>
#include <iostream>
using namespace gay::pizza::stable::diffusion;
int main() {
ModelInfo info;
info.set_name("anything-4.5");
std::cout << info.DebugString() << std::endl;
return 0;
}