diff --git a/weed/storage/volume_info.go b/weed/storage/volume_info.go index 40e7f173a..e83aa9d93 100644 --- a/weed/storage/volume_info.go +++ b/weed/storage/volume_info.go @@ -67,6 +67,7 @@ func NewVolumeInfoFromShort(m *master_pb.VolumeShortInformationMessage) (vi Volu Id: needle.VolumeId(m.Id), Collection: internVolumeString(m.Collection), Version: needle.Version(m.Version), + DiskId: m.DiskId, } rp, e := super_block.NewReplicaPlacementFromByte(byte(m.ReplicaPlacement)) if e != nil { diff --git a/weed/storage/volume_info_test.go b/weed/storage/volume_info_test.go index a01e1fc9d..6e155705c 100644 --- a/weed/storage/volume_info_test.go +++ b/weed/storage/volume_info_test.go @@ -3,6 +3,7 @@ package storage import ( "testing" + "github.com/seaweedfs/seaweedfs/weed/pb/master_pb" "github.com/seaweedfs/seaweedfs/weed/storage/needle" ) @@ -25,3 +26,17 @@ func TestSortVolumeInfos(t *testing.T) { } } } + +func TestNewVolumeInfoFromShortKeepsDiskId(t *testing.T) { + vi, err := NewVolumeInfoFromShort(&master_pb.VolumeShortInformationMessage{ + Id: 7, + Version: 3, + DiskId: 3, + }) + if err != nil { + t.Fatal(err) + } + if vi.DiskId != 3 { + t.Fatalf("DiskId = %d, want 3", vi.DiskId) + } +}