htwkmooc/poseidon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Unwrap Nomad error for allocation exec

Browse Source 
* This will allow us to inspect whether the websocket connection was closed normally
This commit is contained in:
Sebastian Serth
2021-12-10 10:01:31 +01:00
parent dce895faff
commit ebbbfdb9be
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
internal/nomad/api_querier.go
View File

@ -100,11 +100,14 @@ func (nc *nomadAPIClient) Execute(runnerID string,
return 1, fmt.Errorf("error retrieving allocation info: %w", err)
}
exitCode, err := nc.client.Allocations().Exec(ctx, allocation, TaskName, tty, command, stdin, stdout, stderr, nil, nil)
if err != nil && websocket.IsCloseError(err, websocket.CloseNormalClosure) {
log.WithError(err).Info("The exit code could not be received.")
return 0, nil
} else if err != nil {
return 1, fmt.Errorf("error executing command in allocation: %w", err)
if err != nil {
rootCause := errors.Unwrap(err)
if rootCause != nil && websocket.IsCloseError(rootCause, websocket.CloseNormalClosure) {
log.WithError(err).Info("The exit code could not be received.")
return 0, nil
} else {
return 1, fmt.Errorf("error executing command in allocation: %w", err)
}
}
return exitCode, nil
}