Fix comments

Signed-off-by: naglera <[email protected]>
naglera · Oct 29, 2024 · 4d8126b · 4d8126b
1 parent b82704f
commit 4d8126b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/replication.c b/src/replication.c
@@ -2833,7 +2833,8 @@ int readIntoReplDataBlock(connection *conn, replDataBufBlock *data_block, size_t
     }
     if (nread == 0) {
         serverLog(LL_VERBOSE, "Provisional primary closed connection");
-        if (server.loading_rio) server.loading_rio->flags |= RIO_FLAG_CLOSE_ASAP;
+        /* Signal ongoing RDB load to terminate gracefully */
+        if (server.loading_rio) rioCloseASAP(server.loading_rio);
         cancelReplicationHandshake(1);
         return C_ERR;
     }

diff --git a/src/rio.h b/src/rio.h
@@ -157,6 +157,10 @@ static inline int rioFlush(rio *r) {
     return r->flush(r);
 }
 
+static inline void rioCloseASAP(rio *r) {
+    r->flags |= RIO_FLAG_CLOSE_ASAP;
+}
+
 /* This function allows to know if there was a read error in any past
  * operation, since the rio stream was created or since the last call
  * to rioClearError(). */