cluster.go 20 KB

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