Skip to content

Commit fe2bbde

Browse files
committed
feat(tdp): format dates
Add special case for close unix ts.
1 parent b642a14 commit fe2bbde

File tree

1 file changed

+18
-4
lines changed

1 file changed

+18
-4
lines changed

tdp/tdp.go

Lines changed: 18 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@ import (
88
"reflect"
99
"strconv"
1010
"strings"
11+
"time"
1112
)
1213

1314
// options for formatting.
@@ -28,14 +29,14 @@ const (
2829
noIdent = ""
2930
)
3031

31-
func formatValue(b *strings.Builder, prefix string, opt options, v reflect.Value) {
32+
func formatValue(b *strings.Builder, prefix, fieldName string, opt options, v reflect.Value) {
3233
switch v.Kind() {
3334
case reflect.Struct, reflect.Ptr, reflect.Interface:
3435
i, ok := v.Interface().(Object)
3536
if ok {
3637
format(b, prefix+defaultIdent, opt, i)
3738
} else if v.CanAddr() {
38-
formatValue(b, prefix, opt, v.Addr())
39+
formatValue(b, prefix, fieldName, opt, v.Addr())
3940
}
4041
case reflect.Slice:
4142
if buf, ok := v.Interface().([]byte); ok {
@@ -49,9 +50,22 @@ func formatValue(b *strings.Builder, prefix string, opt options, v reflect.Value
4950
b.WriteString(prefix)
5051
b.WriteString(defaultIdent)
5152
b.WriteString("- ")
52-
formatValue(b, prefix+defaultIdent, opt, vi)
53+
formatValue(b, prefix+defaultIdent, fieldName, opt, vi)
5354
b.WriteRune('\n')
5455
}
56+
case reflect.Int:
57+
// Special case for date.
58+
var (
59+
now = time.Now()
60+
max = now.AddDate(0, 0, 7).Unix()
61+
min = now.AddDate(-2, 0, 0).Unix()
62+
)
63+
i := v.Int()
64+
if i > min && i < max && strings.Contains(fieldName, "date") {
65+
b.WriteString(time.Unix(i, 0).UTC().Format(time.RFC3339))
66+
} else {
67+
b.WriteString(strconv.FormatInt(i, 10))
68+
}
5569
default:
5670
b.WriteString(fmt.Sprint(v.Interface()))
5771
}
@@ -91,7 +105,7 @@ func format(b *strings.Builder, prefix string, opt options, obj Object) {
91105
b.WriteString(f.SchemaName)
92106
b.WriteString(": ")
93107

94-
formatValue(b, prefix, opt, v.FieldByName(f.Name))
108+
formatValue(b, prefix, f.SchemaName, opt, v.FieldByName(f.Name))
95109
}
96110
}
97111

0 commit comments

Comments
 (0)