Browse Source

Thread-safe fixes:

1. avoid sharing []byte
2. switch to use ReadAt()
pull/2/head
Chris Lu 11 years ago
parent
commit
3dbebfd1e1
  1. 2
      go/operation/delete_content.go
  2. 50
      go/storage/needle_map.go

2
go/operation/delete_content.go

@ -18,6 +18,6 @@ func Delete(url string) error {
glog.V(0).Infoln("failing to delete", url) glog.V(0).Infoln("failing to delete", url)
return err return err
} }
_, err = http.DefaultClient.Do(req)
_, err = client.Do(req)
return err return err
} }

50
go/storage/needle_map.go

@ -1,7 +1,6 @@
package storage package storage
import ( import (
"bufio"
"code.google.com/p/weed-fs/go/glog" "code.google.com/p/weed-fs/go/glog"
"code.google.com/p/weed-fs/go/util" "code.google.com/p/weed-fs/go/util"
"fmt" "fmt"
@ -35,16 +34,12 @@ type NeedleMap struct {
indexFile *os.File indexFile *os.File
m CompactMap m CompactMap
//transient
bytes []byte
mapMetric mapMetric
} }
func NewNeedleMap(file *os.File) *NeedleMap { func NewNeedleMap(file *os.File) *NeedleMap {
nm := &NeedleMap{ nm := &NeedleMap{
m: NewCompactMap(), m: NewCompactMap(),
bytes: make([]byte, 16),
indexFile: file, indexFile: file,
} }
return nm return nm
@ -64,14 +59,14 @@ func LoadNeedleMap(file *os.File) (*NeedleMap, error) {
nm.FileByteCounter = nm.FileByteCounter + uint64(size) nm.FileByteCounter = nm.FileByteCounter + uint64(size)
if offset > 0 { if offset > 0 {
oldSize := nm.m.Set(Key(key), offset, size) oldSize := nm.m.Set(Key(key), offset, size)
glog.V(4).Infoln("reading key", key, "offset", offset, "size", size, "oldSize", oldSize)
glog.V(3).Infoln("reading key", key, "offset", offset, "size", size, "oldSize", oldSize)
if oldSize > 0 { if oldSize > 0 {
nm.DeletionCounter++ nm.DeletionCounter++
nm.DeletionByteCounter = nm.DeletionByteCounter + uint64(oldSize) nm.DeletionByteCounter = nm.DeletionByteCounter + uint64(oldSize)
} }
} else { } else {
oldSize := nm.m.Delete(Key(key)) oldSize := nm.m.Delete(Key(key))
glog.V(4).Infoln("removing key", key, "offset", offset, "size", size, "oldSize", oldSize)
glog.V(3).Infoln("removing key", key, "offset", offset, "size", size, "oldSize", oldSize)
nm.DeletionCounter++ nm.DeletionCounter++
nm.DeletionByteCounter = nm.DeletionByteCounter + uint64(oldSize) nm.DeletionByteCounter = nm.DeletionByteCounter + uint64(oldSize)
} }
@ -83,17 +78,19 @@ func LoadNeedleMap(file *os.File) (*NeedleMap, error) {
// walks through the index file, calls fn function with each key, offset, size // walks through the index file, calls fn function with each key, offset, size
// stops with the error returned by the fn function // stops with the error returned by the fn function
func walkIndexFile(r io.Reader, fn func(key uint64, offset, size uint32) error) error {
br := bufio.NewReaderSize(r, 1024*1024)
func walkIndexFile(r *os.File, fn func(key uint64, offset, size uint32) error) error {
var readerOffset int64
bytes := make([]byte, 16*RowsToRead) bytes := make([]byte, 16*RowsToRead)
count, e := br.Read(bytes)
count, e := r.ReadAt(bytes, readerOffset)
glog.V(3).Infoln("file", r.Name(), "readerOffset", readerOffset, "count", count, "e", e)
readerOffset += int64(count)
var ( var (
key uint64 key uint64
offset, size uint32 offset, size uint32
i int i int
) )
for count > 0 && e == nil {
for count > 0 && e == nil || e == io.EOF {
for i = 0; i+16 <= count; i += 16 { for i = 0; i+16 <= count; i += 16 {
key = util.BytesToUint64(bytes[i : i+8]) key = util.BytesToUint64(bytes[i : i+8])
offset = util.BytesToUint32(bytes[i+8 : i+12]) offset = util.BytesToUint32(bytes[i+8 : i+12])
@ -102,33 +99,29 @@ func walkIndexFile(r io.Reader, fn func(key uint64, offset, size uint32) error)
return e return e
} }
} }
if count%16 != 0 {
copy(bytes[:count-i], bytes[i:count])
i = count - i
count, e = br.Read(bytes[i:])
count += i
} else {
count, e = br.Read(bytes)
}
}
if e == io.EOF { if e == io.EOF {
return nil return nil
} }
count, e = r.ReadAt(bytes, readerOffset)
glog.V(3).Infoln("file", r.Name(), "readerOffset", readerOffset, "count", count, "e", e)
readerOffset += int64(count)
}
return e return e
} }
func (nm *NeedleMap) Put(key uint64, offset uint32, size uint32) (int, error) { func (nm *NeedleMap) Put(key uint64, offset uint32, size uint32) (int, error) {
oldSize := nm.m.Set(Key(key), offset, size) oldSize := nm.m.Set(Key(key), offset, size)
util.Uint64toBytes(nm.bytes[0:8], key)
util.Uint32toBytes(nm.bytes[8:12], offset)
util.Uint32toBytes(nm.bytes[12:16], size)
bytes := make([]byte, 16)
util.Uint64toBytes(bytes[0:8], key)
util.Uint32toBytes(bytes[8:12], offset)
util.Uint32toBytes(bytes[12:16], size)
nm.FileCounter++ nm.FileCounter++
nm.FileByteCounter = nm.FileByteCounter + uint64(size) nm.FileByteCounter = nm.FileByteCounter + uint64(size)
if oldSize > 0 { if oldSize > 0 {
nm.DeletionCounter++ nm.DeletionCounter++
nm.DeletionByteCounter = nm.DeletionByteCounter + uint64(oldSize) nm.DeletionByteCounter = nm.DeletionByteCounter + uint64(oldSize)
} }
return nm.indexFile.Write(nm.bytes)
return nm.indexFile.Write(bytes)
} }
func (nm *NeedleMap) Get(key uint64) (element *NeedleValue, ok bool) { func (nm *NeedleMap) Get(key uint64) (element *NeedleValue, ok bool) {
element, ok = nm.m.Get(Key(key)) element, ok = nm.m.Get(Key(key))
@ -140,10 +133,11 @@ func (nm *NeedleMap) Delete(key uint64) error {
if err != nil { if err != nil {
return fmt.Errorf("cannot get position of indexfile: %s", err) return fmt.Errorf("cannot get position of indexfile: %s", err)
} }
util.Uint64toBytes(nm.bytes[0:8], key)
util.Uint32toBytes(nm.bytes[8:12], 0)
util.Uint32toBytes(nm.bytes[12:16], 0)
if _, err = nm.indexFile.Write(nm.bytes); err != nil {
bytes := make([]byte, 16)
util.Uint64toBytes(bytes[0:8], key)
util.Uint32toBytes(bytes[8:12], 0)
util.Uint32toBytes(bytes[12:16], 0)
if _, err = nm.indexFile.Write(bytes); err != nil {
plus := "" plus := ""
if e := nm.indexFile.Truncate(offset); e != nil { if e := nm.indexFile.Truncate(offset); e != nil {
plus = "\ncouldn't truncate index file: " + e.Error() plus = "\ncouldn't truncate index file: " + e.Error()

Loading…
Cancel
Save