re
/
buntdb
forked from mirror/buntdb
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Share time.Now calls

This commit is contained in:
tidwall 2021-07-31 09:17:41 -07:00
parent d4c361c59c
commit 0bf8239554
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
buntdb.go
View File

@ -201,10 +201,11 @@ func (db *DB) Save(wr io.Writer) error {
defer db.mu.RUnlock() defer db.mu.RUnlock()
// use a buffered writer and flush every 4MB // use a buffered writer and flush every 4MB
var buf []byte var buf []byte
now := time.Now()
// iterated through every item in the database and write to the buffer // iterated through every item in the database and write to the buffer
btreeAscend(db.keys, func(item interface{}) bool { btreeAscend(db.keys, func(item interface{}) bool {
dbi := item.(*dbItem) dbi := item.(*dbItem)
buf = dbi.writeSetTo(buf) buf = dbi.writeSetTo(buf, now)
if len(buf) > 1024*1024*4 { if len(buf) > 1024*1024*4 {
// flush when buffer is over 4MB // flush when buffer is over 4MB
_, err = wr.Write(buf) _, err = wr.Write(buf)
@ -684,6 +685,7 @@ func (db *DB) Shrink() error {
} }
done = true done = true
var n int var n int
now := time.Now()
btreeAscendGreaterOrEqual(db.keys, &dbItem{key: pivot}, btreeAscendGreaterOrEqual(db.keys, &dbItem{key: pivot},
func(item interface{}) bool { func(item interface{}) bool {
dbi := item.(*dbItem) dbi := item.(*dbItem)
@ -693,7 +695,7 @@ func (db *DB) Shrink() error {
done = false done = false
return false return false
} }
buf = dbi.writeSetTo(buf) buf = dbi.writeSetTo(buf, now)
n++ n++
return true return true
}, },
@ -1177,12 +1179,13 @@ func (tx *Tx) Commit() error {
if tx.wc.rbkeys != nil { if tx.wc.rbkeys != nil {
tx.db.buf = append(tx.db.buf, "*1\r\n$7\r\nflushdb\r\n"...) tx.db.buf = append(tx.db.buf, "*1\r\n$7\r\nflushdb\r\n"...)
} }
now := time.Now()
// Each committed record is written to disk // Each committed record is written to disk
for key, item := range tx.wc.commitItems { for key, item := range tx.wc.commitItems {
if item == nil { if item == nil {
tx.db.buf = (&dbItem{key: key}).writeDeleteTo(tx.db.buf) tx.db.buf = (&dbItem{key: key}).writeDeleteTo(tx.db.buf)
} else { } else {
tx.db.buf = item.writeSetTo(tx.db.buf) tx.db.buf = item.writeSetTo(tx.db.buf, now)
} }
} }
// Flushing the buffer only once per transaction. // Flushing the buffer only once per transaction.
@ -1257,14 +1260,14 @@ type dbItem struct {
func appendArray(buf []byte, count int) []byte { func appendArray(buf []byte, count int) []byte {
buf = append(buf, '*') buf = append(buf, '*')
buf = append(buf, strconv.FormatInt(int64(count), 10)...) buf = strconv.AppendInt(buf, int64(count), 10)
buf = append(buf, '\r', '\n') buf = append(buf, '\r', '\n')
return buf return buf
} }
func appendBulkString(buf []byte, s string) []byte { func appendBulkString(buf []byte, s string) []byte {
buf = append(buf, '$') buf = append(buf, '$')
buf = append(buf, strconv.FormatInt(int64(len(s)), 10)...) buf = strconv.AppendInt(buf, int64(len(s)), 10)
buf = append(buf, '\r', '\n') buf = append(buf, '\r', '\n')
buf = append(buf, s...) buf = append(buf, s...)
buf = append(buf, '\r', '\n') buf = append(buf, '\r', '\n')
@ -1272,9 +1275,9 @@ func appendBulkString(buf []byte, s string) []byte {
} }
// writeSetTo writes an item as a single SET record to the a bufio Writer. // writeSetTo writes an item as a single SET record to the a bufio Writer.
func (dbi *dbItem) writeSetTo(buf []byte) []byte { func (dbi *dbItem) writeSetTo(buf []byte, now time.Time) []byte {
if dbi.opts != nil && dbi.opts.ex { if dbi.opts != nil && dbi.opts.ex {
ex := time.Until(dbi.opts.exat) / time.Second ex := dbi.opts.exat.Sub(now) / time.Second
buf = appendArray(buf, 5) buf = appendArray(buf, 5)
buf = appendBulkString(buf, "set") buf = appendBulkString(buf, "set")
buf = appendBulkString(buf, dbi.key) buf = appendBulkString(buf, dbi.key)