add diskcheck

node/src/node.rs

@@ -34,8 +34,8 @@ use starcoin_rpc_server::module::{PubSubService, PubSubServiceFactory};
 use starcoin_rpc_server::service::RpcService;
 use starcoin_service_registry::bus::{Bus, BusService};
 use starcoin_service_registry::{
-    ActorService, RegistryAsyncService, RegistryService, ServiceContext, ServiceFactory,
-    ServiceHandler, ServiceRef,
+    ActorService, EventHandler, RegistryAsyncService, RegistryService, ServiceContext,
+    ServiceFactory, ServiceHandler, ServiceRef,
 };
 use starcoin_state_service::ChainStateService;
 use starcoin_storage::block_info::BlockInfoStore;
@@ -53,7 +53,7 @@ use starcoin_sync::sync::SyncService;
 use starcoin_sync::txn_sync::TxnSyncService;
 use starcoin_sync::verified_rpc_client::VerifiedRpcClient;
 use starcoin_txpool::TxPoolActorService;
-use starcoin_types::system_events::SystemStarted;
+use starcoin_types::system_events::{SystemShutdown, SystemStarted};
 use std::sync::Arc;
 use std::time::{Duration, SystemTime};
 
@@ -69,7 +69,23 @@ impl ServiceFactory<Self> for NodeService {
     }
 }
 
-impl ActorService for NodeService {}
+impl ActorService for NodeService {
+    fn started(&mut self, ctx: &mut ServiceContext<Self>) -> Result<()> {
+        ctx.subscribe::<SystemShutdown>();
+        Ok(())
+    }
+
+    fn stopped(&mut self, ctx: &mut ServiceContext<Self>) -> Result<()> {
+        ctx.unsubscribe::<SystemShutdown>();
+        Ok(())
+    }
+}
+
+impl EventHandler<Self, SystemShutdown> for NodeService {
+    fn handle_event(&mut self, _: SystemShutdown, ctx: &mut ServiceContext<Self>) {
+        ctx.broadcast(NodeRequest::ShutdownSystem);
+    }
+}
 
 impl ServiceHandler<Self, NodeRequest> for NodeService {
     fn handle(

sync/src/block_connector/block_connector_service.rs

@@ -18,7 +18,7 @@ use starcoin_storage::{BlockStore, Storage};
 use starcoin_sync_api::PeerNewBlock;
 use starcoin_types::block::ExecutedBlock;
 use starcoin_types::sync_status::SyncStatus;
-use starcoin_types::system_events::{MinedBlock, SyncStatusChangeEvent};
+use starcoin_types::system_events::{MinedBlock, SyncStatusChangeEvent, SystemShutdown};
 use std::sync::Arc;
 use sysinfo::{DiskExt, System, SystemExt};
 use txpool::TxPoolService;
@@ -86,7 +86,7 @@ impl BlockConnectorService {
                         } else if DISK_CHECKPOINT_FOR_WARN > disk.available_space() {
                             return Some(Ok(disk.available_space() / 1024 / 1024));
                         }
-                        
+
                         break;
                     }
                 }
@@ -128,9 +128,9 @@ impl ActorService for BlockConnectorService {
         ctx.subscribe::<MinedBlock>();
 
         ctx.run_interval(std::time::Duration::from_secs(3), move |ctx| {
-            ctx.notify(crate::tasks::BlockDiskCheckEvent{});
+            ctx.notify(crate::tasks::BlockDiskCheckEvent {});
         });
-        
+
         Ok(())
     }
 
@@ -145,7 +145,7 @@ impl EventHandler<Self, BlockDiskCheckEvent> for BlockConnectorService {
     fn handle_event(
         &mut self,
         _: BlockDiskCheckEvent,
-        _ctx: &mut ServiceContext<BlockConnectorService>,
+        ctx: &mut ServiceContext<BlockConnectorService>,
     ) {
         if let Some(res) = self.check_disk_space() {
             match res {
@@ -154,13 +154,10 @@ impl EventHandler<Self, BlockDiskCheckEvent> for BlockConnectorService {
                 }
                 Err(e) => {
                     error!("{}", e);
-                    //TODO: exit the starcoin
-                    //ctx.notify(NodeRequest::ShutdownSystem);
+                    ctx.broadcast(SystemShutdown);
                 }
             }
         }
-
-        error!("Check the disk space");
     }
 }
 

sync/src/tasks/mod.rs

@@ -387,7 +387,7 @@ pub struct BlockConnectedEvent {
 }
 
 #[derive(Clone, Debug)]
-pub struct BlockDiskCheckEvent{}
+pub struct BlockDiskCheckEvent {}
 
 pub trait BlockConnectedEventHandle: Send + Clone + std::marker::Unpin {
     fn handle(&mut self, event: BlockConnectedEvent) -> Result<()>;

types/src/system_events.rs

@@ -23,6 +23,9 @@ pub struct MinedBlock(pub Arc<Block>);
 #[derive(Clone, Debug)]
 pub struct SystemStarted;
 
+#[derive(Clone, Debug)]
+pub struct SystemShutdown;
+
 #[derive(Clone, Debug)]
 pub struct SyncStatusChangeEvent(pub SyncStatus);