lidavidm commented on a change in pull request #12464:
URL: https://github.com/apache/arrow/pull/12464#discussion_r833420401
##########
File path: cpp/src/arrow/compute/kernels/scalar_temporal_unary.cc
##########
@@ -1143,6 +1145,123 @@ struct Strftime {
};
#endif
+// ----------------------------------------------------------------------
+// Convert string representations of timestamps in arbitrary format to
timestamps
+
+const std::string GetZone(const std::string& format) {
+ // Check for use of %z or %Z
+ size_t cur = 0;
+ std::string zone = "";
+ while (cur < format.size() - 1) {
+ if (format[cur] == '%') {
+ if (format[cur + 1] == 'z') {
+ zone = "UTC";
+ break;
+ }
+ cur++;
+ }
+ cur++;
+ }
+ return zone;
+}
+
+template <typename Duration, typename InType>
+struct Strptime {
+ const std::shared_ptr<TimestampParser> parser;
+ const TimeUnit::type unit;
+ const std::string zone;
+ const bool error_is_null;
+
+ static Result<Strptime> Make(KernelContext* ctx, const DataType& type) {
+ const StrptimeOptions& options = StrptimeState::Get(ctx);
+
+ return Strptime{TimestampParser::MakeStrptime(options.format),
+ std::move(options.unit), GetZone(options.format),
+ options.error_is_null};
+ }
+
+ static Status Call(KernelContext* ctx, const Scalar& in, Scalar* out) {
+ ARROW_ASSIGN_OR_RAISE(auto self, Make(ctx, *in.type));
+
+ if (in.is_valid) {
+ auto s = internal::UnboxScalar<InType>::Unbox(in);
+ int64_t result;
+ if ((*self.parser)(s.data(), s.size(), self.unit, &result)) {
+ *checked_cast<TimestampScalar*>(out) =
+ TimestampScalar(result, timestamp(self.unit, self.zone));
+ } else {
+ if (self.error_is_null) {
+ out->is_valid = false;
+ } else {
+ return Status::Invalid("Failed to parse string: '", s,
+ "' as a scalar of type ",
+ TimestampType(self.unit).ToString());
+ }
+ }
+ } else {
+ out->is_valid = false;
+ }
+ return Status::OK();
+ }
+
+ static Status Call(KernelContext* ctx, const ArrayData& in, ArrayData* out) {
+ ARROW_ASSIGN_OR_RAISE(auto self, Make(ctx, *in.type));
+ int64_t* out_data = out->GetMutableValues<int64_t>(1);
+
+ if (self.error_is_null) {
+ arrow::internal::BitmapWriter out_writer(out->buffers[0]->mutable_data(),
Review comment:
Arrays with no input nulls
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
