client context out of utils

Author: lia-viam

src/viam/sdk/common/client_helper.cpp

@@ -2,8 +2,12 @@
 
 #include <cstdlib>
 
+#include <grpcpp/client_context.h>
+
 #include <boost/log/trivial.hpp>
 
+#include <viam/sdk/common/private/version_metadata.hpp>
+
 namespace viam {
 namespace sdk {
 
@@ -16,5 +20,37 @@ namespace client_helper_details {
 }
 
 }  // namespace client_helper_details
+
+ClientContext::ClientContext() : wrapped_context_(std::make_unique<grpc::ClientContext>()) {
+    set_client_ctx_authority_();
+    add_viam_client_version_();
+}
+
+ClientContext::~ClientContext() = default;
+
+ClientContext::operator const grpc::ClientContext*() const {
+    return wrapped_context_.get();
+}
+
+ClientContext::operator grpc::ClientContext*() {
+    return wrapped_context_.get();
+}
+
+void ClientContext::try_cancel() {
+    wrapped_context_->TryCancel();
+}
+
+void ClientContext::set_debug_key(const std::string& debug_key) {
+    wrapped_context_->AddMetadata("dtname", debug_key);
+}
+
+void ClientContext::set_client_ctx_authority_() {
+    wrapped_context_->set_authority("viam-placeholder");
+}
+
+void ClientContext::add_viam_client_version_() {
+    wrapped_context_->AddMetadata("viam_client", impl::k_version);
+}
+
 }  // namespace sdk
 }  // namespace viam

src/viam/sdk/common/client_helper.hpp

@@ -4,7 +4,6 @@
 #include <viam/sdk/common/grpc_client_fwd.hpp>
 #include <viam/sdk/common/private/utils.hpp>
 #include <viam/sdk/common/proto_value.hpp>
-#include <viam/sdk/common/utils.hpp>
 
 namespace grpc {
 
@@ -21,6 +20,29 @@ namespace client_helper_details {
 
 }  // namespace client_helper_details
 
+// the authority on a grpc::ClientContext is sometimes set to an invalid uri on mac, causing
+// `rust-utils` to fail to process gRPC requests. This class provides a convenience wrapper around a
+// grpc ClientContext that allows us to make any necessary modifications to authority or else where
+// to avoid runtime issues.
+// For more details, see https://viam.atlassian.net/browse/RSDK-5194.
+class ClientContext {
+   public:
+    ClientContext();
+    ~ClientContext();
+
+    void try_cancel();
+
+    operator grpc::ClientContext*();
+    operator const grpc::ClientContext*() const;
+
+    void set_debug_key(const std::string& debug_key);
+
+   private:
+    void set_client_ctx_authority_();
+    void add_viam_client_version_();
+    std::unique_ptr<grpc::ClientContext> wrapped_context_;
+};
+
 // Method type for a gRPC call that returns a response message type.
 template <typename StubType, typename RequestType, typename ResponseType>
 using SyncMethodType = ::grpc::Status (StubType::*)(::grpc::ClientContext*,

src/viam/sdk/common/utils.cpp

@@ -18,7 +18,6 @@
 #include <viam/api/common/v1/common.pb.h>
 
 #include <viam/sdk/common/private/utils.hpp>
-#include <viam/sdk/common/private/version_metadata.hpp>
 #include <viam/sdk/components/component.hpp>
 #include <viam/sdk/registry/registry.hpp>
 
@@ -151,37 +150,6 @@ ProtoStruct with_debug_entry(ProtoStruct&& map) {
     return map;
 }
 
-ClientContext::ClientContext() : wrapped_context_(std::make_unique<grpc::ClientContext>()) {
-    set_client_ctx_authority_();
-    add_viam_client_version_();
-}
-
-ClientContext::~ClientContext() = default;
-
-ClientContext::operator const grpc::ClientContext*() const {
-    return wrapped_context_.get();
-}
-
-ClientContext::operator grpc::ClientContext*() {
-    return wrapped_context_.get();
-}
-
-void ClientContext::try_cancel() {
-    wrapped_context_->TryCancel();
-}
-
-void ClientContext::set_debug_key(const std::string& debug_key) {
-    wrapped_context_->AddMetadata("dtname", debug_key);
-}
-
-void ClientContext::set_client_ctx_authority_() {
-    wrapped_context_->set_authority("viam-placeholder");
-}
-
-void ClientContext::add_viam_client_version_() {
-    wrapped_context_->AddMetadata("viam_client", impl::k_version);
-}
-
 bool from_dm_from_extra(const ProtoStruct& extra) {
     auto pos = extra.find("fromDataManagement");
     if (pos != extra.end()) {

src/viam/sdk/common/utils.hpp

@@ -4,7 +4,6 @@
 
 #include <boost/optional/optional.hpp>
 
-#include <viam/sdk/common/grpc_client_fwd.hpp>
 #include <viam/sdk/common/proto_value.hpp>
 #include <viam/sdk/components/component.hpp>
 #include <viam/sdk/resource/resource_api.hpp>
@@ -82,29 +81,6 @@ struct from_proto<common::v1::ResponseMetadata> {
 std::vector<unsigned char> string_to_bytes(std::string const& s);
 std::string bytes_to_string(std::vector<unsigned char> const& b);
 
-// the authority on a grpc::ClientContext is sometimes set to an invalid uri on mac, causing
-// `rust-utils` to fail to process gRPC requests. This class provides a convenience wrapper around a
-// grpc ClientContext that allows us to make any necessary modifications to authority or else where
-// to avoid runtime issues.
-// For more details, see https://viam.atlassian.net/browse/RSDK-5194.
-class ClientContext {
-   public:
-    ClientContext();
-    ~ClientContext();
-
-    void try_cancel();
-
-    operator grpc::ClientContext*();
-    operator const grpc::ClientContext*() const;
-
-    void set_debug_key(const std::string& debug_key);
-
-   private:
-    void set_client_ctx_authority_();
-    void add_viam_client_version_();
-    std::unique_ptr<grpc::ClientContext> wrapped_context_;
-};
-
 /// @brief Given a fully qualified resource name, returns remote name (or "" if no remote name
 /// exists) and short name
 std::pair<std::string, std::string> long_name_to_remote_and_short(const std::string& long_name);

src/viam/sdk/robot/client.cpp

@@ -19,6 +19,7 @@
 #include <viam/api/robot/v1/robot.grpc.pb.h>
 #include <viam/api/robot/v1/robot.pb.h>
 
+#include <viam/sdk/common/client_helper.hpp>
 #include <viam/sdk/common/private/repeated_ptr_convert.hpp>
 #include <viam/sdk/common/proto_value.hpp>
 #include <viam/sdk/common/utils.hpp>

src/viam/sdk/services/private/mlmodel_client.cpp

@@ -16,7 +16,7 @@
 
 #include <grpcpp/channel.h>
 
-#include <viam/sdk/common/utils.hpp>
+#include <viam/sdk/common/client_helper.hpp>
 #include <viam/sdk/services/private/mlmodel.hpp>
 
 #include <viam/sdk/common/exception.hpp>