command-timeout-with-channel.go

//CommandTimeoutWithChannel command timeout using channel and goroutine
func CommandTimeoutWithChannel(command string, timeout time.Duration) (string, error) {
	cmd := exec.Command("/bin/bash", "-c", command)
	var out bytes.Buffer
	cmd.Stdout = &out

	err := cmd.Start()
	if err != nil {
		log.Printf("Error : %s", err.Error())
		return err.Error(), err
	}

	done := make(chan error)

	go func() { done <- cmd.Wait() }()

	timer := time.After(timeout)

	select {
	case <-timer:
		cmd.Process.Kill()
		fmt.Println("Command timed out")
		return "Command timed out", errors.New("Command Timeout")
	case err := <-done:
		if err != nil {
			return err.Error(), err
		}
		return out.String(), nil
	}
}