status-im
/
consul
mirror of https://github.com/status-im/consul.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

consul/state: cleanup

This commit is contained in:
Ryan Uber 2015-08-29 12:09:14 -07:00 committed by James Phillips
parent e148bb55d8
commit 9f281cbf80
3 changed files with 8 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
consul/state/state_store.go
View File

@ -180,13 +180,10 @@ func (s *StateStore) deleteNodeTxn(idx uint64, nodeID string, tx *memdb.Txn) err
} }
for service := services.Next(); service != nil; service = services.Next() { for service := services.Next(); service != nil; service = services.Next() {
svc := service.(*structs.ServiceNode) svc := service.(*structs.ServiceNode)
if err := s.deleteNodeServiceTxn(idx, nodeID, svc.ServiceID, tx); err != nil { if err := s.deleteServiceTxn(idx, nodeID, svc.ServiceID, tx); err != nil {
return err return err
} }
} }
if err := tx.Insert("index", &IndexEntry{"services", idx}); err != nil {
return fmt.Errorf("failed updating index: %s", err)
}
// Delete all checks associated with the node and update the check index // Delete all checks associated with the node and update the check index
checks, err := tx.Get("checks", "node", nodeID) checks, err := tx.Get("checks", "node", nodeID)
@ -199,9 +196,6 @@ func (s *StateStore) deleteNodeTxn(idx uint64, nodeID string, tx *memdb.Txn) err
return err return err
} }
} }
if err := tx.Insert("index", &IndexEntry{"checks", idx}); err != nil {
return fmt.Errorf("failed updating index: %s", err)
}
// Delete the node and update the index // Delete the node and update the index
if err := tx.Delete("nodes", node); err != nil { if err := tx.Delete("nodes", node); err != nil {
@ -327,14 +321,13 @@ func (s *StateStore) NodeServices(nodeID string) (*structs.NodeServices, error)
return ns, nil return ns, nil
} }
// DeleteNodeService is used to delete a given service associated // DeleteService is used to delete a given service associated with a node.
// with the given node. func (s *StateStore) DeleteService(idx uint64, nodeID, serviceID string) error {
func (s *StateStore) DeleteNodeService(idx uint64, nodeID, serviceID string) error {
tx := s.db.Txn(true) tx := s.db.Txn(true)
defer tx.Abort() defer tx.Abort()
// Call the service deletion // Call the service deletion
if err := s.deleteNodeServiceTxn(idx, nodeID, serviceID, tx); err != nil { if err := s.deleteServiceTxn(idx, nodeID, serviceID, tx); err != nil {
return err return err
} }
@ -342,9 +335,9 @@ func (s *StateStore) DeleteNodeService(idx uint64, nodeID, serviceID string) err
return nil return nil
} }
// deleteNodeServiceTxn is the inner method called to remove a service // deleteServiceTxn is the inner method called to remove a service
// registration within an existing transaction. // registration within an existing transaction.
func (s *StateStore) deleteNodeServiceTxn(idx uint64, nodeID, serviceID string, tx *memdb.Txn) error { func (s *StateStore) deleteServiceTxn(idx uint64, nodeID, serviceID string, tx *memdb.Txn) error {
// Look up the service // Look up the service
service, err := tx.First("services", "id", nodeID, serviceID) service, err := tx.First("services", "id", nodeID, serviceID)
if err != nil { if err != nil {

4
consul/state/state_store_test.go
View File

@ -304,7 +304,7 @@ func TestStateStore_EnsureService_NodeServices(t *testing.T) {
} }
} }
func TestStateStore_DeleteNodeService(t *testing.T) { func TestStateStore_DeleteService(t *testing.T) {
s := testStateStore(t) s := testStateStore(t)
// Register a node with one service // Register a node with one service
@ -321,7 +321,7 @@ func TestStateStore_DeleteNodeService(t *testing.T) {
testRegisterCheck(t, s, 3, "node1", "service1", "check1") testRegisterCheck(t, s, 3, "node1", "service1", "check1")
// Delete the service // Delete the service
if err := s.DeleteNodeService(4, "node1", "service1"); err != nil { if err := s.DeleteService(4, "node1", "service1"); err != nil {
t.Fatalf("err: %s", err) t.Fatalf("err: %s", err)
} }

17
consul/state_store_test.go
View File

@ -14,23 +14,6 @@ func testStateStore() (*StateStore, error) {
return NewStateStore(nil, os.Stderr) return NewStateStore(nil, os.Stderr)
} }
func testNodeRegister(t *testing.T, idx uint64, db *StateStore, nodeID string) {
node := &structs.Node{Node: nodeID}
if err := db.EnsureNode(idx, node); err != nil {
t.Fatalf("err: %s", err)
}
tx := db.Txn(false)
defer tx.Abort()
n, err := tx.First("nodes", "id", nodeID)
if err != nil {
t.Fatalf("err: %s", err)
}
if result := n.(*structs.Node); result != node {
t.Fatalf("bad node: %#v", result)
}
}
func TestEnsureRegistration(t *testing.T) { func TestEnsureRegistration(t *testing.T) {
store, err := testStateStore() store, err := testStateStore()
if err != nil { if err != nil {