47b744f1c0
ipfs-cluster-service now has a migration subcommand that upgrades persistant state snapshots with an out-of-date format version to the newest version of raft state. If all cluster members shutdown with consistent state, upgrade ipfs-cluster, and run the state upgrade command, the new version of cluster will be compatible with persistent storage. ipfs-cluster now validates its persistent state upon loading it and exits with a clear error in the case the state format version is not up to date. Raft snapshotting is enforced on all shutdowns and the json backup is no longer run. This commit makes use of recent changes to libp2p-raft allowing raft states to implement their own marshaling strategies. Now mapstate handles the logic for its (de)serialization. In the interest of supporting various potential upgrade formats the state serialization begins with a varint (right now one byte) describing the version. Some go tests are modified and a go test is added to cover new ipfs-cluster raft snapshot reading functions. Sharness tests are added to cover the state upgrade command.
131 lines
2.7 KiB
Go
131 lines
2.7 KiB
Go
package mapstate
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
msgpack "github.com/multiformats/go-multicodec/msgpack"
|
|
|
|
cid "github.com/ipfs/go-cid"
|
|
peer "github.com/libp2p/go-libp2p-peer"
|
|
|
|
"github.com/ipfs/ipfs-cluster/api"
|
|
)
|
|
|
|
var testCid1, _ = cid.Decode("QmP63DkAFEnDYNjDYBpyNDfttu1fvUw99x1brscPzpqmmq")
|
|
var testPeerID1, _ = peer.IDB58Decode("QmXZrtE5jQwXNqCJMfHUTQkvhQ4ZAnqMnmzFMJfLewuabc")
|
|
|
|
var c = api.Pin{
|
|
Cid: testCid1,
|
|
Allocations: []peer.ID{testPeerID1},
|
|
ReplicationFactor: -1,
|
|
}
|
|
|
|
func TestAdd(t *testing.T) {
|
|
ms := NewMapState()
|
|
ms.Add(c)
|
|
if !ms.Has(c.Cid) {
|
|
t.Error("should have added it")
|
|
}
|
|
}
|
|
|
|
func TestRm(t *testing.T) {
|
|
ms := NewMapState()
|
|
ms.Add(c)
|
|
ms.Rm(c.Cid)
|
|
if ms.Has(c.Cid) {
|
|
t.Error("should have removed it")
|
|
}
|
|
}
|
|
|
|
func TestGet(t *testing.T) {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
t.Fatal("paniced")
|
|
}
|
|
}()
|
|
ms := NewMapState()
|
|
ms.Add(c)
|
|
get := ms.Get(c.Cid)
|
|
if get.Cid.String() != c.Cid.String() ||
|
|
get.Allocations[0] != c.Allocations[0] ||
|
|
get.ReplicationFactor != c.ReplicationFactor {
|
|
t.Error("returned something different")
|
|
}
|
|
}
|
|
|
|
func TestList(t *testing.T) {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
t.Fatal("paniced")
|
|
}
|
|
}()
|
|
ms := NewMapState()
|
|
ms.Add(c)
|
|
list := ms.List()
|
|
if list[0].Cid.String() != c.Cid.String() ||
|
|
list[0].Allocations[0] != c.Allocations[0] ||
|
|
list[0].ReplicationFactor != c.ReplicationFactor {
|
|
t.Error("returned something different")
|
|
}
|
|
}
|
|
|
|
func TestMarshalUnmarshal(t *testing.T) {
|
|
ms := NewMapState()
|
|
ms.Add(c)
|
|
b, err := ms.Marshal()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
ms2 := NewMapState()
|
|
err = ms2.Unmarshal(b)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if ms.Version != ms2.Version {
|
|
t.Fatal(err)
|
|
}
|
|
get := ms2.Get(c.Cid)
|
|
if get.Allocations[0] != testPeerID1 {
|
|
t.Error("expected different peer id")
|
|
}
|
|
}
|
|
|
|
func TestMigrateFromV1(t *testing.T) {
|
|
// Construct the bytes of a v1 state
|
|
var v1State mapStateV1
|
|
v1State.PinMap = map[string]struct{}{
|
|
c.Cid.String(): {}}
|
|
v1State.Version = 1
|
|
buf := new(bytes.Buffer)
|
|
enc := msgpack.Multicodec(msgpack.DefaultMsgpackHandle()).Encoder(buf)
|
|
err := enc.Encode(v1State)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
vCodec := make([]byte, 1)
|
|
vCodec[0] = byte(v1State.Version)
|
|
v1Bytes := append(vCodec, buf.Bytes()...)
|
|
|
|
// Unmarshal first to check this is v1
|
|
ms := NewMapState()
|
|
err = ms.Unmarshal(v1Bytes)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if ms.Version != 1 {
|
|
t.Error("unmarshal picked up the wrong version")
|
|
}
|
|
// Migrate state to current version
|
|
r := bytes.NewBuffer(v1Bytes)
|
|
err = ms.Restore(r)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
get := ms.Get(c.Cid)
|
|
if get.ReplicationFactor != -1 || !get.Cid.Equals(c.Cid) {
|
|
t.Error("expected something different")
|
|
t.Logf("%+v", get)
|
|
}
|
|
}
|