forked from TrueCloudLab/frostfs-api-go
60e9c3d0d3
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
106 lines
2.2 KiB
Go
106 lines
2.2 KiB
Go
package accounting
|
|
|
|
import (
|
|
accounting "github.com/nspcc-dev/neofs-api-go/v2/accounting/grpc"
|
|
"github.com/nspcc-dev/neofs-api-go/v2/refs"
|
|
"github.com/nspcc-dev/neofs-api-go/v2/service"
|
|
grpcService "github.com/nspcc-dev/neofs-api-go/v2/service/grpc"
|
|
)
|
|
|
|
func BalanceRequestBodyToGRPCMessage(b *BalanceRequestBody) *accounting.BalanceRequest_Body {
|
|
if b == nil {
|
|
return nil
|
|
}
|
|
|
|
m := new(accounting.BalanceRequest_Body)
|
|
|
|
m.SetOwnerId(
|
|
refs.OwnerIDToGRPCMessage(b.GetOwnerID()),
|
|
)
|
|
|
|
return m
|
|
}
|
|
|
|
func BalanceRequestBodyFromGRPCMessage(m *accounting.BalanceRequest_Body) *BalanceRequestBody {
|
|
if m == nil {
|
|
return nil
|
|
}
|
|
|
|
b := new(BalanceRequestBody)
|
|
|
|
b.SetOwnerID(
|
|
refs.OwnerIDFromGRPCMessage(m.GetOwnerId()),
|
|
)
|
|
|
|
return b
|
|
}
|
|
|
|
func headersToGRPC(
|
|
src interface {
|
|
GetRequestMetaHeader() *service.RequestMetaHeader
|
|
GetRequestVerificationHeader() *service.RequestVerificationHeader
|
|
},
|
|
dst interface {
|
|
SetMetaHeader(*grpcService.RequestMetaHeader)
|
|
SetVerifyHeader(*grpcService.RequestVerificationHeader)
|
|
},
|
|
) {
|
|
dst.SetMetaHeader(
|
|
service.RequestMetaHeaderToGRPCMessage(src.GetRequestMetaHeader()),
|
|
)
|
|
|
|
dst.SetVerifyHeader(
|
|
service.RequestVerificationHeaderToGRPCMessage(src.GetRequestVerificationHeader()),
|
|
)
|
|
}
|
|
|
|
func headersFromGRPC(
|
|
src interface {
|
|
GetMetaHeader() *grpcService.RequestMetaHeader
|
|
GetVerifyHeader() *grpcService.RequestVerificationHeader
|
|
},
|
|
dst interface {
|
|
SetRequestMetaHeader(*service.RequestMetaHeader)
|
|
SetRequestVerificationHeader(*service.RequestVerificationHeader)
|
|
},
|
|
) {
|
|
dst.SetRequestMetaHeader(
|
|
service.RequestMetaHeaderFromGRPCMessage(src.GetMetaHeader()),
|
|
)
|
|
|
|
dst.SetRequestVerificationHeader(
|
|
service.RequestVerificationHeaderFromGRPCMessage(src.GetVerifyHeader()),
|
|
)
|
|
}
|
|
|
|
func BalanceRequestToGRPCMessage(b *BalanceRequest) *accounting.BalanceRequest {
|
|
if b == nil {
|
|
return nil
|
|
}
|
|
|
|
m := new(accounting.BalanceRequest)
|
|
|
|
m.SetBody(
|
|
BalanceRequestBodyToGRPCMessage(b.GetBody()),
|
|
)
|
|
|
|
headersToGRPC(b, m)
|
|
|
|
return m
|
|
}
|
|
|
|
func BalanceRequestFromGRPCMessage(m *accounting.BalanceRequest) *BalanceRequest {
|
|
if m == nil {
|
|
return nil
|
|
}
|
|
|
|
b := new(BalanceRequest)
|
|
|
|
b.SetBody(
|
|
BalanceRequestBodyFromGRPCMessage(m.GetBody()),
|
|
)
|
|
|
|
headersFromGRPC(m, b)
|
|
|
|
return b
|
|
}
|