/
server.go
118 lines (100 loc) · 2.34 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 pithy
import (
"errors"
"net"
"net/http"
"strings"
"sync"
"github.com/gorilla/mux"
)
// apiVersionMatcher can get the api version from the path
const apiVersionMatcher = "/v{version:[0-9.]+}"
// constant errors
var (
ErrInvalidServerRouter = errors.New("Invalid server router")
ErrInvalidListenAddress = errors.New("Invalid listen address")
)
type HTTPServer struct {
listener net.Listener
wg sync.WaitGroup
svr *http.Server
router SwappableHandler
}
func NewHTTPServer(address string) *HTTPServer {
svr := &HTTPServer{
svr: &http.Server{
Addr: address,
},
}
svr.svr.Handler = &svr.router
return svr
}
func (s *HTTPServer) EnablePprof() {
s.router.enablePprof = true
}
// Serve with the speicified address
func (s *HTTPServer) Serve() error {
// Is router already set ?
if nil == s.router.router {
return ErrInvalidServerRouter
}
if "" == s.svr.Addr {
return ErrInvalidListenAddress
}
var ls net.Listener
var err error
if strings.HasSuffix(s.svr.Addr, ".sock") {
ls, err = net.Listen("unix", s.svr.Addr)
if nil != err {
return err
}
} else {
ls, err = net.Listen("tcp", s.svr.Addr)
if nil != err {
return err
}
}
s.listener = ls
s.wg.Add(1)
go s.serve()
return nil
}
// GetListener get the listener of the server
func (s *HTTPServer) GetListener() net.Listener {
return s.listener
}
// Stop stops the server
func (s *HTTPServer) Stop() {
s.listener.Close()
s.wg.Wait()
}
// AppendRouters create mux and update handler of server
func (s *HTTPServer) AppendRouters(rts ...APIRouter) {
if nil == s.router.router {
s.router.router = mux.NewRouter()
}
for _, v := range rts {
fn := wrapHTTPHandler(v.Handler())
Debugf("Register http router [%s]%s", v.Method(), v.Path())
path := v.Path()
if v.Version() {
path = apiVersionMatcher + v.Path()
}
s.router.router.Path(path).Methods(v.Method()).Handler(fn)
}
}
// SetNotFoundHandler -
func (s *HTTPServer) SetNotFoundHandler(h APIFunc) {
if nil == s.router.router {
return
}
s.router.router.NotFoundHandler = wrapHTTPHandler(h)
}
func (s *HTTPServer) serve() {
defer s.wg.Done()
Infoln("HTTP serve @ ", s.listener.Addr().String())
s.svr.Handler = &s.router
if err := s.svr.Serve(s.listener); !strings.Contains(err.Error(), "use of closed network connection") {
Errorln("HTTP server stop serve with error : ", err)
}
}