Преглед на файлове

Reset with_auth in controller before each sending

jamesge преди 4 години
родител
ревизия
4e519089a4
променени са 2 файла, в които са добавени 6 реда и са изтрити 0 реда
  1. 4 0
      src/brpc/details/controller_private_accessor.h
  2. 2 0
      src/brpc/policy/redis_protocol.cpp

+ 4 - 0
src/brpc/details/controller_private_accessor.h

@@ -128,6 +128,10 @@ public:
         _cntl->add_flag(Controller::FLAGS_REQUEST_WITH_AUTH);
     }
 
+    void clear_with_auth() {
+        _cntl->clear_flag(Controller::FLAGS_REQUEST_WITH_AUTH);
+    }
+
     std::string& protocol_param() { return _cntl->protocol_param(); }
     const std::string& protocol_param() const { return _cntl->protocol_param(); }
 

+ 2 - 0
src/brpc/policy/redis_protocol.cpp

@@ -329,6 +329,8 @@ void PackRedisRequest(butil::IOBuf* buf,
         }
         buf->append(auth_str);
         ControllerPrivateAccessor(cntl).add_with_auth();
+    } else {
+        ControllerPrivateAccessor(cntl).clear_with_auth();
     }
 
     buf->append(request);