-
Notifications
You must be signed in to change notification settings - Fork 1
/
net_http.go
53 lines (47 loc) · 1.18 KB
/
net_http.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
package app
import (
"net/http"
"sync"
netHttp "github.com/cnlisea/ant/app/net/http"
"github.com/cnlisea/ant/logs"
)
func (a *App) NetHttpRegister(name string, ip string, port uint16, discoverySoftState *bool, handler http.Handler) error {
if a.httpServers == nil {
a.httpServers = make(map[string]*netHttp.Server, 1)
}
s := a.httpServers[name]
if s == nil {
s = netHttp.NewServer(ip, port)
a.httpServers[name] = s
}
if discoverySoftState != nil {
a._DiscoverySoftStateAdd(_DiscoverySoftStateHttpNamePrefix, name, *discoverySoftState)
}
return s.SetHandler(handler)
}
func (a *App) NetHttpAllStart(ws *sync.WaitGroup) bool {
var (
name string
s *netHttp.Server
)
for name, s = range a.httpServers {
ws.Add(1)
go func(name string, server *netHttp.Server) {
logs.Info("http server run",
logs.String("name", name),
logs.String("ip", server.Ip),
logs.Uint16("port", server.Port))
if err := server.Run(); err != nil {
a.Close()
logs.Err("http server run fail", logs.String("name", name), logs.Error("err", err))
}
ws.Done()
}(name, s)
}
return true
}
func (a *App) NetHttpAllClose() {
for _, s := range a.httpServers {
s.Close()
}
}