forked from TrueCloudLab/frostfs-node
[#306] private: Implement setters on generated proto messages
Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
This commit is contained in:
parent
85ec633938
commit
e75ddb0549
2 changed files with 75 additions and 0 deletions
60
pkg/services/private/service.go
Normal file
60
pkg/services/private/service.go
Normal file
|
@ -0,0 +1,60 @@
|
|||
package private
|
||||
|
||||
// SetBody sets health check request body.
|
||||
func (m *HealthCheckRequest) SetBody(v *HealthCheckRequest_Body) {
|
||||
if m != nil {
|
||||
m.Body = v
|
||||
}
|
||||
}
|
||||
|
||||
// SetSignature sets signature of the health check request body.
|
||||
func (m *HealthCheckRequest) SetSignature(body *Signature) {
|
||||
if m != nil {
|
||||
m.Signature = body
|
||||
}
|
||||
}
|
||||
|
||||
// ReadSignedData marshals request body to buf.
|
||||
func (m *HealthCheckRequest) ReadSignedData(buf []byte) ([]byte, error) {
|
||||
_, err := m.GetBody().MarshalTo(buf)
|
||||
|
||||
return buf, err
|
||||
}
|
||||
|
||||
// SignedDataSize returns binary size of the request body.
|
||||
func (m *HealthCheckRequest) SignedDataSize() int {
|
||||
return m.GetBody().Size()
|
||||
}
|
||||
|
||||
// SetStatus sets health status of storage node.
|
||||
func (m *HealthCheckResponse_Body) SetStatus(v HealthStatus) {
|
||||
if m != nil {
|
||||
m.Status = v
|
||||
}
|
||||
}
|
||||
|
||||
// SetBody sets health check response body.
|
||||
func (m *HealthCheckResponse) SetBody(v *HealthCheckResponse_Body) {
|
||||
if m != nil {
|
||||
m.Body = v
|
||||
}
|
||||
}
|
||||
|
||||
// SetSignature sets signature of the health check response body.
|
||||
func (m *HealthCheckResponse) SetSignature(v *Signature) {
|
||||
if m != nil {
|
||||
m.Signature = v
|
||||
}
|
||||
}
|
||||
|
||||
// ReadSignedData marshals response body to buf.
|
||||
func (m *HealthCheckResponse) ReadSignedData(buf []byte) ([]byte, error) {
|
||||
_, err := m.GetBody().MarshalTo(buf)
|
||||
|
||||
return buf, err
|
||||
}
|
||||
|
||||
// SignedDataSize returns binary size of the response body.
|
||||
func (m *HealthCheckResponse) SignedDataSize() int {
|
||||
return m.GetBody().Size()
|
||||
}
|
15
pkg/services/private/types.go
Normal file
15
pkg/services/private/types.go
Normal file
|
@ -0,0 +1,15 @@
|
|||
package private
|
||||
|
||||
// SetKey sets public key used for signing.
|
||||
func (m *Signature) SetKey(v []byte) {
|
||||
if m != nil {
|
||||
m.Key = v
|
||||
}
|
||||
}
|
||||
|
||||
// SetSign sets binary signature.
|
||||
func (m *Signature) SetSign(v []byte) {
|
||||
if m != nil {
|
||||
m.Sign = v
|
||||
}
|
||||
}
|
Loading…
Reference in a new issue