From af6665fe4f9e8fa6d7a9d4ad85907d59f27ddb56 Mon Sep 17 00:00:00 2001
From: Chris Tsang <tyt2y7@gmail.com>
Date: Mon, 4 Oct 2021 12:12:58 +0800
Subject: [PATCH] Fix clippy warnings

---
 src/database/connection.rs   | 5 ++++-
 src/database/stream/query.rs | 5 ++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/database/connection.rs b/src/database/connection.rs
index d2283caf..2c65fa28 100644
--- a/src/database/connection.rs
+++ b/src/database/connection.rs
@@ -3,7 +3,10 @@ use crate::{
     StatementBuilder, TransactionError,
 };
 use sea_query::{MysqlQueryBuilder, PostgresQueryBuilder, QueryBuilder, SqliteQueryBuilder};
-use std::{future::Future, pin::Pin, sync::Arc};
+use std::{future::Future, pin::Pin};
+
+#[cfg(feature = "mock")]
+use std::sync::Arc;
 
 #[cfg_attr(not(feature = "mock"), derive(Clone))]
 pub enum DatabaseConnection {
diff --git a/src/database/stream/query.rs b/src/database/stream/query.rs
index 73668da0..8383659a 100644
--- a/src/database/stream/query.rs
+++ b/src/database/stream/query.rs
@@ -1,4 +1,7 @@
-use std::{pin::Pin, sync::Arc, task::Poll};
+use std::{pin::Pin, task::Poll};
+
+#[cfg(feature = "mock")]
+use std::sync::Arc;
 
 use futures::Stream;
 #[cfg(feature = "sqlx-dep")]