-
Notifications
You must be signed in to change notification settings - Fork 32
/
server.go
84 lines (70 loc) · 1.76 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
package tunnel
import (
"sync"
"github.com/VaalaCat/frp-panel/services/server"
)
type ServerController interface {
Add(serverID string, serverHandler server.ServerHandler)
Get(serverID string) server.ServerHandler
Delete(serverID string)
Set(serverID string, serverHandler server.ServerHandler)
Run(serverID string) // 不阻塞
Stop(serverID string)
List() []string
}
type serverController struct {
servers *sync.Map
}
var (
serverControllerInstance *serverController
)
func NewServerController() ServerController {
return &serverController{
servers: &sync.Map{},
}
}
func GetServerController() ServerController {
if serverControllerInstance == nil {
serverControllerInstance = NewServerController().(*serverController)
}
return serverControllerInstance
}
func (c *serverController) Add(serverID string, serverHandler server.ServerHandler) {
c.servers.Store(serverID, serverHandler)
}
func (c *serverController) Get(serverID string) server.ServerHandler {
v, ok := c.servers.Load(serverID)
if !ok {
return nil
}
return v.(server.ServerHandler)
}
func (c *serverController) Delete(serverID string) {
c.Stop(serverID)
c.servers.Delete(serverID)
}
func (c *serverController) Set(serverID string, serverHandler server.ServerHandler) {
c.servers.Store(serverID, serverHandler)
}
func (c *serverController) Run(serverID string) {
v, ok := c.servers.Load(serverID)
if !ok {
return
}
go v.(server.ServerHandler).Run()
}
func (c *serverController) Stop(serverID string) {
v, ok := c.servers.Load(serverID)
if !ok {
return
}
v.(server.ServerHandler).Stop()
}
func (c *serverController) List() []string {
keys := make([]string, 0)
c.servers.Range(func(key, value interface{}) bool {
keys = append(keys, key.(string))
return true
})
return keys
}