|
|
@ -8,6 +8,7 @@ import ( |
|
|
|
"sync/atomic" |
|
|
|
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/glog" |
|
|
|
"github.com/chrislusf/seaweedfs/weed/storage/erasure_coding" |
|
|
|
"github.com/chrislusf/seaweedfs/weed/storage/needle" |
|
|
|
) |
|
|
|
|
|
|
@ -19,10 +20,12 @@ type Node interface { |
|
|
|
ReserveOneVolume(r int64) (*DataNode, error) |
|
|
|
UpAdjustMaxVolumeCountDelta(maxVolumeCountDelta int64) |
|
|
|
UpAdjustVolumeCountDelta(volumeCountDelta int64) |
|
|
|
UpAdjustEcShardCountDelta(ecShardCountDelta int64) |
|
|
|
UpAdjustActiveVolumeCountDelta(activeVolumeCountDelta int64) |
|
|
|
UpAdjustMaxVolumeId(vid needle.VolumeId) |
|
|
|
|
|
|
|
GetVolumeCount() int64 |
|
|
|
GetEcShardCount() int64 |
|
|
|
GetActiveVolumeCount() int64 |
|
|
|
GetMaxVolumeCount() int64 |
|
|
|
GetMaxVolumeId() needle.VolumeId |
|
|
@ -40,10 +43,11 @@ type Node interface { |
|
|
|
GetValue() interface{} //get reference to the topology,dc,rack,datanode
|
|
|
|
} |
|
|
|
type NodeImpl struct { |
|
|
|
id NodeId |
|
|
|
volumeCount int64 |
|
|
|
activeVolumeCount int64 |
|
|
|
ecShardCount int64 |
|
|
|
maxVolumeCount int64 |
|
|
|
id NodeId |
|
|
|
parent Node |
|
|
|
sync.RWMutex // lock children
|
|
|
|
children map[NodeId]Node |
|
|
@ -128,6 +132,9 @@ func (n *NodeImpl) Id() NodeId { |
|
|
|
return n.id |
|
|
|
} |
|
|
|
func (n *NodeImpl) FreeSpace() int64 { |
|
|
|
if n.ecShardCount > 0 { |
|
|
|
return n.maxVolumeCount - n.volumeCount - n.ecShardCount/erasure_coding.DataShardsCount - 1 |
|
|
|
} |
|
|
|
return n.maxVolumeCount - n.volumeCount |
|
|
|
} |
|
|
|
func (n *NodeImpl) SetParent(node Node) { |
|
|
@ -184,6 +191,12 @@ func (n *NodeImpl) UpAdjustVolumeCountDelta(volumeCountDelta int64) { //can be n |
|
|
|
n.parent.UpAdjustVolumeCountDelta(volumeCountDelta) |
|
|
|
} |
|
|
|
} |
|
|
|
func (n *NodeImpl) UpAdjustEcShardCountDelta(ecShardCountDelta int64) { //can be negative
|
|
|
|
atomic.AddInt64(&n.ecShardCount, ecShardCountDelta) |
|
|
|
if n.parent != nil { |
|
|
|
n.parent.UpAdjustEcShardCountDelta(ecShardCountDelta) |
|
|
|
} |
|
|
|
} |
|
|
|
func (n *NodeImpl) UpAdjustActiveVolumeCountDelta(activeVolumeCountDelta int64) { //can be negative
|
|
|
|
atomic.AddInt64(&n.activeVolumeCount, activeVolumeCountDelta) |
|
|
|
if n.parent != nil { |
|
|
@ -204,6 +217,9 @@ func (n *NodeImpl) GetMaxVolumeId() needle.VolumeId { |
|
|
|
func (n *NodeImpl) GetVolumeCount() int64 { |
|
|
|
return n.volumeCount |
|
|
|
} |
|
|
|
func (n *NodeImpl) GetEcShardCount() int64 { |
|
|
|
return n.ecShardCount |
|
|
|
} |
|
|
|
func (n *NodeImpl) GetActiveVolumeCount() int64 { |
|
|
|
return n.activeVolumeCount |
|
|
|
} |
|
|
@ -219,6 +235,7 @@ func (n *NodeImpl) LinkChildNode(node Node) { |
|
|
|
n.UpAdjustMaxVolumeCountDelta(node.GetMaxVolumeCount()) |
|
|
|
n.UpAdjustMaxVolumeId(node.GetMaxVolumeId()) |
|
|
|
n.UpAdjustVolumeCountDelta(node.GetVolumeCount()) |
|
|
|
n.UpAdjustEcShardCountDelta(node.GetEcShardCount()) |
|
|
|
n.UpAdjustActiveVolumeCountDelta(node.GetActiveVolumeCount()) |
|
|
|
node.SetParent(n) |
|
|
|
glog.V(0).Infoln(n, "adds child", node.Id()) |
|
|
@ -233,6 +250,7 @@ func (n *NodeImpl) UnlinkChildNode(nodeId NodeId) { |
|
|
|
node.SetParent(nil) |
|
|
|
delete(n.children, node.Id()) |
|
|
|
n.UpAdjustVolumeCountDelta(-node.GetVolumeCount()) |
|
|
|
n.UpAdjustEcShardCountDelta(-node.GetEcShardCount()) |
|
|
|
n.UpAdjustActiveVolumeCountDelta(-node.GetActiveVolumeCount()) |
|
|
|
n.UpAdjustMaxVolumeCountDelta(-node.GetMaxVolumeCount()) |
|
|
|
glog.V(0).Infoln(n, "removes", node.Id()) |
|
|
|