-
Notifications
You must be signed in to change notification settings - Fork 0
/
serve.go
101 lines (77 loc) · 2.05 KB
/
serve.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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
package main
import (
"database/sql"
"log"
"net/http"
gsp "github.com/fluxynet/go-scratch-prod"
taskRepositoryG "github.com/fluxynet/go-scratch-prod/gorm/tasks"
taskRepositoryS "github.com/fluxynet/go-scratch-prod/sql/tasks"
taskService "github.com/fluxynet/go-scratch-prod/task"
"github.com/fluxynet/go-scratch-prod/web"
taskHttp "github.com/fluxynet/go-scratch-prod/web/task"
"github.com/go-chi/chi/v5"
_ "github.com/mattn/go-sqlite3"
"gorm.io/driver/sqlite"
"gorm.io/gorm"
)
type Config struct {
IsDev bool
ListenAddr string
DSN string
}
func main() {
router := chi.NewRouter()
cfg, err := loadConfig()
if err != nil {
log.Fatalln(err.Error())
}
if cfg.IsDev {
router.Mount("/api", devRouter(*cfg))
} else {
router.Mount("/api", prodRouter(*cfg))
}
router.Get("/status", web.TimestampHandler)
log.Println("Server started on: " + cfg.ListenAddr)
http.ListenAndServe(cfg.ListenAddr, router)
}
func loadConfig() (*Config, error) {
c := Config{
IsDev: true,
ListenAddr: ":9000",
DSN: ":memory:",
}
//TODO load config from environment
return &c, nil
}
func gormTaskRepo(cfg Config) gsp.TaskRepo {
db, err := gorm.Open(sqlite.Open(cfg.DSN))
if err != nil {
log.Fatalln("failed to connect to database: " + err.Error())
}
db.AutoMigrate(&gsp.Task{})
return taskRepositoryG.New(db)
}
func sqliteTaskRepo(cfg Config) gsp.TaskRepo {
db, err := sql.Open("sqlite", cfg.DSN)
if err != nil {
log.Fatalln("failed to connect to database: " + err.Error())
}
return taskRepositoryS.New(db)
}
func prodRouter(cfg Config) http.Handler {
r := chi.NewRouter()
taskSvc := taskService.New(gormTaskRepo(cfg))
taskApi := taskHttp.New(taskSvc)
r.Post("/tasks", taskApi.Create)
r.Put("/tasks/{id}", taskApi.Update)
r.Put("/tasks/{id}/done", taskApi.MarkDone)
r.Put("/tasks/{id}/pending", taskApi.MarkPending)
r.Get("/tasks", taskApi.List)
r.Get("/tasks/done", taskApi.ListDone)
r.Get("/tasks/pending", taskApi.ListPending)
return r
}
func devRouter(cfg Config) http.Handler {
p := prodRouter(cfg)
return p
}