Delete lock info when unlocking.

hashicorp · Sep 8, 2017 · b8f4f6d · b8f4f6d
1 parent b896348
commit b8f4f6d
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/backend/remote-state/etcdv3/client.go b/backend/remote-state/etcdv3/client.go
@@ -116,6 +116,17 @@ func (c *RemoteClient) Unlock(id string) error {
 	return c.unlock(id)
 }
 
+func (c *RemoteClient) deleteLockInfo(info *state.LockInfo) error {
+	res, err := c.Client.KV.Delete(context.TODO(), c.Key+lockInfoSuffix)
+	if err != nil {
+		return err
+	}
+	if res.Deleted == 0 {
+		return fmt.Errorf("No keys deleted for %s when deleting lock info.", c.Key+lockInfoSuffix)
+	}
+	return nil
+}
+
 func (c *RemoteClient) getLockInfo() (*state.LockInfo, error) {
 	res, err := c.Client.KV.Get(context.TODO(), c.Key+lockInfoSuffix)
 	if err != nil {
@@ -181,6 +192,9 @@ func (c *RemoteClient) unlock(id string) error {
 
 	var errs error
 
+	if err := c.deleteLockInfo(c.info); err != nil {
+		errs = multierror.Append(errs, err)
+	}
 	if err := c.etcdMutex.Unlock(context.TODO()); err != nil {
 		errs = multierror.Append(errs, err)
 	}