/
acceptor.go
75 lines (48 loc) · 1.26 KB
/
acceptor.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
package nexus
import (
"github.com/davyxu/actornet/actor"
"github.com/davyxu/actornet/proto"
"github.com/davyxu/cellnet"
"github.com/davyxu/cellnet/socket"
)
// 启动本机的listen
func Listen(address string, domain string) {
peer := socket.NewAcceptor(nil)
peer.Start(address)
cellnet.RegisterMessage(peer, "coredef.SessionAccepted", func(ev *cellnet.Event) {
sendDomains(ev.Ses)
})
shareInit(peer)
}
func sendDomains(ses cellnet.Session) {
var msg proto.DomainSyncACK
actor.VisitDomains(func(domain *actor.Domain) bool {
msg.DomainNames = append(msg.DomainNames, domain.Name)
return true
})
ses.Send(&msg)
}
func addDomains(domainNames []string, ses cellnet.Session) {
for _, name := range domainNames {
if actor.GetDomain(name) != nil {
log.Errorf("Duplicate remote domain: %s", name)
continue
}
domain := actor.CreateRemoteDomain(name)
domain.RemoteContext = ses
}
}
func removeDomains(ses cellnet.Session) {
var removeNames []string
actor.VisitDomains(func(domain *actor.Domain) bool {
if domain.RemoteContext != nil {
if domain.RemoteContext.(cellnet.Session) == ses {
removeNames = append(removeNames, domain.Name)
}
}
return true
})
for _, name := range removeNames {
actor.DestroyDomain(name)
}
}