server.go 30 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015
  1. /*
  2. Copyright 2014 CoreOS, Inc.
  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. http://www.apache.org/licenses/LICENSE-2.0
  7. Unless required by applicable law or agreed to in writing, software
  8. distributed under the License is distributed on an "AS IS" BASIS,
  9. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  10. See the License for the specific language governing permissions and
  11. limitations under the License.
  12. */
  13. package etcdserver
  14. import (
  15. "encoding/json"
  16. "errors"
  17. "fmt"
  18. "io/ioutil"
  19. "log"
  20. "math/rand"
  21. "net/http"
  22. "os"
  23. "path"
  24. "regexp"
  25. "sort"
  26. "sync/atomic"
  27. "time"
  28. "github.com/coreos/etcd/Godeps/_workspace/src/golang.org/x/net/context"
  29. "github.com/coreos/etcd/discovery"
  30. "github.com/coreos/etcd/etcdserver/etcdhttp/httptypes"
  31. pb "github.com/coreos/etcd/etcdserver/etcdserverpb"
  32. "github.com/coreos/etcd/etcdserver/stats"
  33. "github.com/coreos/etcd/pkg/fileutil"
  34. "github.com/coreos/etcd/pkg/pbutil"
  35. "github.com/coreos/etcd/pkg/types"
  36. "github.com/coreos/etcd/pkg/wait"
  37. "github.com/coreos/etcd/raft"
  38. "github.com/coreos/etcd/raft/raftpb"
  39. "github.com/coreos/etcd/rafthttp"
  40. "github.com/coreos/etcd/snap"
  41. "github.com/coreos/etcd/store"
  42. "github.com/coreos/etcd/wal"
  43. )
  44. const (
  45. // owner can make/remove files inside the directory
  46. privateDirMode = 0700
  47. defaultSyncTimeout = time.Second
  48. DefaultSnapCount = 10000
  49. // TODO: calculate based on heartbeat interval
  50. defaultPublishRetryInterval = 5 * time.Second
  51. StoreAdminPrefix = "/0"
  52. StoreKeysPrefix = "/1"
  53. purgeFileInterval = 30 * time.Second
  54. )
  55. var (
  56. ErrUnknownMethod = errors.New("etcdserver: unknown method")
  57. ErrStopped = errors.New("etcdserver: server stopped")
  58. ErrIDRemoved = errors.New("etcdserver: ID removed")
  59. ErrIDExists = errors.New("etcdserver: ID exists")
  60. ErrIDNotFound = errors.New("etcdserver: ID not found")
  61. ErrPeerURLexists = errors.New("etcdserver: peerURL exists")
  62. ErrCanceled = errors.New("etcdserver: request cancelled")
  63. ErrTimeout = errors.New("etcdserver: request timed out")
  64. storeMembersPrefix = path.Join(StoreAdminPrefix, "members")
  65. storeRemovedMembersPrefix = path.Join(StoreAdminPrefix, "removed_members")
  66. storeMemberAttributeRegexp = regexp.MustCompile(path.Join(storeMembersPrefix, "[[:xdigit:]]{1,16}", attributesSuffix))
  67. )
  68. func init() {
  69. rand.Seed(time.Now().UnixNano())
  70. }
  71. type Response struct {
  72. Event *store.Event
  73. Watcher store.Watcher
  74. err error
  75. }
  76. type Server interface {
  77. // Start performs any initialization of the Server necessary for it to
  78. // begin serving requests. It must be called before Do or Process.
  79. // Start must be non-blocking; any long-running server functionality
  80. // should be implemented in goroutines.
  81. Start()
  82. // Stop terminates the Server and performs any necessary finalization.
  83. // Do and Process cannot be called after Stop has been invoked.
  84. Stop()
  85. // ID returns the ID of the Server.
  86. ID() types.ID
  87. // Do takes a request and attempts to fulfill it, returning a Response.
  88. Do(ctx context.Context, r pb.Request) (Response, error)
  89. // Process takes a raft message and applies it to the server's raft state
  90. // machine, respecting any timeout of the given context.
  91. Process(ctx context.Context, m raftpb.Message) error
  92. // AddMember attempts to add a member into the cluster. It will return
  93. // ErrIDRemoved if member ID is removed from the cluster, or return
  94. // ErrIDExists if member ID exists in the cluster.
  95. AddMember(ctx context.Context, memb Member) error
  96. // RemoveMember attempts to remove a member from the cluster. It will
  97. // return ErrIDRemoved if member ID is removed from the cluster, or return
  98. // ErrIDNotFound if member ID is not in the cluster.
  99. RemoveMember(ctx context.Context, id uint64) error
  100. // UpdateMember attempts to update a existing member in the cluster. It will
  101. // return ErrIDNotFound if the member ID does not exist.
  102. UpdateMember(ctx context.Context, updateMemb Member) error
  103. }
  104. type RaftTimer interface {
  105. Index() uint64
  106. Term() uint64
  107. }
  108. // EtcdServer is the production implementation of the Server interface
  109. type EtcdServer struct {
  110. cfg *ServerConfig
  111. w wait.Wait
  112. done chan struct{}
  113. stop chan struct{}
  114. id types.ID
  115. attributes Attributes
  116. Cluster *Cluster
  117. node raft.Node
  118. raftStorage *raft.MemoryStorage
  119. storage Storage
  120. store store.Store
  121. stats *stats.ServerStats
  122. lstats *stats.LeaderStats
  123. // transport specifies the transport to send and receive msgs to members.
  124. // Sending messages MUST NOT block. It is okay to drop messages, since
  125. // clients should timeout and reissue their messages.
  126. // If transport is nil, server will panic.
  127. transport rafthttp.Transporter
  128. Ticker <-chan time.Time
  129. SyncTicker <-chan time.Time
  130. snapCount uint64 // number of entries to trigger a snapshot
  131. // Cache of the latest raft index and raft term the server has seen
  132. raftIndex uint64
  133. raftTerm uint64
  134. raftLead uint64
  135. }
  136. // NewServer creates a new EtcdServer from the supplied configuration. The
  137. // configuration is considered static for the lifetime of the EtcdServer.
  138. func NewServer(cfg *ServerConfig) (*EtcdServer, error) {
  139. st := store.New()
  140. var w *wal.WAL
  141. var n raft.Node
  142. var s *raft.MemoryStorage
  143. var id types.ID
  144. walVersion, err := wal.DetectVersion(cfg.DataDir)
  145. if err != nil {
  146. return nil, err
  147. }
  148. if walVersion == wal.WALUnknown {
  149. return nil, fmt.Errorf("unknown wal version in data dir %s", cfg.DataDir)
  150. }
  151. haveWAL := walVersion != wal.WALNotExist
  152. ss := snap.New(cfg.SnapDir())
  153. switch {
  154. case !haveWAL && !cfg.NewCluster:
  155. us := getOtherPeerURLs(cfg.Cluster, cfg.Name)
  156. existingCluster, err := GetClusterFromPeers(us)
  157. if err != nil {
  158. return nil, fmt.Errorf("cannot fetch cluster info from peer urls: %v", err)
  159. }
  160. if err := ValidateClusterAndAssignIDs(cfg.Cluster, existingCluster); err != nil {
  161. return nil, fmt.Errorf("error validating peerURLs %s: %v", existingCluster, err)
  162. }
  163. cfg.Cluster.SetID(existingCluster.id)
  164. cfg.Cluster.SetStore(st)
  165. cfg.Print()
  166. id, n, s, w = startNode(cfg, nil)
  167. case !haveWAL && cfg.NewCluster:
  168. if err := cfg.VerifyBootstrapConfig(); err != nil {
  169. return nil, err
  170. }
  171. m := cfg.Cluster.MemberByName(cfg.Name)
  172. if isBootstrapped(cfg.Cluster, cfg.Name) {
  173. return nil, fmt.Errorf("member %s has already been bootstrapped", m.ID)
  174. }
  175. if cfg.ShouldDiscover() {
  176. s, err := discovery.JoinCluster(cfg.DiscoveryURL, cfg.DiscoveryProxy, m.ID, cfg.Cluster.String())
  177. if err != nil {
  178. return nil, err
  179. }
  180. if cfg.Cluster, err = NewClusterFromString(cfg.Cluster.token, s); err != nil {
  181. return nil, err
  182. }
  183. }
  184. cfg.Cluster.SetStore(st)
  185. cfg.PrintWithInitial()
  186. id, n, s, w = startNode(cfg, cfg.Cluster.MemberIDs())
  187. case haveWAL:
  188. if walVersion != wal.WALv0_5 {
  189. if err := upgradeWAL(cfg, walVersion); err != nil {
  190. return nil, err
  191. }
  192. }
  193. if cfg.ShouldDiscover() {
  194. log.Printf("etcdserver: discovery token ignored since a cluster has already been initialized. Valid log found at %q", cfg.WALDir())
  195. }
  196. var index uint64
  197. snapshot, err := ss.Load()
  198. if err != nil && err != snap.ErrNoSnapshot {
  199. return nil, err
  200. }
  201. if snapshot != nil {
  202. if err := st.Recovery(snapshot.Data); err != nil {
  203. log.Panicf("etcdserver: recovered store from snapshot error: %v", err)
  204. }
  205. log.Printf("etcdserver: recovered store from snapshot at index %d", snapshot.Metadata.Index)
  206. index = snapshot.Metadata.Index
  207. }
  208. cfg.Cluster = NewClusterFromStore(cfg.Cluster.token, st)
  209. cfg.Print()
  210. if snapshot != nil {
  211. log.Printf("etcdserver: loaded cluster information from store: %s", cfg.Cluster)
  212. }
  213. if !cfg.ForceNewCluster {
  214. id, n, s, w = restartNode(cfg, index+1, snapshot)
  215. } else {
  216. id, n, s, w = restartAsStandaloneNode(cfg, index+1, snapshot)
  217. }
  218. default:
  219. return nil, fmt.Errorf("unsupported bootstrap config")
  220. }
  221. sstats := &stats.ServerStats{
  222. Name: cfg.Name,
  223. ID: id.String(),
  224. }
  225. lstats := stats.NewLeaderStats(id.String())
  226. srv := &EtcdServer{
  227. cfg: cfg,
  228. store: st,
  229. node: n,
  230. raftStorage: s,
  231. id: id,
  232. attributes: Attributes{Name: cfg.Name, ClientURLs: cfg.ClientURLs.StringSlice()},
  233. Cluster: cfg.Cluster,
  234. storage: NewStorage(w, ss),
  235. stats: sstats,
  236. lstats: lstats,
  237. Ticker: time.Tick(100 * time.Millisecond),
  238. SyncTicker: time.Tick(500 * time.Millisecond),
  239. snapCount: cfg.SnapCount,
  240. }
  241. tr := &rafthttp.Transport{
  242. RoundTripper: cfg.Transport,
  243. ID: id,
  244. ClusterID: cfg.Cluster.ID(),
  245. Processor: srv,
  246. ServerStats: sstats,
  247. LeaderStats: lstats,
  248. }
  249. tr.Start()
  250. // add all the remote members into sendhub
  251. for _, m := range cfg.Cluster.Members() {
  252. if m.Name != cfg.Name {
  253. tr.AddPeer(m.ID, m.PeerURLs)
  254. }
  255. }
  256. srv.transport = tr
  257. return srv, nil
  258. }
  259. // Start prepares and starts server in a new goroutine. It is no longer safe to
  260. // modify a server's fields after it has been sent to Start.
  261. // It also starts a goroutine to publish its server information.
  262. func (s *EtcdServer) Start() {
  263. s.start()
  264. go s.publish(defaultPublishRetryInterval)
  265. go s.purgeFile()
  266. }
  267. // start prepares and starts server in a new goroutine. It is no longer safe to
  268. // modify a server's fields after it has been sent to Start.
  269. // This function is just used for testing.
  270. func (s *EtcdServer) start() {
  271. if s.snapCount == 0 {
  272. log.Printf("etcdserver: set snapshot count to default %d", DefaultSnapCount)
  273. s.snapCount = DefaultSnapCount
  274. }
  275. s.w = wait.New()
  276. s.done = make(chan struct{})
  277. s.stop = make(chan struct{})
  278. s.stats.Initialize()
  279. // TODO: if this is an empty log, writes all peer infos
  280. // into the first entry
  281. go s.run()
  282. }
  283. func (s *EtcdServer) purgeFile() {
  284. var serrc, werrc <-chan error
  285. if s.cfg.MaxSnapFiles > 0 {
  286. serrc = fileutil.PurgeFile(s.cfg.SnapDir(), "snap", s.cfg.MaxSnapFiles, purgeFileInterval, s.done)
  287. }
  288. if s.cfg.MaxWALFiles > 0 {
  289. werrc = fileutil.PurgeFile(s.cfg.WALDir(), "wal", s.cfg.MaxWALFiles, purgeFileInterval, s.done)
  290. }
  291. select {
  292. case e := <-werrc:
  293. log.Fatalf("etcdserver: failed to purge wal file %v", e)
  294. case e := <-serrc:
  295. log.Fatalf("etcdserver: failed to purge snap file %v", e)
  296. case <-s.done:
  297. return
  298. }
  299. }
  300. func (s *EtcdServer) ID() types.ID { return s.id }
  301. func (s *EtcdServer) RaftHandler() http.Handler { return s.transport.Handler() }
  302. func (s *EtcdServer) Process(ctx context.Context, m raftpb.Message) error {
  303. if s.Cluster.IsIDRemoved(types.ID(m.From)) {
  304. log.Printf("etcdserver: reject message from removed member %s", types.ID(m.From).String())
  305. return httptypes.NewHTTPError(http.StatusForbidden, "cannot process message from removed member")
  306. }
  307. if m.Type == raftpb.MsgApp {
  308. s.stats.RecvAppendReq(types.ID(m.From).String(), m.Size())
  309. }
  310. return s.node.Step(ctx, m)
  311. }
  312. func (s *EtcdServer) run() {
  313. var syncC <-chan time.Time
  314. var shouldstop bool
  315. shouldstopC := s.transport.ShouldStopNotify()
  316. // load initial state from raft storage
  317. snap, err := s.raftStorage.Snapshot()
  318. if err != nil {
  319. log.Panicf("etcdserver: get snapshot from raft storage error: %v", err)
  320. }
  321. // snapi indicates the index of the last submitted snapshot request
  322. snapi := snap.Metadata.Index
  323. appliedi := snap.Metadata.Index
  324. confState := snap.Metadata.ConfState
  325. defer func() {
  326. s.node.Stop()
  327. s.transport.Stop()
  328. if err := s.storage.Close(); err != nil {
  329. log.Panicf("etcdserver: close storage error: %v", err)
  330. }
  331. close(s.done)
  332. }()
  333. for {
  334. select {
  335. case <-s.Ticker:
  336. s.node.Tick()
  337. case rd := <-s.node.Ready():
  338. if rd.SoftState != nil {
  339. atomic.StoreUint64(&s.raftLead, rd.SoftState.Lead)
  340. if rd.RaftState == raft.StateLeader {
  341. syncC = s.SyncTicker
  342. // TODO: remove the nil checking
  343. // current test utility does not provide the stats
  344. if s.stats != nil {
  345. s.stats.BecomeLeader()
  346. }
  347. } else {
  348. syncC = nil
  349. }
  350. }
  351. // apply snapshot to storage if it is more updated than current snapi
  352. if !raft.IsEmptySnap(rd.Snapshot) && rd.Snapshot.Metadata.Index > snapi {
  353. if err := s.storage.SaveSnap(rd.Snapshot); err != nil {
  354. log.Fatalf("etcdserver: save snapshot error: %v", err)
  355. }
  356. s.raftStorage.ApplySnapshot(rd.Snapshot)
  357. snapi = rd.Snapshot.Metadata.Index
  358. log.Printf("etcdserver: saved incoming snapshot at index %d", snapi)
  359. }
  360. if err := s.storage.Save(rd.HardState, rd.Entries); err != nil {
  361. log.Fatalf("etcdserver: save state and entries error: %v", err)
  362. }
  363. s.raftStorage.Append(rd.Entries)
  364. s.send(rd.Messages)
  365. // recover from snapshot if it is more updated than current applied
  366. if !raft.IsEmptySnap(rd.Snapshot) && rd.Snapshot.Metadata.Index > appliedi {
  367. if err := s.store.Recovery(rd.Snapshot.Data); err != nil {
  368. log.Panicf("recovery store error: %v", err)
  369. }
  370. s.Cluster.Recover()
  371. appliedi = rd.Snapshot.Metadata.Index
  372. log.Printf("etcdserver: recovered from incoming snapshot at index %d", snapi)
  373. }
  374. // TODO(bmizerany): do this in the background, but take
  375. // care to apply entries in a single goroutine, and not
  376. // race them.
  377. if len(rd.CommittedEntries) != 0 {
  378. firsti := rd.CommittedEntries[0].Index
  379. if firsti > appliedi+1 {
  380. log.Panicf("etcdserver: first index of committed entry[%d] should <= appliedi[%d] + 1", firsti, appliedi)
  381. }
  382. var ents []raftpb.Entry
  383. if appliedi+1-firsti < uint64(len(rd.CommittedEntries)) {
  384. ents = rd.CommittedEntries[appliedi+1-firsti:]
  385. }
  386. if len(ents) > 0 {
  387. if appliedi, shouldstop = s.apply(ents, &confState); shouldstop {
  388. return
  389. }
  390. }
  391. }
  392. s.node.Advance()
  393. if appliedi-snapi > s.snapCount {
  394. log.Printf("etcdserver: start to snapshot (applied: %d, lastsnap: %d)", appliedi, snapi)
  395. s.snapshot(appliedi, &confState)
  396. snapi = appliedi
  397. }
  398. case <-syncC:
  399. s.sync(defaultSyncTimeout)
  400. case <-shouldstopC:
  401. return
  402. case <-s.stop:
  403. return
  404. }
  405. }
  406. }
  407. // Stop stops the server gracefully, and shuts down the running goroutine.
  408. // Stop should be called after a Start(s), otherwise it will block forever.
  409. func (s *EtcdServer) Stop() {
  410. select {
  411. case s.stop <- struct{}{}:
  412. case <-s.done:
  413. return
  414. }
  415. <-s.done
  416. }
  417. // StopNotify returns a channel that receives a empty struct
  418. // when the server is stopped.
  419. func (s *EtcdServer) StopNotify() <-chan struct{} { return s.done }
  420. // Do interprets r and performs an operation on s.store according to r.Method
  421. // and other fields. If r.Method is "POST", "PUT", "DELETE", or a "GET" with
  422. // Quorum == true, r will be sent through consensus before performing its
  423. // respective operation. Do will block until an action is performed or there is
  424. // an error.
  425. func (s *EtcdServer) Do(ctx context.Context, r pb.Request) (Response, error) {
  426. if r.ID == 0 {
  427. log.Panicf("request ID should never be 0")
  428. }
  429. if r.Method == "GET" && r.Quorum {
  430. r.Method = "QGET"
  431. }
  432. switch r.Method {
  433. case "POST", "PUT", "DELETE", "QGET":
  434. data, err := r.Marshal()
  435. if err != nil {
  436. return Response{}, err
  437. }
  438. ch := s.w.Register(r.ID)
  439. s.node.Propose(ctx, data)
  440. select {
  441. case x := <-ch:
  442. resp := x.(Response)
  443. return resp, resp.err
  444. case <-ctx.Done():
  445. s.w.Trigger(r.ID, nil) // GC wait
  446. return Response{}, parseCtxErr(ctx.Err())
  447. case <-s.done:
  448. return Response{}, ErrStopped
  449. }
  450. case "GET":
  451. switch {
  452. case r.Wait:
  453. wc, err := s.store.Watch(r.Path, r.Recursive, r.Stream, r.Since)
  454. if err != nil {
  455. return Response{}, err
  456. }
  457. return Response{Watcher: wc}, nil
  458. default:
  459. ev, err := s.store.Get(r.Path, r.Recursive, r.Sorted)
  460. if err != nil {
  461. return Response{}, err
  462. }
  463. return Response{Event: ev}, nil
  464. }
  465. case "HEAD":
  466. ev, err := s.store.Get(r.Path, r.Recursive, r.Sorted)
  467. if err != nil {
  468. return Response{}, err
  469. }
  470. return Response{Event: ev}, nil
  471. default:
  472. return Response{}, ErrUnknownMethod
  473. }
  474. }
  475. func (s *EtcdServer) SelfStats() []byte { return s.stats.JSON() }
  476. func (s *EtcdServer) LeaderStats() []byte {
  477. lead := atomic.LoadUint64(&s.raftLead)
  478. if lead != uint64(s.id) {
  479. return nil
  480. }
  481. return s.lstats.JSON()
  482. }
  483. func (s *EtcdServer) StoreStats() []byte { return s.store.JsonStats() }
  484. func (s *EtcdServer) AddMember(ctx context.Context, memb Member) error {
  485. // TODO: move Member to protobuf type
  486. b, err := json.Marshal(memb)
  487. if err != nil {
  488. return err
  489. }
  490. cc := raftpb.ConfChange{
  491. ID: GenID(),
  492. Type: raftpb.ConfChangeAddNode,
  493. NodeID: uint64(memb.ID),
  494. Context: b,
  495. }
  496. return s.configure(ctx, cc)
  497. }
  498. func (s *EtcdServer) RemoveMember(ctx context.Context, id uint64) error {
  499. cc := raftpb.ConfChange{
  500. ID: GenID(),
  501. Type: raftpb.ConfChangeRemoveNode,
  502. NodeID: id,
  503. }
  504. return s.configure(ctx, cc)
  505. }
  506. func (s *EtcdServer) UpdateMember(ctx context.Context, memb Member) error {
  507. b, err := json.Marshal(memb)
  508. if err != nil {
  509. return err
  510. }
  511. cc := raftpb.ConfChange{
  512. ID: GenID(),
  513. Type: raftpb.ConfChangeUpdateNode,
  514. NodeID: uint64(memb.ID),
  515. Context: b,
  516. }
  517. return s.configure(ctx, cc)
  518. }
  519. // Implement the RaftTimer interface
  520. func (s *EtcdServer) Index() uint64 { return atomic.LoadUint64(&s.raftIndex) }
  521. func (s *EtcdServer) Term() uint64 { return atomic.LoadUint64(&s.raftTerm) }
  522. // Only for testing purpose
  523. // TODO: add Raft server interface to expose raft related info:
  524. // Index, Term, Lead, Committed, Applied, LastIndex, etc.
  525. func (s *EtcdServer) Lead() uint64 { return atomic.LoadUint64(&s.raftLead) }
  526. // configure sends a configuration change through consensus and
  527. // then waits for it to be applied to the server. It
  528. // will block until the change is performed or there is an error.
  529. func (s *EtcdServer) configure(ctx context.Context, cc raftpb.ConfChange) error {
  530. ch := s.w.Register(cc.ID)
  531. if err := s.node.ProposeConfChange(ctx, cc); err != nil {
  532. s.w.Trigger(cc.ID, nil)
  533. return err
  534. }
  535. select {
  536. case x := <-ch:
  537. if err, ok := x.(error); ok {
  538. return err
  539. }
  540. if x != nil {
  541. log.Panicf("return type should always be error")
  542. }
  543. return nil
  544. case <-ctx.Done():
  545. s.w.Trigger(cc.ID, nil) // GC wait
  546. return parseCtxErr(ctx.Err())
  547. case <-s.done:
  548. return ErrStopped
  549. }
  550. }
  551. // sync proposes a SYNC request and is non-blocking.
  552. // This makes no guarantee that the request will be proposed or performed.
  553. // The request will be cancelled after the given timeout.
  554. func (s *EtcdServer) sync(timeout time.Duration) {
  555. ctx, cancel := context.WithTimeout(context.Background(), timeout)
  556. req := pb.Request{
  557. Method: "SYNC",
  558. ID: GenID(),
  559. Time: time.Now().UnixNano(),
  560. }
  561. data := pbutil.MustMarshal(&req)
  562. // There is no promise that node has leader when do SYNC request,
  563. // so it uses goroutine to propose.
  564. go func() {
  565. s.node.Propose(ctx, data)
  566. cancel()
  567. }()
  568. }
  569. // publish registers server information into the cluster. The information
  570. // is the JSON representation of this server's member struct, updated with the
  571. // static clientURLs of the server.
  572. // The function keeps attempting to register until it succeeds,
  573. // or its server is stopped.
  574. func (s *EtcdServer) publish(retryInterval time.Duration) {
  575. b, err := json.Marshal(s.attributes)
  576. if err != nil {
  577. log.Printf("etcdserver: json marshal error: %v", err)
  578. return
  579. }
  580. req := pb.Request{
  581. ID: GenID(),
  582. Method: "PUT",
  583. Path: MemberAttributesStorePath(s.id),
  584. Val: string(b),
  585. }
  586. for {
  587. ctx, cancel := context.WithTimeout(context.Background(), retryInterval)
  588. _, err := s.Do(ctx, req)
  589. cancel()
  590. switch err {
  591. case nil:
  592. log.Printf("etcdserver: published %+v to cluster %s", s.attributes, s.Cluster.ID())
  593. return
  594. case ErrStopped:
  595. log.Printf("etcdserver: aborting publish because server is stopped")
  596. return
  597. default:
  598. log.Printf("etcdserver: publish error: %v", err)
  599. }
  600. }
  601. }
  602. func getExpirationTime(r *pb.Request) time.Time {
  603. var t time.Time
  604. if r.Expiration != 0 {
  605. t = time.Unix(0, r.Expiration)
  606. }
  607. return t
  608. }
  609. func (s *EtcdServer) send(ms []raftpb.Message) {
  610. for _, m := range ms {
  611. if !s.Cluster.IsIDRemoved(types.ID(m.To)) {
  612. m.To = 0
  613. }
  614. }
  615. s.transport.Send(ms)
  616. }
  617. // apply takes entries received from Raft (after it has been committed) and
  618. // applies them to the current state of the EtcdServer.
  619. // The given entries should not be empty.
  620. func (s *EtcdServer) apply(es []raftpb.Entry, confState *raftpb.ConfState) (uint64, bool) {
  621. var applied uint64
  622. for i := range es {
  623. e := es[i]
  624. switch e.Type {
  625. case raftpb.EntryNormal:
  626. var r pb.Request
  627. pbutil.MustUnmarshal(&r, e.Data)
  628. s.w.Trigger(r.ID, s.applyRequest(r))
  629. case raftpb.EntryConfChange:
  630. var cc raftpb.ConfChange
  631. pbutil.MustUnmarshal(&cc, e.Data)
  632. shouldstop, err := s.applyConfChange(cc, confState)
  633. s.w.Trigger(cc.ID, err)
  634. if shouldstop {
  635. return applied, true
  636. }
  637. default:
  638. log.Panicf("entry type should be either EntryNormal or EntryConfChange")
  639. }
  640. atomic.StoreUint64(&s.raftIndex, e.Index)
  641. atomic.StoreUint64(&s.raftTerm, e.Term)
  642. applied = e.Index
  643. }
  644. return applied, false
  645. }
  646. // applyRequest interprets r as a call to store.X and returns a Response interpreted
  647. // from store.Event
  648. func (s *EtcdServer) applyRequest(r pb.Request) Response {
  649. f := func(ev *store.Event, err error) Response {
  650. return Response{Event: ev, err: err}
  651. }
  652. expr := getExpirationTime(&r)
  653. switch r.Method {
  654. case "POST":
  655. return f(s.store.Create(r.Path, r.Dir, r.Val, true, expr))
  656. case "PUT":
  657. exists, existsSet := getBool(r.PrevExist)
  658. switch {
  659. case existsSet:
  660. if exists {
  661. return f(s.store.Update(r.Path, r.Val, expr))
  662. }
  663. return f(s.store.Create(r.Path, r.Dir, r.Val, false, expr))
  664. case r.PrevIndex > 0 || r.PrevValue != "":
  665. return f(s.store.CompareAndSwap(r.Path, r.PrevValue, r.PrevIndex, r.Val, expr))
  666. default:
  667. if storeMemberAttributeRegexp.MatchString(r.Path) {
  668. id := mustParseMemberIDFromKey(path.Dir(r.Path))
  669. var attr Attributes
  670. if err := json.Unmarshal([]byte(r.Val), &attr); err != nil {
  671. log.Panicf("unmarshal %s should never fail: %v", r.Val, err)
  672. }
  673. s.Cluster.UpdateAttributes(id, attr)
  674. }
  675. return f(s.store.Set(r.Path, r.Dir, r.Val, expr))
  676. }
  677. case "DELETE":
  678. switch {
  679. case r.PrevIndex > 0 || r.PrevValue != "":
  680. return f(s.store.CompareAndDelete(r.Path, r.PrevValue, r.PrevIndex))
  681. default:
  682. return f(s.store.Delete(r.Path, r.Dir, r.Recursive))
  683. }
  684. case "QGET":
  685. return f(s.store.Get(r.Path, r.Recursive, r.Sorted))
  686. case "SYNC":
  687. s.store.DeleteExpiredKeys(time.Unix(0, r.Time))
  688. return Response{}
  689. default:
  690. // This should never be reached, but just in case:
  691. return Response{err: ErrUnknownMethod}
  692. }
  693. }
  694. // applyConfChange applies a ConfChange to the server. It is only
  695. // invoked with a ConfChange that has already passed through Raft
  696. func (s *EtcdServer) applyConfChange(cc raftpb.ConfChange, confState *raftpb.ConfState) (bool, error) {
  697. if err := s.Cluster.ValidateConfigurationChange(cc); err != nil {
  698. cc.NodeID = raft.None
  699. s.node.ApplyConfChange(cc)
  700. return false, err
  701. }
  702. *confState = *s.node.ApplyConfChange(cc)
  703. switch cc.Type {
  704. case raftpb.ConfChangeAddNode:
  705. m := new(Member)
  706. if err := json.Unmarshal(cc.Context, m); err != nil {
  707. log.Panicf("unmarshal member should never fail: %v", err)
  708. }
  709. if cc.NodeID != uint64(m.ID) {
  710. log.Panicf("nodeID should always be equal to member ID")
  711. }
  712. s.Cluster.AddMember(m)
  713. if m.ID == s.id {
  714. log.Printf("etcdserver: added local member %s %v to cluster %s", m.ID, m.PeerURLs, s.Cluster.ID())
  715. } else {
  716. s.transport.AddPeer(m.ID, m.PeerURLs)
  717. log.Printf("etcdserver: added member %s %v to cluster %s", m.ID, m.PeerURLs, s.Cluster.ID())
  718. }
  719. case raftpb.ConfChangeRemoveNode:
  720. id := types.ID(cc.NodeID)
  721. s.Cluster.RemoveMember(id)
  722. if id == s.id {
  723. log.Printf("etcdserver: removed local member %s from cluster %s", id, s.Cluster.ID())
  724. log.Println("etcdserver: the data-dir used by this member must be removed so that this host can be re-added with a new member ID")
  725. return true, nil
  726. } else {
  727. s.transport.RemovePeer(id)
  728. log.Printf("etcdserver: removed member %s from cluster %s", id, s.Cluster.ID())
  729. }
  730. case raftpb.ConfChangeUpdateNode:
  731. m := new(Member)
  732. if err := json.Unmarshal(cc.Context, m); err != nil {
  733. log.Panicf("unmarshal member should never fail: %v", err)
  734. }
  735. if cc.NodeID != uint64(m.ID) {
  736. log.Panicf("nodeID should always be equal to member ID")
  737. }
  738. s.Cluster.UpdateRaftAttributes(m.ID, m.RaftAttributes)
  739. if m.ID == s.id {
  740. log.Printf("etcdserver: update local member %s %v in cluster %s", m.ID, m.PeerURLs, s.Cluster.ID())
  741. } else {
  742. s.transport.UpdatePeer(m.ID, m.PeerURLs)
  743. log.Printf("etcdserver: update member %s %v in cluster %s", m.ID, m.PeerURLs, s.Cluster.ID())
  744. }
  745. }
  746. return false, nil
  747. }
  748. // TODO: non-blocking snapshot
  749. func (s *EtcdServer) snapshot(snapi uint64, confState *raftpb.ConfState) {
  750. d, err := s.store.Save()
  751. // TODO: current store will never fail to do a snapshot
  752. // what should we do if the store might fail?
  753. if err != nil {
  754. log.Panicf("etcdserver: store save should never fail: %v", err)
  755. }
  756. err = s.raftStorage.Compact(snapi, confState, d)
  757. if err != nil {
  758. // the snapshot was done asynchronously with the progress of raft.
  759. // raft might have already got a newer snapshot and called compact.
  760. if err == raft.ErrCompacted {
  761. return
  762. }
  763. log.Panicf("etcdserver: unexpected compaction error %v", err)
  764. }
  765. log.Printf("etcdserver: compacted log at index %d", snapi)
  766. if err := s.storage.Cut(); err != nil {
  767. log.Panicf("etcdserver: rotate wal file should never fail: %v", err)
  768. }
  769. snap, err := s.raftStorage.Snapshot()
  770. if err != nil {
  771. log.Panicf("etcdserver: snapshot error: %v", err)
  772. }
  773. if err := s.storage.SaveSnap(snap); err != nil {
  774. log.Fatalf("etcdserver: save snapshot error: %v", err)
  775. }
  776. log.Printf("etcdserver: saved snapshot at index %d", snap.Metadata.Index)
  777. }
  778. // for testing
  779. func (s *EtcdServer) PauseSending() {
  780. hub := s.transport.(*rafthttp.Transport)
  781. hub.Pause()
  782. }
  783. func (s *EtcdServer) ResumeSending() {
  784. hub := s.transport.(*rafthttp.Transport)
  785. hub.Resume()
  786. }
  787. func startNode(cfg *ServerConfig, ids []types.ID) (id types.ID, n raft.Node, s *raft.MemoryStorage, w *wal.WAL) {
  788. var err error
  789. member := cfg.Cluster.MemberByName(cfg.Name)
  790. metadata := pbutil.MustMarshal(
  791. &pb.Metadata{
  792. NodeID: uint64(member.ID),
  793. ClusterID: uint64(cfg.Cluster.ID()),
  794. },
  795. )
  796. if err := os.MkdirAll(cfg.SnapDir(), privateDirMode); err != nil {
  797. log.Fatalf("etcdserver create snapshot directory error: %v", err)
  798. }
  799. if w, err = wal.Create(cfg.WALDir(), metadata); err != nil {
  800. log.Fatalf("etcdserver: create wal error: %v", err)
  801. }
  802. peers := make([]raft.Peer, len(ids))
  803. for i, id := range ids {
  804. ctx, err := json.Marshal((*cfg.Cluster).Member(id))
  805. if err != nil {
  806. log.Panicf("marshal member should never fail: %v", err)
  807. }
  808. peers[i] = raft.Peer{ID: uint64(id), Context: ctx}
  809. }
  810. id = member.ID
  811. log.Printf("etcdserver: start member %s in cluster %s", id, cfg.Cluster.ID())
  812. s = raft.NewMemoryStorage()
  813. n = raft.StartNode(uint64(id), peers, 10, 1, s)
  814. return
  815. }
  816. func restartNode(cfg *ServerConfig, index uint64, snapshot *raftpb.Snapshot) (types.ID, raft.Node, *raft.MemoryStorage, *wal.WAL) {
  817. w, id, cid, st, ents := readWAL(cfg.WALDir(), index)
  818. cfg.Cluster.SetID(cid)
  819. log.Printf("etcdserver: restart member %s in cluster %s at commit index %d", id, cfg.Cluster.ID(), st.Commit)
  820. s := raft.NewMemoryStorage()
  821. if snapshot != nil {
  822. s.ApplySnapshot(*snapshot)
  823. }
  824. s.SetHardState(st)
  825. s.Append(ents)
  826. n := raft.RestartNode(uint64(id), 10, 1, s)
  827. return id, n, s, w
  828. }
  829. // isBootstrapped tries to check if the given member has been bootstrapped
  830. // in the given cluster.
  831. func isBootstrapped(cl *Cluster, member string) bool {
  832. us := getOtherPeerURLs(cl, member)
  833. rcl, err := getClusterFromPeers(us, false)
  834. if err != nil {
  835. return false
  836. }
  837. id := cl.MemberByName(member).ID
  838. m := rcl.Member(id)
  839. if m == nil {
  840. return false
  841. }
  842. if len(m.ClientURLs) > 0 {
  843. return true
  844. }
  845. return false
  846. }
  847. // GetClusterFromPeers takes a set of URLs representing etcd peers, and
  848. // attempts to construct a Cluster by accessing the members endpoint on one of
  849. // these URLs. The first URL to provide a response is used. If no URLs provide
  850. // a response, or a Cluster cannot be successfully created from a received
  851. // response, an error is returned.
  852. func GetClusterFromPeers(urls []string) (*Cluster, error) {
  853. return getClusterFromPeers(urls, true)
  854. }
  855. // If logerr is true, it prints out more error messages.
  856. func getClusterFromPeers(urls []string, logerr bool) (*Cluster, error) {
  857. cc := &http.Client{
  858. Transport: &http.Transport{
  859. ResponseHeaderTimeout: 500 * time.Millisecond,
  860. },
  861. Timeout: time.Second,
  862. }
  863. for _, u := range urls {
  864. resp, err := cc.Get(u + "/members")
  865. if err != nil {
  866. if logerr {
  867. log.Printf("etcdserver: could not get cluster response from %s: %v", u, err)
  868. }
  869. continue
  870. }
  871. b, err := ioutil.ReadAll(resp.Body)
  872. if err != nil {
  873. if logerr {
  874. log.Printf("etcdserver: could not read the body of cluster response: %v", err)
  875. }
  876. continue
  877. }
  878. var membs []*Member
  879. if err := json.Unmarshal(b, &membs); err != nil {
  880. if logerr {
  881. log.Printf("etcdserver: could not unmarshal cluster response: %v", err)
  882. }
  883. continue
  884. }
  885. id, err := types.IDFromString(resp.Header.Get("X-Etcd-Cluster-ID"))
  886. if err != nil {
  887. if logerr {
  888. log.Printf("etcdserver: could not parse the cluster ID from cluster res: %v", err)
  889. }
  890. continue
  891. }
  892. return NewClusterFromMembers("", id, membs), nil
  893. }
  894. return nil, fmt.Errorf("etcdserver: could not retrieve cluster information from the given urls")
  895. }
  896. // getOtherPeerURLs returns peer urls of other members in the cluster. The
  897. // returned list is sorted in ascending lexicographical order.
  898. func getOtherPeerURLs(cl ClusterInfo, self string) []string {
  899. us := make([]string, 0)
  900. for _, m := range cl.Members() {
  901. if m.Name == self {
  902. continue
  903. }
  904. us = append(us, m.PeerURLs...)
  905. }
  906. sort.Strings(us)
  907. return us
  908. }
  909. // TODO: move the function to /id pkg maybe?
  910. // GenID generates a random id that is not equal to 0.
  911. func GenID() (n uint64) {
  912. for n == 0 {
  913. n = uint64(rand.Int63())
  914. }
  915. return
  916. }
  917. func parseCtxErr(err error) error {
  918. switch err {
  919. case context.Canceled:
  920. return ErrCanceled
  921. case context.DeadlineExceeded:
  922. return ErrTimeout
  923. default:
  924. return err
  925. }
  926. }
  927. func getBool(v *bool) (vv bool, set bool) {
  928. if v == nil {
  929. return false, false
  930. }
  931. return *v, true
  932. }