caisin/sea-orm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

cargo fmt

Browse Source
This commit is contained in:
Billy Chan 2022-06-28 16:13:38 +08:00
parent 6bc76698cc
commit 80d35a9d18
No known key found for this signature in database
GPG Key ID: A2D690CAC7DF3CC7
3 changed files with 38 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/database/stream/metric.rs
View File

@ -1,8 +1,8 @@
use std::{time::Duration, pin::Pin, task::Poll};
use std::{pin::Pin, task::Poll, time::Duration};
use futures::Stream;
use crate::{QueryResult, DbErr, Statement};
use crate::{DbErr, QueryResult, Statement};
pub(crate) struct MetricStream<'a> {
metric_callback: &'a Option<crate::metric::Callback>,
@ -12,7 +12,12 @@ pub(crate) struct MetricStream<'a> {
}
impl<'a> MetricStream<'a> {
pub(crate) fn new<S>(metric_callback: &'a Option<crate::metric::Callback>, stmt: &'a Statement, elapsed: Option<Duration>, stream: S) -> Self
pub(crate) fn new<S>(
metric_callback: &'a Option<crate::metric::Callback>,
stmt: &'a Statement,
elapsed: Option<Duration>,
stream: S,
) -> Self
where
S: Stream<Item = Result<QueryResult, DbErr>> + 'a + Send,
{
@ -33,7 +38,10 @@ impl<'a> Stream for MetricStream<'a> {
cx: &mut std::task::Context<'_>,
) -> Poll<Option<Self::Item>> {
let this = self.get_mut();
let _start = this.metric_callback.is_some().then(std::time::SystemTime::now);
let _start = this
.metric_callback
.is_some()
.then(std::time::SystemTime::now);
let res = Pin::new(&mut this.stream).poll_next(cx);
if let (Some(_start), Some(elapsed)) = (_start, &mut this.elapsed) {
*elapsed += _start.elapsed().unwrap_or_default();

11
src/database/stream/query.rs
View File

@ -127,7 +127,8 @@ impl QueryStream {
InnerConnection::MySql(c) => {
let query = crate::driver::sqlx_mysql::sqlx_query(stmt);
let _start = _metric_callback.is_some().then(SystemTime::now);
let stream = c.fetch(query)
let stream = c
.fetch(query)
.map_ok(Into::into)
.map_err(crate::sqlx_error_to_query_err);
let elapsed = _start.map(|s| s.elapsed().unwrap_or_default());
@ -137,7 +138,8 @@ impl QueryStream {
InnerConnection::Postgres(c) => {
let query = crate::driver::sqlx_postgres::sqlx_query(stmt);
let _start = _metric_callback.is_some().then(SystemTime::now);
let stream = c.fetch(query)
let stream = c
.fetch(query)
.map_ok(Into::into)
.map_err(crate::sqlx_error_to_query_err);
let elapsed = _start.map(|s| s.elapsed().unwrap_or_default());
@ -147,7 +149,8 @@ impl QueryStream {
InnerConnection::Sqlite(c) => {
let query = crate::driver::sqlx_sqlite::sqlx_query(stmt);
let _start = _metric_callback.is_some().then(SystemTime::now);
let stream = c.fetch(query)
let stream = c
.fetch(query)
.map_ok(Into::into)
.map_err(crate::sqlx_error_to_query_err);
let elapsed = _start.map(|s| s.elapsed().unwrap_or_default());
@ -159,7 +162,7 @@ impl QueryStream {
let stream = c.fetch(stmt);
let elapsed = _start.map(|s| s.elapsed().unwrap_or_default());
MetricStream::new(_metric_callback, stmt, elapsed, stream)
},
}
#[allow(unreachable_patterns)]
_ => unreachable!(),
},

11
src/database/stream/transaction.rs
View File

@ -51,7 +51,8 @@ impl<'a> TransactionStream<'a> {
InnerConnection::MySql(c) => {
let query = crate::driver::sqlx_mysql::sqlx_query(stmt);
let _start = _metric_callback.is_some().then(SystemTime::now);
let stream = c.fetch(query)
let stream = c
.fetch(query)
.map_ok(Into::into)
.map_err(crate::sqlx_error_to_query_err);
let elapsed = _start.map(|s| s.elapsed().unwrap_or_default());
@ -61,7 +62,8 @@ impl<'a> TransactionStream<'a> {
InnerConnection::Postgres(c) => {
let query = crate::driver::sqlx_postgres::sqlx_query(stmt);
let _start = _metric_callback.is_some().then(SystemTime::now);
let stream = c.fetch(query)
let stream = c
.fetch(query)
.map_ok(Into::into)
.map_err(crate::sqlx_error_to_query_err);
let elapsed = _start.map(|s| s.elapsed().unwrap_or_default());
@ -71,7 +73,8 @@ impl<'a> TransactionStream<'a> {
InnerConnection::Sqlite(c) => {
let query = crate::driver::sqlx_sqlite::sqlx_query(stmt);
let _start = _metric_callback.is_some().then(SystemTime::now);
let stream = c.fetch(query)
let stream = c
.fetch(query)
.map_ok(Into::into)
.map_err(crate::sqlx_error_to_query_err);
let elapsed = _start.map(|s| s.elapsed().unwrap_or_default());
@ -83,7 +86,7 @@ impl<'a> TransactionStream<'a> {
let stream = c.fetch(stmt);
let elapsed = _start.map(|s| s.elapsed().unwrap_or_default());
MetricStream::new(_metric_callback, stmt, elapsed, stream)
},
}
#[allow(unreachable_patterns)]
_ => unreachable!(),
},