1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
package main
import (
"fmt"
"os"
"time"
"github.com/joho/godotenv"
"gorm.io/driver/postgres"
"gorm.io/gorm"
"airlines/pkg/model"
)
func main() {
// run with MAKE only
err := godotenv.Load("../../.env")
if err != nil {
fmt.Println(err)
}
dsn := fmt.Sprintf("postgres://%s:%s@%s:%s/%s", os.Getenv("DB_USER"), os.Getenv("DB_PASSWORD"), os.Getenv("DB_HOST"), os.Getenv("DB_PORT"), os.Getenv("DB_NAME"))
db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
if err != nil {
fmt.Println(err)
return
}
// db.Migrator().DropTable(&model.User{}, &model.Card{}, &model.Flight{}) // ;-)
if err := db.AutoMigrate(&model.User{}, &model.Card{}, &model.Flight{}); err != nil {
return
}
jinzhuFlights := []model.Flight{
{
Number: "SU1111",
From: "North pole",
To: "Mars",
Departure: time.Now(),
Arrival: time.Now().AddDate(0, 0, 162),
},
}
user := model.User{
Name: "Jinzhu",
Age: 18,
Birthday: time.Now(),
Flights: jinzhuFlights,
Cards: []model.Card{
{
Prefix: "AA",
Number: 123,
Bonusprogramm: "AbObA airlines SuPrEmE",
Flights: jinzhuFlights,
},
},
}
// Create a single record
result := db.Create(&user)
fmt.Println(result.Error)
}
|
