diff options
Diffstat (limited to 'go/topology/topology.go')
| -rw-r--r-- | go/topology/topology.go | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/go/topology/topology.go b/go/topology/topology.go index 6c5bde304..b1fa3f2a2 100644 --- a/go/topology/topology.go +++ b/go/topology/topology.go @@ -108,8 +108,13 @@ func (t *Topology) NextVolumeId() storage.VolumeId { return next } -func (t *Topology) PickForWrite(collectionName string, rp *storage.ReplicaPlacement, count int, dataCenter string) (string, int, *DataNode, error) { - vid, count, datanodes, err := t.GetVolumeLayout(collectionName, rp).PickForWrite(count, dataCenter) +func (t *Topology) HasWriableVolume(option *VolumeGrowOption) bool { + vl := t.GetVolumeLayout(option.Collection, option.ReplicaPlacement) + return vl.GetActiveVolumeCount(option) > 0 +} + +func (t *Topology) PickForWrite(count int, option *VolumeGrowOption) (string, int, *DataNode, error) { + vid, count, datanodes, err := t.GetVolumeLayout(option.Collection, option.ReplicaPlacement).PickForWrite(count, option) if err != nil || datanodes.Length() == 0 { return "", 0, nil, errors.New("No writable volumes avalable!") } |