mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 05:18:52 +08:00
291 lines
12 KiB
Go
291 lines
12 KiB
Go
// Licensed to the LF AI & Data foundation under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
package importutil
|
|
|
|
import (
|
|
"errors"
|
|
"path"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/milvus-io/milvus/api/schemapb"
|
|
"github.com/milvus-io/milvus/internal/storage"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_NewBinlogParser(t *testing.T) {
|
|
// nil schema
|
|
parser, err := NewBinlogParser(nil, 2, 1024, nil, nil, 0)
|
|
assert.Nil(t, parser)
|
|
assert.NotNil(t, err)
|
|
|
|
// nil chunkmanager
|
|
parser, err = NewBinlogParser(sampleSchema(), 2, 1024, nil, nil, 0)
|
|
assert.Nil(t, parser)
|
|
assert.NotNil(t, err)
|
|
|
|
// nil flushfunc
|
|
parser, err = NewBinlogParser(sampleSchema(), 2, 1024, &MockChunkManager{}, nil, 0)
|
|
assert.Nil(t, parser)
|
|
assert.NotNil(t, err)
|
|
|
|
// succeed
|
|
flushFunc := func(fields map[storage.FieldID]storage.FieldData, shardID int) error {
|
|
return nil
|
|
}
|
|
parser, err = NewBinlogParser(sampleSchema(), 2, 1024, &MockChunkManager{}, flushFunc, 0)
|
|
assert.NotNil(t, parser)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func Test_BinlogParserConstructHolders(t *testing.T) {
|
|
flushFunc := func(fields map[storage.FieldID]storage.FieldData, shardID int) error {
|
|
return nil
|
|
}
|
|
|
|
chunkManager := &MockChunkManager{
|
|
listResult: make(map[string][]string),
|
|
}
|
|
|
|
insertPath := "insertPath"
|
|
deltaPath := "deltaPath"
|
|
|
|
// the first segment has 12 fields, each field has 2 binlog files
|
|
seg1Files := []string{
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/0/435978159903735800",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/1/435978159903735801",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/102/435978159903735802",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/103/435978159903735803",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/104/435978159903735804",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/105/435978159903735805",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/106/435978159903735806",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/107/435978159903735807",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/108/435978159903735808",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/109/435978159903735809",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/110/435978159903735810",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/111/435978159903735811",
|
|
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/0/425978159903735800",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/1/425978159903735801",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/102/425978159903735802",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/103/425978159903735803",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/104/425978159903735804",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/105/425978159903735805",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/106/425978159903735806",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/107/425978159903735807",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/108/425978159903735808",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/109/425978159903735809",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/110/425978159903735810",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/111/425978159903735811",
|
|
}
|
|
|
|
// the second segment has 12 fields, each field has 1 binlog file
|
|
seg2Files := []string{
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483009/0/435978159903735811",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483009/1/435978159903735812",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483009/102/435978159903735802",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483009/103/435978159903735803",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483009/104/435978159903735804",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483009/105/435978159903735805",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483009/106/435978159903735806",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483009/107/435978159903735807",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483009/108/435978159903735808",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483009/109/435978159903735809",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483009/110/435978159903735810",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483009/111/435978159903735811",
|
|
}
|
|
|
|
chunkManager.listResult[insertPath] = append(chunkManager.listResult[insertPath], seg1Files...)
|
|
chunkManager.listResult[insertPath] = append(chunkManager.listResult[insertPath], seg2Files...)
|
|
|
|
// the segment has a delta log file
|
|
chunkManager.listResult[deltaPath] = []string{
|
|
"backup/bak1/data/delta_log/435978159196147009/435978159196147010/435978159261483009/434574382554415105",
|
|
}
|
|
|
|
parser, err := NewBinlogParser(sampleSchema(), 2, 1024, chunkManager, flushFunc, 0)
|
|
assert.NotNil(t, parser)
|
|
assert.Nil(t, err)
|
|
|
|
holders, err := parser.constructSegmentHolders(insertPath, deltaPath)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 2, len(holders))
|
|
|
|
// verify the first segment
|
|
holder := holders[0]
|
|
assert.Equal(t, int64(435978159261483008), holder.segmentID)
|
|
assert.Equal(t, 12, len(holder.fieldFiles))
|
|
for i := 0; i < 12; i++ {
|
|
fieldPath := path.Dir(seg1Files[i])
|
|
fieldStrID := path.Base(fieldPath)
|
|
fieldID, _ := strconv.ParseInt(fieldStrID, 10, 64)
|
|
logFiles, ok := holder.fieldFiles[fieldID]
|
|
assert.True(t, ok)
|
|
assert.Equal(t, 2, len(logFiles))
|
|
|
|
// verify logs under each field is sorted
|
|
log1 := logFiles[0]
|
|
logID1 := path.Base(log1)
|
|
ID1, _ := strconv.ParseInt(logID1, 10, 64)
|
|
log2 := logFiles[1]
|
|
logID2 := path.Base(log2)
|
|
ID2, _ := strconv.ParseInt(logID2, 10, 64)
|
|
assert.LessOrEqual(t, ID1, ID2)
|
|
}
|
|
assert.Equal(t, 0, len(holder.deltaFiles))
|
|
|
|
// verify the second segment
|
|
holder = holders[1]
|
|
assert.Equal(t, int64(435978159261483009), holder.segmentID)
|
|
assert.Equal(t, len(seg2Files), len(holder.fieldFiles))
|
|
for i := 0; i < len(seg2Files); i++ {
|
|
fieldPath := path.Dir(seg2Files[i])
|
|
fieldStrID := path.Base(fieldPath)
|
|
fieldID, _ := strconv.ParseInt(fieldStrID, 10, 64)
|
|
logFiles, ok := holder.fieldFiles[fieldID]
|
|
assert.True(t, ok)
|
|
assert.Equal(t, 1, len(logFiles))
|
|
assert.Equal(t, seg2Files[i], logFiles[0])
|
|
}
|
|
assert.Equal(t, 1, len(holder.deltaFiles))
|
|
assert.Equal(t, chunkManager.listResult[deltaPath][0], holder.deltaFiles[0])
|
|
}
|
|
|
|
func Test_BinlogParserConstructHoldersFailed(t *testing.T) {
|
|
flushFunc := func(fields map[storage.FieldID]storage.FieldData, shardID int) error {
|
|
return nil
|
|
}
|
|
|
|
chunkManager := &MockChunkManager{
|
|
listErr: errors.New("error"),
|
|
listResult: make(map[string][]string),
|
|
}
|
|
|
|
parser, err := NewBinlogParser(sampleSchema(), 2, 1024, chunkManager, flushFunc, 0)
|
|
assert.NotNil(t, parser)
|
|
assert.Nil(t, err)
|
|
|
|
insertPath := "insertPath"
|
|
deltaPath := "deltaPath"
|
|
|
|
// chunkManager return error
|
|
holders, err := parser.constructSegmentHolders(insertPath, deltaPath)
|
|
assert.NotNil(t, err)
|
|
assert.Nil(t, holders)
|
|
|
|
// parse field id error(insert log)
|
|
chunkManager.listErr = nil
|
|
chunkManager.listResult[insertPath] = []string{
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/illegal/435978159903735811",
|
|
}
|
|
holders, err = parser.constructSegmentHolders(insertPath, deltaPath)
|
|
assert.NotNil(t, err)
|
|
assert.Nil(t, holders)
|
|
|
|
// parse segment id error(insert log)
|
|
chunkManager.listResult[insertPath] = []string{
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/illegal/0/435978159903735811",
|
|
}
|
|
holders, err = parser.constructSegmentHolders(insertPath, deltaPath)
|
|
assert.NotNil(t, err)
|
|
assert.Nil(t, holders)
|
|
|
|
// parse segment id error(delta log)
|
|
chunkManager.listResult[insertPath] = []string{}
|
|
chunkManager.listResult[deltaPath] = []string{
|
|
"backup/bak1/data/delta_log/435978159196147009/435978159196147010/illegal/434574382554415105",
|
|
}
|
|
holders, err = parser.constructSegmentHolders(insertPath, deltaPath)
|
|
assert.NotNil(t, err)
|
|
assert.Nil(t, holders)
|
|
}
|
|
|
|
func Test_BinlogParserParseFilesFailed(t *testing.T) {
|
|
flushFunc := func(fields map[storage.FieldID]storage.FieldData, shardID int) error {
|
|
return nil
|
|
}
|
|
|
|
parser, err := NewBinlogParser(sampleSchema(), 2, 1024, &MockChunkManager{}, flushFunc, 0)
|
|
assert.NotNil(t, parser)
|
|
assert.Nil(t, err)
|
|
|
|
err = parser.parseSegmentFiles(nil)
|
|
assert.NotNil(t, err)
|
|
|
|
parser.collectionSchema = nil
|
|
err = parser.parseSegmentFiles(&SegmentFilesHolder{})
|
|
assert.NotNil(t, err)
|
|
}
|
|
|
|
func Test_BinlogParserParse(t *testing.T) {
|
|
flushFunc := func(fields map[storage.FieldID]storage.FieldData, shardID int) error {
|
|
return nil
|
|
}
|
|
|
|
chunkManager := &MockChunkManager{}
|
|
|
|
schema := &schemapb.CollectionSchema{
|
|
Name: "schema",
|
|
Description: "schema",
|
|
AutoID: true,
|
|
Fields: []*schemapb.FieldSchema{
|
|
{
|
|
FieldID: 101,
|
|
Name: "id",
|
|
IsPrimaryKey: true,
|
|
DataType: schemapb.DataType_Int64,
|
|
},
|
|
},
|
|
}
|
|
|
|
parser, err := NewBinlogParser(schema, 2, 1024, chunkManager, flushFunc, 0)
|
|
assert.NotNil(t, parser)
|
|
assert.Nil(t, err)
|
|
|
|
// zero paths
|
|
err = parser.Parse(nil)
|
|
assert.NotNil(t, err)
|
|
|
|
// one empty path
|
|
paths := []string{
|
|
"insertPath",
|
|
}
|
|
err = parser.Parse(paths)
|
|
assert.Nil(t, err)
|
|
|
|
// two empty paths
|
|
paths = append(paths, "deltaPath")
|
|
err = parser.Parse(paths)
|
|
assert.Nil(t, err)
|
|
|
|
// wrong path
|
|
chunkManager.listResult = make(map[string][]string)
|
|
chunkManager.listResult["insertPath"] = []string{
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/illegal/101/435978159903735811",
|
|
}
|
|
err = parser.Parse(paths)
|
|
assert.NotNil(t, err)
|
|
|
|
// file not found
|
|
chunkManager.listResult["insertPath"] = []string{
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/0/435978159903735811",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/1/435978159903735811",
|
|
"backup/bak1/data/insert_log/435978159196147009/435978159196147010/435978159261483008/101/435978159903735811",
|
|
}
|
|
err = parser.Parse(paths)
|
|
assert.NotNil(t, err)
|
|
}
|