aboutsummaryrefslogtreecommitdiffstats
path: root/util/draft/date.cpp
blob: 11b080820fd95e360c424bdcf0c6daf18e0dc8fa (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
#include "date.h"

#include <util/string/cast.h>
#include <util/generic/yexception.h>
#include <util/datetime/base.h> 

time_t GetDateStart(time_t ts) {
    tm dateTm;
    memset(&dateTm, 0, sizeof(tm));
    localtime_r(&ts, &dateTm);

    dateTm.tm_isdst = -1;

    dateTm.tm_sec = 0;
    dateTm.tm_min = 0;
    dateTm.tm_hour = 0;
    return mktime(&dateTm);
}

static time_t ParseDate(const char* date, const char* format) {
    tm dateTm;
    memset(&dateTm, 0, sizeof(tm));
    if (!strptime(date, format, &dateTm)) {
        ythrow yexception() << "Invalid date string and format: " << date << ", " << format;
    }
    return mktime(&dateTm);
}

static time_t ParseDate(const char* dateStr) {
    if (strlen(dateStr) != 8) {
        ythrow yexception() << "Invalid date string: " << dateStr;
    }

    return ParseDate(dateStr, "%Y%m%d");
}

template <>
TDate FromStringImpl<TDate>(const char* data, size_t len) {
    return TDate(ParseDate(TString(data, len).data()));
}

TDate::TDate(const char* yyyymmdd)
    : Timestamp(GetDateStart(ParseDate(yyyymmdd)))
{
}

TDate::TDate(const TString& yyyymmdd)
    : Timestamp(GetDateStart(ParseDate(yyyymmdd.c_str())))
{
}

TDate::TDate(time_t ts)
    : Timestamp(GetDateStart(ts))
{
}

TDate::TDate(const TString& date, const TString& format)
    : Timestamp(GetDateStart(ParseDate(date.data(), format.data())))
{
}

TDate::TDate(unsigned year, unsigned month, unsigned monthDay) {
    tm dateTm;
    Zero(dateTm);
    dateTm.tm_year = year - 1900;
    dateTm.tm_mon = month - 1;
    dateTm.tm_mday = monthDay;
    dateTm.tm_isdst = -1;
    Timestamp = mktime(&dateTm);
    if (Timestamp == (time_t)-1) {
        ythrow yexception() << "Invalid TDate args:(" << year << ',' << month << ',' << monthDay << ')';
    }
}

time_t TDate::GetStartUTC() const {
    tm dateTm;
    localtime_r(&Timestamp, &dateTm);
    dateTm.tm_isdst = -1;
    dateTm.tm_sec = 0;
    dateTm.tm_min = 0;
    dateTm.tm_hour = 0;
    return TimeGM(&dateTm);
}

TString TDate::ToStroka(const char* format) const {
    tm dateTm;
    localtime_r(&Timestamp, &dateTm);
    return Strftime(format, &dateTm);
}

unsigned TDate::GetWeekDay() const {
    tm dateTm;
    localtime_r(&Timestamp, &dateTm);
    return (unsigned)dateTm.tm_wday;
}

unsigned TDate::GetYear() const {
    tm dateTm;
    localtime_r(&Timestamp, &dateTm);
    return ((unsigned)dateTm.tm_year) + 1900;
}

unsigned TDate::GetMonth() const {
    tm dateTm;
    localtime_r(&Timestamp, &dateTm);
    return ((unsigned)dateTm.tm_mon) + 1;
}

unsigned TDate::GetMonthDay() const {
    tm dateTm;
    localtime_r(&Timestamp, &dateTm);
    return (unsigned)dateTm.tm_mday;
}