p3 starter code

p3/client.py

+import grpc, threading, os, sys
+import table_pb2_grpc, table_pb2
+
+SERVER = "localhost:5440"
+
+class TableClient:
+    def __init__(self, file, num_threads=1):
+        self.channel = grpc.insecure_channel(SERVER)
+        self.stub = table_pb2_grpc.TableStub(channel=self.channel)
+        self.num_threads = num_threads
+        self.read_workload_file(file)
+
+    def Purge(self):
+        resp = self.stub.Purge(table_pb2.PurgeReq())
+
+    def Upload(self, path):
+        with open(path, "rb") as f:
+            data = f.read()
+            resp = self.stub.Upload(table_pb2.UploadReq(csv_data=data))
+            print(resp.error if resp.error else "success")
+
+    def ColSum(self, format, column_name):
+        resp = self.stub.ColSum(table_pb2.ColSumReq(column=column_name, format=format))
+        print(resp.error if resp.error else f"{resp.total}")
+
+    def exec_command(self, line):
+        command = line.strip().split(" ")
+        if len(command) == 1:
+            self.Purge()
+        elif len(command) == 2:
+            self.Upload(command[1])
+        elif len(command) == 3:
+            self.ColSum("parquet" if command[1] == "p" else "csv", command[2])
+        else:
+            print(f"Skipping invalid command: {line}")
+
+    def read_workload_file(self, file):
+        with open(file, "r") as f:
+            for line in f.readlines():
+                self.exec_command(line)
+
+
+def main():
+    if len(sys.argv) < 2:
+        print("Expecting command: python3 client.py INPUT [NUM_THREADS]; Missing INPUT")
+        sys.exit(1)
+    workload_spec = sys.argv[1]
+
+    num_threads = 1
+    if len(sys.argv) > 2:
+        num_threads = int(sys.argv[2])
+
+    client = TableClient(workload_spec, num_threads)
+
+
+if __name__ == "__main__":
+    main()