8ea9993577
Added Marshal, Unmarshal, MarshalJSON, UnmarshalJSON methods to ContainerContext Signed-off-by: Angira Kekteeva <kira@nspcc.ru>
139 lines
3.8 KiB
Go
139 lines
3.8 KiB
Go
package session
|
|
|
|
import (
|
|
cid "github.com/nspcc-dev/neofs-api-go/pkg/container/id"
|
|
"github.com/nspcc-dev/neofs-api-go/v2/session"
|
|
)
|
|
|
|
// ContainerContext represents NeoFS API v2-compatible
|
|
// context of the container session.
|
|
//
|
|
// It is a wrapper over session.ContainerSessionContext
|
|
// which allows to abstract from details of the message
|
|
// structure.
|
|
type ContainerContext session.ContainerSessionContext
|
|
|
|
// NewContainerContext creates and returns blank ContainerSessionContext.
|
|
//
|
|
// Defaults:
|
|
// - not bound to any operation;
|
|
// - applied to all containers.
|
|
func NewContainerContext() *ContainerContext {
|
|
v2 := new(session.ContainerSessionContext)
|
|
v2.SetWildcard(true)
|
|
|
|
return ContainerContextFromV2(v2)
|
|
}
|
|
|
|
// ContainerContextFromV2 wraps session.ContainerSessionContext
|
|
// into ContainerContext.
|
|
func ContainerContextFromV2(v *session.ContainerSessionContext) *ContainerContext {
|
|
return (*ContainerContext)(v)
|
|
}
|
|
|
|
// ToV2 converts ContainerContext to session.ContainerSessionContext
|
|
// message structure.
|
|
func (x *ContainerContext) ToV2() *session.ContainerSessionContext {
|
|
return (*session.ContainerSessionContext)(x)
|
|
}
|
|
|
|
// ApplyTo specifies which container the ContainerContext applies to.
|
|
//
|
|
// If id is nil, ContainerContext is applied to all containers.
|
|
func (x *ContainerContext) ApplyTo(id *cid.ID) {
|
|
v2 := (*session.ContainerSessionContext)(x)
|
|
|
|
v2.SetWildcard(id == nil)
|
|
v2.SetContainerID(id.ToV2())
|
|
}
|
|
|
|
// ActOnAllContainers is a helper function that conveniently
|
|
// applies ContainerContext to all containers.
|
|
func ApplyToAllContainers(c *ContainerContext) {
|
|
c.ApplyTo(nil)
|
|
}
|
|
|
|
// Container returns identifier of the container
|
|
// to which the ContainerContext applies.
|
|
//
|
|
// Returns nil if ContainerContext is applied to
|
|
// all containers.
|
|
func (x *ContainerContext) Container() *cid.ID {
|
|
v2 := (*session.ContainerSessionContext)(x)
|
|
|
|
if v2.Wildcard() {
|
|
return nil
|
|
}
|
|
|
|
return cid.NewFromV2(v2.ContainerID())
|
|
}
|
|
|
|
func (x *ContainerContext) forVerb(v session.ContainerSessionVerb) {
|
|
(*session.ContainerSessionContext)(x).
|
|
SetVerb(v)
|
|
}
|
|
|
|
func (x *ContainerContext) isForVerb(v session.ContainerSessionVerb) bool {
|
|
return (*session.ContainerSessionContext)(x).
|
|
Verb() == v
|
|
}
|
|
|
|
// ForPut binds the ContainerContext to
|
|
// PUT operation.
|
|
func (x *ContainerContext) ForPut() {
|
|
x.forVerb(session.ContainerVerbPut)
|
|
}
|
|
|
|
// IsForPut checks if ContainerContext is bound to
|
|
// PUT operation.
|
|
func (x *ContainerContext) IsForPut() bool {
|
|
return x.isForVerb(session.ContainerVerbPut)
|
|
}
|
|
|
|
// ForDelete binds the ContainerContext to
|
|
// DELETE operation.
|
|
func (x *ContainerContext) ForDelete() {
|
|
x.forVerb(session.ContainerVerbDelete)
|
|
}
|
|
|
|
// IsForDelete checks if ContainerContext is bound to
|
|
// DELETE operation.
|
|
func (x *ContainerContext) IsForDelete() bool {
|
|
return x.isForVerb(session.ContainerVerbDelete)
|
|
}
|
|
|
|
// ForSetEACL binds the ContainerContext to
|
|
// SETEACL operation.
|
|
func (x *ContainerContext) ForSetEACL() {
|
|
x.forVerb(session.ContainerVerbSetEACL)
|
|
}
|
|
|
|
// IsForSetEACL checks if ContainerContext is bound to
|
|
// SETEACL operation.
|
|
func (x *ContainerContext) IsForSetEACL() bool {
|
|
return x.isForVerb(session.ContainerVerbSetEACL)
|
|
}
|
|
|
|
// Marshal marshals ContainerContext into a protobuf binary form.
|
|
func (x *ContainerContext) Marshal(bs ...[]byte) ([]byte, error) {
|
|
var buf []byte
|
|
if len(bs) > 0 {
|
|
buf = bs[0]
|
|
}
|
|
return x.ToV2().StableMarshal(buf)
|
|
}
|
|
|
|
// Unmarshal unmarshals protobuf binary representation of ContainerContext.
|
|
func (x *ContainerContext) Unmarshal(data []byte) error {
|
|
return x.ToV2().Unmarshal(data)
|
|
}
|
|
|
|
// MarshalJSON encodes ContainerContext to protobuf JSON format.
|
|
func (x *ContainerContext) MarshalJSON() ([]byte, error) {
|
|
return x.ToV2().MarshalJSON()
|
|
}
|
|
|
|
// UnmarshalJSON decodes ContainerContext from protobuf JSON format.
|
|
func (x *ContainerContext) UnmarshalJSON(data []byte) error {
|
|
return x.ToV2().UnmarshalJSON(data)
|
|
}
|