git-svn-id: https://weed-fs.googlecode.com/svn/trunk@4 282b0af5-e82d-9cf1-ede4-77906d7719d0
This commit is contained in:
@@ -3,13 +3,17 @@ package directory
|
||||
import (
|
||||
"gob"
|
||||
"os"
|
||||
"rand"
|
||||
"log"
|
||||
)
|
||||
|
||||
type Machine struct {
|
||||
Server string //<server name/ip>[:port]
|
||||
}
|
||||
type Mapper struct {
|
||||
dir string
|
||||
fileName string
|
||||
Virtual2physical map[uint32][]uint32
|
||||
Virtual2physical map[uint32][]Machine
|
||||
}
|
||||
|
||||
func NewMapper(dirname string, filename string) (m *Mapper) {
|
||||
@@ -21,17 +25,21 @@ func NewMapper(dirname string, filename string) (m *Mapper) {
|
||||
if e != nil {
|
||||
log.Fatalf("Mapping File Read [ERROR] %s\n", e)
|
||||
} else {
|
||||
m.Virtual2physical = make(map[uint32][]uint32)
|
||||
m.Virtual2physical = make(map[uint32][]Machine)
|
||||
decoder := gob.NewDecoder(dataFile)
|
||||
decoder.Decode(m.Virtual2physical)
|
||||
dataFile.Close()
|
||||
}
|
||||
return
|
||||
}
|
||||
func (m *Mapper) Get(vid uint32) []uint32 {
|
||||
func (m *Mapper) PickForWrite() []Machine {
|
||||
vid := uint32(rand.Intn(len(m.Virtual2physical)))
|
||||
return m.Virtual2physical[vid]
|
||||
}
|
||||
func (m *Mapper) Get(vid uint32) []Machine {
|
||||
return m.Virtual2physical[vid]
|
||||
}
|
||||
func (m *Mapper) Add(vid uint32, pids ...uint32) {
|
||||
func (m *Mapper) Add(vid uint32, pids ...Machine) {
|
||||
m.Virtual2physical[vid] = append(m.Virtual2physical[vid], pids...)
|
||||
}
|
||||
func (m *Mapper) Save() {
|
||||
@@ -41,7 +49,7 @@ func (m *Mapper) Save() {
|
||||
log.Fatalf("Mapping File Save [ERROR] %s\n", e)
|
||||
}
|
||||
defer dataFile.Close()
|
||||
m.Virtual2physical = make(map[uint32][]uint32)
|
||||
m.Virtual2physical = make(map[uint32][]Machine)
|
||||
encoder := gob.NewEncoder(dataFile)
|
||||
encoder.Encode(m.Virtual2physical)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user