fix: Connections leaks in database connections

Author: xuzifu666

pkg/datasource/sql/conn_at_test.go

@@ -116,6 +116,7 @@ func TestATConn_ExecContext(t *testing.T) {
 
 		conn, err := db.Conn(context.Background())
 		assert.NoError(t, err)
+		defer conn.Close()
 
 		_, err = conn.ExecContext(ctx, "SELECT 1")
 		assert.NoError(t, err)
@@ -139,6 +140,7 @@ func TestATConn_ExecContext(t *testing.T) {
 
 		conn, err := db.Conn(context.Background())
 		assert.NoError(t, err)
+		defer conn.Close()
 
 		_, err = conn.ExecContext(context.Background(), "SELECT 1")
 		assert.NoError(t, err)

pkg/datasource/sql/conn_xa_test.go

@@ -199,6 +199,7 @@ func TestXAConn_ExecContext(t *testing.T) {
 
 		conn, err := db.Conn(context.Background())
 		assert.NoError(t, err)
+		defer conn.Close()
 
 		_, err = conn.ExecContext(ctx, "SELECT 1")
 		assert.NoError(t, err)
@@ -225,6 +226,7 @@ func TestXAConn_ExecContext(t *testing.T) {
 
 		conn, err := db.Conn(context.Background())
 		assert.NoError(t, err)
+		defer conn.Close()
 
 		_, err = conn.ExecContext(context.Background(), "SELECT 1")
 		assert.NoError(t, err)

pkg/datasource/sql/datasource/mysql/meta_cache.go

@@ -65,6 +65,7 @@ func (c *TableMetaCache) GetTableMeta(ctx context.Context, dbName, tableName str
 		return nil, err
 	}
 
+	defer conn.Close()
 	tableMeta, err := c.tableMetaCache.GetTableMeta(ctx, dbName, tableName, conn)
 	if err != nil {
 		return nil, err

pkg/datasource/sql/db.go

@@ -223,6 +223,7 @@ func (db *DBResource) ConnectionForXA(ctx context.Context, xaXid XAXid) (*XAConn
 	if err != nil {
 		return nil, fmt.Errorf("create xa resoruce err:%w", err)
 	}
+	defer newDriverConn.Close()
 	xaConn := &XAConn{
 		Conn: &Conn{
 			targetConn: newDriverConn,

pkg/datasource/sql/driver_test.go

@@ -91,6 +91,7 @@ func Test_seataATDriver_Open(t *testing.T) {
 	conn, err := db.Conn(context.Background())
 	assert.NotNil(t, err)
 	assert.Nil(t, conn)
+	conn.Close()
 }
 
 func Test_seataATDriver_OpenConnector(t *testing.T) {

pkg/datasource/sql/undo/base/undo.go

@@ -253,6 +253,7 @@ func (m *BaseUndoLogManager) Undo(ctx context.Context, dbType types.DBType, xid
 	if err != nil {
 		return err
 	}
+	defer conn.Close()
 
 	tx, err := conn.BeginTx(ctx, &sql.TxOptions{})
 	if err != nil {

pkg/datasource/sql/undo_test.go

@@ -39,6 +39,7 @@ func TestBatchDeleteUndoLogs(t *testing.T) {
 
 		sqlConn, err := db.Conn(context.Background())
 		assert.Nil(t, err)
+		defer sqlConn.Close()
 
 		undoLogManager := new(base.BaseUndoLogManager)
 
@@ -86,6 +87,7 @@ func TestHasUndoLogTable(t *testing.T) {
 		ctx := context.Background()
 		sqlConn, err := db.Conn(ctx)
 		assert.Nil(t, err)
+		defer sqlConn.Close()
 
 		undoLogManager := new(base.BaseUndoLogManager)