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] vid = vid[0:commaSep]
} }
volumeId, _ := strconv.ParseUint(vid, 10, 64) volumeId, _ := strconv.ParseUint(vid, 10, 64)
machine, e := mapper.Get(uint32(volumeId)) machine, e := mapper.Get(topology.VolumeId(volumeId))
if e == nil { if e == nil {
writeJson(w, r, machine.Server) writeJson(w, r, machine.Server)
} else { } else {

View file

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

View file

@ -32,8 +32,8 @@ type Mapper struct {
volumeLock sync.Mutex volumeLock sync.Mutex
sequenceLock sync.Mutex sequenceLock sync.Mutex
Machines []*Machine Machines []*Machine
vid2machineId map[uint32]int //machineId is +1 of the index of []*Machine, to detect not found entries vid2machineId map[topology.VolumeId]int //machineId is +1 of the index of []*Machine, to detect not found entries
Writers []uint32 // transient array of Writers volume id Writers []topology.VolumeId // transient array of Writers volume id
FileIdSequence uint64 FileIdSequence uint64
fileIdCounter 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) { func NewMapper(dirname string, filename string, volumeSizeLimit uint64) (m *Mapper) {
m = &Mapper{dir: dirname, fileName: filename} m = &Mapper{dir: dirname, fileName: filename}
m.vid2machineId = make(map[uint32]int) m.vid2machineId = make(map[topology.VolumeId]int)
m.volumeSizeLimit = volumeSizeLimit m.volumeSizeLimit = volumeSizeLimit
m.Writers = *new([]uint32) m.Writers = *new([]topology.VolumeId)
m.Machines = *new([]*Machine) m.Machines = *new([]*Machine)
seqFile, se := os.OpenFile(path.Join(m.dir, m.fileName+".seq"), os.O_RDONLY, 0644) 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 m.fileIdCounter = m.fileIdCounter - count
return m.FileIdSequence - m.fileIdCounter, int(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] machineId := m.vid2machineId[vid]
if machineId <= 0 { if machineId <= 0 {
return nil, errors.New("invalid volume id " + strconv.FormatUint(uint64(vid), 10)) 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 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! //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 _, machine_entry := range m.Machines {
for _, v := range machine_entry.Volumes { for _, v := range machine_entry.Volumes {
if uint64(v.Size) < m.volumeSizeLimit { if uint64(v.Size) < m.volumeSizeLimit {

View file

@ -64,7 +64,7 @@ func (s *Store) Status() *[]*topology.VolumeInfo {
stats := new([]*topology.VolumeInfo) stats := new([]*topology.VolumeInfo)
for k, v := range s.volumes { for k, v := range s.volumes {
s := new(topology.VolumeInfo) 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) *stats = append(*stats, s)
} }
return stats return stats
@ -73,7 +73,7 @@ func (s *Store) Join(mserver string) {
stats := new([]*topology.VolumeInfo) stats := new([]*topology.VolumeInfo)
for k, v := range s.volumes { for k, v := range s.volumes {
s := new(topology.VolumeInfo) 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) *stats = append(*stats, s)
} }
bytes, _ := json.Marshal(stats) bytes, _ := json.Marshal(stats)

View file

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