cluster.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810
  1. // Copyright 2016 The etcd Authors
  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. "crypto/tls"
  17. "fmt"
  18. "io/ioutil"
  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. "golang.org/x/net/context"
  32. "google.golang.org/grpc"
  33. "github.com/coreos/etcd/client"
  34. "github.com/coreos/etcd/clientv3"
  35. "github.com/coreos/etcd/etcdserver"
  36. "github.com/coreos/etcd/etcdserver/api"
  37. "github.com/coreos/etcd/etcdserver/api/v2http"
  38. "github.com/coreos/etcd/etcdserver/api/v3rpc"
  39. pb "github.com/coreos/etcd/etcdserver/etcdserverpb"
  40. "github.com/coreos/etcd/pkg/testutil"
  41. "github.com/coreos/etcd/pkg/transport"
  42. "github.com/coreos/etcd/pkg/types"
  43. "github.com/coreos/etcd/rafthttp"
  44. "github.com/coreos/pkg/capnslog"
  45. )
  46. const (
  47. tickDuration = 10 * time.Millisecond
  48. clusterName = "etcd"
  49. requestTimeout = 20 * time.Second
  50. )
  51. var (
  52. electionTicks = 10
  53. // integration test uses well-known ports to listen for each running member,
  54. // which ensures restarted member could listen on specific port again.
  55. nextListenPort int64 = 21000
  56. testTLSInfo = transport.TLSInfo{
  57. KeyFile: "./fixtures/server.key.insecure",
  58. CertFile: "./fixtures/server.crt",
  59. TrustedCAFile: "./fixtures/ca.crt",
  60. ClientCertAuth: true,
  61. }
  62. plog = capnslog.NewPackageLogger("github.com/coreos/etcd", "integration")
  63. )
  64. type ClusterConfig struct {
  65. Size int
  66. PeerTLS *transport.TLSInfo
  67. ClientTLS *transport.TLSInfo
  68. DiscoveryURL string
  69. UseGRPC bool
  70. QuotaBackendBytes int64
  71. }
  72. type cluster struct {
  73. cfg *ClusterConfig
  74. Members []*member
  75. }
  76. func init() {
  77. // manually enable v3 capability since we know the cluster members all support v3.
  78. api.EnableCapability(api.V3rpcCapability)
  79. }
  80. func (c *cluster) fillClusterForMembers() error {
  81. if c.cfg.DiscoveryURL != "" {
  82. // cluster will be discovered
  83. return nil
  84. }
  85. addrs := make([]string, 0)
  86. for _, m := range c.Members {
  87. scheme := "http"
  88. if m.PeerTLSInfo != nil {
  89. scheme = "https"
  90. }
  91. for _, l := range m.PeerListeners {
  92. addrs = append(addrs, fmt.Sprintf("%s=%s://%s", m.Name, scheme, l.Addr().String()))
  93. }
  94. }
  95. clusterStr := strings.Join(addrs, ",")
  96. var err error
  97. for _, m := range c.Members {
  98. m.InitialPeerURLsMap, err = types.NewURLsMap(clusterStr)
  99. if err != nil {
  100. return err
  101. }
  102. }
  103. return nil
  104. }
  105. func newCluster(t *testing.T, cfg *ClusterConfig) *cluster {
  106. c := &cluster{cfg: cfg}
  107. ms := make([]*member, cfg.Size)
  108. for i := 0; i < cfg.Size; i++ {
  109. ms[i] = c.mustNewMember(t)
  110. }
  111. c.Members = ms
  112. if err := c.fillClusterForMembers(); err != nil {
  113. t.Fatal(err)
  114. }
  115. return c
  116. }
  117. // NewCluster returns an unlaunched cluster of the given size which has been
  118. // set to use static bootstrap.
  119. func NewCluster(t *testing.T, size int) *cluster {
  120. return newCluster(t, &ClusterConfig{Size: size})
  121. }
  122. // NewClusterByConfig returns an unlaunched cluster defined by a cluster configuration
  123. func NewClusterByConfig(t *testing.T, cfg *ClusterConfig) *cluster {
  124. return newCluster(t, cfg)
  125. }
  126. func (c *cluster) Launch(t *testing.T) {
  127. errc := make(chan error)
  128. for _, m := range c.Members {
  129. // Members are launched in separate goroutines because if they boot
  130. // using discovery url, they have to wait for others to register to continue.
  131. go func(m *member) {
  132. errc <- m.Launch()
  133. }(m)
  134. }
  135. for range c.Members {
  136. if err := <-errc; err != nil {
  137. t.Fatalf("error setting up member: %v", err)
  138. }
  139. }
  140. // wait cluster to be stable to receive future client requests
  141. c.waitMembersMatch(t, c.HTTPMembers())
  142. c.waitVersion()
  143. }
  144. func (c *cluster) URL(i int) string {
  145. return c.Members[i].ClientURLs[0].String()
  146. }
  147. // URLs returns a list of all active client URLs in the cluster
  148. func (c *cluster) URLs() []string {
  149. urls := make([]string, 0)
  150. for _, m := range c.Members {
  151. select {
  152. case <-m.s.StopNotify():
  153. continue
  154. default:
  155. }
  156. for _, u := range m.ClientURLs {
  157. urls = append(urls, u.String())
  158. }
  159. }
  160. return urls
  161. }
  162. // HTTPMembers returns a list of all active members as client.Members
  163. func (c *cluster) HTTPMembers() []client.Member {
  164. ms := []client.Member{}
  165. for _, m := range c.Members {
  166. pScheme, cScheme := "http", "http"
  167. if m.PeerTLSInfo != nil {
  168. pScheme = "https"
  169. }
  170. if m.ClientTLSInfo != nil {
  171. cScheme = "https"
  172. }
  173. cm := client.Member{Name: m.Name}
  174. for _, ln := range m.PeerListeners {
  175. cm.PeerURLs = append(cm.PeerURLs, pScheme+"://"+ln.Addr().String())
  176. }
  177. for _, ln := range m.ClientListeners {
  178. cm.ClientURLs = append(cm.ClientURLs, cScheme+"://"+ln.Addr().String())
  179. }
  180. ms = append(ms, cm)
  181. }
  182. return ms
  183. }
  184. func (c *cluster) mustNewMember(t *testing.T) *member {
  185. m := mustNewMember(t,
  186. memberConfig{
  187. name: c.name(rand.Int()),
  188. peerTLS: c.cfg.PeerTLS,
  189. clientTLS: c.cfg.ClientTLS,
  190. quotaBackendBytes: c.cfg.QuotaBackendBytes,
  191. })
  192. m.DiscoveryURL = c.cfg.DiscoveryURL
  193. if c.cfg.UseGRPC {
  194. if err := m.listenGRPC(); err != nil {
  195. t.Fatal(err)
  196. }
  197. }
  198. return m
  199. }
  200. func (c *cluster) addMember(t *testing.T) {
  201. m := c.mustNewMember(t)
  202. scheme := "http"
  203. if c.cfg.PeerTLS != nil {
  204. scheme = "https"
  205. }
  206. // send add request to the cluster
  207. var err error
  208. for i := 0; i < len(c.Members); i++ {
  209. clientURL := c.URL(i)
  210. peerURL := scheme + "://" + m.PeerListeners[0].Addr().String()
  211. if err = c.addMemberByURL(t, clientURL, peerURL); err == nil {
  212. break
  213. }
  214. }
  215. if err != nil {
  216. t.Fatalf("add member failed on all members error: %v", err)
  217. }
  218. m.InitialPeerURLsMap = types.URLsMap{}
  219. for _, mm := range c.Members {
  220. m.InitialPeerURLsMap[mm.Name] = mm.PeerURLs
  221. }
  222. m.InitialPeerURLsMap[m.Name] = m.PeerURLs
  223. m.NewCluster = false
  224. if err := m.Launch(); err != nil {
  225. t.Fatal(err)
  226. }
  227. c.Members = append(c.Members, m)
  228. // wait cluster to be stable to receive future client requests
  229. c.waitMembersMatch(t, c.HTTPMembers())
  230. }
  231. func (c *cluster) addMemberByURL(t *testing.T, clientURL, peerURL string) error {
  232. cc := mustNewHTTPClient(t, []string{clientURL}, c.cfg.ClientTLS)
  233. ma := client.NewMembersAPI(cc)
  234. ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
  235. if _, err := ma.Add(ctx, peerURL); err != nil {
  236. return err
  237. }
  238. cancel()
  239. // wait for the add node entry applied in the cluster
  240. members := append(c.HTTPMembers(), client.Member{PeerURLs: []string{peerURL}, ClientURLs: []string{}})
  241. c.waitMembersMatch(t, members)
  242. return nil
  243. }
  244. func (c *cluster) AddMember(t *testing.T) {
  245. c.addMember(t)
  246. }
  247. func (c *cluster) RemoveMember(t *testing.T, id uint64) {
  248. // send remove request to the cluster
  249. cc := mustNewHTTPClient(t, c.URLs(), c.cfg.ClientTLS)
  250. ma := client.NewMembersAPI(cc)
  251. ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
  252. if err := ma.Remove(ctx, types.ID(id).String()); err != nil {
  253. t.Fatalf("unexpected remove error %v", err)
  254. }
  255. cancel()
  256. newMembers := make([]*member, 0)
  257. for _, m := range c.Members {
  258. if uint64(m.s.ID()) != id {
  259. newMembers = append(newMembers, m)
  260. } else {
  261. select {
  262. case <-m.s.StopNotify():
  263. m.Terminate(t)
  264. // 1s stop delay + election timeout + 1s disk and network delay + connection write timeout
  265. // TODO: remove connection write timeout by selecting on http response closeNotifier
  266. // blocking on https://github.com/golang/go/issues/9524
  267. case <-time.After(time.Second + time.Duration(electionTicks)*tickDuration + time.Second + rafthttp.ConnWriteTimeout):
  268. t.Fatalf("failed to remove member %s in time", m.s.ID())
  269. }
  270. }
  271. }
  272. c.Members = newMembers
  273. c.waitMembersMatch(t, c.HTTPMembers())
  274. }
  275. func (c *cluster) Terminate(t *testing.T) {
  276. for _, m := range c.Members {
  277. m.Terminate(t)
  278. }
  279. }
  280. func (c *cluster) waitMembersMatch(t *testing.T, membs []client.Member) {
  281. for _, u := range c.URLs() {
  282. cc := mustNewHTTPClient(t, []string{u}, c.cfg.ClientTLS)
  283. ma := client.NewMembersAPI(cc)
  284. for {
  285. ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
  286. ms, err := ma.List(ctx)
  287. cancel()
  288. if err == nil && isMembersEqual(ms, membs) {
  289. break
  290. }
  291. time.Sleep(tickDuration)
  292. }
  293. }
  294. return
  295. }
  296. func (c *cluster) WaitLeader(t *testing.T) int { return c.waitLeader(t, c.Members) }
  297. func (c *cluster) waitLeader(t *testing.T, membs []*member) int {
  298. possibleLead := make(map[uint64]bool)
  299. var lead uint64
  300. for _, m := range membs {
  301. possibleLead[uint64(m.s.ID())] = true
  302. }
  303. for lead == 0 || !possibleLead[lead] {
  304. lead = 0
  305. for _, m := range membs {
  306. select {
  307. case <-m.s.StopNotify():
  308. continue
  309. default:
  310. }
  311. if lead != 0 && lead != m.s.Lead() {
  312. lead = 0
  313. time.Sleep(10 * tickDuration)
  314. break
  315. }
  316. lead = m.s.Lead()
  317. }
  318. }
  319. for i, m := range membs {
  320. if uint64(m.s.ID()) == lead {
  321. return i
  322. }
  323. }
  324. return -1
  325. }
  326. func (c *cluster) waitVersion() {
  327. for _, m := range c.Members {
  328. for {
  329. if m.s.ClusterVersion() != nil {
  330. break
  331. }
  332. time.Sleep(tickDuration)
  333. }
  334. }
  335. }
  336. func (c *cluster) name(i int) string {
  337. return fmt.Sprint("node", i)
  338. }
  339. // isMembersEqual checks whether two members equal except ID field.
  340. // The given wmembs should always set ID field to empty string.
  341. func isMembersEqual(membs []client.Member, wmembs []client.Member) bool {
  342. sort.Sort(SortableMemberSliceByPeerURLs(membs))
  343. sort.Sort(SortableMemberSliceByPeerURLs(wmembs))
  344. for i := range membs {
  345. membs[i].ID = ""
  346. }
  347. return reflect.DeepEqual(membs, wmembs)
  348. }
  349. func newLocalListener(t *testing.T) net.Listener {
  350. port := atomic.AddInt64(&nextListenPort, 1)
  351. l, err := net.Listen("tcp", "127.0.0.1:"+strconv.FormatInt(port, 10))
  352. if err != nil {
  353. t.Fatal(err)
  354. }
  355. return l
  356. }
  357. func newListenerWithAddr(t *testing.T, addr string) net.Listener {
  358. var err error
  359. var l net.Listener
  360. // TODO: we want to reuse a previous closed port immediately.
  361. // a better way is to set SO_REUSExx instead of doing retry.
  362. for i := 0; i < 5; i++ {
  363. l, err = net.Listen("tcp", addr)
  364. if err == nil {
  365. break
  366. }
  367. time.Sleep(500 * time.Millisecond)
  368. }
  369. if err != nil {
  370. t.Fatal(err)
  371. }
  372. return l
  373. }
  374. type member struct {
  375. etcdserver.ServerConfig
  376. PeerListeners, ClientListeners []net.Listener
  377. grpcListener net.Listener
  378. // PeerTLSInfo enables peer TLS when set
  379. PeerTLSInfo *transport.TLSInfo
  380. // ClientTLSInfo enables client TLS when set
  381. ClientTLSInfo *transport.TLSInfo
  382. raftHandler *testutil.PauseableHandler
  383. s *etcdserver.EtcdServer
  384. hss []*httptest.Server
  385. grpcServer *grpc.Server
  386. grpcAddr string
  387. }
  388. func (m *member) GRPCAddr() string { return m.grpcAddr }
  389. type memberConfig struct {
  390. name string
  391. peerTLS *transport.TLSInfo
  392. clientTLS *transport.TLSInfo
  393. quotaBackendBytes int64
  394. }
  395. // mustNewMember return an inited member with the given name. If peerTLS is
  396. // set, it will use https scheme to communicate between peers.
  397. func mustNewMember(t *testing.T, mcfg memberConfig) *member {
  398. var err error
  399. m := &member{}
  400. peerScheme, clientScheme := "http", "http"
  401. if mcfg.peerTLS != nil {
  402. peerScheme = "https"
  403. }
  404. if mcfg.clientTLS != nil {
  405. clientScheme = "https"
  406. }
  407. pln := newLocalListener(t)
  408. m.PeerListeners = []net.Listener{pln}
  409. m.PeerURLs, err = types.NewURLs([]string{peerScheme + "://" + pln.Addr().String()})
  410. if err != nil {
  411. t.Fatal(err)
  412. }
  413. m.PeerTLSInfo = mcfg.peerTLS
  414. cln := newLocalListener(t)
  415. m.ClientListeners = []net.Listener{cln}
  416. m.ClientURLs, err = types.NewURLs([]string{clientScheme + "://" + cln.Addr().String()})
  417. if err != nil {
  418. t.Fatal(err)
  419. }
  420. m.ClientTLSInfo = mcfg.clientTLS
  421. m.Name = mcfg.name
  422. m.DataDir, err = ioutil.TempDir(os.TempDir(), "etcd")
  423. if err != nil {
  424. t.Fatal(err)
  425. }
  426. clusterStr := fmt.Sprintf("%s=%s://%s", mcfg.name, peerScheme, pln.Addr().String())
  427. m.InitialPeerURLsMap, err = types.NewURLsMap(clusterStr)
  428. if err != nil {
  429. t.Fatal(err)
  430. }
  431. m.InitialClusterToken = clusterName
  432. m.NewCluster = true
  433. m.BootstrapTimeout = 10 * time.Millisecond
  434. if m.PeerTLSInfo != nil {
  435. m.ServerConfig.PeerTLSInfo = *m.PeerTLSInfo
  436. }
  437. m.ElectionTicks = electionTicks
  438. m.TickMs = uint(tickDuration / time.Millisecond)
  439. m.QuotaBackendBytes = mcfg.quotaBackendBytes
  440. return m
  441. }
  442. // listenGRPC starts a grpc server over a unix domain socket on the member
  443. func (m *member) listenGRPC() error {
  444. // prefix with localhost so cert has right domain
  445. m.grpcAddr = "localhost:" + m.Name + ".sock"
  446. if err := os.RemoveAll(m.grpcAddr); err != nil {
  447. return err
  448. }
  449. l, err := net.Listen("unix", m.grpcAddr)
  450. if err != nil {
  451. return fmt.Errorf("listen failed on grpc socket %s (%v)", m.grpcAddr, err)
  452. }
  453. m.grpcAddr = "unix://" + m.grpcAddr
  454. m.grpcListener = l
  455. return nil
  456. }
  457. // NewClientV3 creates a new grpc client connection to the member
  458. func NewClientV3(m *member) (*clientv3.Client, error) {
  459. if m.grpcAddr == "" {
  460. return nil, fmt.Errorf("member not configured for grpc")
  461. }
  462. cfg := clientv3.Config{
  463. Endpoints: []string{m.grpcAddr},
  464. DialTimeout: 5 * time.Second,
  465. }
  466. if m.ClientTLSInfo != nil {
  467. tls, err := m.ClientTLSInfo.ClientConfig()
  468. if err != nil {
  469. return nil, err
  470. }
  471. cfg.TLS = tls
  472. }
  473. return clientv3.New(cfg)
  474. }
  475. // Clone returns a member with the same server configuration. The returned
  476. // member will not set PeerListeners and ClientListeners.
  477. func (m *member) Clone(t *testing.T) *member {
  478. mm := &member{}
  479. mm.ServerConfig = m.ServerConfig
  480. var err error
  481. clientURLStrs := m.ClientURLs.StringSlice()
  482. mm.ClientURLs, err = types.NewURLs(clientURLStrs)
  483. if err != nil {
  484. // this should never fail
  485. panic(err)
  486. }
  487. peerURLStrs := m.PeerURLs.StringSlice()
  488. mm.PeerURLs, err = types.NewURLs(peerURLStrs)
  489. if err != nil {
  490. // this should never fail
  491. panic(err)
  492. }
  493. clusterStr := m.InitialPeerURLsMap.String()
  494. mm.InitialPeerURLsMap, err = types.NewURLsMap(clusterStr)
  495. if err != nil {
  496. // this should never fail
  497. panic(err)
  498. }
  499. mm.InitialClusterToken = m.InitialClusterToken
  500. mm.ElectionTicks = m.ElectionTicks
  501. mm.PeerTLSInfo = m.PeerTLSInfo
  502. mm.ClientTLSInfo = m.ClientTLSInfo
  503. return mm
  504. }
  505. // Launch starts a member based on ServerConfig, PeerListeners
  506. // and ClientListeners.
  507. func (m *member) Launch() error {
  508. plog.Printf("launching %s (%s)", m.Name, m.grpcAddr)
  509. var err error
  510. if m.s, err = etcdserver.NewServer(&m.ServerConfig); err != nil {
  511. return fmt.Errorf("failed to initialize the etcd server: %v", err)
  512. }
  513. m.s.SyncTicker = time.Tick(500 * time.Millisecond)
  514. m.s.Start()
  515. m.raftHandler = &testutil.PauseableHandler{Next: v2http.NewPeerHandler(m.s)}
  516. for _, ln := range m.PeerListeners {
  517. hs := &httptest.Server{
  518. Listener: ln,
  519. Config: &http.Server{Handler: m.raftHandler},
  520. }
  521. if m.PeerTLSInfo == nil {
  522. hs.Start()
  523. } else {
  524. hs.TLS, err = m.PeerTLSInfo.ServerConfig()
  525. if err != nil {
  526. return err
  527. }
  528. hs.StartTLS()
  529. }
  530. m.hss = append(m.hss, hs)
  531. }
  532. for _, ln := range m.ClientListeners {
  533. hs := &httptest.Server{
  534. Listener: ln,
  535. Config: &http.Server{Handler: v2http.NewClientHandler(m.s, m.ServerConfig.ReqTimeout())},
  536. }
  537. if m.ClientTLSInfo == nil {
  538. hs.Start()
  539. } else {
  540. hs.TLS, err = m.ClientTLSInfo.ServerConfig()
  541. if err != nil {
  542. return err
  543. }
  544. hs.StartTLS()
  545. }
  546. m.hss = append(m.hss, hs)
  547. }
  548. if m.grpcListener != nil {
  549. var (
  550. tlscfg *tls.Config
  551. )
  552. if m.ClientTLSInfo != nil && !m.ClientTLSInfo.Empty() {
  553. tlscfg, err = m.ClientTLSInfo.ServerConfig()
  554. if err != nil {
  555. return err
  556. }
  557. }
  558. m.grpcServer = v3rpc.Server(m.s, tlscfg)
  559. go m.grpcServer.Serve(m.grpcListener)
  560. }
  561. plog.Printf("launched %s (%s)", m.Name, m.grpcAddr)
  562. return nil
  563. }
  564. func (m *member) WaitOK(t *testing.T) {
  565. cc := mustNewHTTPClient(t, []string{m.URL()}, m.ClientTLSInfo)
  566. kapi := client.NewKeysAPI(cc)
  567. for {
  568. ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
  569. _, err := kapi.Get(ctx, "/", nil)
  570. if err != nil {
  571. time.Sleep(tickDuration)
  572. continue
  573. }
  574. cancel()
  575. break
  576. }
  577. for m.s.Leader() == 0 {
  578. time.Sleep(tickDuration)
  579. }
  580. }
  581. func (m *member) URL() string { return m.ClientURLs[0].String() }
  582. func (m *member) Pause() {
  583. m.raftHandler.Pause()
  584. m.s.PauseSending()
  585. }
  586. func (m *member) Resume() {
  587. m.raftHandler.Resume()
  588. m.s.ResumeSending()
  589. }
  590. // Close stops the member's etcdserver and closes its connections
  591. func (m *member) Close() {
  592. if m.grpcServer != nil {
  593. m.grpcServer.Stop()
  594. m.grpcServer = nil
  595. }
  596. m.s.Stop()
  597. for _, hs := range m.hss {
  598. hs.CloseClientConnections()
  599. hs.Close()
  600. }
  601. }
  602. // Stop stops the member, but the data dir of the member is preserved.
  603. func (m *member) Stop(t *testing.T) {
  604. plog.Printf("stopping %s (%s)", m.Name, m.grpcAddr)
  605. m.Close()
  606. m.hss = nil
  607. plog.Printf("stopped %s (%s)", m.Name, m.grpcAddr)
  608. }
  609. // StopNotify unblocks when a member stop completes
  610. func (m *member) StopNotify() <-chan struct{} {
  611. return m.s.StopNotify()
  612. }
  613. // Restart starts the member using the preserved data dir.
  614. func (m *member) Restart(t *testing.T) error {
  615. plog.Printf("restarting %s (%s)", m.Name, m.grpcAddr)
  616. newPeerListeners := make([]net.Listener, 0)
  617. for _, ln := range m.PeerListeners {
  618. newPeerListeners = append(newPeerListeners, newListenerWithAddr(t, ln.Addr().String()))
  619. }
  620. m.PeerListeners = newPeerListeners
  621. newClientListeners := make([]net.Listener, 0)
  622. for _, ln := range m.ClientListeners {
  623. newClientListeners = append(newClientListeners, newListenerWithAddr(t, ln.Addr().String()))
  624. }
  625. m.ClientListeners = newClientListeners
  626. if m.grpcListener != nil {
  627. if err := m.listenGRPC(); err != nil {
  628. t.Fatal(err)
  629. }
  630. }
  631. err := m.Launch()
  632. plog.Printf("restarted %s (%s)", m.Name, m.grpcAddr)
  633. return err
  634. }
  635. // Terminate stops the member and removes the data dir.
  636. func (m *member) Terminate(t *testing.T) {
  637. plog.Printf("terminating %s (%s)", m.Name, m.grpcAddr)
  638. m.Close()
  639. if err := os.RemoveAll(m.ServerConfig.DataDir); err != nil {
  640. t.Fatal(err)
  641. }
  642. plog.Printf("terminated %s (%s)", m.Name, m.grpcAddr)
  643. }
  644. func mustNewHTTPClient(t *testing.T, eps []string, tls *transport.TLSInfo) client.Client {
  645. cfgtls := transport.TLSInfo{}
  646. if tls != nil {
  647. cfgtls = *tls
  648. }
  649. cfg := client.Config{Transport: mustNewTransport(t, cfgtls), Endpoints: eps}
  650. c, err := client.New(cfg)
  651. if err != nil {
  652. t.Fatal(err)
  653. }
  654. return c
  655. }
  656. func mustNewTransport(t *testing.T, tlsInfo transport.TLSInfo) *http.Transport {
  657. // tick in integration test is short, so 1s dial timeout could play well.
  658. tr, err := transport.NewTimeoutTransport(tlsInfo, time.Second, rafthttp.ConnReadTimeout, rafthttp.ConnWriteTimeout)
  659. if err != nil {
  660. t.Fatal(err)
  661. }
  662. return tr
  663. }
  664. type SortableMemberSliceByPeerURLs []client.Member
  665. func (p SortableMemberSliceByPeerURLs) Len() int { return len(p) }
  666. func (p SortableMemberSliceByPeerURLs) Less(i, j int) bool {
  667. return p[i].PeerURLs[0] < p[j].PeerURLs[0]
  668. }
  669. func (p SortableMemberSliceByPeerURLs) Swap(i, j int) { p[i], p[j] = p[j], p[i] }
  670. type ClusterV3 struct {
  671. *cluster
  672. clients []*clientv3.Client
  673. }
  674. // NewClusterV3 returns a launched cluster with a grpc client connection
  675. // for each cluster member.
  676. func NewClusterV3(t *testing.T, cfg *ClusterConfig) *ClusterV3 {
  677. cfg.UseGRPC = true
  678. clus := &ClusterV3{cluster: NewClusterByConfig(t, cfg)}
  679. for _, m := range clus.Members {
  680. client, err := NewClientV3(m)
  681. if err != nil {
  682. t.Fatalf("cannot create client: %v", err)
  683. }
  684. clus.clients = append(clus.clients, client)
  685. }
  686. clus.Launch(t)
  687. return clus
  688. }
  689. func (c *ClusterV3) Terminate(t *testing.T) {
  690. for _, client := range c.clients {
  691. if err := client.Close(); err != nil {
  692. t.Error(err)
  693. }
  694. }
  695. c.cluster.Terminate(t)
  696. }
  697. func (c *ClusterV3) RandClient() *clientv3.Client {
  698. return c.clients[rand.Intn(len(c.clients))]
  699. }
  700. func (c *ClusterV3) Client(i int) *clientv3.Client {
  701. return c.clients[i]
  702. }
  703. type grpcAPI struct {
  704. // Cluster is the cluster API for the client's connection.
  705. Cluster pb.ClusterClient
  706. // KV is the keyvalue API for the client's connection.
  707. KV pb.KVClient
  708. // Lease is the lease API for the client's connection.
  709. Lease pb.LeaseClient
  710. // Watch is the watch API for the client's connection.
  711. Watch pb.WatchClient
  712. // Maintenance is the maintenance API for the client's connection.
  713. Maintenance pb.MaintenanceClient
  714. }
  715. func toGRPC(c *clientv3.Client) grpcAPI {
  716. return grpcAPI{
  717. pb.NewClusterClient(c.ActiveConnection()),
  718. pb.NewKVClient(c.ActiveConnection()),
  719. pb.NewLeaseClient(c.ActiveConnection()),
  720. pb.NewWatchClient(c.ActiveConnection()),
  721. pb.NewMaintenanceClient(c.ActiveConnection()),
  722. }
  723. }