Mirrors/go-prompt
7
0
Fork 0
Code Issues Pull Requests Releases 1 Wiki Activity

Retry syscall.Write in Flush

Browse Source
This commit is contained in:
c-bata 2018-10-18 01:58:55 +09:00
parent b1e823dbd5
commit cd2b4dd322
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
output_posix.go
View File

@ -3,9 +3,12 @@
package prompt package prompt
import ( import (
"log"
"syscall" "syscall"
) )
const flushMaxRetryCount = 3
// PosixWriter is a ConsoleWriter implementation for POSIX environment. // PosixWriter is a ConsoleWriter implementation for POSIX environment.
// To control terminal emulator, this outputs VT100 escape sequences. // To control terminal emulator, this outputs VT100 escape sequences.
type PosixWriter struct { type PosixWriter struct {
@ -17,9 +20,15 @@ type PosixWriter struct {
func (w *PosixWriter) Flush() error { func (w *PosixWriter) Flush() error {
l := len(w.buffer) l := len(w.buffer)
offset := 0 offset := 0
retry := 0
for { for {
n, err := syscall.Write(w.fd, w.buffer[offset:]) n, err := syscall.Write(w.fd, w.buffer[offset:])
if err != nil { if err != nil {
log.Printf("[DEBUG] flush error: %s", err)
if retry < flushMaxRetryCount {
retry++
continue
}
return err return err
} }
offset += n offset += n