From 4d441725ad26d25eb531fc777bbb0fdd665283bc Mon Sep 17 00:00:00 2001
From: siyuan0322 <siyuan0322@gmail.com>
Date: Sat, 20 Feb 2021 11:14:41 +0800
Subject: [PATCH] optimize source chain operator

---
 .../runtime/src/dataflow/operator/sourcestep/builder.rs        | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/interactive_engine/src/executor/runtime/src/dataflow/operator/sourcestep/builder.rs b/interactive_engine/src/executor/runtime/src/dataflow/operator/sourcestep/builder.rs
index e07b736c3bb4..3ded634e93cc 100644
--- a/interactive_engine/src/executor/runtime/src/dataflow/operator/sourcestep/builder.rs
+++ b/interactive_engine/src/executor/runtime/src/dataflow/operator/sourcestep/builder.rs
@@ -85,7 +85,8 @@ fn build_graph_source_operator<VV, VVI, EE, EEI, F>(source: &query_flow::SourceO
         if op_type == OperatorType::V ||
             op_type == OperatorType::E ||
             op_type == OperatorType::V_COUNT ||
-            op_type == OperatorType::E_COUNT {
+            op_type == OperatorType::E_COUNT ||
+            op_type == OperatorType::SOURCE_CHAIN {
             let primary_key_list = parse_from_bytes::<VertexPrimaryKeyListProto>(source.get_base().get_argument().get_payload()).unwrap();
             for primary_key in primary_key_list.get_primary_keys() {
                 let label_id = primary_key.get_label_id();