124 lines
2.8 KiB
Go
124 lines
2.8 KiB
Go
package blobtree
|
|
|
|
import (
|
|
"context"
|
|
"encoding/binary"
|
|
"os"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/common"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/util/logicerr"
|
|
apistatus "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/client/status"
|
|
oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
|
|
)
|
|
|
|
func (b *BlobTree) Delete(_ context.Context, prm common.DeletePrm) (common.DeleteRes, error) {
|
|
if b.cfg.readOnly {
|
|
return common.DeleteRes{}, common.ErrReadOnly
|
|
}
|
|
|
|
if len(prm.StorageID) == storageIDLength {
|
|
return b.deleteFromIdx(prm.Address, binary.LittleEndian.Uint64(prm.StorageID))
|
|
}
|
|
return b.findAndDelete(prm.Address)
|
|
}
|
|
|
|
func (b *BlobTree) deleteFromIdx(addr oid.Address, idx uint64) (common.DeleteRes, error) {
|
|
dir := b.getDirectoryPath(addr)
|
|
path := b.getFilePath(dir, idx)
|
|
|
|
b.fileLock.Lock(path)
|
|
defer b.fileLock.Unlock(path)
|
|
|
|
records, err := b.readFileContent(path)
|
|
if err != nil {
|
|
return common.DeleteRes{}, err
|
|
}
|
|
|
|
deleteIdx := -1
|
|
for i := range records {
|
|
if records[i].Address.Equals(addr) {
|
|
deleteIdx = i
|
|
break
|
|
}
|
|
}
|
|
|
|
if deleteIdx == -1 {
|
|
return common.DeleteRes{}, logicerr.Wrap(new(apistatus.ObjectNotFound))
|
|
}
|
|
|
|
if len(records) == 1 {
|
|
err = os.Remove(path)
|
|
if err == nil {
|
|
b.dispatcher.ReturnIdx(dir, idx)
|
|
// decrease files metric
|
|
}
|
|
return common.DeleteRes{}, err
|
|
}
|
|
|
|
records = append(records[:idx], records[idx+1:]...)
|
|
size, err := b.writeToTmpAndRename(records, path)
|
|
if err != nil {
|
|
return common.DeleteRes{}, err
|
|
}
|
|
if size < b.cfg.targetFileSizeBytes {
|
|
b.dispatcher.ReturnIdx(dir, idx)
|
|
}
|
|
return common.DeleteRes{}, nil
|
|
}
|
|
|
|
func (b *BlobTree) findAndDelete(addr oid.Address) (common.DeleteRes, error) {
|
|
dir := b.getDirectoryPath(addr)
|
|
idx, err := b.findFileIdx(dir, addr)
|
|
if err != nil {
|
|
return common.DeleteRes{}, err
|
|
}
|
|
return b.deleteFromIdx(addr, idx)
|
|
}
|
|
|
|
func (b *BlobTree) findFileIdx(dir string, addr oid.Address) (uint64, error) {
|
|
entities, err := os.ReadDir(dir)
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
return 0, logicerr.Wrap(new(apistatus.ObjectNotFound))
|
|
}
|
|
return 0, err
|
|
}
|
|
for _, entity := range entities {
|
|
if entity.IsDir() {
|
|
continue
|
|
}
|
|
if b.isTempFile(entity.Name()) {
|
|
continue
|
|
}
|
|
idx, err := b.parseIdx(entity.Name())
|
|
if err != nil {
|
|
continue
|
|
}
|
|
path := b.getFilePath(dir, idx)
|
|
contains, err := b.fileContainsObject(path, addr)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
if contains {
|
|
return idx, nil
|
|
}
|
|
}
|
|
return 0, logicerr.Wrap(new(apistatus.ObjectNotFound))
|
|
}
|
|
|
|
func (b *BlobTree) fileContainsObject(path string, addr oid.Address) (bool, error) {
|
|
b.fileLock.RLock(path)
|
|
defer b.fileLock.RUnlock(path)
|
|
|
|
records, err := b.readFileContent(path)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
for i := range records {
|
|
if records[i].Address.Equals(addr) {
|
|
return true, nil
|
|
}
|
|
}
|
|
return false, nil
|
|
}
|