Skip to content

Commit 8d20221

Browse files
authored
Merge pull request #1671 from ydb-platform/cleanup-database-sql
fixed cleanup for database/sql driver
2 parents 887888a + afc803b commit 8d20221

File tree

2 files changed

+6
-9
lines changed

2 files changed

+6
-9
lines changed

tests/integration/database_sql_metrics_test.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,11 +35,12 @@ func TestDatabaseSqlMetrics(t *testing.T) {
3535
[]string{"database.sql.conns{}", "database.sql.tx{}"},
3636
xslices.Keys(registry.gauges.data),
3737
)
38-
require.EqualValues(t, 1, registry.gauges.data["database.sql.conns{}"].gauges["database.sql.conns{}"].value)
38+
require.NotNil(t, registry.gauges.data["database.sql.conns{}"].gauges)
3939

4040
cc1, err := db.Conn(ctx)
4141
require.NoError(t, err)
4242
require.NotNil(t, cc1)
43+
require.NotNil(t, registry.gauges.data["database.sql.conns{}"].gauges["database.sql.conns{}"])
4344
require.EqualValues(t, 1, registry.gauges.data["database.sql.conns{}"].gauges["database.sql.conns{}"].value)
4445
require.Empty(t, registry.gauges.data["database.sql.tx{}"].gauges)
4546

tests/integration/helpers_test.go

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -145,21 +145,17 @@ func (scope *scopeT) driverNamed(name string, opts ...ydb.Option) *ydb.Driver {
145145
func (scope *scopeT) SQLDriver(opts ...ydb.ConnectorOption) *sql.DB {
146146
f := func() (*fixenv.GenericResult[*sql.DB], error) {
147147
driver := scope.Driver()
148-
scope.Logf("Create sql db connector")
148+
scope.Logf("Create database/sql connector for YDB")
149149
connector, err := ydb.Connector(driver, opts...)
150150
if err != nil {
151151
return nil, err
152152
}
153153

154154
db := sql.OpenDB(connector)
155155

156-
scope.Logf("Ping db")
157-
err = db.PingContext(scope.Ctx)
158-
if err != nil {
159-
return nil, err
160-
}
161-
162-
return fixenv.NewGenericResult(db), nil
156+
return fixenv.NewGenericResultWithCleanup(db, func() {
157+
scope.Require.NoError(db.Close())
158+
}), nil
163159
}
164160
return fixenv.CacheResult(scope.Env, f)
165161
}

0 commit comments

Comments
 (0)