changed protobuf schema and stub methods

Comcast · Mar 19, 2019 · a2282a3 · a2282a3
1 parent 5dab43a
commit a2282a3
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 0 deletions.
diff --git a/api/protos/service.proto b/api/protos/service.proto
@@ -101,6 +101,10 @@ enum Op {
   Remove = 4;
   // Returns bool
   SafeToRemove = 5;
+  // Sets to maintenance mode. Returns OpResult
+  SetMaintenance = 7;
+  // Unsets maintenance mode Returns OpResult
+  UnsetMaintenance = 8;
 }
 
 // Datacenter related API's

diff --git a/src/disk_manager.rs b/src/disk_manager.rs
@@ -226,11 +226,43 @@ fn listen(
                     }
                 };
             }
+
+            Op::SetMaintenance => {
+                match set_maintenance_mode(&mut responder) {
+                    Ok(_) => {
+                        info!("Set maintenance operation finished");
+                    }
+                    Err(e) => {
+                        error!("Error when setting to maintenance mode: {:?}", e);
+                    }
+                };
+            }
+
+            Op::UnsetMaintenance => {
+                match unset_maintenance_mode(&mut responder) {
+                    Ok(_) => {
+                        info!("Unset maintenance operation finished");
+                    }
+                    Err(e) => {
+                        error!("Error when unsetting the maintenance mode: {:?}", e);
+                    }
+                };
+            }
         };
         thread::sleep(Duration::from_millis(10));
     }
 }
 
+fn set_maintenance_mode(sock: &mut Socket) -> BynarResult<()> {
+    //TODO implement the feature
+    Ok(())
+}
+
+fn unset_maintenance_mode(sock: &mut Socket) -> BynarResult<()> {
+    //TODO implement the feature
+    Ok(())
+}
+
 fn respond_to_client(result: &OpResult, s: &mut Socket) -> BynarResult<()> {
     let encoded = result.write_to_bytes()?;
     let msg = Message::from_slice(&encoded)?;