pkg/util/context.go - develop - pyo/updater

package util

import (
	"context"

	"github.com/jackc/pgx/v5"
)

func TxToContext(ctx context.Context, tx pgx.Tx) context.Context {
	return context.WithValue(ctx, txCtxPoolKey, tx)
}

type txCtxKey struct{}

var txCtxPoolKey = txCtxKey{}

func TxFromCtx(ctx context.Context) pgx.Tx {
	tx, ok := ctx.Value(txCtxPoolKey).(pgx.Tx)
	if ok {
		return tx
	}
	return nil
}