unit test failure fixed

pkg/parcacol/querier.go

@@ -969,6 +969,20 @@ func (q *Querier) SymbolizeArrowRecord(
 			defer valuePerSecondColumn.Release()
 		}
 
+		indices = schema.FieldIndices(profile.ColumnTimestamp)
+		var timestampColumn *array.Int64
+		if len(indices) != 1 {
+			timestampColumn = arrowutils.MakeNullArray(q.pool, arrow.PrimitiveTypes.Int64, valueColumn.Len()).(*array.Int64)
+			defer timestampColumn.Release()
+		}
+
+		indices = schema.FieldIndices(profile.ColumnDuration)
+		var durationColumn *array.Int64
+		if len(indices) != 1 {
+			durationColumn = arrowutils.MakeNullArray(q.pool, arrow.PrimitiveTypes.Int64, valueColumn.Len()).(*array.Int64)
+			defer durationColumn.Release()
+		}
+
 		profileLabels := []arrow.Field{}
 		profileLabelColumns := []arrow.Array{}
 		for i, field := range schema.Fields() {
@@ -984,14 +998,17 @@ func (q *Querier) SymbolizeArrowRecord(
 		}
 		defer locationsRecord.Release()
 
-		columns := make([]arrow.Array, len(profileLabels)+3) // +3 for stacktrace locations, value and diff
+		columns := make([]arrow.Array, len(profileLabels)+5) // +5 for stacktrace locations, value, diff, timestamp and duration
 		copy(columns, profileLabelColumns)
-		columns[len(columns)-3] = locationsRecord.Column(0)
-		columns[len(columns)-2] = valueColumn
+		columns[len(columns)-5] = locationsRecord.Column(0)
+		columns[len(columns)-4] = valueColumn
 
 		diffColumn := CreateDiffColumn(q.pool, int(r.NumRows()))
 		defer diffColumn.Release()
-		columns[len(columns)-1] = diffColumn
+		columns[len(columns)-3] = diffColumn
+
+		columns[len(columns)-2] = timestampColumn
+		columns[len(columns)-1] = durationColumn
 
 		res[i] = array.NewRecord(profile.ArrowSchema(profileLabels), columns, r.NumRows())
 	}

pkg/profile/profile.go

@@ -76,7 +76,7 @@ func ArrowSamplesField(profileLabelFields []arrow.Field) []arrow.Field {
 	fields := make([]arrow.Field, numFields)
 	copy(fields, profileLabelFields)
 	fields[numFields-5] = LocationsField
-	fields[numFields-3] = arrow.Field{
+	fields[numFields-4] = arrow.Field{
 		Name: "value",
 		Type: arrow.PrimitiveTypes.Int64,
 	}