feat: Add SendError overload for capturing rb

Signed-off-by: Vladislav Oleshko <[email protected]>
dragonflydb · Aug 3, 2023 · fe43785 · fe43785
1 parent a21a5a9
commit fe43785
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/facade/reply_capture.cc b/src/facade/reply_capture.cc
@@ -20,6 +20,13 @@ void CapturingReplyBuilder::SendError(std::string_view str, std::string_view typ
   Capture(Error{str, type});
 }
 
+void CapturingReplyBuilder::SendError(ErrorReply error) {
+  SKIP_LESS(ReplyMode::ONLY_ERR);
+
+  string message = visit([](auto&& str) -> string { return string{move(str)}; }, error.message);
+  Capture(Error{move(message), error.kind});
+}
+
 void CapturingReplyBuilder::SendMGetResponse(absl::Span<const OptResp> arr) {
   SKIP_LESS(ReplyMode::FULL);
   Capture(vector<OptResp>{arr.begin(), arr.end()});

diff --git a/src/facade/reply_capture.h b/src/facade/reply_capture.h
@@ -24,6 +24,7 @@ class CapturingReplyBuilder : public RedisReplyBuilder {
 
  public:
   void SendError(std::string_view str, std::string_view type = {}) override;
+  void SendError(ErrorReply error) override;
   void SendMGetResponse(absl::Span<const OptResp>) override;
 
   // SendStored -> SendSimpleString("OK")