Make constant per query

acvictor · Apr 3, 2024 · d3664f6 · d3664f6
1 parent 55a0168
commit d3664f6
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/velox/functions/sparksql/DateTimeFunctions.h b/velox/functions/sparksql/DateTimeFunctions.h
@@ -111,13 +111,22 @@ template <typename T>
 struct CurrentTimestampFunction {
   VELOX_DEFINE_FUNCTION_TYPES(T);
 
-  FOLLY_ALWAYS_INLINE void call(out_type<Timestamp>& result) {
+  FOLLY_ALWAYS_INLINE void initialize(
+      const std::vector<TypePtr>& /*inputTypes*/,
+      const core::QueryConfig& /*config*/) {
     auto now = std::chrono::system_clock::now();
     auto epoch = std::chrono::duration_cast<std::chrono::microseconds>(
                      now.time_since_epoch())
                      .count();
-    result = Timestamp::fromMicros(epoch);
+    currentTimestamp_ = Timestamp::fromMicros(epoch);
+  }
+
+  FOLLY_ALWAYS_INLINE void call(out_type<Timestamp>& result) {
+    result = currentTimestamp_;
   }
+
+ protected:
+  Timestamp currentTimestamp_;
 };
 
 template <typename T>