1. Fixed mongo goroutine leak

2. Fixed linters error

gorm/example_test.go

@@ -20,10 +20,11 @@ import (
 // Example demonstrates the implementation of the Repository pattern by trm.Manager.
 func Example() {
 	db, err := gorm.Open(sqlite.Open("file:test.db?mode=memory"))
-
 	checkErr(err)
+
 	sqlDB, err := db.DB()
 	checkErr(err)
+
 	defer sqlDB.Close()
 
 	// Migrate the schema

gorm/factory.go

@@ -17,7 +17,7 @@ func NewDefaultFactory(db *gorm.DB) trm.TrFactory {
 		s, _ := trms.(Settings)
 
 		// TODO do update TRM config by settings gorm nested transaction
-		//db.DisableNestedTransaction = true
+		// db.DisableNestedTransaction = true
 
 		return NewTransaction(ctx, s.TxOpts(), db)
 	}

gorm/goroutine_leak_test.go

@@ -0,0 +1,14 @@
+//go:build go1.20
+// +build go1.20
+
+package gorm
+
+import (
+	"testing"
+
+	"go.uber.org/goleak"
+)
+
+func TestMain(m *testing.M) {
+	goleak.VerifyTestMain(m)
+}

gorm/transaction.go

@@ -18,10 +18,10 @@ import (
 
 // Transaction is trm.Transaction for sqlx.Tx.
 type Transaction struct {
-	tx            *gorm.DB
-	txMutex       sync.Mutex
-	active        *drivers.IsClose
-	activeClosure *drivers.IsClose
+	tx              *gorm.DB
+	txMutex         sync.Mutex
+	isClosed        *drivers.IsClosed
+	isClosedClosure *drivers.IsClosed
 }
 
 // NewTransaction creates trm.Transaction for sqlx.Tx.
@@ -31,10 +31,10 @@ func NewTransaction(
 	db *gorm.DB,
 ) (context.Context, *Transaction, error) {
 	t := &Transaction{
-		tx:            nil,
-		txMutex:       sync.Mutex{},
-		active:        drivers.NewIsClosed(),
-		activeClosure: drivers.NewIsClosed(),
+		tx:              nil,
+		txMutex:         sync.Mutex{},
+		isClosed:        drivers.NewIsClosed(),
+		isClosedClosure: drivers.NewIsClosed(),
 	}
 
 	var err error
@@ -50,9 +50,9 @@ func NewTransaction(
 
 			wg.Done()
 
-			<-t.activeClosure.Closed()
+			<-t.isClosedClosure.Closed()
 
-			return t.activeClosure.Err()
+			return t.isClosedClosure.Err()
 		}, opts)
 
 		t.txMutex.Lock()
@@ -67,7 +67,7 @@ func NewTransaction(
 				err = t.tx.Error
 			}
 
-			t.active.CloseWithCause(err)
+			t.isClosed.CloseWithCause(err)
 		} else {
 			wg.Done()
 		}
@@ -92,8 +92,8 @@ func (t *Transaction) awaitDone(ctx context.Context) {
 	select {
 	case <-ctx.Done():
 		// Rollback will be called by context.Err()
-		t.activeClosure.Close()
-	case <-t.active.Closed():
+		t.isClosedClosure.Close()
+	case <-t.isClosed.Closed():
 	}
 }
 
@@ -114,34 +114,34 @@ func (t *Transaction) Begin(ctx context.Context, s trm.Settings) (context.Contex
 // Commit closes the trm.Transaction.
 func (t *Transaction) Commit(_ context.Context) error {
 	select {
-	case <-t.active.Closed():
+	case <-t.isClosed.Closed():
 		t.txMutex.Lock()
 		defer t.txMutex.Unlock()
 
 		return t.tx.Commit().Error
 	default:
-		t.activeClosure.Close()
+		t.isClosedClosure.Close()
 
-		<-t.active.Closed()
+		<-t.isClosed.Closed()
 
-		return t.active.Err()
+		return t.isClosed.Err()
 	}
 }
 
 // Rollback the trm.Transaction.
 func (t *Transaction) Rollback(_ context.Context) error {
 	select {
-	case <-t.active.Closed():
+	case <-t.isClosed.Closed():
 		t.txMutex.Lock()
 		defer t.txMutex.Unlock()
 
 		return t.tx.Rollback().Error
 	default:
-		t.activeClosure.CloseWithCause(drivers.ErrRollbackTr)
+		t.isClosedClosure.CloseWithCause(drivers.ErrRollbackTr)
 
-		<-t.active.Closed()
+		<-t.isClosed.Closed()
 
-		err := t.active.Err()
+		err := t.isClosed.Err()
 		if errors.Is(err, drivers.ErrRollbackTr) {
 			return nil
 		}
@@ -152,9 +152,10 @@ func (t *Transaction) Rollback(_ context.Context) error {
 
 // IsActive returns true if the transaction started but not committed or rolled back.
 func (t *Transaction) IsActive() bool {
-	return t.active.IsActive()
+	return t.isClosed.IsActive()
 }
 
+// Closed returns a channel that's closed when transaction committed or rolled back.
 func (t *Transaction) Closed() <-chan struct{} {
-	return t.active.Closed()
+	return t.isClosed.Closed()
 }

gorm/transaction_test.go

@@ -13,7 +13,6 @@ import (
 	"github.com/DATA-DOG/go-sqlmock"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
-	"go.uber.org/goleak"
 	"gorm.io/driver/mysql"
 	"gorm.io/gorm"
 
@@ -25,10 +24,6 @@ import (
 	"github.com/avito-tech/go-transaction-manager/trm/settings"
 )
 
-func TestMain(m *testing.M) {
-	goleak.VerifyTestMain(m)
-}
-
 func TestTransaction(t *testing.T) {
 	t.Parallel()
 
@@ -272,10 +267,11 @@ func TestTransaction_awaitDone_byRollback(t *testing.T) {
 	require.NoError(t, err)
 
 	f := NewDefaultFactory(dbgorm)
-	ctx, _ := context.WithCancel(context.Background())
+	ctx, _ := context.WithCancel(context.Background()) //nolint:govet
 
 	wg := sync.WaitGroup{}
 	wg.Add(1)
+
 	go func() {
 		defer wg.Done()
 

internal/test/sql.go

@@ -8,12 +8,18 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
+// NewDBMock returns a new sql.DB and sqlmock.
+//
+//nolint:ireturn
 func NewDBMock() (*sql.DB, sqlmock.Sqlmock) {
 	db, dbmock, _ := sqlmock.New()
 
 	return db, dbmock
 }
 
+// NewDBMockWithClose returns a new sql.DB and sqlmock, and close it after test.
+//
+//nolint:ireturn
 func NewDBMockWithClose(t *testing.T) (*sql.DB, sqlmock.Sqlmock) {
 	db, dbmock := NewDBMock()
 

mongo/goroutine_leak_test.go

@@ -0,0 +1,14 @@
+//go:build go1.20
+// +build go1.20
+
+package mongo
+
+import (
+	"testing"
+
+	"go.uber.org/goleak"
+)
+
+func TestMain(m *testing.M) {
+	goleak.VerifyTestMain(m)
+}

mongo/transaction.go

@@ -3,16 +3,17 @@ package mongo
 
 import (
 	"context"
-	"sync/atomic"
 
 	"go.mongodb.org/mongo-driver/mongo"
 	"go.mongodb.org/mongo-driver/mongo/options"
+
+	"github.com/avito-tech/go-transaction-manager/trm/drivers"
 )
 
 // Transaction is trm.Transaction for mongo.Client.
 type Transaction struct {
 	session  mongo.Session
-	isActive int64
+	isClosed *drivers.IsClosed
 }
 
 // NewTransaction creates trm.Transaction for mongo.Client.
@@ -33,7 +34,7 @@ func NewTransaction(
 		return ctx, nil, err
 	}
 
-	tr := &Transaction{session: s, isActive: 1}
+	tr := &Transaction{session: s, isClosed: drivers.NewIsClosed()}
 
 	go tr.awaitDone(ctx)
 
@@ -45,9 +46,11 @@ func (t *Transaction) awaitDone(ctx context.Context) {
 		return
 	}
 
-	<-ctx.Done()
-
-	t.deactivate()
+	select {
+	case <-ctx.Done():
+		t.isClosed.Close()
+	case <-t.isClosed.Closed():
+	}
 }
 
 // Transaction returns the real transaction mongo.Session.
@@ -57,7 +60,7 @@ func (t *Transaction) Transaction() interface{} {
 
 // Commit the trm.Transaction.
 func (t *Transaction) Commit(ctx context.Context) error {
-	defer t.deactivate()
+	defer t.isClosed.Close()
 
 	defer t.session.EndSession(ctx)
 
@@ -66,7 +69,7 @@ func (t *Transaction) Commit(ctx context.Context) error {
 
 // Rollback the trm.Transaction.
 func (t *Transaction) Rollback(ctx context.Context) error {
-	defer t.deactivate()
+	defer t.isClosed.Close()
 
 	defer t.session.EndSession(ctx)
 
@@ -75,9 +78,10 @@ func (t *Transaction) Rollback(ctx context.Context) error {
 
 // IsActive returns true if the transaction started but not committed or rolled back.
 func (t *Transaction) IsActive() bool {
-	return atomic.LoadInt64(&t.isActive) == 1
+	return t.isClosed.IsActive()
 }
 
-func (t *Transaction) deactivate() {
-	atomic.SwapInt64(&t.isActive, 0)
+// Closed returns a channel that's closed when transaction committed or rolled back.
+func (t *Transaction) Closed() <-chan struct{} {
+	return t.isClosed.Closed()
 }

mongo/transaction_test.go

@@ -14,7 +14,6 @@ import (
 	"go.mongodb.org/mongo-driver/mongo/integration/mtest"
 	"go.mongodb.org/mongo-driver/mongo/options"
 	"go.mongodb.org/mongo-driver/mongo/writeconcern"
-	"go.uber.org/goleak"
 
 	"github.com/avito-tech/go-transaction-manager/internal/mock"
 	"github.com/avito-tech/go-transaction-manager/trm"
@@ -27,10 +26,6 @@ type user struct {
 	ID primitive.ObjectID `bson:"_id,omitempty"`
 }
 
-func TestMain(m *testing.M) {
-	goleak.VerifyTestMain(m)
-}
-
 func TestTransaction(t *testing.T) {
 	t.Parallel()
 
@@ -169,8 +164,10 @@ func TestTransaction(t *testing.T) {
 				manager.WithSettings(f.settings),
 			)
 
-			var tr Transaction
+			var tr trm.Transaction
 			err := m.Do(tt.args.ctx, func(ctx context.Context) error {
+				tr = trmcontext.DefaultManager.Default(ctx)
+
 				var trNested trm.Transaction
 				err := m.Do(ctx, func(ctx context.Context) error {
 					trNested = trmcontext.DefaultManager.Default(ctx)
@@ -186,7 +183,9 @@ func TestTransaction(t *testing.T) {
 
 				return err
 			})
-			require.False(t, false, tr.IsActive())
+			if tr != nil {
+				require.False(t, tr.IsActive())
+			}
 
 			if !tt.wantErr(t, err) {
 				return