memory mode: skip restore rocksdb from checkpoint on eof error.

Author: patpatbear

src/ctrip_swap.h

@@ -2443,7 +2443,7 @@ typedef struct swapRdbLoadCtx {
 
 void swapRdbLoadCtxInit(swapRdbLoadCtx *ctx);
 void swapRdbLoadBegin(swapRdbLoadCtx *ctx);
-void swapRdbLoadEnd(swapRdbLoadCtx *ctx, rio *rdb);
+void swapRdbLoadEnd(swapRdbLoadCtx *ctx, rio *rdb, int eoferr);
 int swapRdbLoadAuxField(swapRdbLoadCtx *ctx, rio *rdb, robj *auxkey, robj *auxval);
 int swapRdbLoadNoKV(swapRdbLoadCtx *ctx, rio *rdb, redisDb *db, int type);
 int swapRdbLoadKVBegin(swapRdbLoadCtx *ctx, rio *rdb);

src/ctrip_swap_rdb.c

@@ -1242,9 +1242,9 @@ void swapRdbLoadKVEnd(swapRdbLoadCtx *ctx) {
     ctx->rordb_object_flags = -1;
 }
 
-void swapRdbLoadEnd(swapRdbLoadCtx *ctx, rio *rdb) {
+void swapRdbLoadEnd(swapRdbLoadCtx *ctx, rio *rdb, int eoferr) {
     if (ctx->rordb && !ctx->rordb_sstloaded) {
-        rordbLoadSSTFinished(rdb);
+        if (!eoferr) rordbLoadSSTFinished(rdb);
         ctx->rordb_sstloaded = 1;
     }
 

src/rdb.c

@@ -2919,7 +2919,7 @@ int rdbLoadRio(rio *rdb, int rdbflags, rdbSaveInfo *rsi) {
 #endif
     }
 #ifdef ENABLE_SWAP
-    swapRdbLoadEnd(ctx,rdb);
+    swapRdbLoadEnd(ctx,rdb,0);
 #endif
     /* Verify the checksum if RDB version is >= 5 */
     if (rdbver >= 5) {
@@ -2958,7 +2958,7 @@ int rdbLoadRio(rio *rdb, int rdbflags, rdbSaveInfo *rsi) {
      * we'll report the error to the caller, so that we can retry. */
 eoferr:
 #ifdef ENABLE_SWAP
-    swapRdbLoadEnd(ctx,rdb);
+    swapRdbLoadEnd(ctx,rdb,1);
 #endif
     serverLog(LL_WARNING,
         "Short read or OOM loading DB. Unrecoverable error, aborting now.");