Mercurial > audlegacy
comparison sqlite/date.c @ 1434:b6b61becdf4e trunk
[svn] - add sqlite/ directory
| author | nenolod |
|---|---|
| date | Thu, 27 Jul 2006 22:41:31 -0700 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| 1433:3cbe3d14ea68 | 1434:b6b61becdf4e |
|---|---|
| 1 /* | |
| 2 ** 2003 October 31 | |
| 3 ** | |
| 4 ** The author disclaims copyright to this source code. In place of | |
| 5 ** a legal notice, here is a blessing: | |
| 6 ** | |
| 7 ** May you do good and not evil. | |
| 8 ** May you find forgiveness for yourself and forgive others. | |
| 9 ** May you share freely, never taking more than you give. | |
| 10 ** | |
| 11 ************************************************************************* | |
| 12 ** This file contains the C functions that implement date and time | |
| 13 ** functions for SQLite. | |
| 14 ** | |
| 15 ** There is only one exported symbol in this file - the function | |
| 16 ** sqlite3RegisterDateTimeFunctions() found at the bottom of the file. | |
| 17 ** All other code has file scope. | |
| 18 ** | |
| 19 ** $Id: date.c,v 1.54 2006/01/31 20:49:13 drh Exp $ | |
| 20 ** | |
| 21 ** NOTES: | |
| 22 ** | |
| 23 ** SQLite processes all times and dates as Julian Day numbers. The | |
| 24 ** dates and times are stored as the number of days since noon | |
| 25 ** in Greenwich on November 24, 4714 B.C. according to the Gregorian | |
| 26 ** calendar system. | |
| 27 ** | |
| 28 ** 1970-01-01 00:00:00 is JD 2440587.5 | |
| 29 ** 2000-01-01 00:00:00 is JD 2451544.5 | |
| 30 ** | |
| 31 ** This implemention requires years to be expressed as a 4-digit number | |
| 32 ** which means that only dates between 0000-01-01 and 9999-12-31 can | |
| 33 ** be represented, even though julian day numbers allow a much wider | |
| 34 ** range of dates. | |
| 35 ** | |
| 36 ** The Gregorian calendar system is used for all dates and times, | |
| 37 ** even those that predate the Gregorian calendar. Historians usually | |
| 38 ** use the Julian calendar for dates prior to 1582-10-15 and for some | |
| 39 ** dates afterwards, depending on locale. Beware of this difference. | |
| 40 ** | |
| 41 ** The conversion algorithms are implemented based on descriptions | |
| 42 ** in the following text: | |
| 43 ** | |
| 44 ** Jean Meeus | |
| 45 ** Astronomical Algorithms, 2nd Edition, 1998 | |
| 46 ** ISBM 0-943396-61-1 | |
| 47 ** Willmann-Bell, Inc | |
| 48 ** Richmond, Virginia (USA) | |
| 49 */ | |
| 50 #include "sqliteInt.h" | |
| 51 #include "os.h" | |
| 52 #include <ctype.h> | |
| 53 #include <stdlib.h> | |
| 54 #include <assert.h> | |
| 55 #include <time.h> | |
| 56 | |
| 57 #ifndef SQLITE_OMIT_DATETIME_FUNCS | |
| 58 | |
| 59 /* | |
| 60 ** A structure for holding a single date and time. | |
| 61 */ | |
| 62 typedef struct DateTime DateTime; | |
| 63 struct DateTime { | |
| 64 double rJD; /* The julian day number */ | |
| 65 int Y, M, D; /* Year, month, and day */ | |
| 66 int h, m; /* Hour and minutes */ | |
| 67 int tz; /* Timezone offset in minutes */ | |
| 68 double s; /* Seconds */ | |
| 69 char validYMD; /* True if Y,M,D are valid */ | |
| 70 char validHMS; /* True if h,m,s are valid */ | |
| 71 char validJD; /* True if rJD is valid */ | |
| 72 char validTZ; /* True if tz is valid */ | |
| 73 }; | |
| 74 | |
| 75 | |
| 76 /* | |
| 77 ** Convert zDate into one or more integers. Additional arguments | |
| 78 ** come in groups of 5 as follows: | |
| 79 ** | |
| 80 ** N number of digits in the integer | |
| 81 ** min minimum allowed value of the integer | |
| 82 ** max maximum allowed value of the integer | |
| 83 ** nextC first character after the integer | |
| 84 ** pVal where to write the integers value. | |
| 85 ** | |
| 86 ** Conversions continue until one with nextC==0 is encountered. | |
| 87 ** The function returns the number of successful conversions. | |
| 88 */ | |
| 89 static int getDigits(const char *zDate, ...){ | |
| 90 va_list ap; | |
| 91 int val; | |
| 92 int N; | |
| 93 int min; | |
| 94 int max; | |
| 95 int nextC; | |
| 96 int *pVal; | |
| 97 int cnt = 0; | |
| 98 va_start(ap, zDate); | |
| 99 do{ | |
| 100 N = va_arg(ap, int); | |
| 101 min = va_arg(ap, int); | |
| 102 max = va_arg(ap, int); | |
| 103 nextC = va_arg(ap, int); | |
| 104 pVal = va_arg(ap, int*); | |
| 105 val = 0; | |
| 106 while( N-- ){ | |
| 107 if( !isdigit(*(u8*)zDate) ){ | |
| 108 goto end_getDigits; | |
| 109 } | |
| 110 val = val*10 + *zDate - '0'; | |
| 111 zDate++; | |
| 112 } | |
| 113 if( val<min || val>max || (nextC!=0 && nextC!=*zDate) ){ | |
| 114 goto end_getDigits; | |
| 115 } | |
| 116 *pVal = val; | |
| 117 zDate++; | |
| 118 cnt++; | |
| 119 }while( nextC ); | |
| 120 end_getDigits: | |
| 121 va_end(ap); | |
| 122 return cnt; | |
| 123 } | |
| 124 | |
| 125 /* | |
| 126 ** Read text from z[] and convert into a floating point number. Return | |
| 127 ** the number of digits converted. | |
| 128 */ | |
| 129 #define getValue sqlite3AtoF | |
| 130 | |
| 131 /* | |
| 132 ** Parse a timezone extension on the end of a date-time. | |
| 133 ** The extension is of the form: | |
| 134 ** | |
| 135 ** (+/-)HH:MM | |
| 136 ** | |
| 137 ** If the parse is successful, write the number of minutes | |
| 138 ** of change in *pnMin and return 0. If a parser error occurs, | |
| 139 ** return 0. | |
| 140 ** | |
| 141 ** A missing specifier is not considered an error. | |
| 142 */ | |
| 143 static int parseTimezone(const char *zDate, DateTime *p){ | |
| 144 int sgn = 0; | |
| 145 int nHr, nMn; | |
| 146 while( isspace(*(u8*)zDate) ){ zDate++; } | |
| 147 p->tz = 0; | |
| 148 if( *zDate=='-' ){ | |
| 149 sgn = -1; | |
| 150 }else if( *zDate=='+' ){ | |
| 151 sgn = +1; | |
| 152 }else{ | |
| 153 return *zDate!=0; | |
| 154 } | |
| 155 zDate++; | |
| 156 if( getDigits(zDate, 2, 0, 14, ':', &nHr, 2, 0, 59, 0, &nMn)!=2 ){ | |
| 157 return 1; | |
| 158 } | |
| 159 zDate += 5; | |
| 160 p->tz = sgn*(nMn + nHr*60); | |
| 161 while( isspace(*(u8*)zDate) ){ zDate++; } | |
| 162 return *zDate!=0; | |
| 163 } | |
| 164 | |
| 165 /* | |
| 166 ** Parse times of the form HH:MM or HH:MM:SS or HH:MM:SS.FFFF. | |
| 167 ** The HH, MM, and SS must each be exactly 2 digits. The | |
| 168 ** fractional seconds FFFF can be one or more digits. | |
| 169 ** | |
| 170 ** Return 1 if there is a parsing error and 0 on success. | |
| 171 */ | |
| 172 static int parseHhMmSs(const char *zDate, DateTime *p){ | |
| 173 int h, m, s; | |
| 174 double ms = 0.0; | |
| 175 if( getDigits(zDate, 2, 0, 24, ':', &h, 2, 0, 59, 0, &m)!=2 ){ | |
| 176 return 1; | |
| 177 } | |
| 178 zDate += 5; | |
| 179 if( *zDate==':' ){ | |
| 180 zDate++; | |
| 181 if( getDigits(zDate, 2, 0, 59, 0, &s)!=1 ){ | |
| 182 return 1; | |
| 183 } | |
| 184 zDate += 2; | |
| 185 if( *zDate=='.' && isdigit((u8)zDate[1]) ){ | |
| 186 double rScale = 1.0; | |
| 187 zDate++; | |
| 188 while( isdigit(*(u8*)zDate) ){ | |
| 189 ms = ms*10.0 + *zDate - '0'; | |
| 190 rScale *= 10.0; | |
| 191 zDate++; | |
| 192 } | |
| 193 ms /= rScale; | |
| 194 } | |
| 195 }else{ | |
| 196 s = 0; | |
| 197 } | |
| 198 p->validJD = 0; | |
| 199 p->validHMS = 1; | |
| 200 p->h = h; | |
| 201 p->m = m; | |
| 202 p->s = s + ms; | |
| 203 if( parseTimezone(zDate, p) ) return 1; | |
| 204 p->validTZ = p->tz!=0; | |
| 205 return 0; | |
| 206 } | |
| 207 | |
| 208 /* | |
| 209 ** Convert from YYYY-MM-DD HH:MM:SS to julian day. We always assume | |
| 210 ** that the YYYY-MM-DD is according to the Gregorian calendar. | |
| 211 ** | |
| 212 ** Reference: Meeus page 61 | |
| 213 */ | |
| 214 static void computeJD(DateTime *p){ | |
| 215 int Y, M, D, A, B, X1, X2; | |
| 216 | |
| 217 if( p->validJD ) return; | |
| 218 if( p->validYMD ){ | |
| 219 Y = p->Y; | |
| 220 M = p->M; | |
| 221 D = p->D; | |
| 222 }else{ | |
| 223 Y = 2000; /* If no YMD specified, assume 2000-Jan-01 */ | |
| 224 M = 1; | |
| 225 D = 1; | |
| 226 } | |
| 227 if( M<=2 ){ | |
| 228 Y--; | |
| 229 M += 12; | |
| 230 } | |
| 231 A = Y/100; | |
| 232 B = 2 - A + (A/4); | |
| 233 X1 = 365.25*(Y+4716); | |
| 234 X2 = 30.6001*(M+1); | |
| 235 p->rJD = X1 + X2 + D + B - 1524.5; | |
| 236 p->validJD = 1; | |
| 237 p->validYMD = 0; | |
| 238 if( p->validHMS ){ | |
| 239 p->rJD += (p->h*3600.0 + p->m*60.0 + p->s)/86400.0; | |
| 240 if( p->validTZ ){ | |
| 241 p->rJD -= p->tz*60/86400.0; | |
| 242 p->validHMS = 0; | |
| 243 p->validTZ = 0; | |
| 244 } | |
| 245 } | |
| 246 } | |
| 247 | |
| 248 /* | |
| 249 ** Parse dates of the form | |
| 250 ** | |
| 251 ** YYYY-MM-DD HH:MM:SS.FFF | |
| 252 ** YYYY-MM-DD HH:MM:SS | |
| 253 ** YYYY-MM-DD HH:MM | |
| 254 ** YYYY-MM-DD | |
| 255 ** | |
| 256 ** Write the result into the DateTime structure and return 0 | |
| 257 ** on success and 1 if the input string is not a well-formed | |
| 258 ** date. | |
| 259 */ | |
| 260 static int parseYyyyMmDd(const char *zDate, DateTime *p){ | |
| 261 int Y, M, D, neg; | |
| 262 | |
| 263 if( zDate[0]=='-' ){ | |
| 264 zDate++; | |
| 265 neg = 1; | |
| 266 }else{ | |
| 267 neg = 0; | |
| 268 } | |
| 269 if( getDigits(zDate,4,0,9999,'-',&Y,2,1,12,'-',&M,2,1,31,0,&D)!=3 ){ | |
| 270 return 1; | |
| 271 } | |
| 272 zDate += 10; | |
| 273 while( isspace(*(u8*)zDate) || 'T'==*(u8*)zDate ){ zDate++; } | |
| 274 if( parseHhMmSs(zDate, p)==0 ){ | |
| 275 /* We got the time */ | |
| 276 }else if( *zDate==0 ){ | |
| 277 p->validHMS = 0; | |
| 278 }else{ | |
| 279 return 1; | |
| 280 } | |
| 281 p->validJD = 0; | |
| 282 p->validYMD = 1; | |
| 283 p->Y = neg ? -Y : Y; | |
| 284 p->M = M; | |
| 285 p->D = D; | |
| 286 if( p->validTZ ){ | |
| 287 computeJD(p); | |
| 288 } | |
| 289 return 0; | |
| 290 } | |
| 291 | |
| 292 /* | |
| 293 ** Attempt to parse the given string into a Julian Day Number. Return | |
| 294 ** the number of errors. | |
| 295 ** | |
| 296 ** The following are acceptable forms for the input string: | |
| 297 ** | |
| 298 ** YYYY-MM-DD HH:MM:SS.FFF +/-HH:MM | |
| 299 ** DDDD.DD | |
| 300 ** now | |
| 301 ** | |
| 302 ** In the first form, the +/-HH:MM is always optional. The fractional | |
| 303 ** seconds extension (the ".FFF") is optional. The seconds portion | |
| 304 ** (":SS.FFF") is option. The year and date can be omitted as long | |
| 305 ** as there is a time string. The time string can be omitted as long | |
| 306 ** as there is a year and date. | |
| 307 */ | |
| 308 static int parseDateOrTime(const char *zDate, DateTime *p){ | |
| 309 memset(p, 0, sizeof(*p)); | |
| 310 if( parseYyyyMmDd(zDate,p)==0 ){ | |
| 311 return 0; | |
| 312 }else if( parseHhMmSs(zDate, p)==0 ){ | |
| 313 return 0; | |
| 314 }else if( sqlite3StrICmp(zDate,"now")==0){ | |
| 315 double r; | |
| 316 sqlite3OsCurrentTime(&r); | |
| 317 p->rJD = r; | |
| 318 p->validJD = 1; | |
| 319 return 0; | |
| 320 }else if( sqlite3IsNumber(zDate, 0, SQLITE_UTF8) ){ | |
| 321 getValue(zDate, &p->rJD); | |
| 322 p->validJD = 1; | |
| 323 return 0; | |
| 324 } | |
| 325 return 1; | |
| 326 } | |
| 327 | |
| 328 /* | |
| 329 ** Compute the Year, Month, and Day from the julian day number. | |
| 330 */ | |
| 331 static void computeYMD(DateTime *p){ | |
| 332 int Z, A, B, C, D, E, X1; | |
| 333 if( p->validYMD ) return; | |
| 334 if( !p->validJD ){ | |
| 335 p->Y = 2000; | |
| 336 p->M = 1; | |
| 337 p->D = 1; | |
| 338 }else{ | |
| 339 Z = p->rJD + 0.5; | |
| 340 A = (Z - 1867216.25)/36524.25; | |
| 341 A = Z + 1 + A - (A/4); | |
| 342 B = A + 1524; | |
| 343 C = (B - 122.1)/365.25; | |
| 344 D = 365.25*C; | |
| 345 E = (B-D)/30.6001; | |
| 346 X1 = 30.6001*E; | |
| 347 p->D = B - D - X1; | |
| 348 p->M = E<14 ? E-1 : E-13; | |
| 349 p->Y = p->M>2 ? C - 4716 : C - 4715; | |
| 350 } | |
| 351 p->validYMD = 1; | |
| 352 } | |
| 353 | |
| 354 /* | |
| 355 ** Compute the Hour, Minute, and Seconds from the julian day number. | |
| 356 */ | |
| 357 static void computeHMS(DateTime *p){ | |
| 358 int Z, s; | |
| 359 if( p->validHMS ) return; | |
| 360 Z = p->rJD + 0.5; | |
| 361 s = (p->rJD + 0.5 - Z)*86400000.0 + 0.5; | |
| 362 p->s = 0.001*s; | |
| 363 s = p->s; | |
| 364 p->s -= s; | |
| 365 p->h = s/3600; | |
| 366 s -= p->h*3600; | |
| 367 p->m = s/60; | |
| 368 p->s += s - p->m*60; | |
| 369 p->validHMS = 1; | |
| 370 } | |
| 371 | |
| 372 /* | |
| 373 ** Compute both YMD and HMS | |
| 374 */ | |
| 375 static void computeYMD_HMS(DateTime *p){ | |
| 376 computeYMD(p); | |
| 377 computeHMS(p); | |
| 378 } | |
| 379 | |
| 380 /* | |
| 381 ** Clear the YMD and HMS and the TZ | |
| 382 */ | |
| 383 static void clearYMD_HMS_TZ(DateTime *p){ | |
| 384 p->validYMD = 0; | |
| 385 p->validHMS = 0; | |
| 386 p->validTZ = 0; | |
| 387 } | |
| 388 | |
| 389 /* | |
| 390 ** Compute the difference (in days) between localtime and UTC (a.k.a. GMT) | |
| 391 ** for the time value p where p is in UTC. | |
| 392 */ | |
| 393 static double localtimeOffset(DateTime *p){ | |
| 394 DateTime x, y; | |
| 395 time_t t; | |
| 396 struct tm *pTm; | |
| 397 x = *p; | |
| 398 computeYMD_HMS(&x); | |
| 399 if( x.Y<1971 || x.Y>=2038 ){ | |
| 400 x.Y = 2000; | |
| 401 x.M = 1; | |
| 402 x.D = 1; | |
| 403 x.h = 0; | |
| 404 x.m = 0; | |
| 405 x.s = 0.0; | |
| 406 } else { | |
| 407 int s = x.s + 0.5; | |
| 408 x.s = s; | |
| 409 } | |
| 410 x.tz = 0; | |
| 411 x.validJD = 0; | |
| 412 computeJD(&x); | |
| 413 t = (x.rJD-2440587.5)*86400.0 + 0.5; | |
| 414 sqlite3OsEnterMutex(); | |
| 415 pTm = localtime(&t); | |
| 416 y.Y = pTm->tm_year + 1900; | |
| 417 y.M = pTm->tm_mon + 1; | |
| 418 y.D = pTm->tm_mday; | |
| 419 y.h = pTm->tm_hour; | |
| 420 y.m = pTm->tm_min; | |
| 421 y.s = pTm->tm_sec; | |
| 422 sqlite3OsLeaveMutex(); | |
| 423 y.validYMD = 1; | |
| 424 y.validHMS = 1; | |
| 425 y.validJD = 0; | |
| 426 y.validTZ = 0; | |
| 427 computeJD(&y); | |
| 428 return y.rJD - x.rJD; | |
| 429 } | |
| 430 | |
| 431 /* | |
| 432 ** Process a modifier to a date-time stamp. The modifiers are | |
| 433 ** as follows: | |
| 434 ** | |
| 435 ** NNN days | |
| 436 ** NNN hours | |
| 437 ** NNN minutes | |
| 438 ** NNN.NNNN seconds | |
| 439 ** NNN months | |
| 440 ** NNN years | |
| 441 ** start of month | |
| 442 ** start of year | |
| 443 ** start of week | |
| 444 ** start of day | |
| 445 ** weekday N | |
| 446 ** unixepoch | |
| 447 ** localtime | |
| 448 ** utc | |
| 449 ** | |
| 450 ** Return 0 on success and 1 if there is any kind of error. | |
| 451 */ | |
| 452 static int parseModifier(const char *zMod, DateTime *p){ | |
| 453 int rc = 1; | |
| 454 int n; | |
| 455 double r; | |
| 456 char *z, zBuf[30]; | |
| 457 z = zBuf; | |
| 458 for(n=0; n<sizeof(zBuf)-1 && zMod[n]; n++){ | |
| 459 z[n] = tolower(zMod[n]); | |
| 460 } | |
| 461 z[n] = 0; | |
| 462 switch( z[0] ){ | |
| 463 case 'l': { | |
| 464 /* localtime | |
| 465 ** | |
| 466 ** Assuming the current time value is UTC (a.k.a. GMT), shift it to | |
| 467 ** show local time. | |
| 468 */ | |
| 469 if( strcmp(z, "localtime")==0 ){ | |
| 470 computeJD(p); | |
| 471 p->rJD += localtimeOffset(p); | |
| 472 clearYMD_HMS_TZ(p); | |
| 473 rc = 0; | |
| 474 } | |
| 475 break; | |
| 476 } | |
| 477 case 'u': { | |
| 478 /* | |
| 479 ** unixepoch | |
| 480 ** | |
| 481 ** Treat the current value of p->rJD as the number of | |
| 482 ** seconds since 1970. Convert to a real julian day number. | |
| 483 */ | |
| 484 if( strcmp(z, "unixepoch")==0 && p->validJD ){ | |
| 485 p->rJD = p->rJD/86400.0 + 2440587.5; | |
| 486 clearYMD_HMS_TZ(p); | |
| 487 rc = 0; | |
| 488 }else if( strcmp(z, "utc")==0 ){ | |
| 489 double c1; | |
| 490 computeJD(p); | |
| 491 c1 = localtimeOffset(p); | |
| 492 p->rJD -= c1; | |
| 493 clearYMD_HMS_TZ(p); | |
| 494 p->rJD += c1 - localtimeOffset(p); | |
| 495 rc = 0; | |
| 496 } | |
| 497 break; | |
| 498 } | |
| 499 case 'w': { | |
| 500 /* | |
| 501 ** weekday N | |
| 502 ** | |
| 503 ** Move the date to the same time on the next occurrence of | |
| 504 ** weekday N where 0==Sunday, 1==Monday, and so forth. If the | |
| 505 ** date is already on the appropriate weekday, this is a no-op. | |
| 506 */ | |
| 507 if( strncmp(z, "weekday ", 8)==0 && getValue(&z[8],&r)>0 | |
| 508 && (n=r)==r && n>=0 && r<7 ){ | |
| 509 int Z; | |
| 510 computeYMD_HMS(p); | |
| 511 p->validTZ = 0; | |
| 512 p->validJD = 0; | |
| 513 computeJD(p); | |
| 514 Z = p->rJD + 1.5; | |
| 515 Z %= 7; | |
| 516 if( Z>n ) Z -= 7; | |
| 517 p->rJD += n - Z; | |
| 518 clearYMD_HMS_TZ(p); | |
| 519 rc = 0; | |
| 520 } | |
| 521 break; | |
| 522 } | |
| 523 case 's': { | |
| 524 /* | |
| 525 ** start of TTTTT | |
| 526 ** | |
| 527 ** Move the date backwards to the beginning of the current day, | |
| 528 ** or month or year. | |
| 529 */ | |
| 530 if( strncmp(z, "start of ", 9)!=0 ) break; | |
| 531 z += 9; | |
| 532 computeYMD(p); | |
| 533 p->validHMS = 1; | |
| 534 p->h = p->m = 0; | |
| 535 p->s = 0.0; | |
| 536 p->validTZ = 0; | |
| 537 p->validJD = 0; | |
| 538 if( strcmp(z,"month")==0 ){ | |
| 539 p->D = 1; | |
| 540 rc = 0; | |
| 541 }else if( strcmp(z,"year")==0 ){ | |
| 542 computeYMD(p); | |
| 543 p->M = 1; | |
| 544 p->D = 1; | |
| 545 rc = 0; | |
| 546 }else if( strcmp(z,"day")==0 ){ | |
| 547 rc = 0; | |
| 548 } | |
| 549 break; | |
| 550 } | |
| 551 case '+': | |
| 552 case '-': | |
| 553 case '0': | |
| 554 case '1': | |
| 555 case '2': | |
| 556 case '3': | |
| 557 case '4': | |
| 558 case '5': | |
| 559 case '6': | |
| 560 case '7': | |
| 561 case '8': | |
| 562 case '9': { | |
| 563 n = getValue(z, &r); | |
| 564 if( n<=0 ) break; | |
| 565 if( z[n]==':' ){ | |
| 566 /* A modifier of the form (+|-)HH:MM:SS.FFF adds (or subtracts) the | |
| 567 ** specified number of hours, minutes, seconds, and fractional seconds | |
| 568 ** to the time. The ".FFF" may be omitted. The ":SS.FFF" may be | |
| 569 ** omitted. | |
| 570 */ | |
| 571 const char *z2 = z; | |
| 572 DateTime tx; | |
| 573 int day; | |
| 574 if( !isdigit(*(u8*)z2) ) z2++; | |
| 575 memset(&tx, 0, sizeof(tx)); | |
| 576 if( parseHhMmSs(z2, &tx) ) break; | |
| 577 computeJD(&tx); | |
| 578 tx.rJD -= 0.5; | |
| 579 day = (int)tx.rJD; | |
| 580 tx.rJD -= day; | |
| 581 if( z[0]=='-' ) tx.rJD = -tx.rJD; | |
| 582 computeJD(p); | |
| 583 clearYMD_HMS_TZ(p); | |
| 584 p->rJD += tx.rJD; | |
| 585 rc = 0; | |
| 586 break; | |
| 587 } | |
| 588 z += n; | |
| 589 while( isspace(*(u8*)z) ) z++; | |
| 590 n = strlen(z); | |
| 591 if( n>10 || n<3 ) break; | |
| 592 if( z[n-1]=='s' ){ z[n-1] = 0; n--; } | |
| 593 computeJD(p); | |
| 594 rc = 0; | |
| 595 if( n==3 && strcmp(z,"day")==0 ){ | |
| 596 p->rJD += r; | |
| 597 }else if( n==4 && strcmp(z,"hour")==0 ){ | |
| 598 p->rJD += r/24.0; | |
| 599 }else if( n==6 && strcmp(z,"minute")==0 ){ | |
| 600 p->rJD += r/(24.0*60.0); | |
| 601 }else if( n==6 && strcmp(z,"second")==0 ){ | |
| 602 p->rJD += r/(24.0*60.0*60.0); | |
| 603 }else if( n==5 && strcmp(z,"month")==0 ){ | |
| 604 int x, y; | |
| 605 computeYMD_HMS(p); | |
| 606 p->M += r; | |
| 607 x = p->M>0 ? (p->M-1)/12 : (p->M-12)/12; | |
| 608 p->Y += x; | |
| 609 p->M -= x*12; | |
| 610 p->validJD = 0; | |
| 611 computeJD(p); | |
| 612 y = r; | |
| 613 if( y!=r ){ | |
| 614 p->rJD += (r - y)*30.0; | |
| 615 } | |
| 616 }else if( n==4 && strcmp(z,"year")==0 ){ | |
| 617 computeYMD_HMS(p); | |
| 618 p->Y += r; | |
| 619 p->validJD = 0; | |
| 620 computeJD(p); | |
| 621 }else{ | |
| 622 rc = 1; | |
| 623 } | |
| 624 clearYMD_HMS_TZ(p); | |
| 625 break; | |
| 626 } | |
| 627 default: { | |
| 628 break; | |
| 629 } | |
| 630 } | |
| 631 return rc; | |
| 632 } | |
| 633 | |
| 634 /* | |
| 635 ** Process time function arguments. argv[0] is a date-time stamp. | |
| 636 ** argv[1] and following are modifiers. Parse them all and write | |
| 637 ** the resulting time into the DateTime structure p. Return 0 | |
| 638 ** on success and 1 if there are any errors. | |
| 639 */ | |
| 640 static int isDate(int argc, sqlite3_value **argv, DateTime *p){ | |
| 641 int i; | |
| 642 if( argc==0 ) return 1; | |
| 643 if( SQLITE_NULL==sqlite3_value_type(argv[0]) || | |
| 644 parseDateOrTime((char*)sqlite3_value_text(argv[0]), p) ) return 1; | |
| 645 for(i=1; i<argc; i++){ | |
| 646 if( SQLITE_NULL==sqlite3_value_type(argv[i]) || | |
| 647 parseModifier((char*)sqlite3_value_text(argv[i]), p) ) return 1; | |
| 648 } | |
| 649 return 0; | |
| 650 } | |
| 651 | |
| 652 | |
| 653 /* | |
| 654 ** The following routines implement the various date and time functions | |
| 655 ** of SQLite. | |
| 656 */ | |
| 657 | |
| 658 /* | |
| 659 ** julianday( TIMESTRING, MOD, MOD, ...) | |
| 660 ** | |
| 661 ** Return the julian day number of the date specified in the arguments | |
| 662 */ | |
| 663 static void juliandayFunc( | |
| 664 sqlite3_context *context, | |
| 665 int argc, | |
| 666 sqlite3_value **argv | |
| 667 ){ | |
| 668 DateTime x; | |
| 669 if( isDate(argc, argv, &x)==0 ){ | |
| 670 computeJD(&x); | |
| 671 sqlite3_result_double(context, x.rJD); | |
| 672 } | |
| 673 } | |
| 674 | |
| 675 /* | |
| 676 ** datetime( TIMESTRING, MOD, MOD, ...) | |
| 677 ** | |
| 678 ** Return YYYY-MM-DD HH:MM:SS | |
| 679 */ | |
| 680 static void datetimeFunc( | |
| 681 sqlite3_context *context, | |
| 682 int argc, | |
| 683 sqlite3_value **argv | |
| 684 ){ | |
| 685 DateTime x; | |
| 686 if( isDate(argc, argv, &x)==0 ){ | |
| 687 char zBuf[100]; | |
| 688 computeYMD_HMS(&x); | |
| 689 sprintf(zBuf, "%04d-%02d-%02d %02d:%02d:%02d",x.Y, x.M, x.D, x.h, x.m, | |
| 690 (int)(x.s)); | |
| 691 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT); | |
| 692 } | |
| 693 } | |
| 694 | |
| 695 /* | |
| 696 ** time( TIMESTRING, MOD, MOD, ...) | |
| 697 ** | |
| 698 ** Return HH:MM:SS | |
| 699 */ | |
| 700 static void timeFunc( | |
| 701 sqlite3_context *context, | |
| 702 int argc, | |
| 703 sqlite3_value **argv | |
| 704 ){ | |
| 705 DateTime x; | |
| 706 if( isDate(argc, argv, &x)==0 ){ | |
| 707 char zBuf[100]; | |
| 708 computeHMS(&x); | |
| 709 sprintf(zBuf, "%02d:%02d:%02d", x.h, x.m, (int)x.s); | |
| 710 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT); | |
| 711 } | |
| 712 } | |
| 713 | |
| 714 /* | |
| 715 ** date( TIMESTRING, MOD, MOD, ...) | |
| 716 ** | |
| 717 ** Return YYYY-MM-DD | |
| 718 */ | |
| 719 static void dateFunc( | |
| 720 sqlite3_context *context, | |
| 721 int argc, | |
| 722 sqlite3_value **argv | |
| 723 ){ | |
| 724 DateTime x; | |
| 725 if( isDate(argc, argv, &x)==0 ){ | |
| 726 char zBuf[100]; | |
| 727 computeYMD(&x); | |
| 728 sprintf(zBuf, "%04d-%02d-%02d", x.Y, x.M, x.D); | |
| 729 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT); | |
| 730 } | |
| 731 } | |
| 732 | |
| 733 /* | |
| 734 ** strftime( FORMAT, TIMESTRING, MOD, MOD, ...) | |
| 735 ** | |
| 736 ** Return a string described by FORMAT. Conversions as follows: | |
| 737 ** | |
| 738 ** %d day of month | |
| 739 ** %f ** fractional seconds SS.SSS | |
| 740 ** %H hour 00-24 | |
| 741 ** %j day of year 000-366 | |
| 742 ** %J ** Julian day number | |
| 743 ** %m month 01-12 | |
| 744 ** %M minute 00-59 | |
| 745 ** %s seconds since 1970-01-01 | |
| 746 ** %S seconds 00-59 | |
| 747 ** %w day of week 0-6 sunday==0 | |
| 748 ** %W week of year 00-53 | |
| 749 ** %Y year 0000-9999 | |
| 750 ** %% % | |
| 751 */ | |
| 752 static void strftimeFunc( | |
| 753 sqlite3_context *context, | |
| 754 int argc, | |
| 755 sqlite3_value **argv | |
| 756 ){ | |
| 757 DateTime x; | |
| 758 int n, i, j; | |
| 759 char *z; | |
| 760 const char *zFmt = (const char*)sqlite3_value_text(argv[0]); | |
| 761 char zBuf[100]; | |
| 762 if( zFmt==0 || isDate(argc-1, argv+1, &x) ) return; | |
| 763 for(i=0, n=1; zFmt[i]; i++, n++){ | |
| 764 if( zFmt[i]=='%' ){ | |
| 765 switch( zFmt[i+1] ){ | |
| 766 case 'd': | |
| 767 case 'H': | |
| 768 case 'm': | |
| 769 case 'M': | |
| 770 case 'S': | |
| 771 case 'W': | |
| 772 n++; | |
| 773 /* fall thru */ | |
| 774 case 'w': | |
| 775 case '%': | |
| 776 break; | |
| 777 case 'f': | |
| 778 n += 8; | |
| 779 break; | |
| 780 case 'j': | |
| 781 n += 3; | |
| 782 break; | |
| 783 case 'Y': | |
| 784 n += 8; | |
| 785 break; | |
| 786 case 's': | |
| 787 case 'J': | |
| 788 n += 50; | |
| 789 break; | |
| 790 default: | |
| 791 return; /* ERROR. return a NULL */ | |
| 792 } | |
| 793 i++; | |
| 794 } | |
| 795 } | |
| 796 if( n<sizeof(zBuf) ){ | |
| 797 z = zBuf; | |
| 798 }else{ | |
| 799 z = sqliteMalloc( n ); | |
| 800 if( z==0 ) return; | |
| 801 } | |
| 802 computeJD(&x); | |
| 803 computeYMD_HMS(&x); | |
| 804 for(i=j=0; zFmt[i]; i++){ | |
| 805 if( zFmt[i]!='%' ){ | |
| 806 z[j++] = zFmt[i]; | |
| 807 }else{ | |
| 808 i++; | |
| 809 switch( zFmt[i] ){ | |
| 810 case 'd': sprintf(&z[j],"%02d",x.D); j+=2; break; | |
| 811 case 'f': { | |
| 812 int s = x.s; | |
| 813 int ms = (x.s - s)*1000.0; | |
| 814 sprintf(&z[j],"%02d.%03d",s,ms); | |
| 815 j += strlen(&z[j]); | |
| 816 break; | |
| 817 } | |
| 818 case 'H': sprintf(&z[j],"%02d",x.h); j+=2; break; | |
| 819 case 'W': /* Fall thru */ | |
| 820 case 'j': { | |
| 821 int nDay; /* Number of days since 1st day of year */ | |
| 822 DateTime y = x; | |
| 823 y.validJD = 0; | |
| 824 y.M = 1; | |
| 825 y.D = 1; | |
| 826 computeJD(&y); | |
| 827 nDay = x.rJD - y.rJD; | |
| 828 if( zFmt[i]=='W' ){ | |
| 829 int wd; /* 0=Monday, 1=Tuesday, ... 6=Sunday */ | |
| 830 wd = ((int)(x.rJD+0.5)) % 7; | |
| 831 sprintf(&z[j],"%02d",(nDay+7-wd)/7); | |
| 832 j += 2; | |
| 833 }else{ | |
| 834 sprintf(&z[j],"%03d",nDay+1); | |
| 835 j += 3; | |
| 836 } | |
| 837 break; | |
| 838 } | |
| 839 case 'J': sprintf(&z[j],"%.16g",x.rJD); j+=strlen(&z[j]); break; | |
| 840 case 'm': sprintf(&z[j],"%02d",x.M); j+=2; break; | |
| 841 case 'M': sprintf(&z[j],"%02d",x.m); j+=2; break; | |
| 842 case 's': { | |
| 843 sprintf(&z[j],"%d",(int)((x.rJD-2440587.5)*86400.0 + 0.5)); | |
| 844 j += strlen(&z[j]); | |
| 845 break; | |
| 846 } | |
| 847 case 'S': sprintf(&z[j],"%02d",(int)(x.s+0.5)); j+=2; break; | |
| 848 case 'w': z[j++] = (((int)(x.rJD+1.5)) % 7) + '0'; break; | |
| 849 case 'Y': sprintf(&z[j],"%04d",x.Y); j+=strlen(&z[j]); break; | |
| 850 case '%': z[j++] = '%'; break; | |
| 851 } | |
| 852 } | |
| 853 } | |
| 854 z[j] = 0; | |
| 855 sqlite3_result_text(context, z, -1, SQLITE_TRANSIENT); | |
| 856 if( z!=zBuf ){ | |
| 857 sqliteFree(z); | |
| 858 } | |
| 859 } | |
| 860 | |
| 861 /* | |
| 862 ** current_time() | |
| 863 ** | |
| 864 ** This function returns the same value as time('now'). | |
| 865 */ | |
| 866 static void ctimeFunc( | |
| 867 sqlite3_context *context, | |
| 868 int argc, | |
| 869 sqlite3_value **argv | |
| 870 ){ | |
| 871 sqlite3_value *pVal = sqlite3ValueNew(); | |
| 872 if( pVal ){ | |
| 873 sqlite3ValueSetStr(pVal, -1, "now", SQLITE_UTF8, SQLITE_STATIC); | |
| 874 timeFunc(context, 1, &pVal); | |
| 875 sqlite3ValueFree(pVal); | |
| 876 } | |
| 877 } | |
| 878 | |
| 879 /* | |
| 880 ** current_date() | |
| 881 ** | |
| 882 ** This function returns the same value as date('now'). | |
| 883 */ | |
| 884 static void cdateFunc( | |
| 885 sqlite3_context *context, | |
| 886 int argc, | |
| 887 sqlite3_value **argv | |
| 888 ){ | |
| 889 sqlite3_value *pVal = sqlite3ValueNew(); | |
| 890 if( pVal ){ | |
| 891 sqlite3ValueSetStr(pVal, -1, "now", SQLITE_UTF8, SQLITE_STATIC); | |
| 892 dateFunc(context, 1, &pVal); | |
| 893 sqlite3ValueFree(pVal); | |
| 894 } | |
| 895 } | |
| 896 | |
| 897 /* | |
| 898 ** current_timestamp() | |
| 899 ** | |
| 900 ** This function returns the same value as datetime('now'). | |
| 901 */ | |
| 902 static void ctimestampFunc( | |
| 903 sqlite3_context *context, | |
| 904 int argc, | |
| 905 sqlite3_value **argv | |
| 906 ){ | |
| 907 sqlite3_value *pVal = sqlite3ValueNew(); | |
| 908 if( pVal ){ | |
| 909 sqlite3ValueSetStr(pVal, -1, "now", SQLITE_UTF8, SQLITE_STATIC); | |
| 910 datetimeFunc(context, 1, &pVal); | |
| 911 sqlite3ValueFree(pVal); | |
| 912 } | |
| 913 } | |
| 914 #endif /* !defined(SQLITE_OMIT_DATETIME_FUNCS) */ | |
| 915 | |
| 916 #ifdef SQLITE_OMIT_DATETIME_FUNCS | |
| 917 /* | |
| 918 ** If the library is compiled to omit the full-scale date and time | |
| 919 ** handling (to get a smaller binary), the following minimal version | |
| 920 ** of the functions current_time(), current_date() and current_timestamp() | |
| 921 ** are included instead. This is to support column declarations that | |
| 922 ** include "DEFAULT CURRENT_TIME" etc. | |
| 923 ** | |
| 924 ** This function uses the C-library functions time(), gmtime() | |
| 925 ** and strftime(). The format string to pass to strftime() is supplied | |
| 926 ** as the user-data for the function. | |
| 927 */ | |
| 928 static void currentTimeFunc( | |
| 929 sqlite3_context *context, | |
| 930 int argc, | |
| 931 sqlite3_value **argv | |
| 932 ){ | |
| 933 time_t t; | |
| 934 char *zFormat = (char *)sqlite3_user_data(context); | |
| 935 char zBuf[20]; | |
| 936 | |
| 937 time(&t); | |
| 938 #ifdef SQLITE_TEST | |
| 939 { | |
| 940 extern int sqlite3_current_time; /* See os_XXX.c */ | |
| 941 if( sqlite3_current_time ){ | |
| 942 t = sqlite3_current_time; | |
| 943 } | |
| 944 } | |
| 945 #endif | |
| 946 | |
| 947 sqlite3OsEnterMutex(); | |
| 948 strftime(zBuf, 20, zFormat, gmtime(&t)); | |
| 949 sqlite3OsLeaveMutex(); | |
| 950 | |
| 951 sqlite3_result_text(context, zBuf, -1, SQLITE_TRANSIENT); | |
| 952 } | |
| 953 #endif | |
| 954 | |
| 955 /* | |
| 956 ** This function registered all of the above C functions as SQL | |
| 957 ** functions. This should be the only routine in this file with | |
| 958 ** external linkage. | |
| 959 */ | |
| 960 void sqlite3RegisterDateTimeFunctions(sqlite3 *db){ | |
| 961 #ifndef SQLITE_OMIT_DATETIME_FUNCS | |
| 962 static const struct { | |
| 963 char *zName; | |
| 964 int nArg; | |
| 965 void (*xFunc)(sqlite3_context*,int,sqlite3_value**); | |
| 966 } aFuncs[] = { | |
| 967 { "julianday", -1, juliandayFunc }, | |
| 968 { "date", -1, dateFunc }, | |
| 969 { "time", -1, timeFunc }, | |
| 970 { "datetime", -1, datetimeFunc }, | |
| 971 { "strftime", -1, strftimeFunc }, | |
| 972 { "current_time", 0, ctimeFunc }, | |
| 973 { "current_timestamp", 0, ctimestampFunc }, | |
| 974 { "current_date", 0, cdateFunc }, | |
| 975 }; | |
| 976 int i; | |
| 977 | |
| 978 for(i=0; i<sizeof(aFuncs)/sizeof(aFuncs[0]); i++){ | |
| 979 sqlite3CreateFunc(db, aFuncs[i].zName, aFuncs[i].nArg, | |
| 980 SQLITE_UTF8, 0, aFuncs[i].xFunc, 0, 0); | |
| 981 } | |
| 982 #else | |
| 983 static const struct { | |
| 984 char *zName; | |
| 985 char *zFormat; | |
| 986 } aFuncs[] = { | |
| 987 { "current_time", "%H:%M:%S" }, | |
| 988 { "current_date", "%Y-%m-%d" }, | |
| 989 { "current_timestamp", "%Y-%m-%d %H:%M:%S" } | |
| 990 }; | |
| 991 int i; | |
| 992 | |
| 993 for(i=0; i<sizeof(aFuncs)/sizeof(aFuncs[0]); i++){ | |
| 994 sqlite3CreateFunc(db, aFuncs[i].zName, 0, SQLITE_UTF8, | |
| 995 aFuncs[i].zFormat, currentTimeFunc, 0, 0); | |
| 996 } | |
| 997 #endif | |
| 998 } |
