forked from TrueCloudLab/frostfs-node
125 lines
2.5 KiB
Go
125 lines
2.5 KiB
Go
package audit
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/nspcc-dev/neofs-sdk-go/container"
|
|
cid "github.com/nspcc-dev/neofs-sdk-go/container/id"
|
|
"github.com/nspcc-dev/neofs-sdk-go/netmap"
|
|
"github.com/nspcc-dev/neofs-sdk-go/object"
|
|
)
|
|
|
|
// Task groups groups the container audit parameters.
|
|
type Task struct {
|
|
reporter Reporter
|
|
|
|
auditContext context.Context
|
|
|
|
cid *cid.ID
|
|
|
|
cnr *container.Container
|
|
|
|
nm *netmap.Netmap
|
|
|
|
cnrNodes netmap.ContainerNodes
|
|
|
|
sgList []*object.ID
|
|
}
|
|
|
|
// WithReporter sets audit report writer.
|
|
func (t *Task) WithReporter(r Reporter) *Task {
|
|
if t != nil {
|
|
t.reporter = r
|
|
}
|
|
|
|
return t
|
|
}
|
|
|
|
// Reporter returns audit report writer.
|
|
func (t *Task) Reporter() Reporter {
|
|
return t.reporter
|
|
}
|
|
|
|
// WithAuditContext sets context of the audit of the current epoch.
|
|
func (t *Task) WithAuditContext(ctx context.Context) *Task {
|
|
if t != nil {
|
|
t.auditContext = ctx
|
|
}
|
|
|
|
return t
|
|
}
|
|
|
|
// AuditContext returns context of the audit of the current epoch.
|
|
func (t *Task) AuditContext() context.Context {
|
|
return t.auditContext
|
|
}
|
|
|
|
// WithContainerID sets identifier of the container under audit.
|
|
func (t *Task) WithContainerID(cid *cid.ID) *Task {
|
|
if t != nil {
|
|
t.cid = cid
|
|
}
|
|
|
|
return t
|
|
}
|
|
|
|
// ContainerID returns identifier of the container under audit.
|
|
func (t *Task) ContainerID() *cid.ID {
|
|
return t.cid
|
|
}
|
|
|
|
// WithContainerStructure sets structure of the container under audit.
|
|
func (t *Task) WithContainerStructure(cnr *container.Container) *Task {
|
|
if t != nil {
|
|
t.cnr = cnr
|
|
}
|
|
|
|
return t
|
|
}
|
|
|
|
// ContainerStructure returns structure of the container under audit.
|
|
func (t *Task) ContainerStructure() *container.Container {
|
|
return t.cnr
|
|
}
|
|
|
|
// WithContainerNodes sets nodes in the container under audit.
|
|
func (t *Task) WithContainerNodes(cnrNodes netmap.ContainerNodes) *Task {
|
|
if t != nil {
|
|
t.cnrNodes = cnrNodes
|
|
}
|
|
|
|
return t
|
|
}
|
|
|
|
// NetworkMap returns network map of audit epoch.
|
|
func (t *Task) NetworkMap() *netmap.Netmap {
|
|
return t.nm
|
|
}
|
|
|
|
// WithNetworkMap sets network map of audit epoch.
|
|
func (t *Task) WithNetworkMap(nm *netmap.Netmap) *Task {
|
|
if t != nil {
|
|
t.nm = nm
|
|
}
|
|
|
|
return t
|
|
}
|
|
|
|
// ContainerNodes returns nodes in the container under audit.
|
|
func (t *Task) ContainerNodes() netmap.ContainerNodes {
|
|
return t.cnrNodes
|
|
}
|
|
|
|
// WithStorageGroupList sets list of storage groups from container under audit.
|
|
func (t *Task) WithStorageGroupList(sgList []*object.ID) *Task {
|
|
if t != nil {
|
|
t.sgList = sgList
|
|
}
|
|
|
|
return t
|
|
}
|
|
|
|
// StorageGroupList returns list of storage groups from container under audit.
|
|
func (t *Task) StorageGroupList() []*object.ID {
|
|
return t.sgList
|
|
}
|