Skip to content

Commit 051dad9

Browse files
authored
fix(interactive): enable more than 256 labels in edge column (#4530)
Fixes
1 parent b2b347d commit 051dad9

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

flex/engines/graph_db/runtime/common/columns/edge_columns.h

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -558,9 +558,9 @@ class SDMLEdgeColumn : public IEdgeColumn {
558558
private:
559559
friend class SDMLEdgeColumnBuilder;
560560
Direction dir_;
561-
std::map<LabelTriplet, int8_t> index_;
561+
std::map<LabelTriplet, label_t> index_;
562562
std::vector<std::pair<LabelTriplet, PropertyType>> edge_labels_;
563-
std::vector<std::tuple<int8_t, vid_t, vid_t, size_t>> edges_;
563+
std::vector<std::tuple<label_t, vid_t, vid_t, size_t>> edges_;
564564
std::vector<std::shared_ptr<EdgePropVecBase>> prop_cols_;
565565
};
566566

@@ -646,9 +646,9 @@ class BDMLEdgeColumn : public IEdgeColumn {
646646

647647
private:
648648
friend class BDMLEdgeColumnBuilder;
649-
std::map<LabelTriplet, int8_t> index_;
649+
std::map<LabelTriplet, label_t> index_;
650650
std::vector<std::pair<LabelTriplet, PropertyType>> labels_;
651-
std::vector<std::tuple<int8_t, vid_t, vid_t, size_t, bool>> edges_;
651+
std::vector<std::tuple<label_t, vid_t, vid_t, size_t, bool>> edges_;
652652
std::vector<std::shared_ptr<EdgePropVecBase>> prop_cols_;
653653
};
654654

@@ -791,7 +791,7 @@ class SDMLEdgeColumnBuilder : public IContextColumnBuilder {
791791
auto index = index_[label];
792792
push_back_opt(index, e.src_, e.dst_, e.prop_);
793793
}
794-
inline void push_back_opt(int8_t index, vid_t src, vid_t dst,
794+
inline void push_back_opt(label_t index, vid_t src, vid_t dst,
795795
const EdgeData& data) {
796796
edges_.emplace_back(index, src, dst, prop_cols_[index]->size());
797797
set_edge_data(prop_cols_[index].get(), prop_cols_[index]->size(), data);
@@ -803,7 +803,7 @@ class SDMLEdgeColumnBuilder : public IContextColumnBuilder {
803803
push_back_opt(index, src, dst, data);
804804
}
805805

806-
inline void push_back_endpoints(int8_t index, vid_t src, vid_t dst) {
806+
inline void push_back_endpoints(label_t index, vid_t src, vid_t dst) {
807807
LOG(FATAL) << "Not implemented";
808808
}
809809

@@ -812,9 +812,9 @@ class SDMLEdgeColumnBuilder : public IContextColumnBuilder {
812812
private:
813813
friend class SDMLEdgeColumn;
814814
Direction dir_;
815-
std::map<LabelTriplet, int8_t> index_;
815+
std::map<LabelTriplet, label_t> index_;
816816
std::vector<std::pair<LabelTriplet, PropertyType>> edge_labels_;
817-
std::vector<std::tuple<int8_t, vid_t, vid_t, size_t>> edges_;
817+
std::vector<std::tuple<label_t, vid_t, vid_t, size_t>> edges_;
818818
std::vector<std::shared_ptr<EdgePropVecBase>> prop_cols_;
819819
};
820820

@@ -848,7 +848,7 @@ class BDMLEdgeColumnBuilder : public IContextColumnBuilder {
848848
auto index = index_[label];
849849
push_back_opt(index, e.src_, e.dst_, e.prop_, e.dir_);
850850
}
851-
inline void push_back_opt(int8_t index, vid_t src, vid_t dst,
851+
inline void push_back_opt(label_t index, vid_t src, vid_t dst,
852852
const EdgeData& data, Direction dir) {
853853
edges_.emplace_back(index, src, dst, prop_cols_[index]->size(),
854854
dir == Direction::kOut);
@@ -861,13 +861,13 @@ class BDMLEdgeColumnBuilder : public IContextColumnBuilder {
861861
push_back_opt(index, src, dst, data, dir);
862862
}
863863

864-
inline void push_back_endpoints(int8_t index, vid_t src, vid_t dst,
864+
inline void push_back_endpoints(label_t index, vid_t src, vid_t dst,
865865
Direction dir) {
866866
edges_.emplace_back(index, src, dst, prop_cols_[index]->size(),
867867
dir == Direction::kOut);
868868
}
869869

870-
inline void push_back_endpoints(int8_t index, vid_t src, vid_t dst,
870+
inline void push_back_endpoints(label_t index, vid_t src, vid_t dst,
871871
bool dir) {
872872
edges_.emplace_back(index, src, dst, prop_cols_[index]->size(), dir);
873873
}
@@ -877,9 +877,9 @@ class BDMLEdgeColumnBuilder : public IContextColumnBuilder {
877877
private:
878878
friend class BDMLEdgeColumn;
879879

880-
std::map<LabelTriplet, int8_t> index_;
880+
std::map<LabelTriplet, label_t> index_;
881881
std::vector<std::pair<LabelTriplet, PropertyType>> labels_;
882-
std::vector<std::tuple<int8_t, vid_t, vid_t, size_t, bool>> edges_;
882+
std::vector<std::tuple<label_t, vid_t, vid_t, size_t, bool>> edges_;
883883
std::vector<std::shared_ptr<EdgePropVecBase>> prop_cols_;
884884
};
885885

0 commit comments

Comments
 (0)