From 8a0e99562a991e2f9dbf9283da98279109be0e98 Mon Sep 17 00:00:00 2001
From: Ben Johnson <benbjohnson@yahoo.com>
Date: Tue, 10 Jul 2018 23:41:02 +0100
Subject: [PATCH] encoding/proto: Fix key fields in protobuf.

---
 encoding/proto/proto.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/encoding/proto/proto.go b/encoding/proto/proto.go
index d8367c468..fcdde95f9 100644
--- a/encoding/proto/proto.go
+++ b/encoding/proto/proto.go
@@ -961,6 +961,7 @@ func encodeColumnAttrSets(a []*pilosa.ColumnAttrSet) []*internal.ColumnAttrSet {
 func encodeColumnAttrSet(set *pilosa.ColumnAttrSet) *internal.ColumnAttrSet {
 	return &internal.ColumnAttrSet{
 		ID:    set.ID,
+		Key:   set.Key,
 		Attrs: encodeAttrs(set.Attrs),
 	}
 }
@@ -972,6 +973,7 @@ func encodeRow(r *pilosa.Row) *internal.Row {
 
 	return &internal.Row{
 		Columns: r.Columns(),
+		Keys:    r.Keys,
 		Attrs:   encodeAttrs(r.Attrs),
 	}
 }