Revert "Add ASAP abort flag to provisional primary for safer replicat…

…ion disconnection handling"

This reverts commit b873d41.

Signed-off-by: naglera <[email protected]>

src/rdb.c

@@ -2921,7 +2921,7 @@ void stopSaving(int success) {
 
 /* Track loading progress in order to serve client's from time to time
    and if needed calculate rdb checksum  */
-int rdbLoadProgressCallback(rio *r, const void *buf, size_t len) {
+void rdbLoadProgressCallback(rio *r, const void *buf, size_t len) {
     if (server.rdb_checksum) rioGenericUpdateChecksum(r, buf, len);
     if (server.loading_process_events_interval_bytes &&
         (r->processed_bytes + len) / server.loading_process_events_interval_bytes >
@@ -2938,7 +2938,6 @@ int rdbLoadProgressCallback(rio *r, const void *buf, size_t len) {
     if (server.repl_state == REPL_STATE_TRANSFER && rioCheckType(r) == RIO_TYPE_CONN) {
         server.stat_net_repl_input_bytes += len;
     }
-    return 0;
 }
 
 /* Save the given functions_ctx to the rdb.

src/replication.c

@@ -2701,7 +2701,6 @@ static int dualChannelReplHandleEndOffsetResponse(connection *conn, sds *err) {
     server.repl_provisional_primary.reploff = reploffset;
     server.repl_provisional_primary.read_reploff = reploffset;
     server.repl_provisional_primary.dbid = dbid;
-    server.repl_provisional_primary.close_asap = 0;
 
     /* Now that we have the snapshot end-offset, we can ask for psync from that offset. Prepare the
      * main connection accordingly.*/
@@ -2834,7 +2833,6 @@ int readIntoReplDataBlock(connection *conn, replDataBufBlock *data_block, size_t
     }
     if (nread == 0) {
         serverLog(LL_VERBOSE, "Provisional primary closed connection");
-        server.repl_provisional_primary.close_asap = 1;
         cancelReplicationHandshake(1);
         return C_ERR;
     }

src/rio.c

@@ -424,9 +424,8 @@ void rioFreeFd(rio *r) {
 
 /* This function can be installed both in memory and file streams when checksum
  * computation is needed. */
-int rioGenericUpdateChecksum(rio *r, const void *buf, size_t len) {
+void rioGenericUpdateChecksum(rio *r, const void *buf, size_t len) {
     r->cksum = crc64(r->cksum, buf, len);
-    return 1;
 }
 
 /* Set the file-based rio object to auto-fsync every 'bytes' file written.

src/rio.h

@@ -57,10 +57,8 @@ struct _rio {
      * all the data that was read or written so far. The method should be
      * designed so that can be called with the current checksum, and the buf
      * and len fields pointing to the new block of data to add to the checksum
-     * computation.
-     * The method should return -1 to indicate that the rio operation should be
-     * terminated, or a non-negative value to continue processing. */
-    int (*update_cksum)(struct _rio *, const void *buf, size_t len);
+     * computation. */
+    void (*update_cksum)(struct _rio *, const void *buf, size_t len);
 
     /* The current checksum and flags (see RIO_FLAG_*) */
     uint64_t cksum, flags;
@@ -142,7 +140,7 @@ static inline size_t rioRead(rio *r, void *buf, size_t len) {
             r->flags |= RIO_FLAG_READ_ERROR;
             return 0;
         }
-        if (r->update_cksum && r->update_cksum(r, buf, bytes_to_read) < 0) return 0;
+        if (r->update_cksum) r->update_cksum(r, buf, bytes_to_read);
         buf = (char *)buf + bytes_to_read;
         len -= bytes_to_read;
         r->processed_bytes += bytes_to_read;
@@ -190,7 +188,7 @@ size_t rioWriteBulkDouble(rio *r, double d);
 struct serverObject;
 int rioWriteBulkObject(rio *r, struct serverObject *obj);
 
-int rioGenericUpdateChecksum(rio *r, const void *buf, size_t len);
+void rioGenericUpdateChecksum(rio *r, const void *buf, size_t len);
 void rioSetAutoSync(rio *r, off_t bytes);
 void rioSetReclaimCache(rio *r, int enabled);
 uint8_t rioCheckType(rio *r);

src/server.h

@@ -2056,7 +2056,6 @@ struct valkeyServer {
         long long reploff;
         long long read_reploff;
         int dbid;
-        uint64_t close_asap : 1;
     } repl_provisional_primary;
     client *cached_primary;             /* Cached primary to be reused for PSYNC. */
     int repl_syncio_timeout;            /* Timeout for synchronous I/O calls */

src/valkey-check-rdb.c

@@ -37,7 +37,7 @@
 #include <sys/stat.h>
 
 void createSharedObjects(void);
-int rdbLoadProgressCallback(rio *r, const void *buf, size_t len);
+void rdbLoadProgressCallback(rio *r, const void *buf, size_t len);
 int rdbCheckMode = 0;
 
 struct {