From 329f86033d4eae2514d2b0c1fdfa207a5e0a4daa Mon Sep 17 00:00:00 2001
From: Todd Gruben <todd@pilosa.com>
Date: Fri, 14 May 2021 11:28:23 -0500
Subject: [PATCH] restore field translate keys

---
 client.go      | 10 ++++++++++
 ctl/restore.go |  5 +++++
 2 files changed, 15 insertions(+)

diff --git a/client.go b/client.go
index cbce63d39..b1ddaab30 100644
--- a/client.go
+++ b/client.go
@@ -91,6 +91,9 @@ type InternalClient interface {
 
 	GetNodeUsage(ctx context.Context, uri *pnet.URI) (map[string]NodeUsage, error)
 	GetPastQueries(ctx context.Context, uri *pnet.URI) ([]PastQueryStatus, error)
+
+	ImportFieldKeys(ctx context.Context, uri *pnet.URI, index, field string, remote bool, rddbdata io.Reader) error
+	ImportIndexKeys(ctx context.Context, uri *pnet.URI, index string, partitionID int, remote bool, rddbdata io.Reader) error
 }
 
 //===============
@@ -274,3 +277,10 @@ func (n nopInternalClient) GetNodeUsage(ctx context.Context, uri *pnet.URI) (map
 func (n nopInternalClient) GetPastQueries(ctx context.Context, uri *pnet.URI) ([]PastQueryStatus, error) {
 	return nil, nil
 }
+func (c nopInternalClient) ImportFieldKeys(ctx context.Context, uri *pnet.URI, index, field string, remote bool, rddbdata io.Reader) error {
+	return nil
+}
+
+func (c nopInternalClient) ImportIndexKeys(ctx context.Context, uri *pnet.URI, index string, partitionID int, remote bool, rddbdata io.Reader) error {
+	return nil
+}
diff --git a/ctl/restore.go b/ctl/restore.go
index 0c26866fd..6c67ad8ef 100644
--- a/ctl/restore.go
+++ b/ctl/restore.go
@@ -180,6 +180,7 @@ func (cmd *RestoreCommand) Run(ctx context.Context) error {
 			vprint.VV("shard %v %v", shard, indexName)
 			url := primary.URI.Path(fmt.Sprintf("/internal/restore/%v/%v", indexName, shard))
 			vprint.VV("%v", url)
+			//TODO (twg) cluster aware client
 			_, err = c.Post(url, "application/octet-stream", tarReader)
 			if err != nil {
 				return err
@@ -193,6 +194,10 @@ func (cmd *RestoreCommand) Run(ctx context.Context) error {
 			switch action := record[4]; action {
 			case "translate":
 				vprint.VV("field keys %v %v", indexName, fieldName)
+				err := cmd.client.ImportFieldKeys(ctx, &primary.URI, indexName, fieldName, false, tarReader)
+				if err != nil {
+					return err
+				}
 			case "attributes":
 				vprint.VV("field attributes %v %v", indexName, fieldName)
 			default: