mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2025-07-31 23:13:26 +08:00
Thread-safe fixes:
1. avoid sharing []byte 2. switch to use ReadAt()
This commit is contained in:
parent
e5e6eeda83
commit
3dbebfd1e1
@ -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
|
||||||
}
|
}
|
||||||
|
@ -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 {
|
func walkIndexFile(r *os.File, fn func(key uint64, offset, size uint32) error) error {
|
||||||
br := bufio.NewReaderSize(r, 1024*1024)
|
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 {
|
if e == io.EOF {
|
||||||
copy(bytes[:count-i], bytes[i:count])
|
return nil
|
||||||
i = count - i
|
|
||||||
count, e = br.Read(bytes[i:])
|
|
||||||
count += i
|
|
||||||
} else {
|
|
||||||
count, e = br.Read(bytes)
|
|
||||||
}
|
}
|
||||||
}
|
count, e = r.ReadAt(bytes, readerOffset)
|
||||||
if e == io.EOF {
|
glog.V(3).Infoln("file", r.Name(), "readerOffset", readerOffset, "count", count, "e", e)
|
||||||
return nil
|
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)
|
bytes := make([]byte, 16)
|
||||||
util.Uint32toBytes(nm.bytes[8:12], offset)
|
util.Uint64toBytes(bytes[0:8], key)
|
||||||
util.Uint32toBytes(nm.bytes[12:16], size)
|
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)
|
bytes := make([]byte, 16)
|
||||||
util.Uint32toBytes(nm.bytes[8:12], 0)
|
util.Uint64toBytes(bytes[0:8], key)
|
||||||
util.Uint32toBytes(nm.bytes[12:16], 0)
|
util.Uint32toBytes(bytes[8:12], 0)
|
||||||
if _, err = nm.indexFile.Write(nm.bytes); err != nil {
|
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…
Reference in New Issue
Block a user