/
main.go
50 lines (45 loc) · 1.26 KB
/
main.go
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
package main
import (
"fmt"
"log"
"net/http"
"os"
"github.com/PacktPublishing/Test-Driven-Development-in-Go/chapter06/db"
"github.com/PacktPublishing/Test-Driven-Development-in-Go/chapter06/handlers"
"github.com/golang-migrate/migrate/v4"
_ "github.com/golang-migrate/migrate/v4/database/postgres"
_ "github.com/golang-migrate/migrate/v4/source/file"
"gorm.io/driver/postgres"
"gorm.io/gorm"
)
func main() {
port, ok := os.LookupEnv("BOOKSWAP_PORT")
if !ok {
log.Fatal("$BOOKSWAP_PORT not found")
}
postgresURL, ok := os.LookupEnv("BOOKSWAP_DB_URL")
if !ok {
log.Fatal("$BOOKSWAP_DB_URL not found")
}
m, err := migrate.New("file://chapter06/db/migrations", postgresURL)
if err != nil {
log.Fatalf("migrate:%v", err)
}
if err := m.Up(); err != nil && err != migrate.ErrNoChange {
log.Fatalf("migration up:%v", err)
}
// defer func() {
// m.Down()
// }()
dbConn, err := gorm.Open(postgres.Open(postgresURL), &gorm.Config{})
if err != nil {
log.Fatalf("db open:%v", err)
}
ps := db.NewPostingService()
b := db.NewBookService(dbConn, ps)
u := db.NewUserService(dbConn, b)
h := handlers.NewHandler(b, u)
router := handlers.ConfigureServer(h)
log.Printf("Listening on :%s...\n", port)
log.Fatal(http.ListenAndServe(fmt.Sprint(":", port), router))
}