From 32842569cb17a0adead9bd5afc926fcac9b1cb19 Mon Sep 17 00:00:00 2001
From: Matthew Nibecker <hello@mattnibecker.com>
Date: Thu, 5 Dec 2024 00:03:57 -0500
Subject: [PATCH] Compute counts when creating View of Dict (#5518)

Closes #5514
---
 vector/view.go | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/vector/view.go b/vector/view.go
index ef9b2838a0..ac1d6d011a 100644
--- a/vector/view.go
+++ b/vector/view.go
@@ -19,6 +19,7 @@ func NewView(val Any, index []uint32) Any {
 		return NewConst(val.val, uint32(len(index)), NewBoolView(val.Nulls, index))
 	case *Dict:
 		index2 := make([]byte, len(index))
+		counts := make([]uint32, val.Any.Len())
 		var nulls *Bool
 		if val.Nulls != nil {
 			nulls = NewBoolEmpty(uint32(len(index)), nil)
@@ -26,14 +27,18 @@ func NewView(val Any, index []uint32) Any {
 				if val.Nulls.Value(idx) {
 					nulls.Set(uint32(k))
 				}
-				index2[k] = val.Index[idx]
+				v := val.Index[idx]
+				index2[k] = v
+				counts[v]++
 			}
 		} else {
 			for k, idx := range index {
-				index2[k] = val.Index[idx]
+				v := val.Index[idx]
+				index2[k] = v
+				counts[v]++
 			}
 		}
-		return NewDict(val.Any, index2, nil, nulls)
+		return NewDict(val.Any, index2, counts, nulls)
 	case *Error:
 		return NewError(val.Typ, NewView(val.Vals, index), NewBoolView(val.Nulls, index))
 	case *Union: