From a1a9103f62ffc7af9dce5599b5938df698343ea1 Mon Sep 17 00:00:00 2001
From: Todd Gruben <todd@pilosa.com>
Date: Fri, 14 May 2021 11:37:54 -0500
Subject: [PATCH] restore column translate keys

---
 ctl/restore.go | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/ctl/restore.go b/ctl/restore.go
index 6c67ad8ef..f1f98432f 100644
--- a/ctl/restore.go
+++ b/ctl/restore.go
@@ -172,8 +172,7 @@ func (cmd *RestoreCommand) Run(ctx context.Context) error {
 		indexName := record[1]
 		switch record[2] {
 		case "shards":
-			sshard := record[3]
-			shard, err := strconv.Atoi(sshard)
+			shard, err := strconv.Atoi(record[3])
 			if err != nil {
 				return err
 			}
@@ -186,7 +185,16 @@ func (cmd *RestoreCommand) Run(ctx context.Context) error {
 				return err
 			}
 		case "translate":
-			vprint.VV("column keys %v", indexName)
+			partitionID, err := strconv.Atoi(record[3])
+			vprint.VV("column keys %v (%v)", indexName, partitionID)
+			if err != nil {
+				return err
+			}
+
+			err = cmd.client.ImportIndexKeys(ctx, &primary.URI, indexName, partitionID, false, tarReader)
+			if err != nil {
+				return err
+			}
 		case "attributes":
 			vprint.VV("column attributes %v", indexName)
 		case "fields":