updated diskwriter node

Author: Calvin Leng

synapse/client/nodes/disk_writer.py

@@ -6,12 +6,13 @@
 class DiskWriter(Node):
     type = NodeType.kDiskWriter
 
-    def __init__(self, filename: str):
+    def __init__(self, filename: str, storage_device_id: int):
         self.filename = filename
+        self.storage_device_id = storage_device_id
 
     def _to_proto(self):
         n = NodeConfig()
-        p = DiskWriterConfig(filename=self.filename)
+        p = DiskWriterConfig(filename=self.filename, storage_device_id=self.storage_device_id)
         n.disk_writer.CopyFrom(p)
         return n
 
@@ -22,4 +23,4 @@ def _from_proto(proto: DiskWriterConfig):
         if not isinstance(proto, DiskWriterConfig):
             raise ValueError("proto is not of type DiskWriterConfig")
 
-        return DiskWriter(filename=proto.filename)
+        return DiskWriter(filename=proto.filename, storage_device_id=proto.storage_device_id)