copr: add projection

components/tidb_query_common/src/metrics.rs

@@ -14,6 +14,7 @@ make_auto_flush_static_metric! {
         batch_stream_aggr,
         batch_limit,
         batch_top_n,
+        batch_projection,
         table_scan,
         index_scan,
         selection,

components/tidb_query_datatype/src/codec/data_type/vector.rs

@@ -37,6 +37,47 @@ impl VectorValue {
         }
     }
 
+    /// Creates a `VectorValue` of length `len` with the given value `scalar`.
+    #[inline]
+    pub fn from_scalar(scalar: &ScalarValue, len: usize) -> Self {
+        macro_rules! expand_convertion {
+            ($val:tt, $( $tp:tt : $chktp:ty ),* ) => {
+                match &$val {
+                    $(
+                        &ScalarValue::$tp(val) => {
+                            let mut v: $chktp = ChunkedVec::with_capacity(len);
+                            match val {
+                                None => {
+                                    for _ in 0..len {
+                                        v.push_null();
+                                    }
+                                },
+                                Some(val) => {
+                                    for _ in 0..len {
+                                        v.push_data(val.clone());
+                                    }
+                                }
+                            }
+                            VectorValue::$tp(v)
+                        }
+                    )*
+                }
+            }
+        }
+        expand_convertion!(
+            scalar,
+            Int: ChunkedVecSized<Int>,
+            Real: ChunkedVecSized<Real>,
+            Decimal: ChunkedVecSized<Decimal>,
+            DateTime: ChunkedVecSized<DateTime>,
+            Duration: ChunkedVecSized<Duration>,
+            Set: ChunkedVecSet,
+            Json: ChunkedVecJson,
+            Enum: ChunkedVecEnum,
+            Bytes: ChunkedVecBytes
+        )
+    }
+
     /// Creates a new empty `VectorValue` with the same eval type.
     #[inline]
     #[must_use]

components/tidb_query_executors/src/lib.rs

@@ -30,6 +30,7 @@ mod fast_hash_aggr_executor;
 mod index_scan_executor;
 pub mod interface;
 mod limit_executor;
+mod projection_executor;
 pub mod runner;
 mod selection_executor;
 mod simple_aggr_executor;
@@ -42,6 +43,7 @@ mod util;
 pub use self::fast_hash_aggr_executor::BatchFastHashAggregationExecutor;
 pub use self::index_scan_executor::BatchIndexScanExecutor;
 pub use self::limit_executor::BatchLimitExecutor;
+pub use self::projection_executor::BatchProjectionExecutor;
 pub use self::selection_executor::BatchSelectionExecutor;
 pub use self::simple_aggr_executor::BatchSimpleAggregationExecutor;
 pub use self::slow_hash_aggr_executor::BatchSlowHashAggregationExecutor;