etcd.go 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744
  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 embed
  15. import (
  16. "context"
  17. "crypto/tls"
  18. "fmt"
  19. "io/ioutil"
  20. defaultLog "log"
  21. "net"
  22. "net/http"
  23. "net/url"
  24. "sort"
  25. "strconv"
  26. "sync"
  27. "time"
  28. "go.uber.org/zap/zapcore"
  29. "github.com/coreos/etcd/etcdserver"
  30. "github.com/coreos/etcd/etcdserver/api/etcdhttp"
  31. "github.com/coreos/etcd/etcdserver/api/v2http"
  32. "github.com/coreos/etcd/etcdserver/api/v2v3"
  33. "github.com/coreos/etcd/etcdserver/api/v3client"
  34. "github.com/coreos/etcd/etcdserver/api/v3rpc"
  35. "github.com/coreos/etcd/pkg/debugutil"
  36. runtimeutil "github.com/coreos/etcd/pkg/runtime"
  37. "github.com/coreos/etcd/pkg/transport"
  38. "github.com/coreos/etcd/pkg/types"
  39. "github.com/coreos/etcd/rafthttp"
  40. "github.com/coreos/pkg/capnslog"
  41. "github.com/grpc-ecosystem/go-grpc-prometheus"
  42. "github.com/soheilhy/cmux"
  43. "go.uber.org/zap"
  44. "google.golang.org/grpc"
  45. "google.golang.org/grpc/keepalive"
  46. )
  47. var plog = capnslog.NewPackageLogger("github.com/coreos/etcd", "embed")
  48. const (
  49. // internal fd usage includes disk usage and transport usage.
  50. // To read/write snapshot, snap pkg needs 1. In normal case, wal pkg needs
  51. // at most 2 to read/lock/write WALs. One case that it needs to 2 is to
  52. // read all logs after some snapshot index, which locates at the end of
  53. // the second last and the head of the last. For purging, it needs to read
  54. // directory, so it needs 1. For fd monitor, it needs 1.
  55. // For transport, rafthttp builds two long-polling connections and at most
  56. // four temporary connections with each member. There are at most 9 members
  57. // in a cluster, so it should reserve 96.
  58. // For the safety, we set the total reserved number to 150.
  59. reservedInternalFDNum = 150
  60. )
  61. // Etcd contains a running etcd server and its listeners.
  62. type Etcd struct {
  63. Peers []*peerListener
  64. Clients []net.Listener
  65. // a map of contexts for the servers that serves client requests.
  66. sctxs map[string]*serveCtx
  67. metricsListeners []net.Listener
  68. Server *etcdserver.EtcdServer
  69. cfg Config
  70. stopc chan struct{}
  71. errc chan error
  72. closeOnce sync.Once
  73. }
  74. type peerListener struct {
  75. net.Listener
  76. serve func() error
  77. close func(context.Context) error
  78. }
  79. // StartEtcd launches the etcd server and HTTP handlers for client/server communication.
  80. // The returned Etcd.Server is not guaranteed to have joined the cluster. Wait
  81. // on the Etcd.Server.ReadyNotify() channel to know when it completes and is ready for use.
  82. func StartEtcd(inCfg *Config) (e *Etcd, err error) {
  83. if err = inCfg.Validate(); err != nil {
  84. return nil, err
  85. }
  86. serving := false
  87. e = &Etcd{cfg: *inCfg, stopc: make(chan struct{})}
  88. cfg := &e.cfg
  89. defer func() {
  90. if e == nil || err == nil {
  91. return
  92. }
  93. if !serving {
  94. // errored before starting gRPC server for serveCtx.serversC
  95. for _, sctx := range e.sctxs {
  96. close(sctx.serversC)
  97. }
  98. }
  99. e.Close()
  100. e = nil
  101. }()
  102. if e.cfg.logger != nil {
  103. e.cfg.logger.Info(
  104. "configuring peer listeners",
  105. zap.Strings("listen-peer-urls", e.cfg.getLPURLs()),
  106. )
  107. }
  108. if e.Peers, err = configurePeerListeners(cfg); err != nil {
  109. return e, err
  110. }
  111. if e.cfg.logger != nil {
  112. e.cfg.logger.Info(
  113. "configuring client listeners",
  114. zap.Strings("listen-client-urls", e.cfg.getLCURLs()),
  115. )
  116. }
  117. if e.sctxs, err = configureClientListeners(cfg); err != nil {
  118. return e, err
  119. }
  120. for _, sctx := range e.sctxs {
  121. e.Clients = append(e.Clients, sctx.l)
  122. }
  123. var (
  124. urlsmap types.URLsMap
  125. token string
  126. )
  127. memberInitialized := true
  128. if !isMemberInitialized(cfg) {
  129. memberInitialized = false
  130. urlsmap, token, err = cfg.PeerURLsMapAndToken("etcd")
  131. if err != nil {
  132. return e, fmt.Errorf("error setting up initial cluster: %v", err)
  133. }
  134. }
  135. // AutoCompactionRetention defaults to "0" if not set.
  136. if len(cfg.AutoCompactionRetention) == 0 {
  137. cfg.AutoCompactionRetention = "0"
  138. }
  139. autoCompactionRetention, err := parseCompactionRetention(cfg.AutoCompactionMode, cfg.AutoCompactionRetention)
  140. if err != nil {
  141. return e, err
  142. }
  143. srvcfg := etcdserver.ServerConfig{
  144. Name: cfg.Name,
  145. ClientURLs: cfg.ACUrls,
  146. PeerURLs: cfg.APUrls,
  147. DataDir: cfg.Dir,
  148. DedicatedWALDir: cfg.WalDir,
  149. SnapCount: cfg.SnapCount,
  150. MaxSnapFiles: cfg.MaxSnapFiles,
  151. MaxWALFiles: cfg.MaxWalFiles,
  152. InitialPeerURLsMap: urlsmap,
  153. InitialClusterToken: token,
  154. DiscoveryURL: cfg.Durl,
  155. DiscoveryProxy: cfg.Dproxy,
  156. NewCluster: cfg.IsNewCluster(),
  157. PeerTLSInfo: cfg.PeerTLSInfo,
  158. TickMs: cfg.TickMs,
  159. ElectionTicks: cfg.ElectionTicks(),
  160. InitialElectionTickAdvance: cfg.InitialElectionTickAdvance,
  161. AutoCompactionRetention: autoCompactionRetention,
  162. AutoCompactionMode: cfg.AutoCompactionMode,
  163. QuotaBackendBytes: cfg.QuotaBackendBytes,
  164. MaxTxnOps: cfg.MaxTxnOps,
  165. MaxRequestBytes: cfg.MaxRequestBytes,
  166. StrictReconfigCheck: cfg.StrictReconfigCheck,
  167. ClientCertAuthEnabled: cfg.ClientTLSInfo.ClientCertAuth,
  168. AuthToken: cfg.AuthToken,
  169. CORS: cfg.CORS,
  170. HostWhitelist: cfg.HostWhitelist,
  171. InitialCorruptCheck: cfg.ExperimentalInitialCorruptCheck,
  172. CorruptCheckTime: cfg.ExperimentalCorruptCheckTime,
  173. PreVote: cfg.PreVote,
  174. Logger: cfg.logger,
  175. LoggerConfig: cfg.loggerConfig,
  176. LoggerCore: cfg.loggerCore,
  177. LoggerWriteSyncer: cfg.loggerWriteSyncer,
  178. Debug: cfg.Debug,
  179. ForceNewCluster: cfg.ForceNewCluster,
  180. }
  181. if e.Server, err = etcdserver.NewServer(srvcfg); err != nil {
  182. return e, err
  183. }
  184. if len(e.cfg.CORS) > 0 {
  185. ss := make([]string, 0, len(e.cfg.CORS))
  186. for v := range e.cfg.CORS {
  187. ss = append(ss, v)
  188. }
  189. sort.Strings(ss)
  190. if e.cfg.logger != nil {
  191. e.cfg.logger.Info("configured CORS", zap.Strings("cors", ss))
  192. } else {
  193. plog.Infof("%s starting with cors %q", e.Server.ID(), ss)
  194. }
  195. }
  196. if len(e.cfg.HostWhitelist) > 0 {
  197. ss := make([]string, 0, len(e.cfg.HostWhitelist))
  198. for v := range e.cfg.HostWhitelist {
  199. ss = append(ss, v)
  200. }
  201. sort.Strings(ss)
  202. if e.cfg.logger != nil {
  203. e.cfg.logger.Info("configured host whitelist", zap.Strings("hosts", ss))
  204. } else {
  205. plog.Infof("%s starting with host whitelist %q", e.Server.ID(), ss)
  206. }
  207. }
  208. // buffer channel so goroutines on closed connections won't wait forever
  209. e.errc = make(chan error, len(e.Peers)+len(e.Clients)+2*len(e.sctxs))
  210. // newly started member ("memberInitialized==false")
  211. // does not need corruption check
  212. if memberInitialized {
  213. if err = e.Server.CheckInitialHashKV(); err != nil {
  214. // set "EtcdServer" to nil, so that it does not block on "EtcdServer.Close()"
  215. // (nothing to close since rafthttp transports have not been started)
  216. e.Server = nil
  217. return e, err
  218. }
  219. }
  220. e.Server.Start()
  221. if err = e.servePeers(); err != nil {
  222. return e, err
  223. }
  224. if err = e.serveClients(); err != nil {
  225. return e, err
  226. }
  227. if err = e.serveMetrics(); err != nil {
  228. return e, err
  229. }
  230. if e.cfg.logger != nil {
  231. e.cfg.logger.Info(
  232. "now serving peer/client/metrics",
  233. zap.String("local-member-id", e.Server.ID().String()),
  234. zap.Strings("initial-advertise-peer-urls", e.cfg.getAPURLs()),
  235. zap.Strings("listen-peer-urls", e.cfg.getLPURLs()),
  236. zap.Strings("advertise-client-urls", e.cfg.getACURLs()),
  237. zap.Strings("listen-client-urls", e.cfg.getLCURLs()),
  238. zap.Strings("listen-metrics-urls", e.cfg.getMetricsURLs()),
  239. )
  240. }
  241. serving = true
  242. return e, nil
  243. }
  244. // Config returns the current configuration.
  245. func (e *Etcd) Config() Config {
  246. return e.cfg
  247. }
  248. // Close gracefully shuts down all servers/listeners.
  249. // Client requests will be terminated with request timeout.
  250. // After timeout, enforce remaning requests be closed immediately.
  251. func (e *Etcd) Close() {
  252. fields := []zapcore.Field{
  253. zap.String("name", e.cfg.Name),
  254. zap.String("data-dir", e.cfg.Dir),
  255. zap.Strings("advertise-peer-urls", e.cfg.getAPURLs()),
  256. zap.Strings("advertise-client-urls", e.cfg.getACURLs()),
  257. }
  258. lg := e.GetLogger()
  259. if lg != nil {
  260. lg.Info("closing etcd server", fields...)
  261. }
  262. defer func() {
  263. if lg != nil {
  264. lg.Info("closed etcd server", fields...)
  265. lg.Sync()
  266. }
  267. }()
  268. e.closeOnce.Do(func() { close(e.stopc) })
  269. // close client requests with request timeout
  270. timeout := 2 * time.Second
  271. if e.Server != nil {
  272. timeout = e.Server.Cfg.ReqTimeout()
  273. }
  274. for _, sctx := range e.sctxs {
  275. for ss := range sctx.serversC {
  276. ctx, cancel := context.WithTimeout(context.Background(), timeout)
  277. stopServers(ctx, ss)
  278. cancel()
  279. }
  280. }
  281. for _, sctx := range e.sctxs {
  282. sctx.cancel()
  283. }
  284. for i := range e.Clients {
  285. if e.Clients[i] != nil {
  286. e.Clients[i].Close()
  287. }
  288. }
  289. for i := range e.metricsListeners {
  290. e.metricsListeners[i].Close()
  291. }
  292. // close rafthttp transports
  293. if e.Server != nil {
  294. e.Server.Stop()
  295. }
  296. // close all idle connections in peer handler (wait up to 1-second)
  297. for i := range e.Peers {
  298. if e.Peers[i] != nil && e.Peers[i].close != nil {
  299. ctx, cancel := context.WithTimeout(context.Background(), time.Second)
  300. e.Peers[i].close(ctx)
  301. cancel()
  302. }
  303. }
  304. }
  305. func stopServers(ctx context.Context, ss *servers) {
  306. shutdownNow := func() {
  307. // first, close the http.Server
  308. ss.http.Shutdown(ctx)
  309. // then close grpc.Server; cancels all active RPCs
  310. ss.grpc.Stop()
  311. }
  312. // do not grpc.Server.GracefulStop with TLS enabled etcd server
  313. // See https://github.com/grpc/grpc-go/issues/1384#issuecomment-317124531
  314. // and https://github.com/coreos/etcd/issues/8916
  315. if ss.secure {
  316. shutdownNow()
  317. return
  318. }
  319. ch := make(chan struct{})
  320. go func() {
  321. defer close(ch)
  322. // close listeners to stop accepting new connections,
  323. // will block on any existing transports
  324. ss.grpc.GracefulStop()
  325. }()
  326. // wait until all pending RPCs are finished
  327. select {
  328. case <-ch:
  329. case <-ctx.Done():
  330. // took too long, manually close open transports
  331. // e.g. watch streams
  332. shutdownNow()
  333. // concurrent GracefulStop should be interrupted
  334. <-ch
  335. }
  336. }
  337. func (e *Etcd) Err() <-chan error { return e.errc }
  338. func configurePeerListeners(cfg *Config) (peers []*peerListener, err error) {
  339. if err = cfg.PeerSelfCert(); err != nil {
  340. if cfg.logger != nil {
  341. cfg.logger.Fatal("failed to get peer self-signed certs", zap.Error(err))
  342. } else {
  343. plog.Fatalf("could not get certs (%v)", err)
  344. }
  345. }
  346. if !cfg.PeerTLSInfo.Empty() {
  347. if cfg.logger != nil {
  348. cfg.logger.Info("starting with peer TLS", zap.String("tls-info", fmt.Sprintf("%+v", cfg.PeerTLSInfo)))
  349. } else {
  350. plog.Infof("peerTLS: %s", cfg.PeerTLSInfo)
  351. }
  352. }
  353. peers = make([]*peerListener, len(cfg.LPUrls))
  354. defer func() {
  355. if err == nil {
  356. return
  357. }
  358. for i := range peers {
  359. if peers[i] != nil && peers[i].close != nil {
  360. if cfg.logger != nil {
  361. cfg.logger.Warn(
  362. "closing peer listener",
  363. zap.String("address", cfg.LPUrls[i].String()),
  364. zap.Error(err),
  365. )
  366. } else {
  367. plog.Info("stopping listening for peers on ", cfg.LPUrls[i].String())
  368. }
  369. ctx, cancel := context.WithTimeout(context.Background(), time.Second)
  370. peers[i].close(ctx)
  371. cancel()
  372. }
  373. }
  374. }()
  375. for i, u := range cfg.LPUrls {
  376. if u.Scheme == "http" {
  377. if !cfg.PeerTLSInfo.Empty() {
  378. if cfg.logger != nil {
  379. cfg.logger.Warn("scheme is HTTP while key and cert files are present; ignoring key and cert files", zap.String("peer-url", u.String()))
  380. } else {
  381. plog.Warningf("The scheme of peer url %s is HTTP while peer key/cert files are presented. Ignored peer key/cert files.", u.String())
  382. }
  383. }
  384. if cfg.PeerTLSInfo.ClientCertAuth {
  385. if cfg.logger != nil {
  386. cfg.logger.Warn("scheme is HTTP while --peer-client-cert-auth is enabled; ignoring client cert auth for this URL", zap.String("peer-url", u.String()))
  387. } else {
  388. plog.Warningf("The scheme of peer url %s is HTTP while client cert auth (--peer-client-cert-auth) is enabled. Ignored client cert auth for this url.", u.String())
  389. }
  390. }
  391. }
  392. peers[i] = &peerListener{close: func(context.Context) error { return nil }}
  393. peers[i].Listener, err = rafthttp.NewListener(u, &cfg.PeerTLSInfo)
  394. if err != nil {
  395. return nil, err
  396. }
  397. // once serve, overwrite with 'http.Server.Shutdown'
  398. peers[i].close = func(context.Context) error {
  399. return peers[i].Listener.Close()
  400. }
  401. }
  402. return peers, nil
  403. }
  404. // configure peer handlers after rafthttp.Transport started
  405. func (e *Etcd) servePeers() (err error) {
  406. ph := etcdhttp.NewPeerHandler(e.Server)
  407. var peerTLScfg *tls.Config
  408. if !e.cfg.PeerTLSInfo.Empty() {
  409. if peerTLScfg, err = e.cfg.PeerTLSInfo.ServerConfig(); err != nil {
  410. return err
  411. }
  412. }
  413. for _, p := range e.Peers {
  414. u := p.Listener.Addr().String()
  415. gs := v3rpc.Server(e.Server, peerTLScfg)
  416. m := cmux.New(p.Listener)
  417. go gs.Serve(m.Match(cmux.HTTP2()))
  418. srv := &http.Server{
  419. Handler: grpcHandlerFunc(gs, ph),
  420. ReadTimeout: 5 * time.Minute,
  421. ErrorLog: defaultLog.New(ioutil.Discard, "", 0), // do not log user error
  422. }
  423. go srv.Serve(m.Match(cmux.Any()))
  424. p.serve = func() error { return m.Serve() }
  425. p.close = func(ctx context.Context) error {
  426. // gracefully shutdown http.Server
  427. // close open listeners, idle connections
  428. // until context cancel or time-out
  429. if e.cfg.logger != nil {
  430. e.cfg.logger.Info(
  431. "stopping serving peer traffic",
  432. zap.String("address", u),
  433. )
  434. }
  435. stopServers(ctx, &servers{secure: peerTLScfg != nil, grpc: gs, http: srv})
  436. if e.cfg.logger != nil {
  437. e.cfg.logger.Info(
  438. "stopped serving peer traffic",
  439. zap.String("address", u),
  440. )
  441. }
  442. return nil
  443. }
  444. }
  445. // start peer servers in a goroutine
  446. for _, pl := range e.Peers {
  447. go func(l *peerListener) {
  448. u := l.Addr().String()
  449. if e.cfg.logger != nil {
  450. e.cfg.logger.Info(
  451. "serving peer traffic",
  452. zap.String("address", u),
  453. )
  454. } else {
  455. plog.Info("listening for peers on ", u)
  456. }
  457. e.errHandler(l.serve())
  458. }(pl)
  459. }
  460. return nil
  461. }
  462. func configureClientListeners(cfg *Config) (sctxs map[string]*serveCtx, err error) {
  463. if err = cfg.ClientSelfCert(); err != nil {
  464. if cfg.logger != nil {
  465. cfg.logger.Fatal("failed to get client self-signed certs", zap.Error(err))
  466. } else {
  467. plog.Fatalf("could not get certs (%v)", err)
  468. }
  469. }
  470. if cfg.EnablePprof {
  471. if cfg.logger != nil {
  472. cfg.logger.Info("pprof is enabled", zap.String("path", debugutil.HTTPPrefixPProf))
  473. } else {
  474. plog.Infof("pprof is enabled under %s", debugutil.HTTPPrefixPProf)
  475. }
  476. }
  477. sctxs = make(map[string]*serveCtx)
  478. for _, u := range cfg.LCUrls {
  479. sctx := newServeCtx(cfg.logger)
  480. if u.Scheme == "http" || u.Scheme == "unix" {
  481. if !cfg.ClientTLSInfo.Empty() {
  482. if cfg.logger != nil {
  483. cfg.logger.Warn("scheme is HTTP while key and cert files are present; ignoring key and cert files", zap.String("client-url", u.String()))
  484. } else {
  485. plog.Warningf("The scheme of client url %s is HTTP while peer key/cert files are presented. Ignored key/cert files.", u.String())
  486. }
  487. }
  488. if cfg.ClientTLSInfo.ClientCertAuth {
  489. if cfg.logger != nil {
  490. cfg.logger.Warn("scheme is HTTP while --client-cert-auth is enabled; ignoring client cert auth for this URL", zap.String("client-url", u.String()))
  491. } else {
  492. plog.Warningf("The scheme of client url %s is HTTP while client cert auth (--client-cert-auth) is enabled. Ignored client cert auth for this url.", u.String())
  493. }
  494. }
  495. }
  496. if (u.Scheme == "https" || u.Scheme == "unixs") && cfg.ClientTLSInfo.Empty() {
  497. return nil, fmt.Errorf("TLS key/cert (--cert-file, --key-file) must be provided for client url %s with HTTPs scheme", u.String())
  498. }
  499. network := "tcp"
  500. addr := u.Host
  501. if u.Scheme == "unix" || u.Scheme == "unixs" {
  502. network = "unix"
  503. addr = u.Host + u.Path
  504. }
  505. sctx.network = network
  506. sctx.secure = u.Scheme == "https" || u.Scheme == "unixs"
  507. sctx.insecure = !sctx.secure
  508. if oldctx := sctxs[addr]; oldctx != nil {
  509. oldctx.secure = oldctx.secure || sctx.secure
  510. oldctx.insecure = oldctx.insecure || sctx.insecure
  511. continue
  512. }
  513. if sctx.l, err = net.Listen(network, addr); err != nil {
  514. return nil, err
  515. }
  516. // net.Listener will rewrite ipv4 0.0.0.0 to ipv6 [::], breaking
  517. // hosts that disable ipv6. So, use the address given by the user.
  518. sctx.addr = addr
  519. if fdLimit, fderr := runtimeutil.FDLimit(); fderr == nil {
  520. if fdLimit <= reservedInternalFDNum {
  521. if cfg.logger != nil {
  522. cfg.logger.Fatal(
  523. "file descriptor limit of etcd process is too low; please set higher",
  524. zap.Uint64("limit", fdLimit),
  525. zap.Int("recommended-limit", reservedInternalFDNum),
  526. )
  527. } else {
  528. plog.Fatalf("file descriptor limit[%d] of etcd process is too low, and should be set higher than %d to ensure internal usage", fdLimit, reservedInternalFDNum)
  529. }
  530. }
  531. sctx.l = transport.LimitListener(sctx.l, int(fdLimit-reservedInternalFDNum))
  532. }
  533. if network == "tcp" {
  534. if sctx.l, err = transport.NewKeepAliveListener(sctx.l, network, nil); err != nil {
  535. return nil, err
  536. }
  537. }
  538. defer func() {
  539. if err == nil {
  540. return
  541. }
  542. sctx.l.Close()
  543. if cfg.logger != nil {
  544. cfg.logger.Warn(
  545. "closing peer listener",
  546. zap.String("address", u.Host),
  547. zap.Error(err),
  548. )
  549. } else {
  550. plog.Info("stopping listening for client requests on ", u.Host)
  551. }
  552. }()
  553. for k := range cfg.UserHandlers {
  554. sctx.userHandlers[k] = cfg.UserHandlers[k]
  555. }
  556. sctx.serviceRegister = cfg.ServiceRegister
  557. if cfg.EnablePprof || cfg.Debug {
  558. sctx.registerPprof()
  559. }
  560. if cfg.Debug {
  561. sctx.registerTrace()
  562. }
  563. sctxs[addr] = sctx
  564. }
  565. return sctxs, nil
  566. }
  567. func (e *Etcd) serveClients() (err error) {
  568. if !e.cfg.ClientTLSInfo.Empty() {
  569. if e.cfg.logger != nil {
  570. e.cfg.logger.Info(
  571. "starting with client TLS",
  572. zap.String("tls-info", fmt.Sprintf("%+v", e.cfg.ClientTLSInfo)),
  573. )
  574. } else {
  575. plog.Infof("ClientTLS: %s", e.cfg.ClientTLSInfo)
  576. }
  577. }
  578. // Start a client server goroutine for each listen address
  579. var h http.Handler
  580. if e.Config().EnableV2 {
  581. if len(e.Config().ExperimentalEnableV2V3) > 0 {
  582. srv := v2v3.NewServer(e.cfg.logger, v3client.New(e.Server), e.cfg.ExperimentalEnableV2V3)
  583. h = v2http.NewClientHandler(srv, e.Server.Cfg.ReqTimeout())
  584. } else {
  585. h = v2http.NewClientHandler(e.Server, e.Server.Cfg.ReqTimeout())
  586. }
  587. } else {
  588. mux := http.NewServeMux()
  589. etcdhttp.HandleBasic(mux, e.Server)
  590. h = mux
  591. }
  592. gopts := []grpc.ServerOption{}
  593. if e.cfg.GRPCKeepAliveMinTime > time.Duration(0) {
  594. gopts = append(gopts, grpc.KeepaliveEnforcementPolicy(keepalive.EnforcementPolicy{
  595. MinTime: e.cfg.GRPCKeepAliveMinTime,
  596. PermitWithoutStream: false,
  597. }))
  598. }
  599. if e.cfg.GRPCKeepAliveInterval > time.Duration(0) &&
  600. e.cfg.GRPCKeepAliveTimeout > time.Duration(0) {
  601. gopts = append(gopts, grpc.KeepaliveParams(keepalive.ServerParameters{
  602. Time: e.cfg.GRPCKeepAliveInterval,
  603. Timeout: e.cfg.GRPCKeepAliveTimeout,
  604. }))
  605. }
  606. // start client servers in a goroutine
  607. for _, sctx := range e.sctxs {
  608. go func(s *serveCtx) {
  609. e.errHandler(s.serve(e.Server, &e.cfg.ClientTLSInfo, h, e.errHandler, gopts...))
  610. }(sctx)
  611. }
  612. return nil
  613. }
  614. func (e *Etcd) serveMetrics() (err error) {
  615. if e.cfg.Metrics == "extensive" {
  616. grpc_prometheus.EnableHandlingTimeHistogram()
  617. }
  618. if len(e.cfg.ListenMetricsUrls) > 0 {
  619. metricsMux := http.NewServeMux()
  620. etcdhttp.HandleMetricsHealth(metricsMux, e.Server)
  621. for _, murl := range e.cfg.ListenMetricsUrls {
  622. tlsInfo := &e.cfg.ClientTLSInfo
  623. if murl.Scheme == "http" {
  624. tlsInfo = nil
  625. }
  626. ml, err := transport.NewListener(murl.Host, murl.Scheme, tlsInfo)
  627. if err != nil {
  628. return err
  629. }
  630. e.metricsListeners = append(e.metricsListeners, ml)
  631. go func(u url.URL, ln net.Listener) {
  632. if e.cfg.logger != nil {
  633. e.cfg.logger.Info(
  634. "serving metrics",
  635. zap.String("address", u.String()),
  636. )
  637. } else {
  638. plog.Info("listening for metrics on ", u.String())
  639. }
  640. e.errHandler(http.Serve(ln, metricsMux))
  641. }(murl, ml)
  642. }
  643. }
  644. return nil
  645. }
  646. func (e *Etcd) errHandler(err error) {
  647. select {
  648. case <-e.stopc:
  649. return
  650. default:
  651. }
  652. select {
  653. case <-e.stopc:
  654. case e.errc <- err:
  655. }
  656. }
  657. // GetLogger returns the logger.
  658. func (e *Etcd) GetLogger() *zap.Logger {
  659. e.cfg.loggerMu.RLock()
  660. l := e.cfg.logger
  661. e.cfg.loggerMu.RUnlock()
  662. return l
  663. }
  664. func parseCompactionRetention(mode, retention string) (ret time.Duration, err error) {
  665. h, err := strconv.Atoi(retention)
  666. if err == nil {
  667. switch mode {
  668. case CompactorModeRevision:
  669. ret = time.Duration(int64(h))
  670. case CompactorModePeriodic:
  671. ret = time.Duration(int64(h)) * time.Hour
  672. }
  673. } else {
  674. // periodic compaction
  675. ret, err = time.ParseDuration(retention)
  676. if err != nil {
  677. return 0, fmt.Errorf("error parsing CompactionRetention: %v", err)
  678. }
  679. }
  680. return ret, nil
  681. }