examples/basic/xorm/data.go - master - githubmirror/ydb-go-sdk

examples/basic/xorm/data.go
219 строк7 KB
size12
enabled nlreturn (#1027)
02 фев 2024, 16:40
02 фев 2024, 16:40c6e45cd
package main

import (
	"time"

	"github.com/google/uuid"
)

func seriesData(id string, released time.Time, title, info, comment string) *Series {
	return &Series{
		ID:          id,
		Title:       title,
		Info:        info,
		ReleaseDate: released,
		Comment:     comment,
	}
}

func seasonData(seriesID, seasonID, title string, first, last time.Time) *Seasons {
	return &Seasons{
		ID:         seasonID,
		SeriesID:   seriesID,
		Title:      title,
		FirstAired: first,
		LastAired:  last,
	}
}

func episodeData(seasonID, episodeID, title string, date time.Time) *Episodes {
	return &Episodes{
		ID:       episodeID,
		SeasonID: seasonID,
		Title:    title,
		AirDate:  date,
	}
}

func getData() (series []*Series, seasons []*Seasons, episodes []*Episodes) {
	for seriesID, fill := range map[string]func(seriesID string) (
		seriesData *Series,
		seasons []*Seasons,
		episodes []*Episodes,
	){
		uuid.New().String(): getDataForITCrowd,
		uuid.New().String(): getDataForSiliconValley,
	} {
		seriesData, seasonsData, episodesData := fill(seriesID)
		series = append(series, seriesData)
		seasons = append(seasons, seasonsData...)
		episodes = append(episodes, episodesData...)
	}

	return
}

func getDataForITCrowd(seriesID string) (series *Series, seasons []*Seasons, episodes []*Episodes) {
	series = seriesData(
		seriesID, date("2006-02-03"), "IT Crowd", ""+
			"The IT Crowd is a British sitcom produced by Channel 4, written by Graham Linehan, produced by "+
			"Ash Atalla and starring Chris O'Dowd, Richard Ayoade, Katherine Parkinson, and Matt Berry.",
		"", // NULL comment.
	)
	for _, season := range []*struct {
		title    string
		first    time.Time
		last     time.Time
		episodes map[string]time.Time
	}{
		{
			"Season 1", date("2006-02-03"), date("2006-03-03"), map[string]time.Time{
				"Yesterday's Jam":             date("2006-02-03"),
				"Calamity Jen":                date("2006-02-03"),
				"Fifty-Fifty":                 date("2006-02-10"),
				"The Red Door":                date("2006-02-17"),
				"The Haunting of Bill Crouse": date("2006-02-24"),
				"Aunt Irma Visits":            date("2006-03-03"),
			},
		},
		{
			"Season 2", date("2007-08-24"), date("2007-09-28"), map[string]time.Time{
				"The Work Outing":            date("2006-08-24"),
				"Return of the Golden Child": date("2007-08-31"),
				"Moss and the German":        date("2007-09-07"),
				"The Dinner Party":           date("2007-09-14"),
				"Smoke and Mirrors":          date("2007-09-21"),
				"Men Without Women":          date("2007-09-28"),
			},
		},
		{
			"Season 3", date("2008-11-21"), date("2008-12-26"), map[string]time.Time{
				"From Hell":       date("2008-11-21"),
				"Are We Not Men?": date("2008-11-28"),
				"Tramps Like Us":  date("2008-12-05"),
				"The Speech":      date("2008-12-12"),
				"Friendface":      date("2008-12-19"),
				"Calendar Geeks":  date("2008-12-26"),
			},
		},
		{
			"Season 4", date("2010-06-25"), date("2010-07-30"), map[string]time.Time{
				"Jen The Fredo":         date("2010-06-25"),
				"The Final Countdown":   date("2010-07-02"),
				"Something Happened":    date("2010-07-09"),
				"Italian For Beginners": date("2010-07-16"),
				"Bad Boys":              date("2010-07-23"),
				"Reynholm vs Reynholm":  date("2010-07-30"),
			},
		},
	} {
		seasonID := uuid.New().String()
		seasons = append(seasons, seasonData(seriesID, seasonID, season.title, season.first, season.last))
		for title, date := range season.episodes {
			episodes = append(episodes, episodeData(seasonID, uuid.New().String(), title, date))
		}
	}

	return series, seasons, episodes
}

func getDataForSiliconValley(seriesID string) (series *Series, seasons []*Seasons, episodes []*Episodes) {
	series = seriesData(
		seriesID, date("2014-04-06"), "Silicon Valley", ""+
			"Silicon Valley is an American comedy television series created by Mike Judge, John Altschuler and "+
			"Dave Krinsky. The series focuses on five young men who founded a startup company in Silicon Valley.",
		"Some comment here",
	)
	for _, season := range []*struct {
		title    string
		first    time.Time
		last     time.Time
		episodes map[string]time.Time
	}{
		{
			"Season 1", date("2014-04-06"), date("2014-06-01"), map[string]time.Time{
				"Minimum Viable Product":        date("2014-04-06"),
				"The Cap Table":                 date("2014-04-13"),
				"Articles of Incorporation":     date("2014-04-20"),
				"Fiduciary Duties":              date("2014-04-27"),
				"Signaling Risk":                date("2014-05-04"),
				"Third Party Insourcing":        date("2014-05-11"),
				"Proof of Concept":              date("2014-05-18"),
				"Optimal Tip-to-Tip Efficiency": date("2014-06-01"),
			},
		},
		{
			"Season 2", date("2015-04-12"), date("2015-06-14"), map[string]time.Time{
				"Sand Hill Shuffle":      date("2015-04-12"),
				"Runaway Devaluation":    date("2015-04-19"),
				"Bad Money":              date("2015-04-26"),
				"The Lady":               date("2015-05-03"),
				"Server Space":           date("2015-05-10"),
				"Homicide":               date("2015-05-17"),
				"Adult Content":          date("2015-05-24"),
				"White Hat/Black Hat":    date("2015-05-31"),
				"Binding Arbitration":    date("2015-06-07"),
				"Two Days of the Condor": date("2015-06-14"),
			},
		},
		{
			"Season 3", date("2016-04-24"), date("2016-06-26"), map[string]time.Time{
				"Founder Friendly":               date("2016-04-24"),
				"Two in the Box":                 date("2016-05-01"),
				"Meinertzhagen's Haversack":      date("2016-05-08"),
				"Maleant Data Systems Solutions": date("2016-05-15"),
				"The Empty Chair":                date("2016-05-22"),
				"Bachmanity Insanity":            date("2016-05-29"),
				"To Build a Better Beta":         date("2016-06-05"),
				"Bachman's Earnings Over-Ride":   date("2016-06-12"),
				"Daily Active Users":             date("2016-06-19"),
				"The Uptick":                     date("2016-06-26"),
			},
		},
		{
			"Season 4", date("2017-04-23"), date("2017-06-25"), map[string]time.Time{
				"Success Failure":       date("2017-04-23"),
				"Terms of Service":      date("2017-04-30"),
				"Intellectual Property": date("2017-05-07"),
				"Teambuilding Exercise": date("2017-05-14"),
				"The Blood Boy":         date("2017-05-21"),
				"Customer Service":      date("2017-05-28"),
				"The Patent Troll":      date("2017-06-04"),
				"The Keenan Vortex":     date("2017-06-11"),
				"Hooli-Con":             date("2017-06-18"),
				"Server Error":          date("2017-06-25"),
			},
		},
		{
			"Season 5", date("2018-03-25"), date("2018-05-13"), map[string]time.Time{
				"Grow Fast or Die Slow":             date("2018-03-25"),
				"Reorientation":                     date("2018-04-01"),
				"Chief Operating Officer":           date("2018-04-08"),
				"Tech Evangelist":                   date("2018-04-15"),
				"Facial Recognition":                date("2018-04-22"),
				"Artificial Emotional Intelligence": date("2018-04-29"),
				"Initial Coin Offering":             date("2018-05-06"),
				"Fifty-One Percent":                 date("2018-05-13"),
			},
		},
	} {
		seasonID := uuid.New().String()
		seasons = append(seasons, seasonData(seriesID, seasonID, season.title, season.first, season.last))
		for title, date := range season.episodes {
			episodes = append(episodes, episodeData(seasonID, uuid.New().String(), title, date))
		}
	}

	return series, seasons, episodes
}

const dateISO8601 = "2006-01-02"

func date(date string) time.Time {
	t, err := time.Parse(dateISO8601, date)
	if err != nil {
		panic(err)
	}

	return t
}