change volume id type to topology.VolumeId

This commit is contained in:
Chris Lu 2012-08-23 22:20:37 -07:00
parent 0121f35c12
commit 31e4dbfe38
6 changed files with 17 additions and 15 deletions

View file

@ -1 +0,0 @@
/weed

View file

@ -39,7 +39,7 @@ func dirLookupHandler(w http.ResponseWriter, r *http.Request) {
vid = vid[0:commaSep]
}
volumeId, _ := strconv.ParseUint(vid, 10, 64)
machine, e := mapper.Get(uint32(volumeId))
machine, e := mapper.Get(topology.VolumeId(volumeId))
if e == nil {
writeJson(w, r, machine.Server)
} else {

View file

@ -4,18 +4,19 @@ import (
"encoding/hex"
"log"
"pkg/storage"
"pkg/topology"
"strconv"
"strings"
"pkg/util"
)
type FileId struct {
VolumeId uint32
VolumeId topology.VolumeId
Key uint64
Hashcode uint32
}
func NewFileId(VolumeId uint32, Key uint64, Hashcode uint32) *FileId {
func NewFileId(VolumeId topology.VolumeId, Key uint64, Hashcode uint32) *FileId {
return &FileId{VolumeId: VolumeId, Key: Key, Hashcode: Hashcode}
}
func ParseFileId(fid string) *FileId{
@ -27,7 +28,7 @@ func ParseFileId(fid string) *FileId{
vid_string, key_hash_string := a[0], a[1]
vid, _ := strconv.ParseUint(vid_string, 10, 64)
key, hash := storage.ParseKeyHash(key_hash_string)
return &FileId{VolumeId: uint32(vid), Key: key, Hashcode: hash}
return &FileId{VolumeId: topology.VolumeId(vid), Key: key, Hashcode: hash}
}
func (n *FileId) String() string {
bytes := make([]byte, 12)

View file

@ -32,8 +32,8 @@ type Mapper struct {
volumeLock sync.Mutex
sequenceLock sync.Mutex
Machines []*Machine
vid2machineId map[uint32]int //machineId is +1 of the index of []*Machine, to detect not found entries
Writers []uint32 // transient array of Writers volume id
vid2machineId map[topology.VolumeId]int //machineId is +1 of the index of []*Machine, to detect not found entries
Writers []topology.VolumeId // transient array of Writers volume id
FileIdSequence uint64
fileIdCounter uint64
@ -47,9 +47,9 @@ func NewMachine(server, publicUrl string, volumes []topology.VolumeInfo) *Machin
func NewMapper(dirname string, filename string, volumeSizeLimit uint64) (m *Mapper) {
m = &Mapper{dir: dirname, fileName: filename}
m.vid2machineId = make(map[uint32]int)
m.vid2machineId = make(map[topology.VolumeId]int)
m.volumeSizeLimit = volumeSizeLimit
m.Writers = *new([]uint32)
m.Writers = *new([]topology.VolumeId)
m.Machines = *new([]*Machine)
seqFile, se := os.OpenFile(path.Join(m.dir, m.fileName+".seq"), os.O_RDONLY, 0644)
@ -102,7 +102,7 @@ func (m *Mapper) NextFileId(c string) (uint64,int) {
m.fileIdCounter = m.fileIdCounter - count
return m.FileIdSequence - m.fileIdCounter, int(count)
}
func (m *Mapper) Get(vid uint32) (*Machine, error) {
func (m *Mapper) Get(vid topology.VolumeId) (*Machine, error) {
machineId := m.vid2machineId[vid]
if machineId <= 0 {
return nil, errors.New("invalid volume id " + strconv.FormatUint(uint64(vid), 10))
@ -134,7 +134,7 @@ func (m *Mapper) Add(machine Machine) {
m.vid2machineId[v.Id] = machineId + 1 //use base 1 indexed, to detect not found cases
}
//setting Writers, copy-on-write because of possible updating, this needs some future work!
var writers []uint32
var writers []topology.VolumeId
for _, machine_entry := range m.Machines {
for _, v := range machine_entry.Volumes {
if uint64(v.Size) < m.volumeSizeLimit {

View file

@ -64,7 +64,7 @@ func (s *Store) Status() *[]*topology.VolumeInfo {
stats := new([]*topology.VolumeInfo)
for k, v := range s.volumes {
s := new(topology.VolumeInfo)
s.Id, s.Size = uint32(k), v.Size()
s.Id, s.Size = topology.VolumeId(k), v.Size()
*stats = append(*stats, s)
}
return stats
@ -73,7 +73,7 @@ func (s *Store) Join(mserver string) {
stats := new([]*topology.VolumeInfo)
for k, v := range s.volumes {
s := new(topology.VolumeInfo)
s.Id, s.Size = uint32(k), v.Size()
s.Id, s.Size = topology.VolumeId(k), v.Size()
*stats = append(*stats, s)
}
bytes, _ := json.Marshal(stats)

View file

@ -4,13 +4,15 @@ import (
)
type VolumeId uint32
type VolumeInfo struct {
Id uint32
Id VolumeId
Size int64
}
type Node struct {
volumes map[uint64]VolumeInfo
volumes map[VolumeId]VolumeInfo
volumeLimit int
Ip string
Port int
PublicUrl string
}