proto/sgx/app.proto

@@ -37,8 +37,8 @@ message NewAppReq {
 
 message AppResource {
   uint32 memory_mb = 1;
-  uint32 disk_size_gb = 2;
-  uint32 vcpus  = 3;
+  uint32 disk_mb = 2;
+  uint32 vcpu  = 3;
   repeated uint32 ports = 4;
 }
 
@@ -65,7 +65,7 @@ message ListAppContractsReq {
 message AppNodeFilters {
   uint32 vcpus = 1;
   uint32 memory_mb = 2;
-  uint32 storage_gb = 3;
+  uint32 storage_mb = 3;
   string country = 4;
   string region = 5;
   string city = 6;
@@ -111,7 +111,7 @@ message AppNodeResources {
   uint32 avail_no_of_port = 2;
   uint32 avail_vcpus = 3;
   uint32 avail_memory_mb = 4;
-  uint32 avail_storage_gb = 5;
+  uint32 avail_storage_mb = 5;
   uint32 max_ports_per_app = 6;
 }
 
@@ -138,7 +138,7 @@ message DaemonMessageApp {
 }
 
 service BrainAppDaemon {
-  rpc RegisterAppNode (RegisterAppNodeReq) returns (stream DelAppReq);
+  rpc RegisterAppNode (RegisterAppNodeReq) returns (stream AppContract);
   rpc BrainMessages (DaemonAuth) returns (stream BrainMessageApp);
   rpc DaemonMessages (stream DaemonMessageApp) returns (common_proto.Empty);
 }

src/sgx/types/brain.rs

@@ -6,8 +6,8 @@ use serde::{Deserialize, Serialize};
 #[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Default)]
 pub struct Resource {
     pub memory_mb: u32,
-    pub disk_size_gb: u32,
-    pub vcpus: u32,
+    pub disk_mb: u32,
+    pub vcpu: u32,
     pub port: Vec<u32>,
 }
 
@@ -15,8 +15,8 @@ impl From<AppResource> for Resource {
     fn from(pb_val: AppResource) -> Self {
         Self {
             memory_mb: pb_val.memory_mb,
-            disk_size_gb: pb_val.disk_size_gb,
-            vcpus: pb_val.vcpus,
+            disk_mb: pb_val.disk_mb,
+            vcpu: pb_val.vcpu,
             port: pb_val.ports,
         }
     }
@@ -25,8 +25,8 @@ impl From<Resource> for AppResource {
     fn from(val: Resource) -> AppResource {
         AppResource {
             memory_mb: val.memory_mb,
-            disk_size_gb: val.disk_size_gb,
-            vcpus: val.vcpus,
+            disk_mb: val.disk_mb,
+            vcpu: val.vcpu,
             ports: val.port,
         }
     }