diff --git a/src/client/src/lib.rs b/src/client/src/lib.rs index 713faba658df..0741c8e1c7a6 100644 --- a/src/client/src/lib.rs +++ b/src/client/src/lib.rs @@ -16,12 +16,12 @@ mod client; pub mod client_manager; +#[cfg(feature = "testing")] +mod database; pub mod error; pub mod load_balance; mod metrics; pub mod region; -#[cfg(feature = "testing")] -mod database; pub use api; use api::v1::greptime_response::Response; diff --git a/src/cmd/src/cli.rs b/src/cmd/src/cli.rs index 9933704ba32d..fc443a169d2b 100644 --- a/src/cmd/src/cli.rs +++ b/src/cmd/src/cli.rs @@ -31,8 +31,8 @@ use async_trait::async_trait; use bench::BenchTableMetadataCommand; use clap::Parser; use common_telemetry::logging::{LoggingOptions, TracingOptions}; -use tracing_appender::non_blocking::WorkerGuard; pub use repl::Repl; +use tracing_appender::non_blocking::WorkerGuard; use upgrade::UpgradeCommand; use self::export::ExportCommand; diff --git a/src/cmd/src/cli/export.rs b/src/cmd/src/cli/export.rs index f3781c6844dc..00d916407458 100644 --- a/src/cmd/src/cli/export.rs +++ b/src/cmd/src/cli/export.rs @@ -17,10 +17,14 @@ use std::path::Path; use std::sync::Arc; use async_trait::async_trait; -use base64::Engine; use base64::engine::general_purpose; +use base64::Engine; use clap::{Parser, ValueEnum}; +use client::DEFAULT_SCHEMA_NAME; +use common_telemetry::{debug, error, info, warn}; use serde_json::Value; +use servers::http::greptime_result_v1::GreptimedbV1Response; +use servers::http::GreptimeQueryOutput; use snafu::{OptionExt, ResultExt}; use tokio::fs::File; use tokio::io::{AsyncWriteExt, BufWriter}; @@ -28,11 +32,6 @@ use tokio::sync::Semaphore; use tokio::time::Instant; use tracing_appender::non_blocking::WorkerGuard; -use client::DEFAULT_SCHEMA_NAME; -use common_telemetry::{debug, error, info, warn}; -use servers::http::greptime_result_v1::GreptimedbV1Response; -use servers::http::GreptimeQueryOutput; - use crate::cli::{Instance, Tool}; use crate::error::{ EmptyResultSnafu, Error, FileIoSnafu, HttpQuerySqlSnafu, InvalidDatabaseNameSnafu, Result, @@ -439,14 +438,13 @@ fn split_database(database: &str) -> Result<(String, Option)> { #[cfg(test)] mod tests { use clap::Parser; - use client::{Client, Database}; use common_catalog::consts::{DEFAULT_CATALOG_NAME, DEFAULT_SCHEMA_NAME}; use common_telemetry::logging::LoggingOptions; - use crate::{App, cli, standalone}; use crate::error::Result as CmdResult; use crate::options::GlobalOptions; + use crate::{cli, standalone, App}; #[tokio::test(flavor = "multi_thread")] async fn test_export_create_table_with_quoted_names() -> CmdResult<()> { diff --git a/tests-integration/src/test_util.rs b/tests-integration/src/test_util.rs index 35ce52e273bd..3b96809a40fe 100644 --- a/tests-integration/src/test_util.rs +++ b/tests-integration/src/test_util.rs @@ -39,6 +39,7 @@ use datanode::config::{ use frontend::instance::Instance; use frontend::service_config::{MysqlOptions, PostgresOptions}; use futures::future::BoxFuture; +use client::Database; use object_store::services::{Azblob, Gcs, Oss, S3}; use object_store::test_util::TempFolder; use object_store::ObjectStore; @@ -56,7 +57,6 @@ use servers::tls::ReloadableTlsServerConfig; use servers::Mode; use session::context::QueryContext; -use crate::database::Database; use crate::standalone::{GreptimeDbStandalone, GreptimeDbStandaloneBuilder}; pub const PEER_PLACEHOLDER_ADDR: &str = "127.0.0.1:3001"; diff --git a/tests-integration/tests/grpc.rs b/tests-integration/tests/grpc.rs index a7956726cbf1..7d1f9d57768f 100644 --- a/tests-integration/tests/grpc.rs +++ b/tests-integration/tests/grpc.rs @@ -22,7 +22,7 @@ use api::v1::{ PromqlRequest, RequestHeader, SemanticType, }; use auth::user_provider_from_option; -use client::{Client, OutputData, DEFAULT_CATALOG_NAME, DEFAULT_SCHEMA_NAME, Database}; +use client::{Client, Database, OutputData, DEFAULT_CATALOG_NAME, DEFAULT_SCHEMA_NAME}; use common_catalog::consts::MITO_ENGINE; use common_grpc::channel_manager::ClientTlsOption; use common_query::Output;