6447ea51d2
This takes advantange of the latest features in go-cid, peer.ID and go-multiaddr and makes the Go types serializable by default. This means we no longer need to copy between Pin <-> PinSerial, or ID <-> IDSerial etc. We can now efficiently binary-encode these types using short field keys and without parsing/stringifying (in many cases it just a cast). We still get the same json output as before (with minor modifications for Cids). This should greatly improve Cluster performance and memory usage when dealing with large collections of items. License: MIT Signed-off-by: Hector Sanjuan <hector@protocol.ai>
118 lines
2.4 KiB
Go
118 lines
2.4 KiB
Go
package descendalloc
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/ipfs/ipfs-cluster/api"
|
|
|
|
cid "github.com/ipfs/go-cid"
|
|
peer "github.com/libp2p/go-libp2p-peer"
|
|
)
|
|
|
|
type testcase struct {
|
|
candidates map[peer.ID]*api.Metric
|
|
current map[peer.ID]*api.Metric
|
|
expected []peer.ID
|
|
}
|
|
|
|
var (
|
|
peer0 = peer.ID("QmUQ6Nsejt1SuZAu8yL8WgqQZHHAYreLVYYa4VPsLUCed7")
|
|
peer1 = peer.ID("QmUZ13osndQ5uL4tPWHXe3iBgBgq9gfewcBMSCAuMBsDJ6")
|
|
peer2 = peer.ID("QmPrSBATWGAN56fiiEWEhKX3L1F3mTghEQR7vQwaeo7zHi")
|
|
peer3 = peer.ID("QmPGDFvBkgWhvzEK9qaTWrWurSwqXNmhnK3hgELPdZZNPa")
|
|
testCid, _ = cid.Decode("QmP63DkAFEnDYNjDYBpyNDfttu1fvUw99x1brscPzpqmmq")
|
|
)
|
|
|
|
var inAMinute = time.Now().Add(time.Minute).UnixNano()
|
|
|
|
var testCases = []testcase{
|
|
{ // regular sort
|
|
candidates: map[peer.ID]*api.Metric{
|
|
peer0: {
|
|
Name: "some-metric",
|
|
Value: "5",
|
|
Expire: inAMinute,
|
|
Valid: true,
|
|
},
|
|
peer1: {
|
|
Name: "some-metric",
|
|
Value: "1",
|
|
Expire: inAMinute,
|
|
Valid: true,
|
|
},
|
|
peer2: {
|
|
Name: "some-metric",
|
|
Value: "3",
|
|
Expire: inAMinute,
|
|
Valid: true,
|
|
},
|
|
peer3: {
|
|
Name: "some-metric",
|
|
Value: "2",
|
|
Expire: inAMinute,
|
|
Valid: true,
|
|
},
|
|
},
|
|
current: map[peer.ID]*api.Metric{},
|
|
expected: []peer.ID{peer1, peer3, peer2, peer0},
|
|
},
|
|
{ // filter invalid
|
|
candidates: map[peer.ID]*api.Metric{
|
|
peer0: {
|
|
Name: "some-metric",
|
|
Value: "1",
|
|
Expire: inAMinute,
|
|
Valid: false,
|
|
},
|
|
peer1: {
|
|
Name: "some-metric",
|
|
Value: "5",
|
|
Expire: inAMinute,
|
|
Valid: true,
|
|
},
|
|
},
|
|
current: map[peer.ID]*api.Metric{},
|
|
expected: []peer.ID{peer1},
|
|
},
|
|
{ // filter bad value
|
|
candidates: map[peer.ID]*api.Metric{
|
|
peer0: {
|
|
Name: "some-metric",
|
|
Value: "abc",
|
|
Expire: inAMinute,
|
|
Valid: true,
|
|
},
|
|
peer1: {
|
|
Name: "some-metric",
|
|
Value: "5",
|
|
Expire: inAMinute,
|
|
Valid: true,
|
|
},
|
|
},
|
|
current: map[peer.ID]*api.Metric{},
|
|
expected: []peer.ID{peer1},
|
|
},
|
|
}
|
|
|
|
func Test(t *testing.T) {
|
|
ctx := context.Background()
|
|
alloc := &DescendAllocator{}
|
|
for i, tc := range testCases {
|
|
t.Logf("Test case %d", i)
|
|
res, err := alloc.Allocate(ctx, testCid, tc.current, tc.candidates, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(res) == 0 {
|
|
t.Fatal("0 allocations")
|
|
}
|
|
for i, r := range res {
|
|
if e := tc.expected[len(res)-i-1]; r != e {
|
|
t.Errorf("Expect r[%d]=%s but got %s", i, r, e)
|
|
}
|
|
}
|
|
}
|
|
}
|