cluster_test.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929
  1. // Copyright 2015 CoreOS, Inc.
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // http://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. package integration
  15. import (
  16. "fmt"
  17. "io/ioutil"
  18. "log"
  19. "math/rand"
  20. "net"
  21. "net/http"
  22. "net/http/httptest"
  23. "os"
  24. "reflect"
  25. "sort"
  26. "strconv"
  27. "strings"
  28. "sync/atomic"
  29. "testing"
  30. "time"
  31. "github.com/coreos/etcd/client"
  32. "github.com/coreos/etcd/clientv3"
  33. "github.com/coreos/etcd/etcdserver"
  34. "github.com/coreos/etcd/etcdserver/api/v3rpc"
  35. "github.com/coreos/etcd/etcdserver/etcdhttp"
  36. "github.com/coreos/etcd/etcdserver/etcdserverpb"
  37. "github.com/coreos/etcd/pkg/testutil"
  38. "github.com/coreos/etcd/pkg/transport"
  39. "github.com/coreos/etcd/pkg/types"
  40. "github.com/coreos/etcd/rafthttp"
  41. "github.com/coreos/etcd/Godeps/_workspace/src/golang.org/x/net/context"
  42. "github.com/coreos/etcd/Godeps/_workspace/src/google.golang.org/grpc"
  43. )
  44. const (
  45. tickDuration = 10 * time.Millisecond
  46. clusterName = "etcd"
  47. requestTimeout = 20 * time.Second
  48. )
  49. var (
  50. electionTicks = 10
  51. // integration test uses well-known ports to listen for each running member,
  52. // which ensures restarted member could listen on specific port again.
  53. nextListenPort int64 = 20000
  54. )
  55. func init() {
  56. // open microsecond-level time log for integration test debugging
  57. log.SetFlags(log.Ltime | log.Lmicroseconds | log.Lshortfile)
  58. if t := os.Getenv("ETCD_ELECTION_TIMEOUT_TICKS"); t != "" {
  59. if i, err := strconv.ParseInt(t, 10, 64); err == nil {
  60. electionTicks = int(i)
  61. }
  62. }
  63. }
  64. func TestClusterOf1(t *testing.T) { testCluster(t, 1) }
  65. func TestClusterOf3(t *testing.T) { testCluster(t, 3) }
  66. func testCluster(t *testing.T, size int) {
  67. defer testutil.AfterTest(t)
  68. c := NewCluster(t, size)
  69. c.Launch(t)
  70. defer c.Terminate(t)
  71. clusterMustProgress(t, c.Members)
  72. }
  73. func TestTLSClusterOf3(t *testing.T) {
  74. defer testutil.AfterTest(t)
  75. c := NewClusterByConfig(t, &clusterConfig{size: 3, usePeerTLS: true})
  76. c.Launch(t)
  77. defer c.Terminate(t)
  78. clusterMustProgress(t, c.Members)
  79. }
  80. func TestClusterOf1UsingDiscovery(t *testing.T) { testClusterUsingDiscovery(t, 1) }
  81. func TestClusterOf3UsingDiscovery(t *testing.T) { testClusterUsingDiscovery(t, 3) }
  82. func testClusterUsingDiscovery(t *testing.T, size int) {
  83. defer testutil.AfterTest(t)
  84. dc := NewCluster(t, 1)
  85. dc.Launch(t)
  86. defer dc.Terminate(t)
  87. // init discovery token space
  88. dcc := mustNewHTTPClient(t, dc.URLs())
  89. dkapi := client.NewKeysAPI(dcc)
  90. ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
  91. if _, err := dkapi.Create(ctx, "/_config/size", fmt.Sprintf("%d", size)); err != nil {
  92. t.Fatal(err)
  93. }
  94. cancel()
  95. c := NewClusterByConfig(
  96. t,
  97. &clusterConfig{size: size, discoveryURL: dc.URL(0) + "/v2/keys"},
  98. )
  99. c.Launch(t)
  100. defer c.Terminate(t)
  101. clusterMustProgress(t, c.Members)
  102. }
  103. func TestTLSClusterOf3UsingDiscovery(t *testing.T) {
  104. defer testutil.AfterTest(t)
  105. dc := NewCluster(t, 1)
  106. dc.Launch(t)
  107. defer dc.Terminate(t)
  108. // init discovery token space
  109. dcc := mustNewHTTPClient(t, dc.URLs())
  110. dkapi := client.NewKeysAPI(dcc)
  111. ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
  112. if _, err := dkapi.Create(ctx, "/_config/size", fmt.Sprintf("%d", 3)); err != nil {
  113. t.Fatal(err)
  114. }
  115. cancel()
  116. c := NewClusterByConfig(t,
  117. &clusterConfig{
  118. size: 3,
  119. usePeerTLS: true,
  120. discoveryURL: dc.URL(0) + "/v2/keys"},
  121. )
  122. c.Launch(t)
  123. defer c.Terminate(t)
  124. clusterMustProgress(t, c.Members)
  125. }
  126. func TestDoubleClusterSizeOf1(t *testing.T) { testDoubleClusterSize(t, 1) }
  127. func TestDoubleClusterSizeOf3(t *testing.T) { testDoubleClusterSize(t, 3) }
  128. func testDoubleClusterSize(t *testing.T, size int) {
  129. defer testutil.AfterTest(t)
  130. c := NewCluster(t, size)
  131. c.Launch(t)
  132. defer c.Terminate(t)
  133. for i := 0; i < size; i++ {
  134. c.AddMember(t)
  135. }
  136. clusterMustProgress(t, c.Members)
  137. }
  138. func TestDoubleTLSClusterSizeOf3(t *testing.T) {
  139. defer testutil.AfterTest(t)
  140. c := NewClusterByConfig(t, &clusterConfig{size: 3, usePeerTLS: true})
  141. c.Launch(t)
  142. defer c.Terminate(t)
  143. for i := 0; i < 3; i++ {
  144. c.AddMember(t)
  145. }
  146. clusterMustProgress(t, c.Members)
  147. }
  148. func TestDecreaseClusterSizeOf3(t *testing.T) { testDecreaseClusterSize(t, 3) }
  149. func TestDecreaseClusterSizeOf5(t *testing.T) { testDecreaseClusterSize(t, 5) }
  150. func testDecreaseClusterSize(t *testing.T, size int) {
  151. defer testutil.AfterTest(t)
  152. c := NewCluster(t, size)
  153. c.Launch(t)
  154. defer c.Terminate(t)
  155. // TODO: remove the last but one member
  156. for i := 0; i < size-1; i++ {
  157. id := c.Members[len(c.Members)-1].s.ID()
  158. c.RemoveMember(t, uint64(id))
  159. c.waitLeader(t, c.Members)
  160. }
  161. clusterMustProgress(t, c.Members)
  162. }
  163. func TestForceNewCluster(t *testing.T) {
  164. c := NewCluster(t, 3)
  165. c.Launch(t)
  166. cc := mustNewHTTPClient(t, []string{c.Members[0].URL()})
  167. kapi := client.NewKeysAPI(cc)
  168. ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
  169. resp, err := kapi.Create(ctx, "/foo", "bar")
  170. if err != nil {
  171. t.Fatalf("unexpected create error: %v", err)
  172. }
  173. cancel()
  174. // ensure create has been applied in this machine
  175. ctx, cancel = context.WithTimeout(context.Background(), requestTimeout)
  176. if _, err = kapi.Watcher("/foo", &client.WatcherOptions{AfterIndex: resp.Node.ModifiedIndex - 1}).Next(ctx); err != nil {
  177. t.Fatalf("unexpected watch error: %v", err)
  178. }
  179. cancel()
  180. c.Members[0].Stop(t)
  181. c.Members[1].Terminate(t)
  182. c.Members[2].Terminate(t)
  183. c.Members[0].ForceNewCluster = true
  184. err = c.Members[0].Restart(t)
  185. if err != nil {
  186. t.Fatalf("unexpected ForceRestart error: %v", err)
  187. }
  188. defer c.Members[0].Terminate(t)
  189. c.waitLeader(t, c.Members[:1])
  190. // use new http client to init new connection
  191. cc = mustNewHTTPClient(t, []string{c.Members[0].URL()})
  192. kapi = client.NewKeysAPI(cc)
  193. // ensure force restart keep the old data, and new cluster can make progress
  194. ctx, cancel = context.WithTimeout(context.Background(), requestTimeout)
  195. if _, err := kapi.Watcher("/foo", &client.WatcherOptions{AfterIndex: resp.Node.ModifiedIndex - 1}).Next(ctx); err != nil {
  196. t.Fatalf("unexpected watch error: %v", err)
  197. }
  198. cancel()
  199. clusterMustProgress(t, c.Members[:1])
  200. }
  201. func TestAddMemberAfterClusterFullRotation(t *testing.T) {
  202. defer testutil.AfterTest(t)
  203. c := NewCluster(t, 3)
  204. c.Launch(t)
  205. defer c.Terminate(t)
  206. // remove all the previous three members and add in three new members.
  207. for i := 0; i < 3; i++ {
  208. c.RemoveMember(t, uint64(c.Members[0].s.ID()))
  209. c.waitLeader(t, c.Members)
  210. c.AddMember(t)
  211. c.waitLeader(t, c.Members)
  212. }
  213. c.AddMember(t)
  214. c.waitLeader(t, c.Members)
  215. clusterMustProgress(t, c.Members)
  216. }
  217. // Ensure we can remove a member then add a new one back immediately.
  218. func TestIssue2681(t *testing.T) {
  219. defer testutil.AfterTest(t)
  220. c := NewCluster(t, 5)
  221. c.Launch(t)
  222. defer c.Terminate(t)
  223. c.RemoveMember(t, uint64(c.Members[4].s.ID()))
  224. c.waitLeader(t, c.Members)
  225. c.AddMember(t)
  226. c.waitLeader(t, c.Members)
  227. clusterMustProgress(t, c.Members)
  228. }
  229. // Ensure we can remove a member after a snapshot then add a new one back.
  230. func TestIssue2746(t *testing.T) {
  231. defer testutil.AfterTest(t)
  232. c := NewCluster(t, 5)
  233. for _, m := range c.Members {
  234. m.SnapCount = 10
  235. }
  236. c.Launch(t)
  237. defer c.Terminate(t)
  238. // force a snapshot
  239. for i := 0; i < 20; i++ {
  240. clusterMustProgress(t, c.Members)
  241. }
  242. c.RemoveMember(t, uint64(c.Members[4].s.ID()))
  243. c.waitLeader(t, c.Members)
  244. c.AddMember(t)
  245. c.waitLeader(t, c.Members)
  246. clusterMustProgress(t, c.Members)
  247. }
  248. // Ensure etcd will not panic when removing a just started member.
  249. func TestIssue2904(t *testing.T) {
  250. defer testutil.AfterTest(t)
  251. // start 1-member cluster to ensure member 0 is the leader of the cluster.
  252. c := NewCluster(t, 1)
  253. c.Launch(t)
  254. defer c.Terminate(t)
  255. c.AddMember(t)
  256. c.Members[1].Stop(t)
  257. // send remove member-1 request to the cluster.
  258. cc := mustNewHTTPClient(t, c.URLs())
  259. ma := client.NewMembersAPI(cc)
  260. ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
  261. // the proposal is not committed because member 1 is stopped, but the
  262. // proposal is appended to leader's raft log.
  263. ma.Remove(ctx, c.Members[1].s.ID().String())
  264. cancel()
  265. // restart member, and expect it to send updateAttr request.
  266. // the log in the leader is like this:
  267. // [..., remove 1, ..., update attr 1, ...]
  268. c.Members[1].Restart(t)
  269. // when the member comes back, it ack the proposal to remove itself,
  270. // and apply it.
  271. <-c.Members[1].s.StopNotify()
  272. // terminate removed member
  273. c.Members[1].Terminate(t)
  274. c.Members = c.Members[:1]
  275. // wait member to be removed.
  276. c.waitMembersMatch(t, c.HTTPMembers())
  277. }
  278. // clusterMustProgress ensures that cluster can make progress. It creates
  279. // a random key first, and check the new key could be got from all client urls
  280. // of the cluster.
  281. func clusterMustProgress(t *testing.T, membs []*member) {
  282. cc := mustNewHTTPClient(t, []string{membs[0].URL()})
  283. kapi := client.NewKeysAPI(cc)
  284. ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
  285. key := fmt.Sprintf("foo%d", rand.Int())
  286. resp, err := kapi.Create(ctx, "/"+key, "bar")
  287. if err != nil {
  288. t.Fatalf("create on %s error: %v", membs[0].URL(), err)
  289. }
  290. cancel()
  291. for i, m := range membs {
  292. u := m.URL()
  293. mcc := mustNewHTTPClient(t, []string{u})
  294. mkapi := client.NewKeysAPI(mcc)
  295. mctx, mcancel := context.WithTimeout(context.Background(), requestTimeout)
  296. if _, err := mkapi.Watcher(key, &client.WatcherOptions{AfterIndex: resp.Node.ModifiedIndex - 1}).Next(mctx); err != nil {
  297. t.Fatalf("#%d: watch on %s error: %v", i, u, err)
  298. }
  299. mcancel()
  300. }
  301. }
  302. type clusterConfig struct {
  303. size int
  304. usePeerTLS bool
  305. discoveryURL string
  306. useV3 bool
  307. useGRPC bool
  308. }
  309. type cluster struct {
  310. cfg *clusterConfig
  311. Members []*member
  312. }
  313. func (c *cluster) fillClusterForMembers() error {
  314. if c.cfg.discoveryURL != "" {
  315. // cluster will be discovered
  316. return nil
  317. }
  318. addrs := make([]string, 0)
  319. for _, m := range c.Members {
  320. scheme := "http"
  321. if !m.PeerTLSInfo.Empty() {
  322. scheme = "https"
  323. }
  324. for _, l := range m.PeerListeners {
  325. addrs = append(addrs, fmt.Sprintf("%s=%s://%s", m.Name, scheme, l.Addr().String()))
  326. }
  327. }
  328. clusterStr := strings.Join(addrs, ",")
  329. var err error
  330. for _, m := range c.Members {
  331. m.InitialPeerURLsMap, err = types.NewURLsMap(clusterStr)
  332. if err != nil {
  333. return err
  334. }
  335. }
  336. return nil
  337. }
  338. func newCluster(t *testing.T, cfg *clusterConfig) *cluster {
  339. c := &cluster{cfg: cfg}
  340. ms := make([]*member, cfg.size)
  341. for i := 0; i < cfg.size; i++ {
  342. ms[i] = c.mustNewMember(t)
  343. }
  344. c.Members = ms
  345. if err := c.fillClusterForMembers(); err != nil {
  346. t.Fatal(err)
  347. }
  348. return c
  349. }
  350. // NewCluster returns an unlaunched cluster of the given size which has been
  351. // set to use static bootstrap.
  352. func NewCluster(t *testing.T, size int) *cluster {
  353. return newCluster(t, &clusterConfig{size: size})
  354. }
  355. // NewClusterByConfig returns an unlaunched cluster defined by a cluster configuration
  356. func NewClusterByConfig(t *testing.T, cfg *clusterConfig) *cluster {
  357. return newCluster(t, cfg)
  358. }
  359. func (c *cluster) Launch(t *testing.T) {
  360. errc := make(chan error)
  361. for _, m := range c.Members {
  362. // Members are launched in separate goroutines because if they boot
  363. // using discovery url, they have to wait for others to register to continue.
  364. go func(m *member) {
  365. errc <- m.Launch()
  366. }(m)
  367. }
  368. for range c.Members {
  369. if err := <-errc; err != nil {
  370. t.Fatalf("error setting up member: %v", err)
  371. }
  372. }
  373. // wait cluster to be stable to receive future client requests
  374. c.waitMembersMatch(t, c.HTTPMembers())
  375. c.waitVersion()
  376. }
  377. func (c *cluster) URL(i int) string {
  378. return c.Members[i].ClientURLs[0].String()
  379. }
  380. func (c *cluster) URLs() []string {
  381. urls := make([]string, 0)
  382. for _, m := range c.Members {
  383. for _, u := range m.ClientURLs {
  384. urls = append(urls, u.String())
  385. }
  386. }
  387. return urls
  388. }
  389. func (c *cluster) HTTPMembers() []client.Member {
  390. ms := make([]client.Member, len(c.Members))
  391. for i, m := range c.Members {
  392. scheme := "http"
  393. if !m.PeerTLSInfo.Empty() {
  394. scheme = "https"
  395. }
  396. ms[i].Name = m.Name
  397. for _, ln := range m.PeerListeners {
  398. ms[i].PeerURLs = append(ms[i].PeerURLs, scheme+"://"+ln.Addr().String())
  399. }
  400. for _, ln := range m.ClientListeners {
  401. ms[i].ClientURLs = append(ms[i].ClientURLs, "http://"+ln.Addr().String())
  402. }
  403. }
  404. return ms
  405. }
  406. func (c *cluster) mustNewMember(t *testing.T) *member {
  407. name := c.name(rand.Int())
  408. m := mustNewMember(t, name, c.cfg.usePeerTLS)
  409. m.DiscoveryURL = c.cfg.discoveryURL
  410. m.V3demo = c.cfg.useV3
  411. if c.cfg.useGRPC {
  412. if err := m.listenGRPC(); err != nil {
  413. t.Fatal(err)
  414. }
  415. }
  416. return m
  417. }
  418. func (c *cluster) addMember(t *testing.T) {
  419. m := c.mustNewMember(t)
  420. scheme := "http"
  421. if c.cfg.usePeerTLS {
  422. scheme = "https"
  423. }
  424. // send add request to the cluster
  425. cc := mustNewHTTPClient(t, []string{c.URL(0)})
  426. ma := client.NewMembersAPI(cc)
  427. ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
  428. peerURL := scheme + "://" + m.PeerListeners[0].Addr().String()
  429. if _, err := ma.Add(ctx, peerURL); err != nil {
  430. t.Fatalf("add member on %s error: %v", c.URL(0), err)
  431. }
  432. cancel()
  433. // wait for the add node entry applied in the cluster
  434. members := append(c.HTTPMembers(), client.Member{PeerURLs: []string{peerURL}, ClientURLs: []string{}})
  435. c.waitMembersMatch(t, members)
  436. m.InitialPeerURLsMap = types.URLsMap{}
  437. for _, mm := range c.Members {
  438. m.InitialPeerURLsMap[mm.Name] = mm.PeerURLs
  439. }
  440. m.InitialPeerURLsMap[m.Name] = m.PeerURLs
  441. m.NewCluster = false
  442. if err := m.Launch(); err != nil {
  443. t.Fatal(err)
  444. }
  445. c.Members = append(c.Members, m)
  446. // wait cluster to be stable to receive future client requests
  447. c.waitMembersMatch(t, c.HTTPMembers())
  448. }
  449. func (c *cluster) AddMember(t *testing.T) {
  450. c.addMember(t)
  451. }
  452. func (c *cluster) RemoveMember(t *testing.T, id uint64) {
  453. // send remove request to the cluster
  454. cc := mustNewHTTPClient(t, c.URLs())
  455. ma := client.NewMembersAPI(cc)
  456. ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
  457. if err := ma.Remove(ctx, types.ID(id).String()); err != nil {
  458. t.Fatalf("unexpected remove error %v", err)
  459. }
  460. cancel()
  461. newMembers := make([]*member, 0)
  462. for _, m := range c.Members {
  463. if uint64(m.s.ID()) != id {
  464. newMembers = append(newMembers, m)
  465. } else {
  466. select {
  467. case <-m.s.StopNotify():
  468. m.Terminate(t)
  469. // 1s stop delay + election timeout + 1s disk and network delay + connection write timeout
  470. // TODO: remove connection write timeout by selecting on http response closeNotifier
  471. // blocking on https://github.com/golang/go/issues/9524
  472. case <-time.After(time.Second + time.Duration(electionTicks)*tickDuration + time.Second + rafthttp.ConnWriteTimeout):
  473. t.Fatalf("failed to remove member %s in time", m.s.ID())
  474. }
  475. }
  476. }
  477. c.Members = newMembers
  478. c.waitMembersMatch(t, c.HTTPMembers())
  479. }
  480. func (c *cluster) Terminate(t *testing.T) {
  481. for _, m := range c.Members {
  482. m.Terminate(t)
  483. }
  484. }
  485. func (c *cluster) waitMembersMatch(t *testing.T, membs []client.Member) {
  486. for _, u := range c.URLs() {
  487. cc := mustNewHTTPClient(t, []string{u})
  488. ma := client.NewMembersAPI(cc)
  489. for {
  490. ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
  491. ms, err := ma.List(ctx)
  492. cancel()
  493. if err == nil && isMembersEqual(ms, membs) {
  494. break
  495. }
  496. time.Sleep(tickDuration)
  497. }
  498. }
  499. return
  500. }
  501. func (c *cluster) waitLeader(t *testing.T, membs []*member) {
  502. possibleLead := make(map[uint64]bool)
  503. var lead uint64
  504. for _, m := range membs {
  505. possibleLead[uint64(m.s.ID())] = true
  506. }
  507. for lead == 0 || !possibleLead[lead] {
  508. lead = 0
  509. for _, m := range membs {
  510. if lead != 0 && lead != m.s.Lead() {
  511. lead = 0
  512. break
  513. }
  514. lead = m.s.Lead()
  515. }
  516. time.Sleep(10 * tickDuration)
  517. }
  518. }
  519. func (c *cluster) waitVersion() {
  520. for _, m := range c.Members {
  521. for {
  522. if m.s.ClusterVersion() != nil {
  523. break
  524. }
  525. time.Sleep(tickDuration)
  526. }
  527. }
  528. }
  529. func (c *cluster) name(i int) string {
  530. return fmt.Sprint("node", i)
  531. }
  532. // isMembersEqual checks whether two members equal except ID field.
  533. // The given wmembs should always set ID field to empty string.
  534. func isMembersEqual(membs []client.Member, wmembs []client.Member) bool {
  535. sort.Sort(SortableMemberSliceByPeerURLs(membs))
  536. sort.Sort(SortableMemberSliceByPeerURLs(wmembs))
  537. for i := range membs {
  538. membs[i].ID = ""
  539. }
  540. return reflect.DeepEqual(membs, wmembs)
  541. }
  542. func newLocalListener(t *testing.T) net.Listener {
  543. port := atomic.AddInt64(&nextListenPort, 1)
  544. l, err := net.Listen("tcp", "127.0.0.1:"+strconv.FormatInt(port, 10))
  545. if err != nil {
  546. t.Fatal(err)
  547. }
  548. return l
  549. }
  550. func newListenerWithAddr(t *testing.T, addr string) net.Listener {
  551. var err error
  552. var l net.Listener
  553. // TODO: we want to reuse a previous closed port immediately.
  554. // a better way is to set SO_REUSExx instead of doing retry.
  555. for i := 0; i < 5; i++ {
  556. l, err = net.Listen("tcp", addr)
  557. if err == nil {
  558. break
  559. }
  560. time.Sleep(500 * time.Millisecond)
  561. }
  562. if err != nil {
  563. t.Fatal(err)
  564. }
  565. return l
  566. }
  567. type member struct {
  568. etcdserver.ServerConfig
  569. PeerListeners, ClientListeners []net.Listener
  570. grpcListener net.Listener
  571. // inited PeerTLSInfo implies to enable peer TLS
  572. PeerTLSInfo transport.TLSInfo
  573. raftHandler *testutil.PauseableHandler
  574. s *etcdserver.EtcdServer
  575. hss []*httptest.Server
  576. grpcServer *grpc.Server
  577. grpcAddr string
  578. }
  579. // mustNewMember return an inited member with the given name. If usePeerTLS is
  580. // true, it will set PeerTLSInfo and use https scheme to communicate between
  581. // peers.
  582. func mustNewMember(t *testing.T, name string, usePeerTLS bool) *member {
  583. var (
  584. testTLSInfo = transport.TLSInfo{
  585. KeyFile: "./fixtures/server.key.insecure",
  586. CertFile: "./fixtures/server.crt",
  587. TrustedCAFile: "./fixtures/ca.crt",
  588. ClientCertAuth: true,
  589. }
  590. err error
  591. )
  592. m := &member{}
  593. peerScheme := "http"
  594. if usePeerTLS {
  595. peerScheme = "https"
  596. }
  597. pln := newLocalListener(t)
  598. m.PeerListeners = []net.Listener{pln}
  599. m.PeerURLs, err = types.NewURLs([]string{peerScheme + "://" + pln.Addr().String()})
  600. if err != nil {
  601. t.Fatal(err)
  602. }
  603. if usePeerTLS {
  604. m.PeerTLSInfo = testTLSInfo
  605. }
  606. cln := newLocalListener(t)
  607. m.ClientListeners = []net.Listener{cln}
  608. m.ClientURLs, err = types.NewURLs([]string{"http://" + cln.Addr().String()})
  609. if err != nil {
  610. t.Fatal(err)
  611. }
  612. m.Name = name
  613. m.DataDir, err = ioutil.TempDir(os.TempDir(), "etcd")
  614. if err != nil {
  615. t.Fatal(err)
  616. }
  617. clusterStr := fmt.Sprintf("%s=%s://%s", name, peerScheme, pln.Addr().String())
  618. m.InitialPeerURLsMap, err = types.NewURLsMap(clusterStr)
  619. if err != nil {
  620. t.Fatal(err)
  621. }
  622. m.InitialClusterToken = clusterName
  623. m.NewCluster = true
  624. m.ServerConfig.PeerTLSInfo = m.PeerTLSInfo
  625. m.ElectionTicks = electionTicks
  626. m.TickMs = uint(tickDuration / time.Millisecond)
  627. return m
  628. }
  629. // startGRPC starts a grpc server over a unix domain socket on the member
  630. func (m *member) listenGRPC() error {
  631. if m.V3demo == false {
  632. return fmt.Errorf("starting grpc server without v3 configured")
  633. }
  634. m.grpcAddr = m.Name + ".sock"
  635. if err := os.RemoveAll(m.grpcAddr); err != nil {
  636. return err
  637. }
  638. l, err := net.Listen("unix", m.grpcAddr)
  639. if err != nil {
  640. return fmt.Errorf("listen failed on grpc socket %s (%v)", m.grpcAddr, err)
  641. }
  642. m.grpcListener = l
  643. return nil
  644. }
  645. // NewClientV3 creates a new grpc client connection to the member
  646. func NewClientV3(m *member) (*clientv3.Client, error) {
  647. if m.grpcAddr == "" {
  648. return nil, fmt.Errorf("member not configured for grpc")
  649. }
  650. f := func(a string, t time.Duration) (net.Conn, error) {
  651. return net.Dial("unix", a)
  652. }
  653. unixdialer := grpc.WithDialer(f)
  654. conn, err := grpc.Dial(m.grpcAddr, grpc.WithInsecure(), unixdialer)
  655. if err != nil {
  656. return nil, err
  657. }
  658. return clientv3.NewFromConn(conn), nil
  659. }
  660. // Clone returns a member with the same server configuration. The returned
  661. // member will not set PeerListeners and ClientListeners.
  662. func (m *member) Clone(t *testing.T) *member {
  663. mm := &member{}
  664. mm.ServerConfig = m.ServerConfig
  665. var err error
  666. clientURLStrs := m.ClientURLs.StringSlice()
  667. mm.ClientURLs, err = types.NewURLs(clientURLStrs)
  668. if err != nil {
  669. // this should never fail
  670. panic(err)
  671. }
  672. peerURLStrs := m.PeerURLs.StringSlice()
  673. mm.PeerURLs, err = types.NewURLs(peerURLStrs)
  674. if err != nil {
  675. // this should never fail
  676. panic(err)
  677. }
  678. clusterStr := m.InitialPeerURLsMap.String()
  679. mm.InitialPeerURLsMap, err = types.NewURLsMap(clusterStr)
  680. if err != nil {
  681. // this should never fail
  682. panic(err)
  683. }
  684. mm.InitialClusterToken = m.InitialClusterToken
  685. mm.ElectionTicks = m.ElectionTicks
  686. mm.PeerTLSInfo = m.PeerTLSInfo
  687. return mm
  688. }
  689. // Launch starts a member based on ServerConfig, PeerListeners
  690. // and ClientListeners.
  691. func (m *member) Launch() error {
  692. var err error
  693. if m.s, err = etcdserver.NewServer(&m.ServerConfig); err != nil {
  694. return fmt.Errorf("failed to initialize the etcd server: %v", err)
  695. }
  696. m.s.SyncTicker = time.Tick(500 * time.Millisecond)
  697. m.s.Start()
  698. m.raftHandler = &testutil.PauseableHandler{Next: etcdhttp.NewPeerHandler(m.s)}
  699. for _, ln := range m.PeerListeners {
  700. hs := &httptest.Server{
  701. Listener: ln,
  702. Config: &http.Server{Handler: m.raftHandler},
  703. }
  704. if m.PeerTLSInfo.Empty() {
  705. hs.Start()
  706. } else {
  707. hs.TLS, err = m.PeerTLSInfo.ServerConfig()
  708. if err != nil {
  709. return err
  710. }
  711. hs.StartTLS()
  712. }
  713. m.hss = append(m.hss, hs)
  714. }
  715. for _, ln := range m.ClientListeners {
  716. hs := &httptest.Server{
  717. Listener: ln,
  718. Config: &http.Server{Handler: etcdhttp.NewClientHandler(m.s, m.ServerConfig.ReqTimeout())},
  719. }
  720. hs.Start()
  721. m.hss = append(m.hss, hs)
  722. }
  723. if m.grpcListener != nil {
  724. m.grpcServer = grpc.NewServer()
  725. etcdserverpb.RegisterKVServer(m.grpcServer, v3rpc.NewKVServer(m.s))
  726. etcdserverpb.RegisterWatchServer(m.grpcServer, v3rpc.NewWatchServer(m.s))
  727. etcdserverpb.RegisterLeaseServer(m.grpcServer, v3rpc.NewLeaseServer(m.s))
  728. go m.grpcServer.Serve(m.grpcListener)
  729. }
  730. return nil
  731. }
  732. func (m *member) WaitOK(t *testing.T) {
  733. cc := mustNewHTTPClient(t, []string{m.URL()})
  734. kapi := client.NewKeysAPI(cc)
  735. for {
  736. ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
  737. _, err := kapi.Get(ctx, "/", nil)
  738. if err != nil {
  739. time.Sleep(tickDuration)
  740. continue
  741. }
  742. cancel()
  743. break
  744. }
  745. for m.s.Leader() == 0 {
  746. time.Sleep(tickDuration)
  747. }
  748. }
  749. func (m *member) URL() string { return m.ClientURLs[0].String() }
  750. func (m *member) Pause() {
  751. m.raftHandler.Pause()
  752. m.s.PauseSending()
  753. }
  754. func (m *member) Resume() {
  755. m.raftHandler.Resume()
  756. m.s.ResumeSending()
  757. }
  758. // Close stops the member's etcdserver and closes its connections
  759. func (m *member) Close() {
  760. if m.grpcServer != nil {
  761. m.grpcServer.Stop()
  762. m.grpcServer = nil
  763. }
  764. m.s.Stop()
  765. for _, hs := range m.hss {
  766. hs.CloseClientConnections()
  767. hs.Close()
  768. }
  769. }
  770. // Stop stops the member, but the data dir of the member is preserved.
  771. func (m *member) Stop(t *testing.T) {
  772. m.Close()
  773. m.hss = nil
  774. }
  775. // Restart starts the member using the preserved data dir.
  776. func (m *member) Restart(t *testing.T) error {
  777. newPeerListeners := make([]net.Listener, 0)
  778. for _, ln := range m.PeerListeners {
  779. newPeerListeners = append(newPeerListeners, newListenerWithAddr(t, ln.Addr().String()))
  780. }
  781. m.PeerListeners = newPeerListeners
  782. newClientListeners := make([]net.Listener, 0)
  783. for _, ln := range m.ClientListeners {
  784. newClientListeners = append(newClientListeners, newListenerWithAddr(t, ln.Addr().String()))
  785. }
  786. m.ClientListeners = newClientListeners
  787. if m.grpcListener != nil {
  788. if err := m.listenGRPC(); err != nil {
  789. t.Fatal(err)
  790. }
  791. }
  792. return m.Launch()
  793. }
  794. // Terminate stops the member and removes the data dir.
  795. func (m *member) Terminate(t *testing.T) {
  796. m.Close()
  797. if err := os.RemoveAll(m.ServerConfig.DataDir); err != nil {
  798. t.Fatal(err)
  799. }
  800. }
  801. func mustNewHTTPClient(t *testing.T, eps []string) client.Client {
  802. cfg := client.Config{Transport: mustNewTransport(t, transport.TLSInfo{}), Endpoints: eps}
  803. c, err := client.New(cfg)
  804. if err != nil {
  805. t.Fatal(err)
  806. }
  807. return c
  808. }
  809. func mustNewTransport(t *testing.T, tlsInfo transport.TLSInfo) *http.Transport {
  810. // tick in integration test is short, so 1s dial timeout could play well.
  811. tr, err := transport.NewTimeoutTransport(tlsInfo, time.Second, rafthttp.ConnReadTimeout, rafthttp.ConnWriteTimeout)
  812. if err != nil {
  813. t.Fatal(err)
  814. }
  815. return tr
  816. }
  817. type SortableMemberSliceByPeerURLs []client.Member
  818. func (p SortableMemberSliceByPeerURLs) Len() int { return len(p) }
  819. func (p SortableMemberSliceByPeerURLs) Less(i, j int) bool {
  820. return p[i].PeerURLs[0] < p[j].PeerURLs[0]
  821. }
  822. func (p SortableMemberSliceByPeerURLs) Swap(i, j int) { p[i], p[j] = p[j], p[i] }