Просмотр исходного кода

Merge pull request #130 from tuna/add-bolt-open-timeout

Add 5 seconds timeout for bolt
Yuxiang Zhang 5 лет назад
Родитель
Сommit
b34238c097
1 измененных файлов с 4 добавлено и 1 удалено
  1. 4 1
      manager/db.go

+ 4 - 1
manager/db.go

@@ -4,6 +4,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"strings"
+	"time"
 
 	"github.com/boltdb/bolt"
 
@@ -26,7 +27,9 @@ type dbAdapter interface {
 
 func makeDBAdapter(dbType string, dbFile string) (dbAdapter, error) {
 	if dbType == "bolt" {
-		innerDB, err := bolt.Open(dbFile, 0600, nil)
+		innerDB, err := bolt.Open(dbFile, 0600, &bolt.Options{
+			Timeout: 5 * time.Second,
+		})
 		if err != nil {
 			return nil, err
 		}