cluster.go 20 KB

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