test/cpp/util/grpc_cli.cc

//

// Copyright 2015 gRPC authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
//

//
// A command line tool to talk to a grpc server.
// Run `grpc_cli help` command to see its usage information.

// Example of talking to grpc interop server:
// grpc _cli call localhost:50051 UnaryCall "response_size:10" \
//    --protofiles=src/proto/grpc/testing/test.proto \
//    --channel_creds_type=insecure

// Options:
//   1. --protofiles, use this flag to provide proto files if the server does
//      does not have the reflection service.
//   2. --proto_path, if your proto file is not under current working directory,
//      use this flag to provide a search root. It should work similar to the
//      counterpart in protoc. This option is valid only when protofiles is
//      provided.
//   3. --metadata specifies metadata to be sent to the server, such as:
//      --metadata="MyHeaderKey1:Value1:MyHeaderKey2:Value2"
//   4. --channel_creds_type, whether to use tls, insecure or platform-specific
//      options.
//   5. --use_auth, if set to true, attach a GoogleDefaultCredentials to the
//   call
//   6. --infile, input filename (defaults to stdin)
//   7. --outfile, output filename (defaults to stdout)
//   8. --binary_input, use the serialized request as input. The serialized
//      request can be generated by calling something like:
//      pr otoc --proto_path=src/proto/grpc/testing/ \
//       --encode=grpc.testing.SimpleRequest \
//       src/proto/grpc/testing/messages.proto \
//       < input.txt > input.bin
//      If this is used and no proto file is provided in the argument list, the
//      method string has to be exact in the form of /package.service/method.
//   9. --binary_output, use binary format response as output, it can
//      be later decoded using protoc:
//      protoc --proto_path=src/proto/grpc/testing/ \
//     --decode=grpc.testing.SimpleResponse \
//     src/proto/grpc/testing/messages.proto \
//     < output.bin > output.txt
//  10. --default_service_config, optional default service config to use
//      on the channel. Note that this may be ignored if the name resolver
//      returns a service config.
//  11. --display_peer_address, on CallMethod commands, log the peer socket
//      address of the connection that each RPC is made on to stderr.
//

#include <fstream>
#include <functional>
#include <iostream>

#include "absl/flags/flag.h"

#include <grpcpp/support/config.h>

#include "test/cpp/util/cli_credentials.h"
#include "test/cpp/util/grpc_tool.h"
#include "test/cpp/util/test_config.h"

ABSL_FLAG(std::string, outfile, "", "Output file (default is stdout)");

static bool SimplePrint(const std::string& outfile, const std::string& output) {
  if (outfile.empty()) {
    std::cout << output << std::flush;
  } else {
    std::ofstream output_file(outfile, std::ios::app | std::ios::binary);
    output_file << output << std::flush;
    output_file.close();
  }
  return true;
}

int main(int argc, char** argv) {
  grpc::testing::InitTest(&argc, &argv, true);

  return grpc::testing::GrpcToolMainLib(
      argc, const_cast<const char**>(argv), grpc::testing::CliCredentials(),
      std::bind(SimplePrint, absl::GetFlag(FLAGS_outfile),
                std::placeholders::_1));
}