forked from ebhomengo/niki
501 lines
16 KiB
Go
501 lines
16 KiB
Go
|
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
|
||
|
},
|
||
|
})
|
||
|
|
||
|
}
|