[#366] Add wait for container removed

Signed-off-by: Denis Kirillov <denis@nspcc.ru>
This commit is contained in:
Denis Kirillov 2022-03-03 16:34:47 +03:00 committed by LeL
parent 20b8e3d249
commit e7b742df4d

View file

@ -163,8 +163,8 @@ func (n *layer) createContainer(ctx context.Context, p *CreateBucketParams) (*ci
return bktInfo.CID, nil return bktInfo.CID, nil
} }
func (n *layer) setContainerEACLTable(ctx context.Context, cid *cid.ID, table *eacl.Table) error { func (n *layer) setContainerEACLTable(ctx context.Context, idCnr *cid.ID, table *eacl.Table) error {
table.SetCID(cid) table.SetCID(idCnr)
boxData, err := GetBoxData(ctx) boxData, err := GetBoxData(ctx)
if err == nil { if err == nil {
@ -175,11 +175,11 @@ func (n *layer) setContainerEACLTable(ctx context.Context, cid *cid.ID, table *e
return err return err
} }
return n.waitEACLPresence(ctx, *cid, table, defaultWaitParams()) return n.waitEACLPresence(ctx, *idCnr, table, defaultWaitParams())
} }
func (n *layer) GetContainerEACL(ctx context.Context, cid *cid.ID) (*eacl.Table, error) { func (n *layer) GetContainerEACL(ctx context.Context, idCnr *cid.ID) (*eacl.Table, error) {
return n.neoFS.ContainerEACL(ctx, *cid) return n.neoFS.ContainerEACL(ctx, *idCnr)
} }
type waitParams struct { type waitParams struct {
@ -194,40 +194,22 @@ func defaultWaitParams() *waitParams {
} }
} }
func (n *layer) waitEACLPresence(ctx context.Context, cid cid.ID, table *eacl.Table, params *waitParams) error { func (n *layer) waitEACLPresence(ctx context.Context, idCnr cid.ID, table *eacl.Table, params *waitParams) error {
exp, err := table.Marshal() exp, err := table.Marshal()
if err != nil { if err != nil {
return fmt.Errorf("couldn't marshal eacl: %w", err) return fmt.Errorf("couldn't marshal eacl: %w", err)
} }
wctx, cancel := context.WithTimeout(ctx, params.WaitTimeout) return waitFor(ctx, params, func(ctx context.Context) bool {
defer cancel() eaclTable, err := n.neoFS.ContainerEACL(ctx, idCnr)
ticker := time.NewTimer(params.PollInterval) if err == nil {
defer ticker.Stop() got, err := eaclTable.Marshal()
wdone := wctx.Done() if err == nil && bytes.Equal(exp, got) {
done := ctx.Done() return true
var eaclTable *eacl.Table
var got []byte
for {
select {
case <-done:
return ctx.Err()
case <-wdone:
return wctx.Err()
case <-ticker.C:
eaclTable, err = n.neoFS.ContainerEACL(ctx, cid)
if err == nil {
got, err = eaclTable.Marshal()
if err != nil {
// not expected, but if occurred - doesn't make sense to continue
return fmt.Errorf("marshal received eACL: %w", err)
} else if bytes.Equal(exp, got) {
return nil
}
} }
ticker.Reset(params.PollInterval)
} }
} return false
})
} }
func (n *layer) deleteContainer(ctx context.Context, idCnr *cid.ID) error { func (n *layer) deleteContainer(ctx context.Context, idCnr *cid.ID) error {
@ -236,5 +218,44 @@ func (n *layer) deleteContainer(ctx context.Context, idCnr *cid.ID) error {
if err == nil { if err == nil {
sessionToken = boxData.Gate.SessionTokenForDelete() sessionToken = boxData.Gate.SessionTokenForDelete()
} }
return n.neoFS.DeleteContainer(ctx, *idCnr, sessionToken)
if err = n.neoFS.DeleteContainer(ctx, *idCnr, sessionToken); err != nil {
return err
}
return n.waitForContainerRemoved(ctx, idCnr, defaultWaitParams())
}
func (n *layer) waitForContainerRemoved(ctx context.Context, idCnr *cid.ID, params *waitParams) error {
return waitFor(ctx, params, func(ctx context.Context) bool {
_, err := n.neoFS.Container(ctx, *idCnr)
// TODO: (neofs-s3-gw#367) handle NeoFS API status error
if err != nil && strings.Contains(err.Error(), "container not found") {
return true
}
return false
})
}
// waitFor await that given condition will be met in waitParams time.
func waitFor(ctx context.Context, params *waitParams, condition func(context.Context) bool) error {
wctx, cancel := context.WithTimeout(ctx, params.WaitTimeout)
defer cancel()
ticker := time.NewTimer(params.PollInterval)
defer ticker.Stop()
wdone := wctx.Done()
done := ctx.Done()
for {
select {
case <-done:
return ctx.Err()
case <-wdone:
return wctx.Err()
case <-ticker.C:
if condition(ctx) {
return nil
}
ticker.Reset(params.PollInterval)
}
}
} }