insert/update functions

database.go

@@ -13,6 +13,7 @@ var (
 	ErrNoResults        = errors.New("no results found") // ErrNoResults indicates query found no results
 	ErrInvalidTableName = errors.New("invalid table")    // ErrInvalidTableName indicates that specified table does not exist
 	ErrNoConnection     = errors.New("no db connection") //
+	ErrExists           = errors.New("key existss")
 	db                  *bbolt.DB
 )
 
@@ -73,6 +74,27 @@ func Save(value any, key, table string) error {
 	})
 }
 
+// Insert saves a value only if key does not exist
+func Insert(value any, key, table string) error {
+	_, err := Get[any](key, table)
+	if err == nil {
+		return ErrExists
+	}
+	if errors.Is(err, ErrNoResults) {
+		return Save(value, key, table)
+	}
+	return err
+}
+
+// Update save a value only if key already exists
+func Update(value any, key, table string) error {
+	_, err := Get[any](key, table)
+	if errors.Is(err, ErrExists) {
+		return Save(value, key, table)
+	}
+	return err
+}
+
 // Get retrieves a value for key in specified table
 func Get[T any](key, table string) (T, error) {
 	var value T

database_test.go

@@ -1,6 +1,7 @@
 package boltdb
 
 import (
+	"errors"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -64,9 +65,61 @@ func TestSave(t *testing.T) {
 		assert.NotNil(t, err)
 	})
 	t.Run("valid", func(t *testing.T) {
-		err := Save("testing", "name", "users")
+		user := User{
+			User: "testing",
+		}
+		err := Save(user, user.User, "users")
+		assert.Nil(t, err)
+	})
+	deleteTestEntries(t)
+	err = Close()
+	assert.Nil(t, err)
+}
+
+func TestInsert(t *testing.T) {
+	err := Initialize("./test.db", tables)
+	assert.Nil(t, err)
+	deleteTestEntries(t)
+	t.Run("valid", func(t *testing.T) {
+		user := User{
+			User: "testing",
+		}
+		err := Insert(user, user.User, "users")
+		assert.Nil(t, err)
+	})
+	t.Run("exists", func(t *testing.T) {
+		user := User{
+			User: "testing",
+		}
+		err := Insert(user, user.User, "users")
+		assert.True(t, errors.Is(err, ErrExists))
+	})
+	deleteTestEntries(t)
+	err = Close()
+	assert.Nil(t, err)
+}
+
+func TestUpdate(t *testing.T) {
+	err := Initialize("./test.db", tables)
+	assert.Nil(t, err)
+	deleteTestEntries(t)
+	t.Run("does not exist", func(t *testing.T) {
+		user := User{
+			User: "testing",
+		}
+		err := Update(user, user.User, "users")
+		assert.True(t, errors.Is(err, ErrNoResults))
+	})
+	t.Run("existing", func(t *testing.T) {
+		user := User{
+			User: "testing",
+		}
+		err := Save(user, user.User, "users")
+		assert.Nil(t, err)
+		err = Update(user, user.User, "users")
 		assert.Nil(t, err)
 	})
+	deleteTestEntries(t)
 	err = Close()
 	assert.Nil(t, err)
 }