fix(be): boltdb backet validation

semaphoreui · May 13, 2021 · 0e7c8b7 · 0e7c8b7
1 parent b0d0561
commit 0e7c8b7
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/192.168.185.10:3306 b/192.168.185.10:3306
diff --git a/db/bolt/BoltDb.go b/db/bolt/BoltDb.go
@@ -216,6 +216,10 @@ func (d *BoltDb) getObjects(bucketID int, props db.ObjectProperties, params db.R
 	return d.db.View(func(tx *bbolt.Tx) error {
 
 		b := tx.Bucket(makeBucketId(props, bucketID))
+		if b == nil {
+			return db.ErrNotFound
+		}
+
 		c := b.Cursor()
 
 		return unmarshalObjects(c, props, params, filter, objects)

diff --git a/db/factory/store.go b/db/factory/store.go
@@ -2,9 +2,12 @@ package factory
 
 import (
 	"github.com/ansible-semaphore/semaphore/db"
-	"github.com/ansible-semaphore/semaphore/db/sql"
+	"github.com/ansible-semaphore/semaphore/db/bolt"
+
+	//"github.com/ansible-semaphore/semaphore/db/sql"
 )
 
 func CreateStore() db.Store {
-	return &sql.SqlDb{}
+	return &bolt.BoltDb{}
+	//return &sql.SqlDb{}
 }