Skip to content

Commit 8023859

Browse files
Rename tracing_statement_logging to record_stmt_in_spans
1 parent b1ec898 commit 8023859

14 files changed

Lines changed: 110 additions & 122 deletions

File tree

sea-orm-sync/src/database/db_connection.rs

Lines changed: 8 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ impl ConnectionTrait for DatabaseConnection {
147147
"sea_orm.execute",
148148
self.get_database_backend(),
149149
stmt.sql.as_str(),
150-
record_stmt = self.get_tracing_statement_logging(),
150+
record_stmt = self.get_record_stmt_in_spans(),
151151
{
152152
match &self.inner {
153153
#[cfg(feature = "sqlx-mysql")]
@@ -219,7 +219,7 @@ impl ConnectionTrait for DatabaseConnection {
219219
"sea_orm.query_one",
220220
self.get_database_backend(),
221221
stmt.sql.as_str(),
222-
record_stmt = self.get_tracing_statement_logging(),
222+
record_stmt = self.get_record_stmt_in_spans(),
223223
{
224224
match &self.inner {
225225
#[cfg(feature = "sqlx-mysql")]
@@ -249,7 +249,7 @@ impl ConnectionTrait for DatabaseConnection {
249249
"sea_orm.query_all",
250250
self.get_database_backend(),
251251
stmt.sql.as_str(),
252-
record_stmt = self.get_tracing_statement_logging(),
252+
record_stmt = self.get_record_stmt_in_spans(),
253253
{
254254
match &self.inner {
255255
#[cfg(feature = "sqlx-mysql")]
@@ -588,22 +588,16 @@ impl DatabaseConnection {
588588

589589
impl DatabaseConnection {
590590
#[expect(unused)]
591-
pub(crate) fn get_tracing_statement_logging(&self) -> bool {
591+
pub(crate) fn get_record_stmt_in_spans(&self) -> bool {
592592
match &self.inner {
593593
#[cfg(feature = "sqlx-mysql")]
594-
DatabaseConnectionType::SqlxMySqlPoolConnection(conn) => conn.tracing_statement_logging,
594+
DatabaseConnectionType::SqlxMySqlPoolConnection(conn) => conn.record_stmt_in_spans,
595595
#[cfg(feature = "sqlx-postgres")]
596-
DatabaseConnectionType::SqlxPostgresPoolConnection(conn) => {
597-
conn.tracing_statement_logging
598-
}
596+
DatabaseConnectionType::SqlxPostgresPoolConnection(conn) => conn.record_stmt_in_spans,
599597
#[cfg(feature = "sqlx-sqlite")]
600-
DatabaseConnectionType::SqlxSqlitePoolConnection(conn) => {
601-
conn.tracing_statement_logging
602-
}
598+
DatabaseConnectionType::SqlxSqlitePoolConnection(conn) => conn.record_stmt_in_spans,
603599
#[cfg(feature = "rusqlite")]
604-
DatabaseConnectionType::RusqliteSharedConnection(conn) => {
605-
conn.tracing_statement_logging
606-
}
600+
DatabaseConnectionType::RusqliteSharedConnection(conn) => conn.record_stmt_in_spans,
607601
DatabaseConnectionType::Disconnected => true,
608602
#[cfg(feature = "mock")]
609603
DatabaseConnectionType::MockDatabaseConnection(_) => true,

sea-orm-sync/src/database/mod.rs

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ pub struct ConnectOptions {
9191
/// Enable SQLx statement logging
9292
pub(crate) sqlx_logging: bool,
9393
/// Record SQL statements in tracing spans
94-
pub(crate) tracing_statement_logging: bool,
94+
pub(crate) record_stmt_in_spans: bool,
9595
/// SQLx statement logging level (ignored if `sqlx_logging` is false)
9696
pub(crate) sqlx_logging_level: log::LevelFilter,
9797
/// SQLx slow statements logging level (ignored if `sqlx_logging` is false)
@@ -233,7 +233,7 @@ impl ConnectOptions {
233233
acquire_timeout: None,
234234
max_lifetime: None,
235235
sqlx_logging: true,
236-
tracing_statement_logging: true,
236+
record_stmt_in_spans: true,
237237
sqlx_logging_level: log::LevelFilter::Info,
238238
sqlx_slow_statements_logging_level: log::LevelFilter::Off,
239239
sqlx_slow_statements_logging_threshold: Duration::from_secs(1),
@@ -348,14 +348,14 @@ impl ConnectOptions {
348348
}
349349

350350
/// Enable recording `db.statement` in tracing spans (default true).
351-
pub fn tracing_statement_logging(&mut self, value: bool) -> &mut Self {
352-
self.tracing_statement_logging = value;
351+
pub fn record_stmt_in_spans(&mut self, value: bool) -> &mut Self {
352+
self.record_stmt_in_spans = value;
353353
self
354354
}
355355

356356
/// Get whether `db.statement` recording in tracing spans is enabled.
357-
pub fn get_tracing_statement_logging(&self) -> bool {
358-
self.tracing_statement_logging
357+
pub fn get_record_stmt_in_spans(&self) -> bool {
358+
self.record_stmt_in_spans
359359
}
360360

361361
/// Set SQLx statement logging level (default INFO).

sea-orm-sync/src/database/transaction.rs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ pub struct DatabaseTransaction {
2323
backend: DbBackend,
2424
open: bool,
2525
metric_callback: Option<crate::metric::Callback>,
26-
tracing_statement_logging: bool,
26+
record_stmt_in_spans: bool,
2727
}
2828

2929
impl std::fmt::Debug for DatabaseTransaction {
@@ -38,7 +38,7 @@ impl DatabaseTransaction {
3838
conn: Arc<Mutex<InnerConnection>>,
3939
backend: DbBackend,
4040
metric_callback: Option<crate::metric::Callback>,
41-
tracing_statement_logging: bool,
41+
record_stmt_in_spans: bool,
4242
isolation_level: Option<IsolationLevel>,
4343
access_mode: Option<AccessMode>,
4444
sqlite_transaction_mode: Option<SqliteTransactionMode>,
@@ -48,7 +48,7 @@ impl DatabaseTransaction {
4848
backend,
4949
open: true,
5050
metric_callback,
51-
tracing_statement_logging,
51+
record_stmt_in_spans,
5252
};
5353

5454
let begin_result: Result<(), DbErr> = super::tracing_spans::with_db_span!(
@@ -324,7 +324,7 @@ impl ConnectionTrait for DatabaseTransaction {
324324
"sea_orm.execute",
325325
self.backend,
326326
stmt.sql.as_str(),
327-
record_stmt = self.tracing_statement_logging,
327+
record_stmt = self.record_stmt_in_spans,
328328
{
329329
#[cfg(not(feature = "sync"))]
330330
let conn = &mut *self.conn.lock();
@@ -440,7 +440,7 @@ impl ConnectionTrait for DatabaseTransaction {
440440
"sea_orm.query_one",
441441
self.backend,
442442
stmt.sql.as_str(),
443-
record_stmt = self.tracing_statement_logging,
443+
record_stmt = self.record_stmt_in_spans,
444444
{
445445
#[cfg(not(feature = "sync"))]
446446
let conn = &mut *self.conn.lock();
@@ -500,7 +500,7 @@ impl ConnectionTrait for DatabaseTransaction {
500500
"sea_orm.query_all",
501501
self.backend,
502502
stmt.sql.as_str(),
503-
record_stmt = self.tracing_statement_logging,
503+
record_stmt = self.record_stmt_in_spans,
504504
{
505505
#[cfg(not(feature = "sync"))]
506506
let conn = &mut *self.conn.lock();
@@ -587,7 +587,7 @@ impl TransactionTrait for DatabaseTransaction {
587587
Arc::clone(&self.conn),
588588
self.backend,
589589
self.metric_callback.clone(),
590-
self.tracing_statement_logging,
590+
self.record_stmt_in_spans,
591591
None,
592592
None,
593593
None,
@@ -604,7 +604,7 @@ impl TransactionTrait for DatabaseTransaction {
604604
Arc::clone(&self.conn),
605605
self.backend,
606606
self.metric_callback.clone(),
607-
self.tracing_statement_logging,
607+
self.record_stmt_in_spans,
608608
isolation_level,
609609
access_mode,
610610
None,
@@ -620,7 +620,7 @@ impl TransactionTrait for DatabaseTransaction {
620620
Arc::clone(&self.conn),
621621
self.backend,
622622
self.metric_callback.clone(),
623-
self.tracing_statement_logging,
623+
self.record_stmt_in_spans,
624624
options.isolation_level,
625625
options.access_mode,
626626
options.sqlite_transaction_mode,

sea-orm-sync/src/driver/rusqlite.rs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ pub struct RusqliteSharedConnection {
4343
pub(crate) conn: Arc<Mutex<State>>,
4444
acquire_timeout: Duration,
4545
metric_callback: Option<crate::metric::Callback>,
46-
pub(crate) tracing_statement_logging: bool,
46+
pub(crate) record_stmt_in_spans: bool,
4747
}
4848

4949
/// A loaned connection that supports nested transactions.
@@ -176,7 +176,7 @@ impl From<RusqliteConnection> for RusqliteSharedConnection {
176176
conn: Arc::new(Mutex::new(State::Idle(conn))),
177177
acquire_timeout: DEFAULT_ACQUIRE_TIMEOUT,
178178
metric_callback: None,
179-
tracing_statement_logging: true,
179+
record_stmt_in_spans: true,
180180
}
181181
}
182182
}
@@ -198,7 +198,7 @@ impl RusqliteConnector {
198198
#[instrument(level = "trace")]
199199
pub fn connect(options: ConnectOptions) -> Result<DatabaseConnection, DbErr> {
200200
let acquire_timeout = options.acquire_timeout.unwrap_or(DEFAULT_ACQUIRE_TIMEOUT);
201-
let tracing_statement_logging = options.get_tracing_statement_logging();
201+
let record_stmt_in_spans = options.get_record_stmt_in_spans();
202202
// TODO handle disable_statement_logging
203203
let after_conn = options.after_connect;
204204

@@ -247,7 +247,7 @@ impl RusqliteConnector {
247247
conn: Arc::new(Mutex::new(State::Idle(conn))),
248248
acquire_timeout,
249249
metric_callback: None,
250-
tracing_statement_logging,
250+
record_stmt_in_spans,
251251
};
252252

253253
#[cfg(feature = "sqlite-use-returning-for-3_35")]
@@ -424,7 +424,7 @@ impl RusqliteSharedConnection {
424424
Arc::new(Mutex::new(InnerConnection::Rusqlite(conn))),
425425
crate::DbBackend::Sqlite,
426426
self.metric_callback.clone(),
427-
self.tracing_statement_logging,
427+
self.record_stmt_in_spans,
428428
isolation_level,
429429
access_mode,
430430
sqlite_transaction_mode,

sea-orm-sync/src/driver/sqlx_mysql.rs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ pub struct SqlxMySqlConnector;
2929
pub struct SqlxMySqlPoolConnection {
3030
pub(crate) pool: MySqlPool,
3131
metric_callback: Option<crate::metric::Callback>,
32-
pub(crate) tracing_statement_logging: bool,
32+
pub(crate) record_stmt_in_spans: bool,
3333
}
3434

3535
impl std::fmt::Debug for SqlxMySqlPoolConnection {
@@ -43,7 +43,7 @@ impl From<MySqlPool> for SqlxMySqlPoolConnection {
4343
SqlxMySqlPoolConnection {
4444
pool,
4545
metric_callback: None,
46-
tracing_statement_logging: true,
46+
record_stmt_in_spans: true,
4747
}
4848
}
4949
}
@@ -63,7 +63,7 @@ impl SqlxMySqlConnector {
6363
/// Add configuration options for the MySQL database
6464
#[instrument(level = "trace")]
6565
pub fn connect(options: ConnectOptions) -> Result<DatabaseConnection, DbErr> {
66-
let tracing_statement_logging = options.get_tracing_statement_logging();
66+
let record_stmt_in_spans = options.get_record_stmt_in_spans();
6767
let mut sqlx_opts = options
6868
.url
6969
.parse::<MySqlConnectOptions>()
@@ -103,7 +103,7 @@ impl SqlxMySqlConnector {
103103
DatabaseConnectionType::SqlxMySqlPoolConnection(SqlxMySqlPoolConnection {
104104
pool,
105105
metric_callback: None,
106-
tracing_statement_logging,
106+
record_stmt_in_spans,
107107
})
108108
.into();
109109

@@ -121,7 +121,7 @@ impl SqlxMySqlConnector {
121121
DatabaseConnectionType::SqlxMySqlPoolConnection(SqlxMySqlPoolConnection {
122122
pool,
123123
metric_callback: None,
124-
tracing_statement_logging: true,
124+
record_stmt_in_spans: true,
125125
})
126126
.into()
127127
}
@@ -212,7 +212,7 @@ impl SqlxMySqlPoolConnection {
212212
DatabaseTransaction::new_mysql(
213213
conn,
214214
self.metric_callback.clone(),
215-
self.tracing_statement_logging,
215+
self.record_stmt_in_spans,
216216
isolation_level,
217217
access_mode,
218218
)
@@ -234,7 +234,7 @@ impl SqlxMySqlPoolConnection {
234234
let transaction = DatabaseTransaction::new_mysql(
235235
conn,
236236
self.metric_callback.clone(),
237-
self.tracing_statement_logging,
237+
self.record_stmt_in_spans,
238238
isolation_level,
239239
access_mode,
240240
)
@@ -344,15 +344,15 @@ impl crate::DatabaseTransaction {
344344
pub(crate) fn new_mysql(
345345
inner: PoolConnection<sqlx::MySql>,
346346
metric_callback: Option<crate::metric::Callback>,
347-
tracing_statement_logging: bool,
347+
record_stmt_in_spans: bool,
348348
isolation_level: Option<IsolationLevel>,
349349
access_mode: Option<AccessMode>,
350350
) -> Result<crate::DatabaseTransaction, DbErr> {
351351
Self::begin(
352352
Arc::new(Mutex::new(crate::InnerConnection::MySql(inner))),
353353
crate::DbBackend::MySql,
354354
metric_callback,
355-
tracing_statement_logging,
355+
record_stmt_in_spans,
356356
isolation_level,
357357
access_mode,
358358
None,

sea-orm-sync/src/driver/sqlx_postgres.rs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ pub struct SqlxPostgresConnector;
2828
pub struct SqlxPostgresPoolConnection {
2929
pub(crate) pool: PgPool,
3030
metric_callback: Option<crate::metric::Callback>,
31-
pub(crate) tracing_statement_logging: bool,
31+
pub(crate) record_stmt_in_spans: bool,
3232
}
3333

3434
impl std::fmt::Debug for SqlxPostgresPoolConnection {
@@ -42,7 +42,7 @@ impl From<PgPool> for SqlxPostgresPoolConnection {
4242
SqlxPostgresPoolConnection {
4343
pool,
4444
metric_callback: None,
45-
tracing_statement_logging: true,
45+
record_stmt_in_spans: true,
4646
}
4747
}
4848
}
@@ -62,7 +62,7 @@ impl SqlxPostgresConnector {
6262
/// Add configuration options for the PostgreSQL database
6363
#[instrument(level = "trace")]
6464
pub fn connect(options: ConnectOptions) -> Result<DatabaseConnection, DbErr> {
65-
let tracing_statement_logging = options.get_tracing_statement_logging();
65+
let record_stmt_in_spans = options.get_record_stmt_in_spans();
6666
let mut sqlx_opts = options
6767
.url
6868
.parse::<PgConnectOptions>()
@@ -137,7 +137,7 @@ impl SqlxPostgresConnector {
137137
DatabaseConnectionType::SqlxPostgresPoolConnection(SqlxPostgresPoolConnection {
138138
pool,
139139
metric_callback: None,
140-
tracing_statement_logging,
140+
record_stmt_in_spans,
141141
})
142142
.into();
143143

@@ -155,7 +155,7 @@ impl SqlxPostgresConnector {
155155
DatabaseConnectionType::SqlxPostgresPoolConnection(SqlxPostgresPoolConnection {
156156
pool,
157157
metric_callback: None,
158-
tracing_statement_logging: true,
158+
record_stmt_in_spans: true,
159159
})
160160
.into()
161161
}
@@ -246,7 +246,7 @@ impl SqlxPostgresPoolConnection {
246246
DatabaseTransaction::new_postgres(
247247
conn,
248248
self.metric_callback.clone(),
249-
self.tracing_statement_logging,
249+
self.record_stmt_in_spans,
250250
isolation_level,
251251
access_mode,
252252
)
@@ -268,7 +268,7 @@ impl SqlxPostgresPoolConnection {
268268
let transaction = DatabaseTransaction::new_postgres(
269269
conn,
270270
self.metric_callback.clone(),
271-
self.tracing_statement_logging,
271+
self.record_stmt_in_spans,
272272
isolation_level,
273273
access_mode,
274274
)
@@ -379,15 +379,15 @@ impl crate::DatabaseTransaction {
379379
pub(crate) fn new_postgres(
380380
inner: PoolConnection<sqlx::Postgres>,
381381
metric_callback: Option<crate::metric::Callback>,
382-
tracing_statement_logging: bool,
382+
record_stmt_in_spans: bool,
383383
isolation_level: Option<IsolationLevel>,
384384
access_mode: Option<AccessMode>,
385385
) -> Result<crate::DatabaseTransaction, DbErr> {
386386
Self::begin(
387387
Arc::new(Mutex::new(crate::InnerConnection::Postgres(inner))),
388388
crate::DbBackend::Postgres,
389389
metric_callback,
390-
tracing_statement_logging,
390+
record_stmt_in_spans,
391391
isolation_level,
392392
access_mode,
393393
None,

0 commit comments

Comments
 (0)