/
app.go
79 lines (74 loc) · 1.8 KB
/
app.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 eye
import (
"fmt"
"github.com/dzhcool/eye/endless"
"log"
"net"
"net/http"
"net/http/fcgi"
"os"
"os/exec"
"strings"
"time"
)
type App struct {
Handlers *ControllerRegistor
Server *http.Server
}
func NewApp() *App {
cr := NewControllerRegister()
return &App{Handlers: cr, Server: &http.Server{}}
}
func (p *App) Run() {
running := make(chan bool, 1)
addr := ":" + os.Getenv("GOPORT")
lnet := os.Getenv("GONET")
gosock := os.Getenv("GOSOCK")
//启动程序监听
if strings.ToUpper(os.Getenv("GRACEFUL")) == "ON" {
go func() {
endless.DefaultReadTimeOut = 5 * time.Second
endless.DefaultWriteTimeOut = 10 * time.Second
err := endless.ListenAndServe(addr, p.Handlers)
if err != nil {
log.Println("[Eey]Listen error:", err)
}
log.Println(fmt.Sprintf("[Eye]Server on %s stopped", addr))
running <- true
}()
} else {
if strings.ToUpper(lnet) == "TCP" {
log.Println("[Eey]Listen as", lnet, addr)
go func() {
server := http.Server{
Addr: addr,
Handler: p.Handlers,
ReadTimeout: 5 * time.Second,
WriteTimeout: 10 * time.Second,
}
err := server.ListenAndServe()
if err != nil {
log.Println("[Eey]Listen error:", err)
}
log.Println(fmt.Sprintf("[Eye]Server on %s stopped", addr))
running <- true
}()
} else {
log.Println("[Eey]Listen as", lnet, gosock)
go func() {
exec.Command("/bin/sh", "-c", "rm "+gosock).Run()
unix, err := net.Listen("unix", gosock)
exec.Command("/bin/sh", "-c", "chmod a+w "+gosock).Run()
if err != nil {
log.Println("[Eey]Listen error:", err)
}
fcgi.Serve(unix, p.Handlers)
log.Println(fmt.Sprintf("[Eye]Server on %s stopped", addr))
running <- true
}()
}
}
<-running
log.Println("[Eye]All servers stopped. Exiting.")
os.Exit(0)
}