mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-12-22 16:17:55 +08:00
f2db746690
* Rename `command_ec_encode_test.go` to `command_ec_common_test.go`. All tests defined in this file are now for `command_ec_common.go`. * Minor code cleanups. - Fix broken `ec.balance` test. - Rework integer ceiling division to not use floats, which can introduce precision errors. * Introduce logic to resolve volume replica placement within EC rebalancing. This will be used to make rebalancing logic topology-aware. * Give shell.EcNode.dc a dedicated DataCenterId type.
76 lines
1.7 KiB
Go
76 lines
1.7 KiB
Go
package super_block
|
|
|
|
import (
|
|
"fmt"
|
|
)
|
|
|
|
type ReplicaPlacement struct {
|
|
SameRackCount int `json:"node,omitempty"`
|
|
DiffRackCount int `json:"rack,omitempty"`
|
|
DiffDataCenterCount int `json:"dc,omitempty"`
|
|
}
|
|
|
|
func NewReplicaPlacementFromString(t string) (*ReplicaPlacement, error) {
|
|
rp := &ReplicaPlacement{}
|
|
switch len(t) {
|
|
case 0:
|
|
t = "000"
|
|
case 1:
|
|
t = "00" + t
|
|
case 2:
|
|
t = "0" + t
|
|
}
|
|
for i, c := range t {
|
|
count := int(c - '0')
|
|
if count < 0 {
|
|
return rp, fmt.Errorf("unknown replication type: %s", t)
|
|
}
|
|
switch i {
|
|
case 0:
|
|
rp.DiffDataCenterCount = count
|
|
case 1:
|
|
rp.DiffRackCount = count
|
|
case 2:
|
|
rp.SameRackCount = count
|
|
}
|
|
}
|
|
value := rp.DiffDataCenterCount*100 + rp.DiffRackCount*10 + rp.SameRackCount
|
|
if value > 255 {
|
|
return rp, fmt.Errorf("unexpected replication type: %s", t)
|
|
}
|
|
return rp, nil
|
|
}
|
|
|
|
func NewReplicaPlacementFromByte(b byte) (*ReplicaPlacement, error) {
|
|
return NewReplicaPlacementFromString(fmt.Sprintf("%03d", b))
|
|
}
|
|
|
|
func (a *ReplicaPlacement) Equals(b *ReplicaPlacement) bool {
|
|
if a == nil || b == nil {
|
|
return false
|
|
}
|
|
return (a.SameRackCount == b.SameRackCount &&
|
|
a.DiffRackCount == b.DiffRackCount &&
|
|
a.DiffDataCenterCount == b.DiffDataCenterCount)
|
|
}
|
|
|
|
func (rp *ReplicaPlacement) Byte() byte {
|
|
if rp == nil {
|
|
return 0
|
|
}
|
|
ret := rp.DiffDataCenterCount*100 + rp.DiffRackCount*10 + rp.SameRackCount
|
|
return byte(ret)
|
|
}
|
|
|
|
func (rp *ReplicaPlacement) String() string {
|
|
b := make([]byte, 3)
|
|
b[0] = byte(rp.DiffDataCenterCount + '0')
|
|
b[1] = byte(rp.DiffRackCount + '0')
|
|
b[2] = byte(rp.SameRackCount + '0')
|
|
return string(b)
|
|
}
|
|
|
|
func (rp *ReplicaPlacement) GetCopyCount() int {
|
|
return rp.DiffDataCenterCount + rp.DiffRackCount + rp.SameRackCount + 1
|
|
}
|