fix panic in bolt db udpate

cenkalti · Feb 20, 2019 · c33df81 · c33df81
1 parent d53610c
commit c33df81
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/internal/resumer/boltdbresumer/boltdbresumer.go b/internal/resumer/boltdbresumer/boltdbresumer.go
@@ -76,33 +76,30 @@ func (r *Resumer) Write(spec *Spec) error {
 
 func (r *Resumer) WriteInfo(value []byte) error {
  return r.db.Update(func(tx *bolt.Tx) error {
- b := tx.Bucket(r.mainBucket)
+ b := tx.Bucket(r.mainBucket).Bucket(r.subBucket)
  if b == nil {
  return nil
  }
- b = b.Bucket(r.subBucket)
  return b.Put(infoKey, value)
  })
 }
 
 func (r *Resumer) WriteBitfield(value []byte) error {
  return r.db.Update(func(tx *bolt.Tx) error {
- b := tx.Bucket(r.mainBucket)
+ b := tx.Bucket(r.mainBucket).Bucket(r.subBucket)
  if b == nil {
  return nil
  }
- b = b.Bucket(r.subBucket)
  return b.Put(bitfieldKey, value)
  })
 }
 
 func (r *Resumer) WriteStats(s resumer.Stats) error {
  return r.db.Update(func(tx *bolt.Tx) error {
- b := tx.Bucket(r.mainBucket)
+ b := tx.Bucket(r.mainBucket).Bucket(r.subBucket)
  if b == nil {
  return nil
  }
- b = b.Bucket(r.subBucket)
  b.Put(bytesDownloadedKey, []byte(strconv.FormatInt(s.BytesDownloaded, 10)))
  b.Put(bytesUploadedKey, []byte(strconv.FormatInt(s.BytesUploaded, 10)))
  b.Put(bytesWastedKey, []byte(strconv.FormatInt(s.BytesWasted, 10)))