go-train-delay-db-insert.go

go-train-delay-db-insert.go

func (db *DB) Insert(data []DelayDataSet) {

	tx := db.Pool.MustBegin()
	bar := processBar(int64(len(data)), "[db] inserting...")

	for _, d := range data {
		tx.MustExec(
			"INSERT INTO delays (date, departure, depart_scheduled, destination, arrival_scheduled, delay) VALUES ($1, $2, $3, $4, $5, $6) ON CONFLICT (date) DO NOTHING",
			d.Date,
			d.Departure.Station,
			d.Departure.Scheduled,
			d.Destination.Station,
			d.Destination.Scheduled,
			d.Mins,
		)
		bar.Add(1)
		time.Sleep(40 * time.Millisecond)
	}

	tx.Commit()

}