forked from TrueCloudLab/frostfs-api-go
[#189] sdk/netmap: Refactor Replica type
Replace alias to v2 type Replica with v2-compatible implementation. Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
This commit is contained in:
parent
eb96fffa5c
commit
6a29d9c360
3 changed files with 83 additions and 1 deletions
|
@ -6,7 +6,6 @@ import (
|
|||
|
||||
// fixme: make types instead of aliases to v2 structures
|
||||
type PlacementPolicy = netmap.PlacementPolicy
|
||||
type Replica = netmap.Replica
|
||||
|
||||
func PlacementPolicyToJSON(p *PlacementPolicy) ([]byte, error) {
|
||||
return netmap.PlacementPolicyToJSON(p)
|
||||
|
|
47
pkg/netmap/replica.go
Normal file
47
pkg/netmap/replica.go
Normal file
|
@ -0,0 +1,47 @@
|
|||
package netmap
|
||||
|
||||
import (
|
||||
"github.com/nspcc-dev/neofs-api-go/v2/netmap"
|
||||
)
|
||||
|
||||
// Replica represents v2-compatible object replica descriptor.
|
||||
type Replica netmap.Replica
|
||||
|
||||
// NewReplica creates and returns new Replica instance.
|
||||
func NewReplica() *Replica {
|
||||
return NewReplicaFromV2(new(netmap.Replica))
|
||||
}
|
||||
|
||||
// NewReplicaFromV2 converts v2 Replica to Replica.
|
||||
func NewReplicaFromV2(f *netmap.Replica) *Replica {
|
||||
return (*Replica)(f)
|
||||
}
|
||||
|
||||
// ToV2 converts Replica to v2 Replica.
|
||||
func (r *Replica) ToV2() *netmap.Replica {
|
||||
return (*netmap.Replica)(r)
|
||||
}
|
||||
|
||||
// Count returns number of object replicas.
|
||||
func (r *Replica) Count() uint32 {
|
||||
return (*netmap.Replica)(r).
|
||||
GetCount()
|
||||
}
|
||||
|
||||
// SetCount sets number of object replicas.
|
||||
func (r *Replica) SetCount(c uint32) {
|
||||
(*netmap.Replica)(r).
|
||||
SetCount(c)
|
||||
}
|
||||
|
||||
// Selector returns name of selector bucket to put replicas.
|
||||
func (r *Replica) Selector() string {
|
||||
return (*netmap.Replica)(r).
|
||||
GetSelector()
|
||||
}
|
||||
|
||||
// SetSelector sets name of selector bucket to put replicas.
|
||||
func (r *Replica) SetSelector(s string) {
|
||||
(*netmap.Replica)(r).
|
||||
SetSelector(s)
|
||||
}
|
36
pkg/netmap/replica_test.go
Normal file
36
pkg/netmap/replica_test.go
Normal file
|
@ -0,0 +1,36 @@
|
|||
package netmap
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/nspcc-dev/neofs-api-go/v2/netmap"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestReplicaFromV2(t *testing.T) {
|
||||
rV2 := new(netmap.Replica)
|
||||
rV2.SetCount(3)
|
||||
rV2.SetSelector("selector")
|
||||
|
||||
r := NewReplicaFromV2(rV2)
|
||||
|
||||
require.Equal(t, rV2, r.ToV2())
|
||||
}
|
||||
|
||||
func TestReplica_Count(t *testing.T) {
|
||||
r := NewReplica()
|
||||
c := uint32(3)
|
||||
|
||||
r.SetCount(c)
|
||||
|
||||
require.Equal(t, c, r.Count())
|
||||
}
|
||||
|
||||
func TestReplica_Selector(t *testing.T) {
|
||||
r := NewReplica()
|
||||
s := "some selector"
|
||||
|
||||
r.SetSelector(s)
|
||||
|
||||
require.Equal(t, s, r.Selector())
|
||||
}
|
Loading…
Reference in a new issue