rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-01-03 20:02:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: make tls certificates/keys reloadable (part 1) (#3335)

Browse Source 
* feat: make tls certificates/keys reloadable (part 1)

* feat: add notify watcher for cert/key files

* test: add unit test for watcher

* fix: correct usage of watcher

* fix: skip watch when tls disabled
This commit is contained in:
Ning Sun
2024-02-26 17:37:54 +08:00
committed by GitHub
parent e859f0e67d
commit 3887d207b6
10 changed files with 309 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests-integration/src/test_util.rs
View File

@@ -50,6 +50,7 @@ use servers::postgres::PostgresServer;
use servers::query_handler::grpc::ServerGrpcQueryHandlerAdapter;
use servers::query_handler::sql::{ServerSqlQueryHandlerAdapter, SqlQueryHandler};
use servers::server::Server;
use servers::tls::ReloadableTlsServerConfig;
use servers::Mode;
use session::context::QueryContext;
@@ -568,7 +569,10 @@ pub async fn setup_mysql_server_with_user_provider(
)),
Arc::new(MysqlSpawnConfig::new(
false,
opts.tls.setup().unwrap().map(Arc::new),
Arc::new(
ReloadableTlsServerConfig::try_new(opts.tls.clone())
.expect("Failed to load certificates and keys"),
),
opts.reject_no_database.unwrap_or(false),
)),
));
@@ -614,9 +618,15 @@ pub async fn setup_pg_server_with_user_provider(
addr: fe_pg_addr.clone(),
..Default::default()
};
let tls_server_config = Arc::new(
ReloadableTlsServerConfig::try_new(opts.tls.clone())
.expect("Failed to load certificates and keys"),
);
let fe_pg_server = Arc::new(Box::new(PostgresServer::new(
ServerSqlQueryHandlerAdapter::arc(fe_instance_ref),
opts.tls.clone(),
opts.tls.should_force_tls(),
tls_server_config,
runtime,
user_provider,
)) as Box<dyn Server>);