ghe0/brain-mock-clients-mocks
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

created update functionality?? sequel

Browse Source
This commit is contained in:
Ramil_Algayev 2024-12-24 19:07:18 +04:00
parent 5e916803b3
commit c0e5acbf24
4 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
cli-mock/brain.proto
View File

@ -97,7 +97,8 @@ service BrainDaemonService {
rpc SendVMConfirmations (stream NewVMConfirmation) returns (Empty); rpc SendVMConfirmations (stream NewVMConfirmation) returns (Empty);
rpc DeletedVMUpdates (NodePubkey) returns (stream DeletedVMUpdate); rpc DeletedVMUpdates (NodePubkey) returns (stream DeletedVMUpdate);
rpc ListVMContracts (ListVMContractsReq) returns (stream VMContract); rpc ListVMContracts (ListVMContractsReq) returns (stream VMContract);
rpc UpdateVMs (NodePubkey) returns (stream UpdateVMRequest); rpc GetUpdateVM (NodePubkey) returns (stream UpdateVMRequest);
rpc SendUpdateVM (stream UpdateVMResp) returns (Empty);
} }
message NodeFilters { message NodeFilters {

4
cli-mock/src/main.rs
View File

@ -138,8 +138,8 @@ async fn delete_vm(mut client: BrainCliServiceClient<Channel>, uuid: &str) -> Re
async fn update_vm(mut client: BrainCliServiceClient<Channel>, uuid: &str) -> Result<UpdateVmResp> { async fn update_vm(mut client: BrainCliServiceClient<Channel>, uuid: &str) -> Result<UpdateVmResp> {
info!("Updatind VM {uuid}"); info!("Updatind VM {uuid}");
let req = UpdateVmRequest { let req = UpdateVmRequest {
uuid: String::new(), uuid: uuid.to_string(),
node_pubkey: String::new(), node_pubkey: SECURE_PUBLIC_KEY.to_string(),
vcpus: 4, vcpus: 4,
memory_mb: 4096, memory_mb: 4096,
disk_size_gb: 40, disk_size_gb: 40,

3
daemon-mock/brain.proto
View File

@ -97,7 +97,8 @@ service BrainDaemonService {
rpc SendVMConfirmations (stream NewVMConfirmation) returns (Empty); rpc SendVMConfirmations (stream NewVMConfirmation) returns (Empty);
rpc DeletedVMUpdates (NodePubkey) returns (stream DeletedVMUpdate); rpc DeletedVMUpdates (NodePubkey) returns (stream DeletedVMUpdate);
rpc ListVMContracts (ListVMContractsReq) returns (stream VMContract); rpc ListVMContracts (ListVMContractsReq) returns (stream VMContract);
rpc UpdateVMs (NodePubkey) returns (stream UpdateVMRequest); rpc GetUpdateVM (NodePubkey) returns (stream UpdateVMRequest);
rpc SendUpdateVM (stream UpdateVMResp) returns (Empty);
} }
message NodeFilters { message NodeFilters {

3
daemon-mock/src/main.rs
View File

@ -123,7 +123,7 @@ async fn listen_for_update_vm_reqs(
debug!("starting listen_for_update_vm_reqs"); debug!("starting listen_for_update_vm_reqs");
let node_pubkey = SECURE_PUBLIC_KEY.clone(); let node_pubkey = SECURE_PUBLIC_KEY.clone();
let mut grpc_stream = client let mut grpc_stream = client
.update_v_ms(NodePubkey { node_pubkey }) .get_update_vm(NodePubkey { node_pubkey })
.await? .await?
.into_inner(); .into_inner();
while let Some(stream_update) = grpc_stream.next().await { while let Some(stream_update) = grpc_stream.next().await {
@ -147,6 +147,7 @@ async fn handle_update_vm_requests(
) { ) {
info!("Started to handle update vm requests."); info!("Started to handle update vm requests.");
while let Some(update_vm) = req.recv().await { while let Some(update_vm) = req.recv().await {
info!("Updating vm: {update_vm:?}");
let response = UpdateVmResp { let response = UpdateVmResp {
uuid: update_vm.uuid.clone(), uuid: update_vm.uuid.clone(),
timestamp: chrono::Utc::now().to_rfc3339(), timestamp: chrono::Utc::now().to_rfc3339(),