seaweedfs/weed/topology/collection.go

79 lines
2.1 KiB
Go
Raw Normal View History

2013-11-12 10:21:22 +00:00
package topology
import (
"fmt"
2020-12-13 08:58:58 +00:00
"github.com/chrislusf/seaweedfs/weed/storage"
2019-04-19 04:43:36 +00:00
"github.com/chrislusf/seaweedfs/weed/storage/needle"
2019-12-23 20:48:20 +00:00
"github.com/chrislusf/seaweedfs/weed/storage/super_block"
"github.com/chrislusf/seaweedfs/weed/util"
2013-11-12 10:21:22 +00:00
)
type Collection struct {
Name string
volumeSizeLimit uint64
replicationAsMin bool
storageType2VolumeLayout *util.ConcurrentReadMap
2013-11-12 10:21:22 +00:00
}
func NewCollection(name string, volumeSizeLimit uint64, replicationAsMin bool) *Collection {
c := &Collection{
Name: name,
volumeSizeLimit: volumeSizeLimit,
replicationAsMin: replicationAsMin,
}
c.storageType2VolumeLayout = util.NewConcurrentReadMap()
2013-11-12 10:21:22 +00:00
return c
}
func (c *Collection) String() string {
return fmt.Sprintf("Name:%s, volumeSizeLimit:%d, storageType2VolumeLayout:%v", c.Name, c.volumeSizeLimit, c.storageType2VolumeLayout)
}
2020-12-13 19:59:32 +00:00
func (c *Collection) GetOrCreateVolumeLayout(rp *super_block.ReplicaPlacement, ttl *needle.TTL, diskType storage.DiskType) *VolumeLayout {
keyString := rp.String()
if ttl != nil {
keyString += ttl.String()
2013-11-12 10:21:22 +00:00
}
2020-12-13 19:59:32 +00:00
if diskType != storage.HardDriveType {
keyString += string(diskType)
2020-12-13 08:58:58 +00:00
}
vl := c.storageType2VolumeLayout.Get(keyString, func() interface{} {
2020-12-13 19:59:32 +00:00
return NewVolumeLayout(rp, ttl, diskType, c.volumeSizeLimit, c.replicationAsMin)
})
return vl.(*VolumeLayout)
2013-11-12 10:21:22 +00:00
}
2020-12-13 19:59:32 +00:00
func (c *Collection) DeleteVolumeLayout(rp *super_block.ReplicaPlacement, ttl *needle.TTL, diskType storage.DiskType) {
keyString := rp.String()
if ttl != nil {
keyString += ttl.String()
}
2020-12-13 19:59:32 +00:00
if diskType != storage.HardDriveType {
keyString += string(diskType)
}
c.storageType2VolumeLayout.Delete(keyString)
}
2019-04-19 04:43:36 +00:00
func (c *Collection) Lookup(vid needle.VolumeId) []*DataNode {
2016-05-30 19:30:26 +00:00
for _, vl := range c.storageType2VolumeLayout.Items() {
2013-11-12 10:21:22 +00:00
if vl != nil {
if list := vl.(*VolumeLayout).Lookup(vid); list != nil {
2013-11-12 10:21:22 +00:00
return list
}
}
}
return nil
}
func (c *Collection) ListVolumeServers() (nodes []*DataNode) {
2016-05-30 19:30:26 +00:00
for _, vl := range c.storageType2VolumeLayout.Items() {
if vl != nil {
if list := vl.(*VolumeLayout).ListVolumeServers(); list != nil {
nodes = append(nodes, list...)
}
}
}
return
}