2023-03-17 07:56:37 +00:00
|
|
|
package shard_test
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
2023-05-15 08:43:44 +00:00
|
|
|
"errors"
|
2023-03-17 07:56:37 +00:00
|
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
objectV2 "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/object"
|
|
|
|
objectCore "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/object"
|
2023-03-23 09:42:58 +00:00
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/internal/testutil"
|
2023-03-17 07:56:37 +00:00
|
|
|
meta "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/metabase"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/shard"
|
2023-06-22 11:55:30 +00:00
|
|
|
writecacheconfig "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/writecache/config"
|
2023-03-17 07:56:37 +00:00
|
|
|
cidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id/test"
|
|
|
|
objectSDK "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object"
|
|
|
|
oid "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id"
|
2023-05-15 08:43:44 +00:00
|
|
|
oidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/object/id/test"
|
2023-03-17 07:56:37 +00:00
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
)
|
|
|
|
|
2023-05-15 08:43:44 +00:00
|
|
|
func Test_GCDropsLockedExpiredSimpleObject(t *testing.T) {
|
2023-05-05 11:08:10 +00:00
|
|
|
t.Parallel()
|
|
|
|
|
2023-03-17 07:56:37 +00:00
|
|
|
epoch := &epochState{
|
|
|
|
Value: 100,
|
|
|
|
}
|
|
|
|
|
2023-06-22 11:55:30 +00:00
|
|
|
wcOpts := writecacheconfig.Options{
|
|
|
|
Type: writecacheconfig.TypeBBolt,
|
|
|
|
}
|
|
|
|
sh := newCustomShard(t, t.TempDir(), false, wcOpts, nil, []meta.Option{meta.WithEpochState(epoch)})
|
2023-03-17 07:56:37 +00:00
|
|
|
|
|
|
|
t.Cleanup(func() {
|
|
|
|
releaseShard(sh, t)
|
|
|
|
})
|
|
|
|
|
|
|
|
cnr := cidtest.ID()
|
|
|
|
|
|
|
|
var objExpirationAttr objectSDK.Attribute
|
|
|
|
objExpirationAttr.SetKey(objectV2.SysAttributeExpEpoch)
|
|
|
|
objExpirationAttr.SetValue("101")
|
|
|
|
|
2023-03-23 09:42:58 +00:00
|
|
|
obj := testutil.GenerateObjectWithCID(cnr)
|
2023-03-17 07:56:37 +00:00
|
|
|
obj.SetAttributes(objExpirationAttr)
|
|
|
|
objID, _ := obj.ID()
|
|
|
|
|
|
|
|
var lockExpirationAttr objectSDK.Attribute
|
|
|
|
lockExpirationAttr.SetKey(objectV2.SysAttributeExpEpoch)
|
|
|
|
lockExpirationAttr.SetValue("103")
|
|
|
|
|
2023-03-23 09:42:58 +00:00
|
|
|
lock := testutil.GenerateObjectWithCID(cnr)
|
2023-04-03 13:11:56 +00:00
|
|
|
lock.SetType(objectSDK.TypeLock)
|
2023-03-17 07:56:37 +00:00
|
|
|
lock.SetAttributes(lockExpirationAttr)
|
|
|
|
lockID, _ := lock.ID()
|
|
|
|
|
|
|
|
var putPrm shard.PutPrm
|
|
|
|
putPrm.SetObject(obj)
|
|
|
|
|
2023-04-12 14:01:29 +00:00
|
|
|
_, err := sh.Put(context.Background(), putPrm)
|
2023-03-17 07:56:37 +00:00
|
|
|
require.NoError(t, err)
|
|
|
|
|
2023-04-12 14:01:29 +00:00
|
|
|
err = sh.Lock(context.Background(), cnr, lockID, []oid.ID{objID})
|
2023-03-17 07:56:37 +00:00
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
putPrm.SetObject(lock)
|
2023-04-12 14:01:29 +00:00
|
|
|
_, err = sh.Put(context.Background(), putPrm)
|
2023-03-17 07:56:37 +00:00
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
epoch.Value = 105
|
|
|
|
sh.NotificationChannel() <- shard.EventNewEpoch(epoch.Value)
|
|
|
|
|
|
|
|
var getPrm shard.GetPrm
|
|
|
|
getPrm.SetAddress(objectCore.AddressOf(obj))
|
|
|
|
require.Eventually(t, func() bool {
|
2023-03-13 11:37:35 +00:00
|
|
|
_, err = sh.Get(context.Background(), getPrm)
|
2023-03-17 07:56:37 +00:00
|
|
|
return shard.IsErrNotFound(err)
|
|
|
|
}, 3*time.Second, 1*time.Second, "expired object must be deleted")
|
|
|
|
}
|
2023-05-15 08:43:44 +00:00
|
|
|
|
|
|
|
func Test_GCDropsLockedExpiredComplexObject(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
|
|
|
|
epoch := &epochState{
|
|
|
|
Value: 100,
|
|
|
|
}
|
|
|
|
|
|
|
|
cnr := cidtest.ID()
|
|
|
|
parentID := oidtest.ID()
|
|
|
|
splitID := objectSDK.NewSplitID()
|
|
|
|
|
|
|
|
var objExpirationAttr objectSDK.Attribute
|
|
|
|
objExpirationAttr.SetKey(objectV2.SysAttributeExpEpoch)
|
|
|
|
objExpirationAttr.SetValue("101")
|
|
|
|
|
|
|
|
var lockExpirationAttr objectSDK.Attribute
|
|
|
|
lockExpirationAttr.SetKey(objectV2.SysAttributeExpEpoch)
|
|
|
|
lockExpirationAttr.SetValue("103")
|
|
|
|
|
|
|
|
parent := testutil.GenerateObjectWithCID(cnr)
|
|
|
|
parent.SetID(parentID)
|
|
|
|
parent.SetPayload(nil)
|
|
|
|
parent.SetAttributes(objExpirationAttr)
|
|
|
|
|
|
|
|
const childCount = 10
|
|
|
|
children := make([]*objectSDK.Object, childCount)
|
|
|
|
childIDs := make([]oid.ID, childCount)
|
|
|
|
for i := range children {
|
|
|
|
children[i] = testutil.GenerateObjectWithCID(cnr)
|
|
|
|
if i != 0 {
|
|
|
|
children[i].SetPreviousID(childIDs[i-1])
|
|
|
|
}
|
|
|
|
if i == len(children)-1 {
|
|
|
|
children[i].SetParent(parent)
|
|
|
|
}
|
|
|
|
children[i].SetSplitID(splitID)
|
|
|
|
children[i].SetPayload([]byte{byte(i), byte(i + 1), byte(i + 2)})
|
|
|
|
childIDs[i], _ = children[i].ID()
|
|
|
|
}
|
|
|
|
|
|
|
|
link := testutil.GenerateObjectWithCID(cnr)
|
|
|
|
link.SetParent(parent)
|
|
|
|
link.SetParentID(parentID)
|
|
|
|
link.SetSplitID(splitID)
|
|
|
|
link.SetChildren(childIDs...)
|
|
|
|
|
|
|
|
linkID, _ := link.ID()
|
|
|
|
|
2023-06-22 11:55:30 +00:00
|
|
|
wcOpts := writecacheconfig.Options{
|
|
|
|
Type: writecacheconfig.TypeBBolt,
|
|
|
|
}
|
|
|
|
sh := newCustomShard(t, t.TempDir(), false, wcOpts, nil, []meta.Option{meta.WithEpochState(epoch)})
|
2023-05-15 08:43:44 +00:00
|
|
|
|
|
|
|
t.Cleanup(func() {
|
|
|
|
releaseShard(sh, t)
|
|
|
|
})
|
|
|
|
|
|
|
|
lock := testutil.GenerateObjectWithCID(cnr)
|
|
|
|
lock.SetType(objectSDK.TypeLock)
|
|
|
|
lock.SetAttributes(lockExpirationAttr)
|
|
|
|
lockID, _ := lock.ID()
|
|
|
|
|
|
|
|
var putPrm shard.PutPrm
|
|
|
|
|
|
|
|
for _, child := range children {
|
|
|
|
putPrm.SetObject(child)
|
|
|
|
_, err := sh.Put(context.Background(), putPrm)
|
|
|
|
require.NoError(t, err)
|
|
|
|
}
|
|
|
|
|
|
|
|
putPrm.SetObject(link)
|
|
|
|
_, err := sh.Put(context.Background(), putPrm)
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
err = sh.Lock(context.Background(), cnr, lockID, append(childIDs, parentID, linkID))
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
putPrm.SetObject(lock)
|
|
|
|
_, err = sh.Put(context.Background(), putPrm)
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
|
|
var getPrm shard.GetPrm
|
|
|
|
getPrm.SetAddress(objectCore.AddressOf(parent))
|
|
|
|
|
|
|
|
_, err = sh.Get(context.Background(), getPrm)
|
|
|
|
var splitInfoError *objectSDK.SplitInfoError
|
|
|
|
require.True(t, errors.As(err, &splitInfoError), "split info must be provided")
|
|
|
|
|
|
|
|
epoch.Value = 105
|
|
|
|
sh.NotificationChannel() <- shard.EventNewEpoch(epoch.Value)
|
|
|
|
|
|
|
|
require.Eventually(t, func() bool {
|
|
|
|
_, err = sh.Get(context.Background(), getPrm)
|
|
|
|
return shard.IsErrNotFound(err)
|
|
|
|
}, 3*time.Second, 1*time.Second, "expired complex object must be deleted on epoch after lock expires")
|
|
|
|
}
|