fix notification issue

1 files changed, 10 insertions, 10 deletions

diff --git a/src/PeerAbstractCommand.cc b/src/PeerAbstractCommand.cc
index 03df3492..65995c98 100644
--- a/src/PeerAbstractCommand.cc
+++ b/src/PeerAbstractCommand.cc
@@ -126,7 +126,7 @@ bool PeerAbstractCommand::prepareForNextPeer(time_t wait) { return true; }
 void PeerAbstractCommand::disableReadCheckSocket()
 {
   if (checkSocketIsReadable_) {
-    e_->deleteSocketForReadCheck(readCheckTarget_, this);
+    setRWCheck(e_, readCheckTarget_, false, true);
     checkSocketIsReadable_ = false;
     readCheckTarget_.reset();
   }
@@ -141,13 +141,13 @@ void PeerAbstractCommand::setReadCheckSocket(
   else {
     if (checkSocketIsReadable_) {
       if (*readCheckTarget_ != *socket) {
-        e_->deleteSocketForReadCheck(readCheckTarget_, this);
-        e_->addSocketForReadCheck(socket, this);
+        setRWCheck(e_, readCheckTarget_, false, true);
+        setRWCheck(e_, socket, true, true);
         readCheckTarget_ = socket;
       }
     }
     else {
-      e_->addSocketForReadCheck(socket, this);
+      setRWCheck(e_, socket, true, true);
       checkSocketIsReadable_ = true;
       readCheckTarget_ = socket;
     }
@@ -157,7 +157,7 @@ void PeerAbstractCommand::setReadCheckSocket(
 void PeerAbstractCommand::disableWriteCheckSocket()
 {
   if (checkSocketIsWritable_) {
-    e_->deleteSocketForWriteCheck(writeCheckTarget_, this);
+    setRWCheck(e_, writeCheckTarget_, false, false);
     checkSocketIsWritable_ = false;
     writeCheckTarget_.reset();
   }
@@ -172,20 +172,20 @@ void PeerAbstractCommand::setWriteCheckSocket(
   else {
     if (checkSocketIsWritable_) {
       if (*writeCheckTarget_ != *socket) {
-        e_->deleteSocketForWriteCheck(writeCheckTarget_, this);
-        e_->addSocketForWriteCheck(socket, this);
+        setRWCheck(e_, writeCheckTarget_, false, false);
+        setRWCheck(e_, socket, true, false);
         writeCheckTarget_ = socket;
       }
     }
     else {
-      e_->addSocketForWriteCheck(socket, this);
+      setRWCheck(e_, socket, true, false);
       checkSocketIsWritable_ = true;
       writeCheckTarget_ = socket;
     }
   }
 }
 
-void PeerAbstractCommand::SetRWCheck(DownloadEngine* e,
+void PeerAbstractCommand::setRWCheck(DownloadEngine* e,
                                      const std::shared_ptr<SocketCore>& socket,
                                      int is_add, int is_read)
 {
@@ -201,7 +201,7 @@ void PeerAbstractCommand::SetRWCheck(DownloadEngine* e,
     }
   }
   else {
-    utp->SetRWCheck(this, is_add, is_read);
+    utp->setRWCheck(this, is_add, is_read);
   }
 }