mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-06 17:18:35 +08:00
3e788f0fbd
issue: #38715 - Current milvus use a serialized index size(compressed) for estimate resource for loading. - Add a new field `MemSize` (before compressing) for index to estimate resource. --------- Signed-off-by: chyezh <chyezh@outlook.com>
56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package model
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
|
|
"github.com/milvus-io/milvus/pkg/proto/indexpb"
|
|
)
|
|
|
|
var (
|
|
segmentID = int64(1)
|
|
buildID = int64(1)
|
|
|
|
segmentIdxPb = &indexpb.SegmentIndex{
|
|
CollectionID: colID,
|
|
PartitionID: partID,
|
|
SegmentID: segmentID,
|
|
NumRows: 1025,
|
|
IndexID: indexID,
|
|
BuildID: buildID,
|
|
NodeID: 0,
|
|
IndexVersion: 0,
|
|
State: commonpb.IndexState_Finished,
|
|
FailReason: "",
|
|
IndexFileKeys: nil,
|
|
Deleted: false,
|
|
CreateTime: 1,
|
|
SerializeSize: 0,
|
|
}
|
|
|
|
indexModel2 = &SegmentIndex{
|
|
CollectionID: colID,
|
|
PartitionID: partID,
|
|
SegmentID: segmentID,
|
|
NumRows: 1025,
|
|
IndexID: indexID,
|
|
BuildID: buildID,
|
|
NodeID: 0,
|
|
IndexState: commonpb.IndexState_Finished,
|
|
FailReason: "",
|
|
IndexVersion: 0,
|
|
IsDeleted: false,
|
|
CreatedUTCTime: 1,
|
|
IndexFileKeys: nil,
|
|
IndexSerializedSize: 0,
|
|
}
|
|
)
|
|
|
|
func TestUnmarshalSegmentIndexModel(t *testing.T) {
|
|
ret := UnmarshalSegmentIndexModel(segmentIdxPb)
|
|
assert.Equal(t, indexModel2.SegmentID, ret.SegmentID)
|
|
assert.Nil(t, UnmarshalSegmentIndexModel(nil))
|
|
}
|