0
0

Add bucket fill percent option for boltdb

This commit is contained in:
Mark Mindenhall 2016-06-13 18:47:38 -06:00
parent fedb46269e
commit d369bd5c3c
2 changed files with 18 additions and 10 deletions

View File

@ -33,11 +33,12 @@ const (
) )
type Store struct { type Store struct {
path string path string
bucket string bucket string
db *bolt.DB db *bolt.DB
noSync bool noSync bool
mo store.MergeOperator fillPercent float64
mo store.MergeOperator
} }
func New(mo store.MergeOperator, config map[string]interface{}) (store.KVStore, error) { func New(mo store.MergeOperator, config map[string]interface{}) (store.KVStore, error) {
@ -53,6 +54,11 @@ func New(mo store.MergeOperator, config map[string]interface{}) (store.KVStore,
noSync, _ := config["nosync"].(bool) noSync, _ := config["nosync"].(bool)
fillPercent, ok := config["fillPercent"].(float64)
if !ok {
fillPercent = bolt.DefaultFillPercent
}
db, err := bolt.Open(path, 0600, nil) db, err := bolt.Open(path, 0600, nil)
if err != nil { if err != nil {
return nil, err return nil, err
@ -69,11 +75,12 @@ func New(mo store.MergeOperator, config map[string]interface{}) (store.KVStore,
} }
rv := Store{ rv := Store{
path: path, path: path,
bucket: bucket, bucket: bucket,
db: db, db: db,
mo: mo, mo: mo,
noSync: noSync, noSync: noSync,
fillPercent: fillPercent,
} }
return &rv, nil return &rv, nil
} }

View File

@ -40,6 +40,7 @@ func (w *Writer) ExecuteBatch(batch store.KVBatch) error {
} }
bucket := tx.Bucket([]byte(w.store.bucket)) bucket := tx.Bucket([]byte(w.store.bucket))
bucket.FillPercent = w.store.fillPercent
for k, mergeOps := range emulatedBatch.Merger.Merges { for k, mergeOps := range emulatedBatch.Merger.Merges {
kb := []byte(k) kb := []byte(k)