Put timeout on opening bolt

master
Bel LaPointe 2020-07-23 20:07:43 -06:00
parent 4416a7e4b3
commit d04ced1e33
3 changed files with 10 additions and 2 deletions

View File

@ -2,7 +2,7 @@
port=57017
if ! curl -sS localhost:$port > /dev/null; then
if false && ! curl -sS localhost:$port > /dev/null; then
prefix=/tmp/whodunit.db
mkdir -p $prefix/data
mongod \
@ -26,3 +26,6 @@ remove() {
}
export DBURI=${DB_URI:-"mongodb://localhost:$port"}
export DRIVERTYPE=boltdb
export DBURI=$(mktemp)

View File

@ -9,6 +9,7 @@ import (
"os"
"regexp"
"strings"
"time"
"github.com/boltdb/bolt"
"go.mongodb.org/mongo-driver/bson"
@ -25,7 +26,9 @@ type BoltDB struct {
func NewBoltDB() *BoltDB {
config := config.New()
db, err := bolt.Open(config.DBURI, os.ModePerm, nil)
db, err := bolt.Open(config.DBURI, os.ModePerm, &bolt.Options{
Timeout: time.Second * 3,
})
if err != nil {
panic(err)
}

View File

@ -25,6 +25,8 @@ func TestJSON(t *testing.T) {
}
os.Args = os.Args[:1]
t.Logf("config: %+v", config.New())
g := storage.NewGraph()
ones := fillDB(t, g)
want := ones[len(ones)-1]