Remove update time

This commit is contained in:
Alex Dadgar
2018-02-26 15:06:01 -08:00
committed by Michael Schurter
parent 762db7c5d7
commit fba20fd58d
8 changed files with 10 additions and 18 deletions

View File

@@ -150,7 +150,7 @@ func TestNodeDrainer_SimpleDrain(t *testing.T) {
}
node, err := state.NodeByID(nil, c1.NodeID())
require.Nil(err)
require.Nil(state.UpdateNodeDrain(node.ModifyIndex+1, node.ID, strategy, 101))
require.Nil(state.UpdateNodeDrain(node.ModifyIndex+1, node.ID, strategy))
// Start node 2
c2 := client.TestClient(t, func(conf *config.Config) {

View File

@@ -328,7 +328,7 @@ func (n *nomadFSM) applyDrainUpdate(buf []byte, index uint64) interface{} {
panic(fmt.Errorf("failed to decode request: %v", err))
}
if err := n.state.UpdateNodeDrain(index, req.NodeID, req.DrainStrategy, req.UpdateTime); err != nil {
if err := n.state.UpdateNodeDrain(index, req.NodeID, req.DrainStrategy); err != nil {
n.logger.Printf("[ERR] nomad.fsm: UpdateNodeDrain failed: %v", err)
return err
}

View File

@@ -301,7 +301,6 @@ func TestFSM_UpdateNodeDrain(t *testing.T) {
req2 := structs.NodeUpdateDrainRequest{
NodeID: node.ID,
DrainStrategy: strategy,
UpdateTime: 101,
}
buf, err = structs.Encode(structs.NodeUpdateDrainRequestType, req2)
require.Nil(err)

View File

@@ -433,9 +433,6 @@ func (n *Node) UpdateDrain(args *structs.NodeUpdateDrainRequest,
return fmt.Errorf("node not found")
}
// Update the timestamp to
args.UpdateTime = time.Now().Unix()
// Commit this update via Raft
_, index, err := n.srv.raftApply(structs.NodeUpdateDrainRequestType, args)
if err != nil {

View File

@@ -2379,7 +2379,7 @@ func TestClientEndpoint_ListNodes_Blocking(t *testing.T) {
Deadline: 10 * time.Second,
},
}
if err := state.UpdateNodeDrain(3, node.ID, s, 101); err != nil {
if err := state.UpdateNodeDrain(3, node.ID, s); err != nil {
t.Fatalf("err: %v", err)
}
})

View File

@@ -616,8 +616,7 @@ func (s *StateStore) UpdateNodeStatus(index uint64, nodeID, status string) error
}
// UpdateNodeDrain is used to update the drain of a node
func (s *StateStore) UpdateNodeDrain(index uint64, nodeID string,
drain *structs.DrainStrategy, updateTime int64) error {
func (s *StateStore) UpdateNodeDrain(index uint64, nodeID string, drain *structs.DrainStrategy) error {
txn := s.db.Txn(true)
defer txn.Abort()
@@ -647,7 +646,6 @@ func (s *StateStore) UpdateNodeDrain(index uint64, nodeID string,
}
copyNode.ModifyIndex = index
copyNode.StatusUpdatedAt = updateTime
// Insert the node
if err := txn.Insert("nodes", copyNode); err != nil {

View File

@@ -705,19 +705,18 @@ func TestStateStore_UpdateNodeDrain_Node(t *testing.T) {
require.Nil(state.UpsertNode(1000, node))
expectedTime := int64(101)
// Create a watchset so we can test that update node drain fires the watch
ws := memdb.NewWatchSet()
_, err := state.NodeByID(ws, node.ID)
require.Nil(err)
expectedDrain := &structs.DrainStrategy{
DrainSpec: structs.DrainSpec{
Deadline: -1 * time.Second,
},
}
// Create a watchset so we can test that update node drain fires the watch
ws := memdb.NewWatchSet()
_, err := state.NodeByID(ws, node.ID)
require.Nil(err)
require.Nil(state.UpdateNodeDrain(1001, node.ID, expectedDrain, expectedTime))
require.Nil(state.UpdateNodeDrain(1001, node.ID, expectedDrain))
require.True(watchFired(ws))
ws = memdb.NewWatchSet()

View File

@@ -307,7 +307,6 @@ type NodeUpdateDrainRequest struct {
NodeID string
Drain bool // TODO Deprecate
DrainStrategy *DrainStrategy
UpdateTime int64
WriteRequest
}