forked from go-eagle/eagle
-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.go
99 lines (88 loc) · 2.04 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
package http
import (
"context"
"errors"
"log"
"net"
"net/http"
"net/url"
"time"
"github.com/willieso/baby-univ-biz-service/pkg/transport"
"github.com/willieso/baby-univ-biz-service/pkg/utils"
)
var (
_ transport.Server = (*Server)(nil)
_ transport.Endpoint = (*Server)(nil)
)
// Server http server struct
type Server struct {
*http.Server
lis net.Listener
network string
address string
readTimeout time.Duration
writeTimeout time.Duration
endpoint *url.URL
// log log.Logger
}
// defaultServer return a default config server
func defaultServer() *Server {
return &Server{
network: "tcp",
address: ":8080",
readTimeout: time.Second,
writeTimeout: time.Second,
}
}
// NewServer create a server
func NewServer(opts ...ServerOption) *Server {
srv := defaultServer()
// apply options
for _, o := range opts {
o(srv)
}
// NOTE: must set server
srv.Server = &http.Server{
ReadTimeout: srv.readTimeout,
WriteTimeout: srv.writeTimeout,
Handler: srv,
}
return srv
}
// ServeHTTP should write reply headers and data to the ResponseWriter and then return.
func (s *Server) ServeHTTP(resp http.ResponseWriter, req *http.Request) {
s.ServeHTTP(resp, req)
}
// Endpoint return a real address to registry endpoint.
// examples:
//
// http://127.0.0.1:8080
func (s *Server) Endpoint() (*url.URL, error) {
addr, err := utils.Extract(s.address, s.lis)
if err != nil {
return nil, err
}
s.endpoint = &url.URL{Scheme: "http", Host: addr}
return s.endpoint, nil
}
// Start start a server
func (s *Server) Start(ctx context.Context) error {
lis, err := net.Listen(s.network, s.address)
if err != nil {
return err
}
s.lis = lis
if _, err := s.Endpoint(); err != nil {
return err
}
log.Printf("[HTTP] server is listening on: %s", lis.Addr().String())
if err := s.Serve(lis); !errors.Is(err, http.ErrServerClosed) {
return err
}
return nil
}
// Stop stop server
func (s *Server) Stop(ctx context.Context) error {
log.Printf("[HTTP] server is stopping")
return s.Shutdown(ctx)
}