timestamp-functions.h

Go to the documentation of this file.

// Copyright 2012 Cloudera Inc.
//
// Licensed 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 IMPALA_EXPRS_TIMESTAMP_FUNCTIONS_H
#define IMPALA_EXPRS_TIMESTAMP_FUNCTIONS_H

#include <boost/date_time/posix_time/posix_time.hpp>
#include <boost/date_time/gregorian/gregorian.hpp>
#include <boost/date_time/time_zone_base.hpp>
#include <boost/date_time/local_time/local_time.hpp>
#include <boost/thread/thread.hpp>

#include "runtime/string-value.h"
#include "runtime/timestamp-value.h"
#include "udf/udf.h"

using namespace impala_udf;

using namespace impala_udf;

namespace impala {

class Expr;
class OpcodeRegistry;
class TupleRow;

class TimestampFunctions {
 public:
  static void UnixAndFromUnixPrepare(FunctionContext* context,
      FunctionContext::FunctionStateScope scope);
  static void UnixAndFromUnixClose(FunctionContext* context,
      FunctionContext::FunctionStateScope scope);

  static BigIntVal Unix(FunctionContext* context, const StringVal& string_val,
      const StringVal& fmt);
  static BigIntVal Unix(FunctionContext* context, const TimestampVal& tv_val);
  static BigIntVal Unix(FunctionContext* context);

  static BigIntVal UnixFromString(FunctionContext* context, const StringVal& sv);

  template <class TIME>
  static StringVal FromUnix(FunctionContext* context, const TIME& unix_time);
  template <class TIME>
  static StringVal FromUnix(FunctionContext* context, const TIME& unix_time,
      const StringVal& fmt);

  static TimestampVal FromUtc(FunctionContext* context,
    const TimestampVal& ts_val, const StringVal& tz_string_val);
  static TimestampVal ToUtc(FunctionContext* context,
      const TimestampVal& ts_val, const StringVal& tz_string_val);

  static StringVal DayName(FunctionContext* context, const TimestampVal& dow);

  static IntVal Year(FunctionContext* context, const TimestampVal& ts_val);
  static IntVal Month(FunctionContext* context, const TimestampVal& ts_val);
  static IntVal DayOfWeek(FunctionContext* context, const TimestampVal& ts_val);
  static IntVal DayOfMonth(FunctionContext* context, const TimestampVal& ts_val);
  static IntVal DayOfYear(FunctionContext* context, const TimestampVal& ts_val);
  static IntVal WeekOfYear(FunctionContext* context, const TimestampVal& ts_val);
  static IntVal Hour(FunctionContext* context, const TimestampVal& ts_val);
  static IntVal Minute(FunctionContext* context, const TimestampVal& ts_val);
  static IntVal Second(FunctionContext* context, const TimestampVal& ts_val);

  static TimestampVal Now(FunctionContext* context);
  static StringVal ToDate(FunctionContext* context, const TimestampVal& ts_val);
  static IntVal DateDiff(FunctionContext* context, const TimestampVal& ts_val1,
      const TimestampVal& ts_val2);

  template <bool ISADD, class VALTYPE, class UNIT>
  static TimestampVal DateAddSub(FunctionContext* context, const TimestampVal& ts_value,
      const VALTYPE& count);

  template <bool ISADD, class VALTYPE, class UNIT>
  static TimestampVal TimeAddSub(FunctionContext* context, const TimestampVal& ts_value,
      const VALTYPE& count);

  static StringValue* CheckFormat(StringValue* format);

  static void ReportBadFormat(FunctionContext* context,
      const StringVal& format, bool is_error);

 private:
  static const char* MONDAY;
  static const char* TUESDAY;
  static const char* WEDNESDAY;
  static const char* THURSDAY;
  static const char* FRIDAY;
  static const char* SATURDAY;
  static const char* SUNDAY;
};

class TimezoneDatabase {
 public:
   TimezoneDatabase();
   ~TimezoneDatabase();

  static boost::local_time::time_zone_ptr FindTimezone(const std::string& tz,
      const TimestampValue& tv);

  static const boost::local_time::time_zone_ptr TIMEZONE_MSK_2011_NODST;

 private:
  static const char* TIMEZONE_DATABASE_STR;
  static boost::local_time::tz_database tz_database_;
  static std::vector<std::string> tz_region_list_;
};

} // namespace impala

#endif