std dt datetime-diff: fix uninitialized field ref when borrowing (#10466)

fixes #10455 

@KAAtheWiseGit, I'm sorry, I didn't mean to block your first PR #10461,
didn't see you had submitted it till I got around to submitting this. If
you want to incoporate useful ideas from this PR into yours, I do not
mind deferring to you.


# Description
Changes made in `datetime-diff`:
* Initialize millisecond and microsecond fields in `$current`, to fix
the error when borrow needs to refer to them.
* Fix `borrow_nanoseconds` to borrow from seconds, not from (unused)
microseconds.
* Added error check to insist that first argument is >= second argument.
`datetime-diff` doesn't represent negative durations correctly (it tries
to borrow out of the year, resulting in negative year and positive all
other fields). We don't currently have a use case requiring negative
durations.
* Add comments so help is a bit clearer (I was surprised that the first
argument, named `$from` was actually supposed to be the *later*
datetime. The order of arguments is reasonable (reminiscent of <later>
<minus> <earlier>), so I just changed the param name to match its
purpose.

Changes made in `pretty-print-duration`:
* changed type of argument from `duration` to `record`. (it's not clear
why Nu was not complaining about this!)
* changed test for skipping a clause from `> 0` to `!= 0`. Even though
`datetime-diff` won't present a negative field in the record, user might
call `pretty-print-duration` with one, might as well handle it. (but I
think `hour:-2` will be rendered as `-2hr`, not `-2hrs`...).
* added help and an example.

# User-Facing Changes
none requiring code changes.

# Tests + Formatting
- 🟢 `toolkit fmt`
- 🟢 `toolkit clippy`
- 🟢 `toolkit test`
- 🟢 `toolkit test stdlib`
- 
- # After Submitting
<!-- If your PR had any user-facing changes, update [the
documentation](https://github.com/nushell/nushell.github.io) after the
PR is merged, if necessary. This will help us keep the docs up to date.
-->
This commit is contained in:
Bob Hyman 2023-09-24 01:53:56 -07:00 committed by GitHub
parent 65c4083ae6
commit bc437da5c7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 84 additions and 40 deletions

View File

@ -79,7 +79,7 @@ def borrow-minute [from: record, current: record] {
def borrow-second [from: record, current: record] { def borrow-second [from: record, current: record] {
mut current = $current mut current = $current
$current.millisecond = $current.millisecond + 1_000 $current.nanosecond = $current.nanosecond + 1_000_000_000
$current.second = $current.second - 1 $current.second = $current.second - 1
if $current.second < 0 { if $current.second < 0 {
$current = (borrow-minute $from $current) $current = (borrow-minute $from $current)
@ -88,37 +88,33 @@ def borrow-second [from: record, current: record] {
$current $current
} }
def borrow-millisecond [from: record, current: record] { # Subtract later from earlier datetime and return the unit differences as a record
mut current = $current # Example:
$current.microsecond = $current.microsecond + 1_000_000 # > dt datetime-diff 2023-05-07T04:08:45+12:00 2019-05-10T09:59:12-07:00
$current.millisecond = $current.millisecond - 1 # ╭─────────────┬────╮
if $current.millisecond < 0 { # │ year │ 3 │
$current = (borrow-second $from $current) # │ month │ 11 │
# │ day │ 26 │
# │ hour │ 23 │
# │ minute │ 9 │
# │ second │ 33 │
# │ millisecond │ 0 │
# │ microsecond │ 0 │
# │ nanosecond │ 0 │
# ╰─────────────┴────╯
export def datetime-diff [
later: datetime, # a later datetime
earlier: datetime # earlier (starting) datetime
] {
if $earlier > $later {
let start = (metadata $later).span.start
let end = (metadata $earlier).span.end
error make {msg: "Incompatible arguments", label: {start:$start, end:$end, text:$"First datetime must be >= second, but was actually ($later - $earlier) less than it."}}
} }
let from_expanded = ($later | date to-timezone utc | date to-record)
let to_expanded = ($earlier | date to-timezone utc | date to-record)
$current mut result = { year: ($from_expanded.year - $to_expanded.year), month: ($from_expanded.month - $to_expanded.month), day:0, hour:0, minute:0, second:0, millisecond:0, microsecond:0, nanosecond:0}
}
def borrow-microsecond [from: record, current: record] {
mut current = $current
$current.nanosecond = $current.nanosecond + 1_000_000_000
$current.microsecond = $current.microsecond - 1
if $current.microsecond < 0 {
$current = (borrow-millisecond $from $current)
}
$current
}
# Subtract two datetimes and return a record with the difference
# Examples
# print (datetime-diff 2023-05-07T04:08:45+12:00 2019-05-10T09:59:12+12:00)
# print (datetime-diff (date now) 2019-05-10T09:59:12-07:00)
export def datetime-diff [from: datetime, to: datetime] {
let from_expanded = ($from | date to-timezone utc | date to-record | merge { millisecond: 0, microsecond: 0})
let to_expanded = ($to | date to-timezone utc | date to-record | merge { millisecond: 0, microsecond: 0})
mut result = { year: ($from_expanded.year - $to_expanded.year), month: ($from_expanded.month - $to_expanded.month)}
if $result.month < 0 { if $result.month < 0 {
$result = (borrow-year $from_expanded $result) $result = (borrow-year $from_expanded $result)
@ -156,65 +152,69 @@ export def datetime-diff [from: datetime, to: datetime] {
$result $result
} }
export def pretty-print-duration [dur: duration] { # Convert record from datetime-diff into humanized string
# Example:
# > dt pretty-print-duration (dt datetime-diff 2023-05-07T04:08:45+12:00 2019-05-10T09:59:12+12:00)
# 3yrs 11months 27days 18hrs 9mins 33secs
export def pretty-print-duration [dur: record] {
mut result = "" mut result = ""
if $dur.year > 0 { if $dur.year != 0 {
if $dur.year > 1 { if $dur.year > 1 {
$result = $"($dur.year)yrs " $result = $"($dur.year)yrs "
} else { } else {
$result = $"($dur.year)yr " $result = $"($dur.year)yr "
} }
} }
if $dur.month > 0 { if $dur.month != 0 {
if $dur.month > 1 { if $dur.month > 1 {
$result = $"($result)($dur.month)months " $result = $"($result)($dur.month)months "
} else { } else {
$result = $"($result)($dur.month)month " $result = $"($result)($dur.month)month "
} }
} }
if $dur.day > 0 { if $dur.day != 0 {
if $dur.day > 1 { if $dur.day > 1 {
$result = $"($result)($dur.day)days " $result = $"($result)($dur.day)days "
} else { } else {
$result = $"($result)($dur.day)day " $result = $"($result)($dur.day)day "
} }
} }
if $dur.hour > 0 { if $dur.hour != 0 {
if $dur.hour > 1 { if $dur.hour > 1 {
$result = $"($result)($dur.hour)hrs " $result = $"($result)($dur.hour)hrs "
} else { } else {
$result = $"($result)($dur.hour)hr " $result = $"($result)($dur.hour)hr "
} }
} }
if $dur.minute > 0 { if $dur.minute != 0 {
if $dur.minute > 1 { if $dur.minute > 1 {
$result = $"($result)($dur.minute)mins " $result = $"($result)($dur.minute)mins "
} else { } else {
$result = $"($result)($dur.minute)min " $result = $"($result)($dur.minute)min "
} }
} }
if $dur.second > 0 { if $dur.second != 0 {
if $dur.second > 1 { if $dur.second > 1 {
$result = $"($result)($dur.second)secs " $result = $"($result)($dur.second)secs "
} else { } else {
$result = $"($result)($dur.second)sec " $result = $"($result)($dur.second)sec "
} }
} }
if $dur.millisecond > 0 { if $dur.millisecond != 0 {
if $dur.millisecond > 1 { if $dur.millisecond > 1 {
$result = $"($result)($dur.millisecond)ms " $result = $"($result)($dur.millisecond)ms "
} else { } else {
$result = $"($result)($dur.millisecond)ms " $result = $"($result)($dur.millisecond)ms "
} }
} }
if $dur.microsecond > 0 { if $dur.microsecond != 0 {
if $dur.microsecond > 1 { if $dur.microsecond > 1 {
$result = $"($result)($dur.microsecond)µs " $result = $"($result)($dur.microsecond)µs "
} else { } else {
$result = $"($result)($dur.microsecond)µs " $result = $"($result)($dur.microsecond)µs "
} }
} }
if $dur.nanosecond > 0 { if $dur.nanosecond != 0 {
if $dur.nanosecond > 1 { if $dur.nanosecond > 1 {
$result = $"($result)($dur.nanosecond)ns " $result = $"($result)($dur.nanosecond)ns "
} else { } else {

View File

@ -0,0 +1,44 @@
use std assert
use std dt *
#[test]
def equal_times [] {
let t1 = (date now)
assert equal (datetime-diff $t1 $t1) ({year:0, month:0, day:0, hour:0, minute:0, second:0, millisecond:0, microsecond:0 nanosecond:0})
}
#[test]
def one_ns_later [] {
let t1 = (date now)
assert equal (datetime-diff ($t1 + 1ns) $t1) ({year:0, month:0, day:0, hour:0, minute:0, second:0, millisecond:0, microsecond:0 nanosecond:1})
}
#[test]
def one_yr_later [] {
let t1 = ('2022-10-1T0:1:2z' | into datetime) # a date for which one year later is 365 days, since duration doesn't support year or month
assert equal (datetime-diff ($t1 + 365day) $t1) ({year:1, month:0, day:0, hour:0, minute:0, second:0, millisecond:0, microsecond:0 nanosecond:0})
}
#[test]
def carry_ripples [] {
let t1 = ('2023-10-9T0:0:0z' | into datetime)
let t2 = ('2022-10-9T0:0:0.000000001z' | into datetime)
assert equal (datetime-diff $t1 $t2) ({year:0, month:11, day:30, hour:23, minute:59, second:59, millisecond:999, microsecond:999 nanosecond:999})
}
#[test]
def earlier_arg_must_be_less_or_equal_later [] {
let t1 = ('2022-10-9T0:0:0.000000001z' | into datetime)
let t2 = ('2023-10-9T0:0:0z' | into datetime)
assert error {|| (datetime-diff $t1 $t2)}
}
#[test]
def pp_skips_zeros [] {
assert equal (pretty-print-duration {year:1, month:0, day:0, hour:0, minute:0, second:0, millisecond:0, microsecond:0 nanosecond:0}) "1yr "
}
#[test]
def pp_doesnt_skip_neg [] { # datetime-diff can't return negative units, but prettyprint shouldn't skip them (if passed handcrafted record)
assert equal (pretty-print-duration {year:-1, month:0, day:0, hour:0, minute:0, second:0, millisecond:0, microsecond:0 nanosecond:0}) "-1yr "
}