milvus/internal/storage/binlog_util.go
congqixia 709594f158
enhance: [2.5] Use v2 package name for pkg module (#40117)
Cherry-pick from master
pr: #39990
Related to #39095

https://go.dev/doc/modules/version-numbers

Update pkg version according to golang dep version convention

Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
2025-02-23 00:46:01 +08:00

42 lines
1.0 KiB
Go

package storage
import (
"fmt"
"strconv"
"strings"
"github.com/milvus-io/milvus/pkg/v2/common"
)
// ParseSegmentIDByBinlog parse segment id from binlog paths
// if path format is not expected, returns error
func ParseSegmentIDByBinlog(rootPath, path string) (UniqueID, error) {
// check path contains rootPath as prefix
if !strings.HasPrefix(path, rootPath) {
return 0, fmt.Errorf("path \"%s\" does not contains rootPath \"%s\"", path, rootPath)
}
p := path[len(rootPath):]
// remove leading "/"
for strings.HasPrefix(p, "/") {
p = p[1:]
}
// binlog path should consist of "[log_type]/collID/partID/segID/fieldID/fileName"
keyStr := strings.Split(p, "/")
logType := keyStr[0]
if logType == common.SegmentDeltaLogPath {
if len(keyStr) == 5 {
return strconv.ParseInt(keyStr[3], 10, 64)
}
return 0, fmt.Errorf("%s is not a valid delta log path", path)
}
// log type are binlog or statslog
if len(keyStr) == 6 {
return strconv.ParseInt(keyStr[len(keyStr)-3], 10, 64)
}
return 0, fmt.Errorf("%s is not a valid binlog path", path)
}