uqbar-dao/nectar
1
1
Fork 0
mirror of https://github.com/uqbar-dao/nectar.git synced 2025-01-02 21:55:41 +03:00
Code Issues Packages Projects Releases Wiki Activity

fd_manager: add helpers

Browse Source
This commit is contained in:
hosted-fornet 2024-09-26 12:14:09 -07:00
parent a4ad477fe9
commit c3e01d2800
1 changed files with 40 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
kinode/src/fd_manager.rs
View File

@ -1,5 +1,5 @@
use lib::types::core::{
FdManagerError, FdManagerRequest, KernelMessage, Message, MessageReceiver, MessageSender,
Address, FdManagerError, FdManagerRequest, KernelMessage, Message, MessageReceiver, MessageSender,
PrintSender, Printout, ProcessId, Request, FD_MANAGER_PROCESS_ID,
};
use std::{collections::HashMap, sync::Arc};
@ -178,8 +178,8 @@ async fn send_cull(
for process_id in state.fds.keys() {
KernelMessage::builder()
.id(rand::random())
.source((our_node.clone(), FD_MANAGER_PROCESS_ID.clone()))
.target((our_node.clone(), process_id.clone()))
.source((our_node, FD_MANAGER_PROCESS_ID.clone()))
.target((our_node, process_id.clone()))
.message(message.clone())
.build()
.unwrap()
@ -202,3 +202,40 @@ fn get_max_fd_limit() -> anyhow::Result<u64> {
Err(anyhow::anyhow!("Failed to get the resource limit."))
}
}
pub async fn send_fd_manager_open(our: &Address, number_opened: u64, send_to_loop: &MessageSender) -> anyhow::Result<()> {
let message = Message::Request(Request {
inherit: false,
expects_response: None,
body: serde_json::to_vec(&FdManagerRequest::OpenFds { number_opened }).unwrap(),
metadata: None,
capabilities: vec![],
});
send_to_fd_manager(our, message, send_to_loop).await?;
Ok(())
}
pub async fn send_fd_manager_close(our: &Address, number_closed: u64, send_to_loop: &MessageSender) -> anyhow::Result<()> {
let message = Message::Request(Request {
inherit: false,
expects_response: None,
body: serde_json::to_vec(&FdManagerRequest::CloseFds { number_closed }).unwrap(),
metadata: None,
capabilities: vec![],
});
send_to_fd_manager(our, message, send_to_loop).await?;
Ok(())
}
async fn send_to_fd_manager(our: &Address, message: Message, send_to_loop: &MessageSender) -> anyhow::Result<()> {
KernelMessage::builder()
.id(rand::random())
.source(our.clone())
.target((our.node.clone(), FD_MANAGER_PROCESS_ID.clone()))
.message(message)
.build()
.unwrap()
.send(send_to_loop)
.await;
Ok(())
}