Use once_cell instead of lazy_static (#416)

`once_cell` provides a neat way of initializing lazy singletons without
macro. This PR use `sync::Lazy` to streamline same pattern proposed in
related rust RFC.

Resolve #406

Cargo.toml

@@ -33,12 +33,12 @@ crossbeam-utils = "0.6.6"
 futures-core-preview = "=0.3.0-alpha.19"
 futures-io-preview = "=0.3.0-alpha.19"
 futures-timer = "1.0.2"
-lazy_static = "1.4.0"
 log = { version = "0.4.8", features = ["kv_unstable"] }
 memchr = "2.2.1"
 mio = "0.6.19"
 mio-uds = "0.6.7"
 num_cpus = "1.10.1"
+once_cell = "1.2.0"
 pin-utils = "0.1.0-alpha.4"
 slab = "0.4.2"
 kv-log-macro = "1.0.4"

src/net/driver/mod.rs

@@ -1,8 +1,8 @@
 use std::fmt;
 use std::sync::{Arc, Mutex};
 
-use lazy_static::lazy_static;
 use mio::{self, Evented};
+use once_cell::sync::Lazy;
 use slab::Slab;
 
 use crate::io;
@@ -100,9 +100,8 @@ impl Reactor {
     // }
 }
 
-lazy_static! {
 /// The state of the global networking driver.
-    static ref REACTOR: Reactor = {
+static REACTOR: Lazy<Reactor> = Lazy::new(|| {
     // Spawn a thread that waits on the poller for new events and wakes up tasks blocked on I/O
     // handles.
     std::thread::Builder::new()
@@ -117,8 +116,7 @@ lazy_static! {
         .expect("cannot start a thread driving blocking tasks");
 
     Reactor::new().expect("cannot initialize reactor")
-    };
+});
-}
 
 /// Waits on the poller for new events and wakes up tasks blocked on I/O handles.
 fn main_loop() -> io::Result<()> {

src/task/blocking.rs

@@ -5,7 +5,7 @@ use std::thread;
 use std::time::Duration;
 
 use crossbeam_channel::{bounded, Receiver, Sender};
-use lazy_static::lazy_static;
+use once_cell::sync::Lazy;
 
 use crate::task::task::{JoinHandle, Tag};
 use crate::utils::abort_on_panic;
@@ -19,16 +19,17 @@ struct Pool {
     receiver: Receiver<async_task::Task<Tag>>,
 }
 
-lazy_static! {
+static POOL: Lazy<Pool> = Lazy::new(|| {
-    static ref POOL: Pool = {
     for _ in 0..2 {
         thread::Builder::new()
             .name("async-blocking-driver".to_string())
-                .spawn(|| abort_on_panic(|| {
+            .spawn(|| {
+                abort_on_panic(|| {
                     for task in &POOL.receiver {
                         task.run();
                     }
-                }))
+                })
+            })
             .expect("cannot start a thread driving blocking tasks");
     }
 
@@ -41,8 +42,7 @@ lazy_static! {
     // reducing bufferbloat.
     let (sender, receiver) = bounded(0);
     Pool { sender, receiver }
-    };
+});
-}
 
 // Create up to MAX_THREADS dynamic blocking task worker threads.
 // Dynamic threads will terminate themselves if they don't

src/task/pool.rs

@@ -3,7 +3,7 @@ use std::thread;
 
 use crossbeam_deque::{Injector, Stealer, Worker};
 use kv_log_macro::trace;
-use lazy_static::lazy_static;
+use once_cell::sync::Lazy;
 
 use super::sleepers::Sleepers;
 use super::task;
@@ -111,8 +111,7 @@ impl Pool {
 
 #[inline]
 pub(crate) fn get() -> &'static Pool {
-    lazy_static! {
+    static POOL: Lazy<Pool> = Lazy::new(|| {
-        static ref POOL: Pool = {
         let num_threads = num_cpus::get().max(1);
         let mut stealers = Vec::new();
 
@@ -132,7 +131,6 @@ pub(crate) fn get() -> &'static Pool {
             stealers,
             sleepers: Sleepers::new(),
         }
-        };
+    });
-    }
     &*POOL
 }

src/task/task_local.rs

@@ -5,7 +5,7 @@ use std::future::Future;
 use std::sync::atomic::{AtomicUsize, Ordering};
 use std::sync::Mutex;
 
-use lazy_static::lazy_static;
+use once_cell::sync::Lazy;
 
 use super::worker;
 use crate::utils::abort_on_panic;
@@ -174,9 +174,7 @@ impl<T: Send + 'static> LocalKey<T> {
     fn key(&self) -> usize {
         #[cold]
         fn init(key: &AtomicUsize) -> usize {
-            lazy_static! {
+            static COUNTER: Lazy<Mutex<usize>> = Lazy::new(|| Mutex::new(1));
-                static ref COUNTER: Mutex<usize> = Mutex::new(1);
-            }
 
             let mut counter = COUNTER.lock().unwrap();
             let prev = key.compare_and_swap(0, *counter, Ordering::AcqRel);