forked from TrueCloudLab/frostfs-api-go
9eb567a53a
Define `ContainerSessionContext` structure, implement getters / setters, JSON and binary encoders, gRPC converters. Support new type of context in SessionTokenBody message. Add test message generator and cover methods with unit tests. Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
129 lines
3 KiB
Go
129 lines
3 KiB
Go
package session
|
|
|
|
import (
|
|
"github.com/nspcc-dev/neofs-api-go/rpc/message"
|
|
session "github.com/nspcc-dev/neofs-api-go/v2/session/grpc"
|
|
"google.golang.org/protobuf/encoding/protojson"
|
|
)
|
|
|
|
func (c *ObjectSessionContext) MarshalJSON() ([]byte, error) {
|
|
return message.MarshalJSON(c)
|
|
}
|
|
|
|
func (c *ObjectSessionContext) UnmarshalJSON(data []byte) error {
|
|
return message.UnmarshalJSON(c, data, new(session.ObjectSessionContext))
|
|
}
|
|
|
|
func (l *TokenLifetime) MarshalJSON() ([]byte, error) {
|
|
return message.MarshalJSON(l)
|
|
}
|
|
|
|
func (l *TokenLifetime) UnmarshalJSON(data []byte) error {
|
|
return message.UnmarshalJSON(l, data, new(session.SessionToken_Body_TokenLifetime))
|
|
}
|
|
|
|
func (t *SessionTokenBody) MarshalJSON() ([]byte, error) {
|
|
return message.MarshalJSON(t)
|
|
}
|
|
|
|
func (t *SessionTokenBody) UnmarshalJSON(data []byte) error {
|
|
msg := new(session.SessionToken_Body)
|
|
|
|
if err := protojson.Unmarshal(data, msg); err != nil {
|
|
return err
|
|
}
|
|
|
|
return t.FromGRPCMessage(msg)
|
|
}
|
|
|
|
func (t *SessionToken) MarshalJSON() ([]byte, error) {
|
|
return message.MarshalJSON(t)
|
|
}
|
|
|
|
func (t *SessionToken) UnmarshalJSON(data []byte) error {
|
|
msg := new(session.SessionToken)
|
|
|
|
if err := protojson.Unmarshal(data, msg); err != nil {
|
|
return err
|
|
}
|
|
|
|
return t.FromGRPCMessage(msg)
|
|
}
|
|
|
|
func (x *XHeader) MarshalJSON() ([]byte, error) {
|
|
return message.MarshalJSON(x)
|
|
}
|
|
|
|
func (x *XHeader) UnmarshalJSON(data []byte) error {
|
|
msg := new(session.XHeader)
|
|
|
|
if err := protojson.Unmarshal(data, msg); err != nil {
|
|
return err
|
|
}
|
|
|
|
return x.FromGRPCMessage(msg)
|
|
}
|
|
|
|
func (r *RequestMetaHeader) MarshalJSON() ([]byte, error) {
|
|
return message.MarshalJSON(r)
|
|
}
|
|
|
|
func (r *RequestMetaHeader) UnmarshalJSON(data []byte) error {
|
|
msg := new(session.RequestMetaHeader)
|
|
|
|
if err := protojson.Unmarshal(data, msg); err != nil {
|
|
return err
|
|
}
|
|
|
|
return r.FromGRPCMessage(msg)
|
|
}
|
|
|
|
func (r *RequestVerificationHeader) MarshalJSON() ([]byte, error) {
|
|
return message.MarshalJSON(r)
|
|
}
|
|
|
|
func (r *RequestVerificationHeader) UnmarshalJSON(data []byte) error {
|
|
msg := new(session.RequestVerificationHeader)
|
|
|
|
if err := protojson.Unmarshal(data, msg); err != nil {
|
|
return err
|
|
}
|
|
|
|
return r.FromGRPCMessage(msg)
|
|
}
|
|
|
|
func (r *ResponseMetaHeader) MarshalJSON() ([]byte, error) {
|
|
return message.MarshalJSON(r)
|
|
}
|
|
|
|
func (r *ResponseMetaHeader) UnmarshalJSON(data []byte) error {
|
|
msg := new(session.ResponseMetaHeader)
|
|
|
|
if err := protojson.Unmarshal(data, msg); err != nil {
|
|
return err
|
|
}
|
|
|
|
return r.FromGRPCMessage(msg)
|
|
}
|
|
|
|
func (r *ResponseVerificationHeader) MarshalJSON() ([]byte, error) {
|
|
return message.MarshalJSON(r)
|
|
}
|
|
|
|
func (r *ResponseVerificationHeader) UnmarshalJSON(data []byte) error {
|
|
msg := new(session.ResponseVerificationHeader)
|
|
|
|
if err := protojson.Unmarshal(data, msg); err != nil {
|
|
return err
|
|
}
|
|
|
|
return r.FromGRPCMessage(msg)
|
|
}
|
|
|
|
func (x *ContainerSessionContext) MarshalJSON() ([]byte, error) {
|
|
return message.MarshalJSON(x)
|
|
}
|
|
|
|
func (x *ContainerSessionContext) UnmarshalJSON(data []byte) error {
|
|
return message.UnmarshalJSON(x, data, new(session.ContainerSessionContext))
|
|
}
|