Browse Source
migrate fix command
migrate fix command
git-svn-id: https://weed-fs.googlecode.com/svn/trunk@63 282b0af5-e82d-9cf1-ede4-77906d7719d0pull/2/head
chris.lu@gmail.com
13 years ago
4 changed files with 86 additions and 48 deletions
-
7weed-fs/src/cmd/weed/command.go
-
69weed-fs/src/cmd/weed/fix.go
-
7weed-fs/src/cmd/weed/version.go
-
45weed-fs/src/cmd/weed/weed.go
@ -0,0 +1,69 @@ |
|||||
|
package main |
||||
|
|
||||
|
import ( |
||||
|
"pkg/storage" |
||||
|
"log" |
||||
|
"os" |
||||
|
"path" |
||||
|
"strconv" |
||||
|
) |
||||
|
|
||||
|
func init() { |
||||
|
cmdFix.Run = runFix // break init cycle
|
||||
|
} |
||||
|
|
||||
|
var cmdFix = &Command{ |
||||
|
UsageLine: "fix -dir=/tmp -volumeId=234 -debug=1", |
||||
|
Short: "run weed tool fix on data file if corrupted", |
||||
|
Long: `Fix runs the WeedFS fix command on the .dat volume file. |
||||
|
|
||||
|
`, |
||||
|
} |
||||
|
|
||||
|
var ( |
||||
|
dir = cmdFix.Flag.String("dir", "/tmp", "data directory to store files") |
||||
|
volumeId = cmdFix.Flag.Int("volumeId", -1, "a non-negative volume id. The volume should already exist in the dir. The volume index file should not exist.") |
||||
|
IsDebug = cmdFix.Flag.Bool("debug", false, "enable debug mode") |
||||
|
|
||||
|
store *storage.Store |
||||
|
) |
||||
|
|
||||
|
func runFix(cmd *Command, args []string) bool { |
||||
|
|
||||
|
if *volumeId == -1 { |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
fileName := strconv.Itoa(*volumeId) |
||||
|
dataFile, e := os.OpenFile(path.Join(*dir, fileName+".dat"), os.O_RDONLY, 0644) |
||||
|
if e != nil { |
||||
|
log.Fatalf("Read Volume [ERROR] %s\n", e) |
||||
|
} |
||||
|
defer dataFile.Close() |
||||
|
indexFile, ie := os.OpenFile(path.Join(*dir, fileName+".idx"), os.O_WRONLY|os.O_CREATE, 0644) |
||||
|
if ie != nil { |
||||
|
log.Fatalf("Create Volume Index [ERROR] %s\n", ie) |
||||
|
} |
||||
|
defer indexFile.Close() |
||||
|
|
||||
|
//skip the volume super block
|
||||
|
dataFile.Seek(storage.SuperBlockSize, 0) |
||||
|
|
||||
|
n, length := storage.ReadNeedle(dataFile) |
||||
|
nm := storage.NewNeedleMap(indexFile) |
||||
|
offset := uint32(storage.SuperBlockSize) |
||||
|
for n != nil { |
||||
|
if *IsDebug { |
||||
|
log.Println("key", n.Key, "volume offset", offset, "data_size", n.Size, "length", length) |
||||
|
} |
||||
|
if n.Size > 0 { |
||||
|
count, pe := nm.Put(n.Key, offset/8, n.Size) |
||||
|
if *IsDebug { |
||||
|
log.Println("saved", count, "with error", pe) |
||||
|
} |
||||
|
} |
||||
|
offset += length |
||||
|
n, length = storage.ReadNeedle(dataFile) |
||||
|
} |
||||
|
return true |
||||
|
} |
Write
Preview
Loading…
Cancel
Save
Reference in new issue