Two fixed for MIGRATE: fix TTL propagation and fix transferring of data bigger than...

diff --git a/src/cluster.c b/src/cluster.c
index 52812e3f00d79fc3fe0ebe64e0069f2c01296816..6f771428a73d4f22c1ef403aa078d82dfcdb47ed 100644 (file)
--- a/src/cluster.c
+++ b/src/cluster.c
@@ -1586,7 +1586,7 @@ void migrateCommand(redisClient *c) {
     int fd;
     long timeout;
     long dbid;
-    time_t ttl;
+    long long ttl;
     robj *o;
     rio cmd, payload;
 
@@ -1624,7 +1624,8 @@ void migrateCommand(redisClient *c) {
     redisAssertWithInfo(c,NULL,rioWriteBulkString(&cmd,"SELECT",6));
     redisAssertWithInfo(c,NULL,rioWriteBulkLongLong(&cmd,dbid));
 
-    ttl = getExpire(c->db,c->argv[3]);
+    ttl = getExpire(c->db,c->argv[3])-mstime();
+    if (ttl < 1) ttl = 1;
     redisAssertWithInfo(c,NULL,rioWriteBulkCount(&cmd,'*',4));
     redisAssertWithInfo(c,NULL,rioWriteBulkString(&cmd,"RESTORE",7));
     redisAssertWithInfo(c,NULL,c->argv[3]->encoding == REDIS_ENCODING_RAW);
@@ -1646,7 +1647,7 @@ void migrateCommand(redisClient *c) {
 
         while ((towrite = sdslen(buf)-pos) > 0) {
             towrite = (towrite > (64*1024) ? (64*1024) : towrite);
-            nwritten = syncWrite(fd,buf+nwritten,towrite,timeout);
+            nwritten = syncWrite(fd,buf+pos,towrite,timeout);
             if (nwritten != (signed)towrite) goto socket_wr_err;
             pos += nwritten;
         }