mirrors/etcd
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

etcdserver: not add sender when it has existed

Browse Source
This commit is contained in:
Yicheng Qin
2014-11-07 13:59:48 -08:00
parent d3fd10798b
commit 46cbfbc630
2 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
etcdserver/sender.go
View File

@ -92,6 +92,9 @@ func (h *sendHub) Stop() {
} }
func (h *sendHub) Add(m *Member) { func (h *sendHub) Add(m *Member) {
if _, ok := h.senders[m.ID]; ok {
return
}
// TODO: considering how to switch between all available peer urls // TODO: considering how to switch between all available peer urls
u := fmt.Sprintf("%s%s", m.PickPeerURL(), raftPrefix) u := fmt.Sprintf("%s%s", m.PickPeerURL(), raftPrefix)
c := &http.Client{Transport: h.tr} c := &http.Client{Transport: h.tr}

6
etcdserver/sender_test.go
View File

@ -61,6 +61,12 @@ func TestSendHubAdd(t *testing.T) {
if s.u != "http://a/raft" { if s.u != "http://a/raft" {
t.Errorf("url = %s, want %s", s.u, "http://a/raft") t.Errorf("url = %s, want %s", s.u, "http://a/raft")
} }
h.Add(m)
ns := h.senders[types.ID(1)]
if s != ns {
t.Errorf("sender = %p, want %p", ns, s)
}
} }
func TestSendHubRemove(t *testing.T) { func TestSendHubRemove(t *testing.T) {