fix: generate topology uuid uniformly in single-master mode (#8405)
* fix: ensure topology uuid is generated in single master setups * ensureTopologyId adds a Hashicorp-aware implementation * simplify
This commit is contained in:
@@ -2,6 +2,7 @@ package weed_server
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/http/httputil"
|
"net/http/httputil"
|
||||||
@@ -40,6 +41,7 @@ import (
|
|||||||
const (
|
const (
|
||||||
SequencerType = "master.sequencer.type"
|
SequencerType = "master.sequencer.type"
|
||||||
SequencerSnowflakeId = "master.sequencer.sequencer_snowflake_id"
|
SequencerSnowflakeId = "master.sequencer.sequencer_snowflake_id"
|
||||||
|
raftApplyTimeout = 1 * time.Second
|
||||||
)
|
)
|
||||||
|
|
||||||
type MasterOption struct {
|
type MasterOption struct {
|
||||||
@@ -227,6 +229,7 @@ func (ms *MasterServer) SetRaftServer(raftServer *RaftServer) {
|
|||||||
|
|
||||||
if ms.Topo.IsLeader() {
|
if ms.Topo.IsLeader() {
|
||||||
glog.V(0).Infof("%s I am the leader!", raftServerName)
|
glog.V(0).Infof("%s I am the leader!", raftServerName)
|
||||||
|
go ms.ensureTopologyId()
|
||||||
} else {
|
} else {
|
||||||
var raftServerLeader string
|
var raftServerLeader string
|
||||||
ms.Topo.RaftServerAccessLock.RLock()
|
ms.Topo.RaftServerAccessLock.RLock()
|
||||||
@@ -242,6 +245,26 @@ func (ms *MasterServer) SetRaftServer(raftServer *RaftServer) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (ms *MasterServer) syncRaftForTopologyId(topologyId string) error {
|
||||||
|
ms.Topo.RaftServerAccessLock.RLock()
|
||||||
|
defer ms.Topo.RaftServerAccessLock.RUnlock()
|
||||||
|
|
||||||
|
if ms.Topo.RaftServer != nil {
|
||||||
|
_, err := ms.Topo.RaftServer.Do(topology.NewMaxVolumeIdCommand(ms.Topo.GetMaxVolumeId(), topologyId))
|
||||||
|
return err
|
||||||
|
} else if ms.Topo.HashicorpRaft != nil {
|
||||||
|
b, err := json.Marshal(topology.NewMaxVolumeIdCommand(ms.Topo.GetMaxVolumeId(), topologyId))
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed marshal NewMaxVolumeIdCommand: %v", err)
|
||||||
|
}
|
||||||
|
if future := ms.Topo.HashicorpRaft.Apply(b, raftApplyTimeout); future.Error() != nil {
|
||||||
|
return future.Error()
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
return fmt.Errorf("no raft server configured")
|
||||||
|
}
|
||||||
|
|
||||||
func (ms *MasterServer) ensureTopologyId() {
|
func (ms *MasterServer) ensureTopologyId() {
|
||||||
ms.topologyIdGenLock.Lock()
|
ms.topologyIdGenLock.Lock()
|
||||||
defer ms.topologyIdGenLock.Unlock()
|
defer ms.topologyIdGenLock.Unlock()
|
||||||
@@ -254,7 +277,7 @@ func (ms *MasterServer) ensureTopologyId() {
|
|||||||
glog.V(1).Infof("lost leadership while sending barrier command for topologyId")
|
glog.V(1).Infof("lost leadership while sending barrier command for topologyId")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if _, err := ms.Topo.RaftServer.Do(topology.NewMaxVolumeIdCommand(ms.Topo.GetMaxVolumeId(), ms.Topo.GetTopologyId())); err != nil {
|
if err := ms.syncRaftForTopologyId(ms.Topo.GetTopologyId()); err != nil {
|
||||||
glog.Errorf("failed to sync raft for topologyId: %v, retrying in 1s", err)
|
glog.Errorf("failed to sync raft for topologyId: %v, retrying in 1s", err)
|
||||||
time.Sleep(time.Second)
|
time.Sleep(time.Second)
|
||||||
continue
|
continue
|
||||||
@@ -273,8 +296,7 @@ func (ms *MasterServer) ensureTopologyId() {
|
|||||||
EnsureTopologyId(ms.Topo, func() bool {
|
EnsureTopologyId(ms.Topo, func() bool {
|
||||||
return ms.Topo.IsLeader()
|
return ms.Topo.IsLeader()
|
||||||
}, func(topologyId string) error {
|
}, func(topologyId string) error {
|
||||||
_, err := ms.Topo.RaftServer.Do(topology.NewMaxVolumeIdCommand(ms.Topo.GetMaxVolumeId(), topologyId))
|
return ms.syncRaftForTopologyId(topologyId)
|
||||||
return err
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -51,6 +51,7 @@ func (ms *MasterServer) collectionDeleteHandler(w http.ResponseWriter, r *http.R
|
|||||||
func (ms *MasterServer) dirStatusHandler(w http.ResponseWriter, r *http.Request) {
|
func (ms *MasterServer) dirStatusHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
m := make(map[string]interface{})
|
m := make(map[string]interface{})
|
||||||
m["Version"] = version.Version()
|
m["Version"] = version.Version()
|
||||||
|
m["TopologyId"] = ms.Topo.GetTopologyId()
|
||||||
m["Topology"] = ms.Topo.ToInfo()
|
m["Topology"] = ms.Topo.ToInfo()
|
||||||
writeJsonQuiet(w, r, http.StatusOK, m)
|
writeJsonQuiet(w, r, http.StatusOK, m)
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user