From c75cad759465d7d0916f5902c8d1fca3cfa664c8 Mon Sep 17 00:00:00 2001
From: HolyLow <jiaming.xie7@gmail.com>
Date: Fri, 27 Dec 2024 13:57:59 +0800
Subject: [PATCH] add S3 storageType to align with java

---
 cpp/celeborn/protocol/PartitionLocation.h | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/cpp/celeborn/protocol/PartitionLocation.h b/cpp/celeborn/protocol/PartitionLocation.h
index 7f9d7c5263..dc0f9ecd75 100644
--- a/cpp/celeborn/protocol/PartitionLocation.h
+++ b/cpp/celeborn/protocol/PartitionLocation.h
@@ -36,12 +36,14 @@ struct StorageInfo {
     SSD = 2,
     HDFS = 3,
     OSS = 4,
+    S3 = 5,
   };
 
   static const int MEMORY_MASK = 0b1;
   static const int LOCAL_DISK_MASK = 0b10;
   static const int HDFS_MASK = 0b100;
   static const int OSS_MASK = 0b1000;
+  static const int S3_MASK = 0b10000;
   static const int ALL_TYPES_AVAILABLE_MASK = 0;
 
   Type type{MEMORY};