atlasexec: change workingDir property instead of creating new client

ariga · Jun 17, 2024 · 954d50a · 954d50a
1 parent b26beac
commit 954d50a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/atlasexec/atlas.go b/atlasexec/atlas.go
@@ -188,9 +188,12 @@ func NewClient(workingDir, execPath string) (_ *Client, err error) {
 //	  })
 //	  return err
 //	})
-func (c Client) WithWorkDir(dir string, fn func(*Client) error) error {
+func (c *Client) WithWorkDir(dir string, fn func(*Client) error) error {
+	wd := c.workingDir
+	defer func() { c.workingDir = wd }()
 	c.workingDir = dir
-	return fn(&c)
+	err := fn(c)
+	return err
 }
 
 // Login runs the 'login' command.