command-timeout-with-context.go

//CommandTimeoutWithContext command timeout with background context
func CommandTimeoutWithContext(command string, timeout time.Duration) (string, error) {
	ctx, cancel := context.WithTimeout(context.Background(), timeout)
	defer cancel()

	cmd := exec.CommandContext(ctx, "/bin/bash", "-c", command)
	out, err := cmd.CombinedOutput()

	if ctx.Err() == context.DeadlineExceeded {
		fmt.Println("Command timed out")
		return "Command timed out", errors.New("Command Timeout")
	}

	if err != nil {
		fmt.Println("Error : ", err.Error())
		return string(out), err
	}
	return string(out), nil
}