libtorch-runtime/grpc/CMakeLists.txt

# compile wenet.proto
set(PROTO_DIR "${CMAKE_CURRENT_SOURCE_DIR}")

include(../cmake-linux/grpc.cmake)

add_custom_command(
  OUTPUT  ${PROTO_DIR}/wenet.pb.cc
          ${PROTO_DIR}/wenet.pb.h
          ${PROTO_DIR}/wenet.grpc.pb.cc
          ${PROTO_DIR}/wenet.grpc.pb.h
  COMMAND ${PROTOBUF_PROTOC}
  ARGS --grpc_out "${PROTO_DIR}"
    --cpp_out "${PROTO_DIR}"
    -I "${PROTO_DIR}"
    --plugin=protoc-gen-grpc=${GRPC_BINARY_DIR}/grpc_cpp_plugin
    wenet.proto)

# grpc_server/client
link_directories(${protobuf_BINARY_DIR}/lib)

add_library(wenet_grpc SHARED
  grpc_client.cc
  grpc_server.cc
  wenet.pb.cc
  wenet.grpc.pb.cc
)
target_link_libraries(wenet_grpc PUBLIC ${REFLECTION}
        ${GRPC_GRPCPP}
        ${PROTOBUF_LIBPROTOBUF} decoder)