This repository has been archived by the owner on Nov 7, 2019. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.go
79 lines (64 loc) · 1.74 KB
/
server.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
package main
import (
"database/sql"
"fmt"
"github.com/datatogether/core"
"github.com/datatogether/sql_datastore"
"github.com/sirupsen/logrus"
"net/http"
"os"
"time"
)
var (
// cfg is the global configuration for the server. It's read in at startup from
// the config.json file and enviornment variables, see config.go for more info.
cfg *config
// When was the last alert sent out?
// Use this value to avoid bombing alerts
lastAlertSent *time.Time
// log output via logrus package
log = logrus.New()
// application database connection
appDB *sql.DB
// elevate default store
store = sql_datastore.DefaultStore
)
func init() {
log.Out = os.Stdout
log.Level = logrus.InfoLevel
log.Formatter = &logrus.TextFormatter{
ForceColors: true,
}
}
func main() {
var err error
cfg, err = initConfig(os.Getenv("GOLANG_ENV"))
if err != nil {
// panic if the server is missing a vital configuration detail
panic(fmt.Errorf("server configuration error: %s", err.Error()))
}
go connectToAppDb()
sql_datastore.SetDB(appDB)
sql_datastore.Register(
&core.Url{},
)
s := &http.Server{}
// connect mux to server
s.Handler = NewServerRoutes()
// print notable config settings
// printConfigInfo()
// fire it up!
fmt.Println("starting server on port", cfg.Port)
// start server wrapped in a log.Fatal b/c http.ListenAndServe will not
// return unless there's an error
log.Fatal(StartServer(cfg, s))
}
// NewServerRoutes returns a Muxer that has all API routes.
// This makes for easy testing using httptest
func NewServerRoutes() *http.ServeMux {
m := http.NewServeMux()
m.HandleFunc("/.well-known/acme-challenge/", CertbotHandler)
m.Handle("/", middleware(HealthCheckHandler))
m.Handle("/urls/", middleware(DownloadUrlHandler))
return m
}