feat(cli): add prof and tokio env config (#3824)

Signed-off-by: Alex Chi <iskyzh@gmail.com>

Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>

src/cmd/src/bin/compactor.rs

@@ -20,14 +20,12 @@ use tikv_jemallocator::Jemalloc;
 static GLOBAL: Jemalloc = Jemalloc;
 
 #[cfg_attr(coverage, no_coverage)]
-#[tokio::main]
-async fn main() {
+fn main() {
     use clap::StructOpt;
 
     let opts = risingwave_compactor::CompactorOpts::parse();
 
-    risingwave_rt::oneshot_common();
     risingwave_rt::init_risingwave_logger(risingwave_rt::LoggerSettings::new_default());
 
-    risingwave_compactor::start(opts).await;
+    risingwave_rt::main_okk(risingwave_compactor::start(opts))
 }

src/cmd/src/bin/compute_node.rs

@@ -20,17 +20,15 @@ use tikv_jemallocator::Jemalloc;
 static GLOBAL: Jemalloc = Jemalloc;
 
 #[cfg_attr(coverage, no_coverage)]
-#[tokio::main]
-async fn main() {
+fn main() {
     use clap::StructOpt;
 
     let opts = risingwave_compute::ComputeNodeOpts::parse();
 
-    risingwave_rt::oneshot_common();
     risingwave_rt::init_risingwave_logger(risingwave_rt::LoggerSettings::new(
         opts.enable_jaeger_tracing,
         false,
     ));
 
-    risingwave_compute::start(opts).await
+    risingwave_rt::main_okk(risingwave_compute::start(opts))
 }

src/cmd/src/bin/ctl.rs

@@ -21,15 +21,12 @@ use tikv_jemallocator::Jemalloc;
 static GLOBAL: Jemalloc = Jemalloc;
 
 #[cfg_attr(coverage, no_coverage)]
-#[tokio::main]
-async fn main() -> Result<()> {
+fn main() -> Result<()> {
     use clap::StructOpt;
 
     let opts = risingwave_ctl::CliOpts::parse();
 
-    risingwave_rt::oneshot_common();
     risingwave_rt::init_risingwave_logger(risingwave_rt::LoggerSettings::new_default());
-    // console_subscriber::init();a
 
-    risingwave_ctl::start(opts).await
+    risingwave_rt::main_okk(risingwave_ctl::start(opts))
 }

src/cmd/src/bin/frontend_node.rs

@@ -20,14 +20,12 @@ use tikv_jemallocator::Jemalloc;
 static GLOBAL: Jemalloc = Jemalloc;
 
 #[cfg_attr(coverage, no_coverage)]
-#[tokio::main(flavor = "multi_thread", worker_threads = 4)]
-async fn main() {
+fn main() {
     use clap::StructOpt;
 
     let opts = risingwave_frontend::FrontendOpts::parse();
 
-    risingwave_rt::oneshot_common();
     risingwave_rt::init_risingwave_logger(risingwave_rt::LoggerSettings::new_default());
 
-    risingwave_frontend::start(opts).await
+    risingwave_rt::main_okk(risingwave_frontend::start(opts))
 }

src/cmd/src/bin/meta_node.rs

@@ -20,14 +20,12 @@ use tikv_jemallocator::Jemalloc;
 static GLOBAL: Jemalloc = Jemalloc;
 
 #[cfg_attr(coverage, no_coverage)]
-#[tokio::main]
-async fn main() {
+fn main() {
     use clap::StructOpt;
 
     let opts = risingwave_meta::MetaNodeOpts::parse();
 
-    risingwave_rt::oneshot_common();
     risingwave_rt::init_risingwave_logger(risingwave_rt::LoggerSettings::new_default());
 
-    risingwave_meta::start(opts).await
+    risingwave_rt::main_okk(risingwave_meta::start(opts))
 }