fix: call query hook when tx is started, committed, or rolled back

README.md

@@ -94,9 +94,9 @@ Projects using Bun:
 
 </details>
 
-## But why?
+## Why another database client?
 
-So you can write queries like this:
+So you can elegantly write complex queries:
 
 ```go
 regionalSales := db.NewSelect().

db.go

@@ -356,7 +356,8 @@ func (db *DB) PrepareContext(ctx context.Context, query string) (Stmt, error) {
 //------------------------------------------------------------------------------
 
 type Tx struct {
-	db *DB
+	ctx context.Context
+	db  *DB
 	*sql.Tx
 }
 
@@ -382,16 +383,33 @@ func (db *DB) Begin() (Tx, error) {
 }
 
 func (db *DB) BeginTx(ctx context.Context, opts *sql.TxOptions) (Tx, error) {
+	ctx, event := db.beforeQuery(ctx, nil, "BEGIN", nil, nil)
 	tx, err := db.DB.BeginTx(ctx, opts)
+	db.afterQuery(ctx, event, nil, err)
 	if err != nil {
 		return Tx{}, err
 	}
 	return Tx{
-		db: db,
-		Tx: tx,
+		ctx: ctx,
+		db:  db,
+		Tx:  tx,
 	}, nil
 }
 
+func (tx Tx) Commit() error {
+	ctx, event := tx.db.beforeQuery(tx.ctx, nil, "COMMIT", nil, nil)
+	err := tx.Tx.Commit()
+	tx.db.afterQuery(ctx, event, nil, err)
+	return err
+}
+
+func (tx Tx) Rollback() error {
+	ctx, event := tx.db.beforeQuery(tx.ctx, nil, "ROLLBACK", nil, nil)
+	err := tx.Tx.Rollback()
+	tx.db.afterQuery(ctx, event, nil, err)
+	return err
+}
+
 func (tx Tx) Exec(query string, args ...interface{}) (sql.Result, error) {
 	return tx.ExecContext(context.TODO(), query, args...)
 }

extra/bundebug/debug.go

@@ -77,7 +77,7 @@ func (h *QueryHook) AfterQuery(ctx context.Context, event *bun.QueryEvent) {
 
 	if !h.verbose {
 		switch event.Err {
-		case nil, sql.ErrNoRows:
+		case nil, sql.ErrNoRows, sql.ErrTxDone:
 			return
 		}
 	}

extra/bunotel/otel.go

@@ -104,7 +104,7 @@ func (h *QueryHook) AfterQuery(ctx context.Context, event *bun.QueryEvent) {
 	}
 
 	switch event.Err {
-	case nil, sql.ErrNoRows:
+	case nil, sql.ErrNoRows, sql.ErrTxDone:
 		// ignore
 	default:
 		span.RecordError(event.Err)