micro-wrapper-sql/tx.go

package wrapper

import (
	"context"
	"database/sql/driver"
	"time"

	"go.unistack.org/micro/v3/tracer"
)

var _ driver.Tx = &wrapperTx{}

// wrapperTx defines a wrapper for driver.Tx
type wrapperTx struct {
	tx   driver.Tx
	span tracer.Span
	opts Options
	conn *wrapperConn
	ctx  context.Context
}

// Commit implements driver.Tx Commit
func (w *wrapperTx) Commit() error {
	if w.span != nil {
		defer w.span.Finish()
	}
	ts := time.Now()
	err := w.tx.Commit()
	td := time.Since(ts)

	if err != nil {
		w.span.AddLabels("error", true)
		w.span.AddLabels("err", err.Error())
	}

	if w.opts.LoggerEnabled {
		w.opts.Logger.Fields(w.opts.LoggerObserver(w.ctx, "Commit", labelUnknown, td, err)...).Log(w.ctx, w.opts.LoggerLevel)
	}

	w.ctx = nil
	w.conn.ctx = nil

	return err
}

// Rollback implements driver.Tx Rollback
func (w *wrapperTx) Rollback() error {
	if w.span != nil {
		defer w.span.Finish()
	}
	ts := time.Now()
	err := w.tx.Rollback()
	td := time.Since(ts)

	if err != nil {
		w.span.AddLabels("error", true)
		w.span.AddLabels("err", err.Error())
	}

	if w.opts.LoggerEnabled {
		w.opts.Logger.Fields(w.opts.LoggerObserver(w.ctx, "Rollback", labelUnknown, td, err)...).Log(w.ctx, w.opts.LoggerLevel)
	}

	w.ctx = nil
	w.conn.ctx = nil

	return err
}