diff --git a/internal/querynode/flow_graph_insert_node.go b/internal/querynode/flow_graph_insert_node.go index 7e025949d6..5a409f7f78 100644 --- a/internal/querynode/flow_graph_insert_node.go +++ b/internal/querynode/flow_graph_insert_node.go @@ -229,7 +229,7 @@ func filterSegmentsByPKs(pks []int64, segment *Segment) ([]int64, error) { buf := make([]byte, 8) res := make([]int64, 0) for _, pk := range pks { - binary.LittleEndian.PutUint64(buf, uint64(pk)) + binary.BigEndian.PutUint64(buf, uint64(pk)) exist := segment.pkFilter.Test(buf) if exist { res = append(res, pk) diff --git a/internal/querynode/flow_graph_insert_node_test.go b/internal/querynode/flow_graph_insert_node_test.go index 2d81fbe0d6..beeda5c534 100644 --- a/internal/querynode/flow_graph_insert_node_test.go +++ b/internal/querynode/flow_graph_insert_node_test.go @@ -401,7 +401,7 @@ func TestGetSegmentsByPKs(t *testing.T) { } pks, err := filterSegmentsByPKs([]int64{0, 1, 2, 3, 4}, segment) assert.Nil(t, err) - assert.Equal(t, len(pks), 1) + assert.Equal(t, len(pks), 3) pks, err = filterSegmentsByPKs([]int64{}, segment) assert.Nil(t, err)