Call RegisterTxDb from NewConnection if needed

core/services/pg/connection.go

@@ -78,7 +78,11 @@ func NewConnection(ctx context.Context, uri string, dialect pgcommon.DialectName
 		// We can happily throw away the original uri here because if we are using
 		// txdb it should have already been set at the point where we called
 		// txdb.Register
-		var err error
+
+		err := pgcommon.RegisterTxDb(ctx, uri)
+		if err != nil {
+			return nil, fmt.Errorf("failed to register txdb: %w", err)
+		}
 		sqldb, err = otelsql.Open(string(dialect), uuid.New().String(), opts...)
 		if err != nil {
 			return nil, fmt.Errorf("failed to open txdb: %w", err)