mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2024-01-19 02:48:24 +00:00
84 lines
2.6 KiB
Plaintext
84 lines
2.6 KiB
Plaintext
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"compress/gzip"
|
|
"crypto/md5"
|
|
"flag"
|
|
"io"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"time"
|
|
"github.com/chrislusf/seaweedfs/weed/glog"
|
|
"github.com/chrislusf/seaweedfs/weed/storage"
|
|
"github.com/chrislusf/seaweedfs/weed/storage/needle"
|
|
"github.com/chrislusf/seaweedfs/weed/storage/super_block"
|
|
"github.com/chrislusf/seaweedfs/weed/util"
|
|
)
|
|
|
|
type VolumeFileScanner4SeeDat struct {
|
|
version needle.Version
|
|
}
|
|
|
|
func (scanner *VolumeFileScanner4SeeDat) VisitSuperBlock(superBlock super_block.SuperBlock) error {
|
|
scanner.version = superBlock.Version
|
|
return nil
|
|
}
|
|
|
|
func (scanner *VolumeFileScanner4SeeDat) ReadNeedleBody() bool {
|
|
return true
|
|
}
|
|
|
|
var (
|
|
files = int64(0)
|
|
filebytes = int64(0)
|
|
diffbytes = int64(0)
|
|
)
|
|
|
|
func Compresssion(data []byte) float64 {
|
|
if len(data) <= 128 {
|
|
return 100.0
|
|
}
|
|
compressed, _ := util.GzipData(data[0:128])
|
|
return float64(len(compressed)*10) / 1280.0
|
|
}
|
|
|
|
func (scanner *VolumeFileScanner4SeeDat) VisitNeedle(n *needle.Needle, offset int64, needleHeader, needleBody []byte) error {
|
|
t := time.Unix(int64(n.AppendAtNs)/int64(time.Second), int64(n.AppendAtNs)%int64(time.Second))
|
|
glog.V(0).Info("----------------------------------------------------------------------------------")
|
|
glog.V(0).Infof("%d,%s%x offset %d size %d(%s) cookie %x appendedAt %v hasmime[%t] mime[%s] (len: %d)",
|
|
*volumeId, n.Id, n.Cookie, offset, n.Size, util.BytesToHumanReadable(uint64(n.Size)), n.Cookie, t, n.HasMime(), string(n.Mime), len(n.Mime))
|
|
r, err := gzip.NewReader(bytes.NewReader(n.Data))
|
|
if err == nil {
|
|
buf := bytes.Buffer{}
|
|
h := md5.New()
|
|
c, _ := io.Copy(&buf, r)
|
|
d := buf.Bytes()
|
|
io.Copy(h, bytes.NewReader(d))
|
|
diff := (int64(n.DataSize) - int64(c))
|
|
diffbytes += diff
|
|
glog.V(0).Infof("was gzip! stored_size: %d orig_size: %d diff: %d(%d) mime:%s compression-of-128: %.2f md5: %x", n.DataSize, c, diff, diffbytes, http.DetectContentType(d), Compresssion(d), h.Sum(nil))
|
|
} else {
|
|
glog.V(0).Infof("no gzip!")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
var (
|
|
_ = ioutil.ReadAll
|
|
volumePath = flag.String("dir", "/tmp", "data directory to store files")
|
|
volumeCollection = flag.String("collection", "", "the volume collection name")
|
|
volumeId = flag.Int("volumeId", -1, "a volume id. The volume should already exist in the dir. The volume index file should not exist.")
|
|
)
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
vid := needle.VolumeId(*volumeId)
|
|
glog.V(0).Info("Starting")
|
|
scanner := &VolumeFileScanner4SeeDat{}
|
|
err := storage.ScanVolumeFile(*volumePath, *volumeCollection, vid, storage.NeedleMapInMemory, scanner)
|
|
if err != nil {
|
|
glog.Fatalf("Reading Volume File [ERROR] %s\n", err)
|
|
}
|
|
}
|