diff options
Diffstat (limited to 'libpod/boltdb_state.go')
-rw-r--r-- | libpod/boltdb_state.go | 52 |
1 files changed, 30 insertions, 22 deletions
diff --git a/libpod/boltdb_state.go b/libpod/boltdb_state.go index afbaecffb..49bdaea39 100644 --- a/libpod/boltdb_state.go +++ b/libpod/boltdb_state.go @@ -5,6 +5,7 @@ import ( "encoding/json" "os" "strings" + "sync" "github.com/boltdb/bolt" "github.com/pkg/errors" @@ -15,6 +16,7 @@ import ( type BoltState struct { valid bool dbPath string + dbLock sync.Mutex namespace string namespaceBytes []byte lockDir string @@ -73,6 +75,12 @@ func NewBoltState(path, lockDir string, runtime *Runtime) (State, error) { if err != nil { return nil, errors.Wrapf(err, "error opening database %s", path) } + // Everywhere else, we use s.closeDBCon(db) to ensure the state's DB + // mutex is also unlocked. + // However, here, the mutex has not been locked, since we just created + // the DB connection, and it hasn't left this function yet - no risk of + // concurrent access. + // As such, just a db.Close() is fine here. defer db.Close() // Perform initial database setup @@ -133,7 +141,7 @@ func (s *BoltState) Refresh() error { if err != nil { return err } - defer db.Close() + defer s.closeDBCon(db) err = db.Update(func(tx *bolt.Tx) error { idBucket, err := getIDBucket(tx) @@ -266,7 +274,7 @@ func (s *BoltState) Container(id string) (*Container, error) { if err != nil { return nil, err } - defer db.Close() + defer s.closeDBCon(db) err = db.View(func(tx *bolt.Tx) error { ctrBucket, err := getCtrBucket(tx) @@ -302,7 +310,7 @@ func (s *BoltState) LookupContainer(idOrName string) (*Container, error) { if err != nil { return nil, err } - defer db.Close() + defer s.closeDBCon(db) err = db.View(func(tx *bolt.Tx) error { idBucket, err := getIDBucket(tx) @@ -390,7 +398,7 @@ func (s *BoltState) HasContainer(id string) (bool, error) { if err != nil { return false, err } - defer db.Close() + defer s.closeDBCon(db) exists := false @@ -455,7 +463,7 @@ func (s *BoltState) RemoveContainer(ctr *Container) error { if err != nil { return err } - defer db.Close() + defer s.closeDBCon(db) err = db.Update(func(tx *bolt.Tx) error { return s.removeContainer(ctr, nil, tx) @@ -486,7 +494,7 @@ func (s *BoltState) UpdateContainer(ctr *Container) error { if err != nil { return err } - defer db.Close() + defer s.closeDBCon(db) err = db.View(func(tx *bolt.Tx) error { ctrBucket, err := getCtrBucket(tx) @@ -555,7 +563,7 @@ func (s *BoltState) SaveContainer(ctr *Container) error { if err != nil { return err } - defer db.Close() + defer s.closeDBCon(db) err = db.Update(func(tx *bolt.Tx) error { ctrBucket, err := getCtrBucket(tx) @@ -612,7 +620,7 @@ func (s *BoltState) ContainerInUse(ctr *Container) ([]string, error) { if err != nil { return nil, err } - defer db.Close() + defer s.closeDBCon(db) err = db.View(func(tx *bolt.Tx) error { ctrBucket, err := getCtrBucket(tx) @@ -663,7 +671,7 @@ func (s *BoltState) AllContainers() ([]*Container, error) { if err != nil { return nil, err } - defer db.Close() + defer s.closeDBCon(db) err = db.View(func(tx *bolt.Tx) error { allCtrsBucket, err := getAllCtrsBucket(tx) @@ -732,7 +740,7 @@ func (s *BoltState) Pod(id string) (*Pod, error) { if err != nil { return nil, err } - defer db.Close() + defer s.closeDBCon(db) err = db.View(func(tx *bolt.Tx) error { podBkt, err := getPodBucket(tx) @@ -767,7 +775,7 @@ func (s *BoltState) LookupPod(idOrName string) (*Pod, error) { if err != nil { return nil, err } - defer db.Close() + defer s.closeDBCon(db) err = db.View(func(tx *bolt.Tx) error { idBucket, err := getIDBucket(tx) @@ -859,7 +867,7 @@ func (s *BoltState) HasPod(id string) (bool, error) { if err != nil { return false, err } - defer db.Close() + defer s.closeDBCon(db) err = db.View(func(tx *bolt.Tx) error { podBkt, err := getPodBucket(tx) @@ -915,7 +923,7 @@ func (s *BoltState) PodHasContainer(pod *Pod, id string) (bool, error) { if err != nil { return false, err } - defer db.Close() + defer s.closeDBCon(db) err = db.View(func(tx *bolt.Tx) error { podBkt, err := getPodBucket(tx) @@ -977,7 +985,7 @@ func (s *BoltState) PodContainersByID(pod *Pod) ([]string, error) { if err != nil { return nil, err } - defer db.Close() + defer s.closeDBCon(db) err = db.View(func(tx *bolt.Tx) error { podBkt, err := getPodBucket(tx) @@ -1039,7 +1047,7 @@ func (s *BoltState) PodContainers(pod *Pod) ([]*Container, error) { if err != nil { return nil, err } - defer db.Close() + defer s.closeDBCon(db) err = db.View(func(tx *bolt.Tx) error { podBkt, err := getPodBucket(tx) @@ -1123,7 +1131,7 @@ func (s *BoltState) AddPod(pod *Pod) error { if err != nil { return err } - defer db.Close() + defer s.closeDBCon(db) err = db.Update(func(tx *bolt.Tx) error { podBkt, err := getPodBucket(tx) @@ -1232,7 +1240,7 @@ func (s *BoltState) RemovePod(pod *Pod) error { if err != nil { return err } - defer db.Close() + defer s.closeDBCon(db) err = db.Update(func(tx *bolt.Tx) error { podBkt, err := getPodBucket(tx) @@ -1327,7 +1335,7 @@ func (s *BoltState) RemovePodContainers(pod *Pod) error { if err != nil { return err } - defer db.Close() + defer s.closeDBCon(db) err = db.Update(func(tx *bolt.Tx) error { podBkt, err := getPodBucket(tx) @@ -1488,7 +1496,7 @@ func (s *BoltState) RemoveContainerFromPod(pod *Pod, ctr *Container) error { if err != nil { return err } - defer db.Close() + defer s.closeDBCon(db) err = db.Update(func(tx *bolt.Tx) error { return s.removeContainer(ctr, pod, tx) @@ -1516,7 +1524,7 @@ func (s *BoltState) UpdatePod(pod *Pod) error { if err != nil { return err } - defer db.Close() + defer s.closeDBCon(db) podID := []byte(pod.ID()) @@ -1576,7 +1584,7 @@ func (s *BoltState) SavePod(pod *Pod) error { if err != nil { return err } - defer db.Close() + defer s.closeDBCon(db) podID := []byte(pod.ID()) @@ -1618,7 +1626,7 @@ func (s *BoltState) AllPods() ([]*Pod, error) { if err != nil { return nil, err } - defer db.Close() + defer s.closeDBCon(db) err = db.View(func(tx *bolt.Tx) error { allPodsBucket, err := getAllPodsBucket(tx) |