|
1 | 1 | package main
|
2 | 2 |
|
3 | 3 | import (
|
4 |
| - "context" |
5 |
| - "flag" |
6 |
| - "github.com/0xJacky/Nginx-UI/server/model" |
7 |
| - "github.com/0xJacky/Nginx-UI/server/router" |
8 |
| - "github.com/0xJacky/Nginx-UI/server/settings" |
9 |
| - tool2 "github.com/0xJacky/Nginx-UI/server/tool" |
10 |
| - "log" |
11 |
| - "mime" |
12 |
| - "net/http" |
13 |
| - "os/signal" |
14 |
| - "syscall" |
15 |
| - "time" |
| 4 | + "context" |
| 5 | + "flag" |
| 6 | + "github.com/0xJacky/Nginx-UI/server/model" |
| 7 | + "github.com/0xJacky/Nginx-UI/server/router" |
| 8 | + "github.com/0xJacky/Nginx-UI/server/settings" |
| 9 | + tool2 "github.com/0xJacky/Nginx-UI/server/tool" |
| 10 | + "github.com/gin-gonic/gin" |
| 11 | + "log" |
| 12 | + "mime" |
| 13 | + "net/http" |
| 14 | + "os/signal" |
| 15 | + "syscall" |
| 16 | + "time" |
16 | 17 | )
|
17 | 18 |
|
18 | 19 | func main() {
|
19 |
| - // Create context that listens for the interrupt signal from the OS. |
20 |
| - ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM) |
21 |
| - defer stop() |
| 20 | + // Create context that listens for the interrupt signal from the OS. |
| 21 | + ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM) |
| 22 | + defer stop() |
22 | 23 |
|
23 |
| - // Hack: fix wrong Content Type of .js file on some OS platforms |
24 |
| - // See https://github.com/golang/go/issues/32350 |
25 |
| - _ = mime.AddExtensionType(".js", "text/javascript; charset=utf-8") |
| 24 | + // Hack: fix wrong Content Type of .js file on some OS platforms |
| 25 | + // See https://github.com/golang/go/issues/32350 |
| 26 | + _ = mime.AddExtensionType(".js", "text/javascript; charset=utf-8") |
26 | 27 |
|
27 |
| - var confPath string |
28 |
| - flag.StringVar(&confPath, "config", "app.ini", "Specify the configuration file") |
29 |
| - flag.Parse() |
| 28 | + var confPath string |
| 29 | + flag.StringVar(&confPath, "config", "app.ini", "Specify the configuration file") |
| 30 | + flag.Parse() |
30 | 31 |
|
31 |
| - settings.Init(confPath) |
32 |
| - log.Printf("nginx config dir path: %s", tool2.GetNginxConfPath("")) |
33 |
| - if "" != settings.ServerSettings.JwtSecret { |
34 |
| - model.Init() |
35 |
| - go tool2.AutoCert() |
36 |
| - } |
| 32 | + gin.SetMode(settings.ServerSettings.RunMode) |
| 33 | + |
| 34 | + settings.Init(confPath) |
| 35 | + log.Printf("nginx config dir path: %s", tool2.GetNginxConfPath("")) |
| 36 | + if "" != settings.ServerSettings.JwtSecret { |
| 37 | + model.Init() |
| 38 | + go tool2.AutoCert() |
| 39 | + } |
37 | 40 |
|
38 |
| - srv := &http.Server{ |
39 |
| - Addr: ":" + settings.ServerSettings.HttpPort, |
40 |
| - Handler: router.InitRouter(), |
41 |
| - } |
| 41 | + srv := &http.Server{ |
| 42 | + Addr: ":" + settings.ServerSettings.HttpPort, |
| 43 | + Handler: router.InitRouter(), |
| 44 | + } |
42 | 45 |
|
43 |
| - // Initializing the server in a goroutine so that |
44 |
| - // it won't block the graceful shutdown handling below |
45 |
| - go func() { |
46 |
| - if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed { |
47 |
| - log.Fatalf("listen: %s\n", err) |
48 |
| - } |
49 |
| - }() |
| 46 | + // Initializing the server in a goroutine so that |
| 47 | + // it won't block the graceful shutdown handling below |
| 48 | + go func() { |
| 49 | + if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed { |
| 50 | + log.Fatalf("listen: %s\n", err) |
| 51 | + } |
| 52 | + }() |
50 | 53 |
|
51 |
| - // Listen for the interrupt signal. |
52 |
| - <-ctx.Done() |
| 54 | + // Listen for the interrupt signal. |
| 55 | + <-ctx.Done() |
53 | 56 |
|
54 |
| - // Restore default behavior on the interrupt signal and notify user of shutdown. |
55 |
| - stop() |
56 |
| - log.Println("shutting down gracefully, press Ctrl+C again to force") |
| 57 | + // Restore default behavior on the interrupt signal and notify user of shutdown. |
| 58 | + stop() |
| 59 | + log.Println("shutting down gracefully, press Ctrl+C again to force") |
57 | 60 |
|
58 |
| - // The context is used to inform the server it has 5 seconds to finish |
59 |
| - // the request it is currently handling |
60 |
| - ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) |
61 |
| - defer cancel() |
62 |
| - if err := srv.Shutdown(ctx); err != nil { |
63 |
| - log.Fatal("Server forced to shutdown: ", err) |
64 |
| - } |
| 61 | + // The context is used to inform the server it has 5 seconds to finish |
| 62 | + // the request it is currently handling |
| 63 | + ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second) |
| 64 | + defer cancel() |
| 65 | + if err := srv.Shutdown(ctx); err != nil { |
| 66 | + log.Fatal("Server forced to shutdown: ", err) |
| 67 | + } |
65 | 68 |
|
66 |
| - log.Println("Server exiting") |
| 69 | + log.Println("Server exiting") |
67 | 70 |
|
68 | 71 | }
|
0 commit comments