niki/vendor/github.com/brianvoe/gofakeit/v6/time.go

501 lines
16 KiB
Go
Raw Permalink Normal View History

2024-04-26 19:30:35 +00:00
package gofakeit
import (
"math/rand"
"strconv"
"strings"
"time"
)
// Date will generate a random time.Time struct
func Date() time.Time { return date(globalFaker.Rand) }
// Date will generate a random time.Time struct
func (f *Faker) Date() time.Time { return date(f.Rand) }
func date(r *rand.Rand) time.Time {
return time.Date(year(r), time.Month(month(r)), day(r), hour(r), minute(r), second(r), nanoSecond(r), time.UTC)
}
// FutureDate will generate a random past time.Time struct
func PastDate() time.Time { return pastDate(globalFaker.Rand) }
// FutureDate will generate a random past time.Time struct
func (f *Faker) PastDate() time.Time { return pastDate(f.Rand) }
func pastDate(r *rand.Rand) time.Time {
return time.Now().Add(time.Hour * -time.Duration(number(r, 1, 12)))
}
// FutureDate will generate a random future time.Time struct
func FutureDate() time.Time { return futureDate(globalFaker.Rand) }
// FutureDate will generate a random future time.Time struct
func (f *Faker) FutureDate() time.Time { return futureDate(f.Rand) }
func futureDate(r *rand.Rand) time.Time {
return time.Now().Add(time.Hour * time.Duration(number(r, 1, 12)))
}
// DateRange will generate a random time.Time struct between a start and end date
func DateRange(start, end time.Time) time.Time { return dateRange(globalFaker.Rand, start, end) }
// DateRange will generate a random time.Time struct between a start and end date
func (f *Faker) DateRange(start, end time.Time) time.Time { return dateRange(f.Rand, start, end) }
func dateRange(r *rand.Rand, start time.Time, end time.Time) time.Time {
return time.Unix(0, int64(number(r, int(start.UnixNano()), int(end.UnixNano())))).UTC()
}
// NanoSecond will generate a random nano second
func NanoSecond() int { return nanoSecond(globalFaker.Rand) }
// NanoSecond will generate a random nano second
func (f *Faker) NanoSecond() int { return nanoSecond(f.Rand) }
func nanoSecond(r *rand.Rand) int { return number(r, 0, 999999999) }
// Second will generate a random second
func Second() int { return second(globalFaker.Rand) }
// Second will generate a random second
func (f *Faker) Second() int { return second(f.Rand) }
func second(r *rand.Rand) int { return number(r, 0, 59) }
// Minute will generate a random minute
func Minute() int { return minute(globalFaker.Rand) }
// Minute will generate a random minute
func (f *Faker) Minute() int { return minute(f.Rand) }
func minute(r *rand.Rand) int { return number(r, 0, 59) }
// Hour will generate a random hour - in military time
func Hour() int { return hour(globalFaker.Rand) }
// Hour will generate a random hour - in military time
func (f *Faker) Hour() int { return hour(f.Rand) }
func hour(r *rand.Rand) int { return number(r, 0, 23) }
// Day will generate a random day between 1 - 31
func Day() int { return day(globalFaker.Rand) }
// Day will generate a random day between 1 - 31
func (f *Faker) Day() int { return day(f.Rand) }
func day(r *rand.Rand) int { return number(r, 1, 31) }
// WeekDay will generate a random weekday string (Monday-Sunday)
func WeekDay() string { return weekDay(globalFaker.Rand) }
// WeekDay will generate a random weekday string (Monday-Sunday)
func (f *Faker) WeekDay() string { return weekDay(f.Rand) }
func weekDay(r *rand.Rand) string { return time.Weekday(number(r, 0, 6)).String() }
// Month will generate a random month int
func Month() int { return month(globalFaker.Rand) }
// Month will generate a random month int
func (f *Faker) Month() int { return month(f.Rand) }
func month(r *rand.Rand) int { return number(r, 1, 12) }
// MonthString will generate a random month string
func MonthString() string { return monthString(globalFaker.Rand) }
// MonthString will generate a random month string
func (f *Faker) MonthString() string { return monthString(f.Rand) }
func monthString(r *rand.Rand) string { return time.Month(number(r, 1, 12)).String() }
// Year will generate a random year between 1900 - current year
func Year() int { return year(globalFaker.Rand) }
// Year will generate a random year between 1900 - current year
func (f *Faker) Year() int { return year(f.Rand) }
func year(r *rand.Rand) int { return number(r, 1900, time.Now().Year()) }
// TimeZone will select a random timezone string
func TimeZone() string { return timeZone(globalFaker.Rand) }
// TimeZone will select a random timezone string
func (f *Faker) TimeZone() string { return timeZone(f.Rand) }
func timeZone(r *rand.Rand) string { return getRandValue(r, []string{"timezone", "text"}) }
// TimeZoneFull will select a random full timezone string
func TimeZoneFull() string { return timeZoneFull(globalFaker.Rand) }
// TimeZoneFull will select a random full timezone string
func (f *Faker) TimeZoneFull() string { return timeZoneFull(f.Rand) }
func timeZoneFull(r *rand.Rand) string { return getRandValue(r, []string{"timezone", "full"}) }
// TimeZoneRegion will select a random region style timezone string, e.g. "America/Chicago"
func TimeZoneRegion() string { return timeZoneRegion(globalFaker.Rand) }
// TimeZoneRegion will select a random region style timezone string, e.g. "America/Chicago"
func (f *Faker) TimeZoneRegion() string { return timeZoneRegion(f.Rand) }
func timeZoneRegion(r *rand.Rand) string { return getRandValue(r, []string{"timezone", "region"}) }
// TimeZoneAbv will select a random timezone abbreviation string
func TimeZoneAbv() string { return timeZoneAbv(globalFaker.Rand) }
// TimeZoneAbv will select a random timezone abbreviation string
func (f *Faker) TimeZoneAbv() string { return timeZoneAbv(f.Rand) }
func timeZoneAbv(r *rand.Rand) string { return getRandValue(r, []string{"timezone", "abr"}) }
// TimeZoneOffset will select a random timezone offset
func TimeZoneOffset() float32 { return timeZoneOffset(globalFaker.Rand) }
// TimeZoneOffset will select a random timezone offset
func (f *Faker) TimeZoneOffset() float32 { return timeZoneOffset(f.Rand) }
func timeZoneOffset(r *rand.Rand) float32 {
value, _ := strconv.ParseFloat(getRandValue(r, []string{"timezone", "offset"}), 32)
return float32(value)
}
// javaDateFormatToGolangDateFormat converts java date format into go date format
func javaDateFormatToGolangDateFormat(format string) string {
format = strings.Replace(format, "ddd", "_2", -1)
format = strings.Replace(format, "dd", "02", -1)
format = strings.Replace(format, "d", "2", -1)
format = strings.Replace(format, "HH", "15", -1)
format = strings.Replace(format, "hh", "03", -1)
format = strings.Replace(format, "h", "3", -1)
format = strings.Replace(format, "mm", "04", -1)
format = strings.Replace(format, "m", "4", -1)
format = strings.Replace(format, "ss", "05", -1)
format = strings.Replace(format, "s", "5", -1)
format = strings.Replace(format, "yyyy", "2006", -1)
format = strings.Replace(format, "yy", "06", -1)
format = strings.Replace(format, "y", "06", -1)
format = strings.Replace(format, "SSS", "000", -1)
format = strings.Replace(format, "a", "pm", -1)
format = strings.Replace(format, "aa", "PM", -1)
format = strings.Replace(format, "MMMM", "January", -1)
format = strings.Replace(format, "MMM", "Jan", -1)
format = strings.Replace(format, "MM", "01", -1)
format = strings.Replace(format, "M", "1", -1)
format = strings.Replace(format, "ZZ", "-0700", -1)
if !strings.Contains(format, "Z07") {
format = strings.Replace(format, "Z", "-07", -1)
}
format = strings.Replace(format, "zz:zz", "Z07:00", -1)
format = strings.Replace(format, "zzzz", "Z0700", -1)
format = strings.Replace(format, "z", "MST", -1)
format = strings.Replace(format, "EEEE", "Monday", -1)
format = strings.Replace(format, "E", "Mon", -1)
return format
}
func addDateTimeLookup() {
AddFuncLookup("date", Info{
Display: "Date",
Category: "time",
Description: "Representation of a specific day, month, and year, often used for chronological reference",
Example: "2006-01-02T15:04:05Z07:00",
Output: "string",
Params: []Param{
{
Field: "format",
Display: "Format",
Type: "string",
Default: "RFC3339",
Options: []string{"ANSIC", "UnixDate", "RubyDate", "RFC822", "RFC822Z", "RFC850", "RFC1123", "RFC1123Z", "RFC3339", "RFC3339Nano"},
Description: "Date time string format output. You may also use golang time format or java time format",
},
},
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
format, err := info.GetString(m, "format")
if err != nil {
return nil, err
}
switch format {
case "ANSIC":
return Date().Format(time.ANSIC), nil
case "UnixDate":
return Date().Format(time.UnixDate), nil
case "RubyDate":
return Date().Format(time.RubyDate), nil
case "RFC822":
return Date().Format(time.RFC822), nil
case "RFC822Z":
return Date().Format(time.RFC822Z), nil
case "RFC850":
return Date().Format(time.RFC850), nil
case "RFC1123":
return Date().Format(time.RFC1123), nil
case "RFC1123Z":
return Date().Format(time.RFC1123Z), nil
case "RFC3339":
return Date().Format(time.RFC3339), nil
case "RFC3339Nano":
return Date().Format(time.RFC3339Nano), nil
default:
if format == "" {
return Date().Format(time.RFC3339), nil
}
return Date().Format(javaDateFormatToGolangDateFormat(format)), nil
}
},
})
AddFuncLookup("daterange", Info{
Display: "DateRange",
Category: "time",
Description: "Random date between two ranges",
Example: "2006-01-02T15:04:05Z07:00",
Output: "string",
Params: []Param{
{
Field: "startdate",
Display: "Start Date",
Type: "string",
Default: "1970-01-01",
Description: "Start date time string",
},
{
Field: "enddate",
Display: "End Date",
Type: "string",
Default: time.Now().Format("2006-01-02"),
Description: "End date time string",
},
{
Field: "format",
Display: "Format",
Type: "string",
Default: "yyyy-MM-dd",
Description: "Date time string format",
},
},
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
format, err := info.GetString(m, "format")
if err != nil {
return nil, err
}
format = javaDateFormatToGolangDateFormat(format)
startdate, err := info.GetString(m, "startdate")
if err != nil {
return nil, err
}
startDateTime, err := time.Parse(format, startdate)
if err != nil {
return nil, err
}
enddate, err := info.GetString(m, "enddate")
if err != nil {
return nil, err
}
endDateTime, err := time.Parse(format, enddate)
if err != nil {
return nil, err
}
return DateRange(startDateTime, endDateTime).Format(format), nil
},
})
AddFuncLookup("pasttime", Info{
Display: "PastTime",
Category: "time",
Description: "Date that has occurred before the current moment in time",
Example: "2007-01-24 13:00:35.820738079 +0000 UTC",
Output: "string",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return pastDate(r), nil
},
})
AddFuncLookup("futuretime", Info{
Display: "FutureTime",
Category: "time",
Description: "Date that has occurred after the current moment in time",
Example: "2107-01-24 13:00:35.820738079 +0000 UTC",
Output: "string",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return futureDate(r), nil
},
})
AddFuncLookup("nanosecond", Info{
Display: "Nanosecond",
Category: "time",
Description: "Unit of time equal to One billionth (10^-9) of a second",
Example: "196446360",
Output: "int",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return nanoSecond(r), nil
},
})
AddFuncLookup("second", Info{
Display: "Second",
Category: "time",
Description: "Unit of time equal to 1/60th of a minute",
Example: "43",
Output: "int",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return second(r), nil
},
})
AddFuncLookup("minute", Info{
Display: "Minute",
Category: "time",
Description: "Unit of time equal to 60 seconds",
Example: "34",
Output: "int",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return minute(r), nil
},
})
AddFuncLookup("hour", Info{
Display: "Hour",
Category: "time",
Description: "Unit of time equal to 60 minutes",
Example: "8",
Output: "int",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return hour(r), nil
},
})
AddFuncLookup("day", Info{
Display: "Day",
Category: "time",
Description: "24-hour period equivalent to one rotation of Earth on its axis",
Example: "12",
Output: "int",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return day(r), nil
},
})
AddFuncLookup("weekday", Info{
Display: "Weekday",
Category: "time",
Description: "Day of the week excluding the weekend",
Example: "Friday",
Output: "string",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return weekDay(r), nil
},
})
AddFuncLookup("month", Info{
Display: "Month",
Category: "time",
Description: "Division of the year, typically 30 or 31 days long",
Example: "1",
Output: "string",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return month(r), nil
},
})
AddFuncLookup("monthstring", Info{
Display: "Month String",
Category: "time",
Description: "String Representation of a month name",
Example: "September",
Output: "string",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return monthString(r), nil
},
})
AddFuncLookup("year", Info{
Display: "Year",
Category: "time",
Description: "Period of 365 days, the time Earth takes to orbit the Sun",
Example: "1900",
Output: "int",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return year(r), nil
},
})
AddFuncLookup("timezone", Info{
Display: "Timezone",
Category: "time",
Description: "Region where the same standard time is used, based on longitudinal divisions of the Earth",
Example: "Kaliningrad Standard Time",
Output: "string",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return timeZone(r), nil
},
})
AddFuncLookup("timezoneabv", Info{
Display: "Timezone Abbreviation",
Category: "time",
Description: "Abbreviated 3-letter word of a timezone",
Example: "KST",
Output: "string",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return timeZoneAbv(r), nil
},
})
AddFuncLookup("timezonefull", Info{
Display: "Timezone Full",
Category: "time",
Description: "Full name of a timezone",
Example: "(UTC+03:00) Kaliningrad, Minsk",
Output: "string",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return timeZoneFull(r), nil
},
})
AddFuncLookup("timezoneoffset", Info{
Display: "Timezone Offset",
Category: "time",
Description: "The difference in hours from Coordinated Universal Time (UTC) for a specific region",
Example: "3",
Output: "float32",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return timeZoneOffset(r), nil
},
})
AddFuncLookup("timezoneregion", Info{
Display: "Timezone Region",
Category: "time",
Description: "Geographic area sharing the same standard time",
Example: "America/Alaska",
Output: "string",
Generate: func(r *rand.Rand, m *MapParams, info *Info) (any, error) {
return timeZoneRegion(r), nil
},
})
}