[Streaming] Streaming data transfer and python integration (#6185)

streaming/src/message/priority_queue.h

@@ -0,0 +1,53 @@
+#ifndef RAY_PRIORITY_QUEUE_H
+#define RAY_PRIORITY_QUEUE_H
+
+#include <algorithm>
+#include <memory>
+#include <vector>
+#include "util/streaming_logging.h"
+
+namespace ray {
+namespace streaming {
+
+template <class T, class C>
+
+class PriorityQueue {
+ private:
+  std::vector<T> merge_vec_;
+  C comparator_;
+
+ public:
+  PriorityQueue(C &comparator) : comparator_(comparator){};
+
+  inline void push(T &&item) {
+    merge_vec_.push_back(std::forward<T>(item));
+    std::push_heap(merge_vec_.begin(), merge_vec_.end(), comparator_);
+  }
+
+  inline void push(const T &item) {
+    merge_vec_.push_back(item);
+    std::push_heap(merge_vec_.begin(), merge_vec_.end(), comparator_);
+  }
+
+  inline void pop() {
+    STREAMING_CHECK(!isEmpty());
+    std::pop_heap(merge_vec_.begin(), merge_vec_.end(), comparator_);
+    merge_vec_.pop_back();
+  }
+
+  inline void makeHeap() {
+    std::make_heap(merge_vec_.begin(), merge_vec_.end(), comparator_);
+  }
+
+  inline T &top() { return merge_vec_.front(); }
+
+  inline uint32_t size() { return merge_vec_.size(); }
+
+  inline bool isEmpty() { return merge_vec_.empty(); }
+
+  std::vector<T> &getRawVector() { return merge_vec_; }
+};
+}  // namespace streaming
+}  // namespace ray
+
+#endif  // RAY_PRIORITY_QUEUE_H