Add View and Update methods

db.go

@@ -25,6 +25,36 @@ func (db DB) Begin(writable bool) (*Tx, error) {
 	}, nil
 }
 
+func (db DB) View(fn func(tx *Tx) error) error {
+	tx, err := db.Begin(false)
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback()
+
+	err = fn(tx)
+	if err != nil {
+		return err
+	}
+
+	return tx.Rollback()
+}
+
+func (db DB) Update(fn func(tx *Tx) error) error {
+	tx, err := db.Begin(true)
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback()
+
+	err = fn(tx)
+	if err != nil {
+		return err
+	}
+
+	return tx.Commit()
+}
+
 type Tx struct {
 	engine.Transaction
 }

db_test.go

@@ -11,13 +11,10 @@ import (
 )
 
 func TestTable(t *testing.T) {
+	t.Run("Table/Insert/NoIndex", func(t *testing.T) {
 		db := genji.New(memory.NewEngine())
 
-	t.Run("Table/Insert/NoIndex", func(t *testing.T) {
-		tx, err := db.Begin(true)
-		require.NoError(t, err)
-		defer tx.Rollback()
-
+		err := db.Update(func(tx *genji.Tx) error {
 			tb, err := tx.CreateTable("test")
 			require.NoError(t, err)
 
@@ -31,13 +28,17 @@ func TestTable(t *testing.T) {
 			m, err := tx.Indexes("test")
 			require.NoError(t, err)
 			require.Empty(t, m)
+
+			return nil
+		})
+		require.NoError(t, err)
 	})
 
 	t.Run("Table/Insert/WithIndex", func(t *testing.T) {
-		tx, err := db.Begin(true)
-		require.NoError(t, err)
-		defer tx.Rollback()
+		db := genji.New(memory.NewEngine())
+		defer db.Close()
 
+		err := db.Update(func(tx *genji.Tx) error {
 			tb, err := tx.CreateTable("test")
 			require.NoError(t, err)
 
@@ -59,8 +60,9 @@ func TestTable(t *testing.T) {
 			v, rid := c.Seek([]byte("John"))
 			require.Equal(t, []byte("John"), v)
 			require.Equal(t, rowid, rid)
-	})
 
-	err := db.Close()
+			return nil
+		})
 		require.NoError(t, err)
+	})
 }