list ec collections also

This commit is contained in:
Chris Lu 2019-05-30 09:17:58 -07:00
parent 866197eee3
commit e941d0c2f4
2 changed files with 15 additions and 3 deletions

View file

@ -19,7 +19,7 @@ func (ms *MasterServer) CollectionList(ctx context.Context, req *master_pb.Colle
collections := ms.Topo.ListCollections()
for _, c := range collections {
resp.Collections = append(resp.Collections, &master_pb.Collection{
Name: c.Name,
Name: c,
})
}

View file

@ -126,9 +126,21 @@ func (t *Topology) GetVolumeLayout(collectionName string, rp *storage.ReplicaPla
}).(*Collection).GetOrCreateVolumeLayout(rp, ttl)
}
func (t *Topology) ListCollections() (ret []*Collection) {
func (t *Topology) ListCollections() (ret []string) {
mapOfCollections := make(map[string]bool)
for _, c := range t.collectionMap.Items() {
ret = append(ret, c.(*Collection))
mapOfCollections[c.(*Collection).Name] = true
}
t.ecShardMapLock.RLock()
for _, ecVolumeLocation := range t.ecShardMap {
mapOfCollections[ecVolumeLocation.Collection] = true
}
t.ecShardMapLock.RUnlock()
for k, _ := range mapOfCollections {
ret = append(ret, k)
}
return ret
}