types/DatetimeIntervalType.cpp

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 **/

#ifndef __STDC_FORMAT_MACROS
#define __STDC_FORMAT_MACROS
#endif

#include "types/DatetimeIntervalType.hpp"

#include <cinttypes>
#include <cstddef>
#include <cstdint>
#include <cstdio>
#include <string>

#include "types/IntervalLit.hpp"
#include "types/IntervalParser.hpp"
#include "types/NullCoercibilityCheckMacro.hpp"
#include "types/Type.hpp"
#include "types/TypeID.hpp"
#include "types/TypedValue.hpp"
#include "utility/CheckSnprintf.hpp"

#include "glog/logging.h"

// NetBSD's libc has snprintf, but it doesn't show up in the std namespace for
// C++.
#ifndef __NetBSD__
using std::snprintf;
#endif

namespace quickstep {

bool DatetimeIntervalType::isCoercibleFrom(const Type &original_type) const {
  QUICKSTEP_NULL_COERCIBILITY_CHECK();
  return (original_type.getTypeID() == kDatetimeInterval);
}

bool DatetimeIntervalType::isSafelyCoercibleFrom(const Type &original_type) const {
  QUICKSTEP_NULL_COERCIBILITY_CHECK();
  return (original_type.getTypeID() == kDatetimeInterval);
}

std::string DatetimeIntervalType::printValueToString(const TypedValue &value) const {
  DCHECK(!value.isNull());

  std::int64_t subseconds = value.getLiteral<DatetimeIntervalLit>().interval_ticks;
  const bool negative_interval = subseconds < 0;
  if (negative_interval) {
    subseconds = -subseconds;
  }

  std::int64_t seconds = subseconds / DatetimeIntervalLit::kTicksPerSecond;
  subseconds -= seconds * DatetimeIntervalLit::kTicksPerSecond;

  const std::int64_t days = seconds / (24 * 60 * 60);
  seconds -= days * (24 * 60 * 60);

  char interval_buf[DatetimeIntervalLit::kPrintingChars + 1];
  std::size_t chars_written = 0;
  int snprintf_result = 0;

  if (negative_interval) {
    interval_buf[0] = '-';
    interval_buf[1] = '\0';
    ++chars_written;
  }

  if (days != 0) {
    snprintf_result = snprintf(interval_buf + chars_written,
                               sizeof(interval_buf) - chars_written,
                               "%" PRId64,
                               days);
    CheckSnprintf(snprintf_result, sizeof(interval_buf), &chars_written);

    if (days == 1) {
      snprintf_result = snprintf(interval_buf + chars_written,
                                 sizeof(interval_buf) - chars_written,
                                 " day ");
    } else {
      snprintf_result = snprintf(interval_buf + chars_written,
                                 sizeof(interval_buf) - chars_written,
                                 " days ");
    }
    CheckSnprintf(snprintf_result, sizeof(interval_buf), &chars_written);
  }

  int hours = seconds / (60 * 60);
  int minutes = (seconds - hours * (60 * 60)) / 60;
  int seconds_remainder = seconds - (60 * hours + minutes) * 60;
  snprintf_result = snprintf(interval_buf + chars_written,
                             sizeof(interval_buf) - chars_written,
                             "%02d:%02d:%02d",
                             hours, minutes, seconds_remainder);
  CheckSnprintf(snprintf_result, sizeof(interval_buf), &chars_written);

  if (subseconds != 0) {
    snprintf_result = snprintf(interval_buf + chars_written,
                               sizeof(interval_buf) - chars_written,
                               ".%06" PRId64,
                               subseconds);
    CheckSnprintf(snprintf_result, sizeof(interval_buf), &chars_written);
  }

  return std::string(interval_buf);
}

void DatetimeIntervalType::printValueToFile(const TypedValue &value,
                                            FILE *file,
                                            const int padding) const {
  // We simply re-use the logic from printValueToString(), as trying to do
  // padding on-the fly with so many different fields is too much of a hassle.
  std::fprintf(file, "%*s", padding, printValueToString(value).c_str());
}

bool DatetimeIntervalType::parseValueFromString(const std::string &value_string,
                                                TypedValue *value) const {
  // Try simple-format parse first.
  std::int64_t count;
  std::string units;
  DatetimeIntervalLit literal;
  if (IntervalParser::ParseSimpleFormatFieldsFromCombinedStringNoExtraWhitespace(
          value_string, &count, &units)
      && IntervalParser::ParseDatetimeIntervalSimpleFormat(count, units, &literal)) {
    *value = TypedValue(literal);
    return true;
  }

  // Try complex format.
  if (IntervalParser::ParseDatetimeIntervalComplexFormat(value_string, &literal)) {
    *value = TypedValue(literal);
    return true;
  }

  return false;
}

}  // namespace quickstep