-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.go
118 lines (98 loc) · 2.18 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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
package server
import (
"log"
"net/http"
"os"
"time"
"github.com/go-chi/chi"
"github.com/go-chi/chi/middleware"
"github.com/go-chi/cors"
v1 "github.com/super-radmir/menuxd_api/internal/server/v1"
)
type config struct {
port string
debug bool
}
var conf config
var fileLog *os.File
// SetConfig set server configuration.
func SetConfig(port string, debug bool) {
conf.port = port
conf.debug = debug
}
var basePath = os.Getenv("XD_BASE_PATH")
func getRoutes() (http.Handler, error) {
cors := cors.New(cors.Options{
AllowedOrigins: []string{"*"},
AllowedMethods: []string{
"GET",
"POST",
"PUT",
"DELETE",
"OPTIONS",
"PATCH",
},
AllowedHeaders: []string{
"Accept",
"Authorization",
"Content-Type",
"X-CSRF-Token",
},
AllowCredentials: true,
MaxAge: 300,
})
r := chi.NewRouter()
r.Use(cors.Handler)
r.Use(middleware.RequestID)
r.Use(middleware.RealIP)
r.Use(setLogger(conf.debug))
r.Use(middleware.Recoverer)
// Configurations
r.Handle("/public/*", http.StripPrefix(
"/public", http.FileServer(http.Dir(basePath+"public")),
))
v1Routes, err := v1.NewAPI()
if err != nil {
return nil, err
}
r.Mount("/api/v1", v1Routes)
r.Handle(
"/docs/*",
http.StripPrefix("/docs/", http.FileServer(http.Dir(basePath+"docs"))),
)
r.Handle("/*", http.FileServer(http.Dir(basePath+"static")))
return r, nil
}
// New inicialize a new server with configuration.
func New() (*http.Server, error) {
r, err := getRoutes()
if err != nil {
return nil, err
}
srv := &http.Server{
Addr: ":" + conf.port,
Handler: r,
ReadTimeout: 10 * time.Second,
WriteTimeout: 10 * time.Second,
IdleTimeout: 120 * time.Second,
}
return srv, nil
}
func setLogger(isDebug bool) func(next http.Handler) http.Handler {
if isDebug {
return middleware.Logger
}
var err error
fileLog, err = os.OpenFile("logs.log", os.O_CREATE|os.O_RDWR|os.O_APPEND, 0666)
if err != nil {
return middleware.Logger
}
logger := middleware.RequestLogger(&middleware.DefaultLogFormatter{Logger: log.New(fileLog, "", log.LstdFlags), NoColor: true})
return logger
}
func Close() error {
if fileLog != nil {
return fileLog.Close()
}
return nil
}