Skip to content

feat(cli/ssh): simplify log file flags #7863

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 16 commits into from
Jun 12, 2023
Merged

feat(cli/ssh): simplify log file flags #7863

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
4b3b2bd
feat(cli/ssh): simplify log file flags
ammario Jun 5, 2023
82ce7a2
Fix race
ammario Jun 6, 2023
ad2c946
Fix race the right way
ammario Jun 6, 2023
1ca5930
Restore directory flag
ammario Jun 6, 2023
d7b26ad
Update tests
ammario Jun 6, 2023
48cb9b2
make
ammario Jun 6, 2023
3c43cfc
Merge remote-tracking branch 'origin/main' into log-race
ammario Jun 7, 2023
a1b68e5
mess around
ammario Jun 8, 2023
23a6833
Fix another race
ammario Jun 8, 2023
f17e7ce
fixup! Fix another race
ammario Jun 8, 2023
5a311b3
Update cli/ssh.go
ammario Jun 8, 2023
65cdbcc
Move CloseStdin into clibase
ammario Jun 9, 2023
d1cb325
Merge remote-tracking branch 'origin/main' into log-race
ammario Jun 9, 2023
8471555
Revert "Move CloseStdin into clibase"
ammario Jun 10, 2023
6593dfe
Merge remote-tracking branch 'origin/main' into log-race
ammario Jun 10, 2023
c5a1968
Spike-ify
ammario Jun 10, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
mess around
  • Loading branch information
@ammario
ammario committed Jun 8, 2023
commit a1b68e55996b986d8b24124a2511c6754865dcdc
21 changes: 14 additions & 7 deletions cli/ssh.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -76,7 +76,7 @@ func (r *RootCmd) ssh() *clibase.Cmd {
if logDirPath != "" {
nonce, err := cryptorand.StringCharset(cryptorand.Lower, 5)
if err != nil {
panic(err)
return xerrors.Errorf("generate nonce: %w", err)
}
logFilePath := filepath.Join(
logDirPath,
Expand Down Expand Up @@ -109,6 +109,19 @@ func (r *RootCmd) ssh() *clibase.Cmd {
client.Logger = logger
}

// This WaitGroup solves for a race condition where we were logging
// while closing the log file in in a defer. It probably solves
// others too.
//
// Its position in this function is important. It must be after
// the logger is created but before any goroutines or wind-down
// defers (e.g. context cancels) are declared.
var wg sync.WaitGroup
defer func() {
cancel()
wg.Wait()
}()

workspace, workspaceAgent, err := getWorkspaceAndAgent(ctx, inv, client, codersdk.Me, inv.Args[0])
if err != nil {
return err
Expand Down Expand Up @@ -169,12 +182,6 @@ func (r *RootCmd) ssh() *clibase.Cmd {
stopPolling := tryPollWorkspaceAutostop(ctx, client, workspace)
defer stopPolling()

// This WaitGroup solves for a race condition where we were logging
// while closing the log file in in a defer. It probably solves
// others too.
var wg sync.WaitGroup
defer wg.Wait()

if stdio {
rawSSH, err := conn.SSH(ctx)
if err != nil {
Expand Down
2 changes: 1 addition & 1 deletion cli/ssh_test.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -714,7 +714,7 @@ type stdioConn struct {
io.Writer
}

func (*stdioConn) Close() (err error) {
func (c *stdioConn) Close() (err error) {
return nil
}

Expand Down