add total byte counts to track accurate flushing of buffers

Signed-off-by: Jeff Carr <jcarr@wit.com>

shell.go

@@ -72,12 +72,17 @@ func Run(cmdline string) int {
 	newerrreader := bufio.NewReader(pstderr)
 	nbrerr := nbreader.NewNBReader(newerrreader, 1024)
 
+	totalCount := 0
+
 	for {
 		time.Sleep(2 * time.Millisecond)   // only check the buffer 500 times a second
 		// log.Println("sleep done")
 
+		// tight loop that reads 1K at a time until buffer is empty
+		for {
 			oneByte := make([]byte, 1024)
 			count, err := nbr.Read(oneByte)
+			totalCount += count
 
 			if (err != nil) {
 				// log.Println("Read() count = ", count, "err = ", err)
@@ -88,9 +93,21 @@ func Run(cmdline string) int {
 			}
 			f.Write([]byte(string(oneByte)))
 			f.Flush()
+			if (count == 0) {
+				break
+			}
+		}
 
-		oneByte = make([]byte, 1024)
+		if (totalCount != 0) {
-		count, err = nbrerr.Read(oneByte)
+			log.Println("totalCount = ", totalCount)
+		}
+
+		//
+		// HANDLE STDERR
+		// HANDLE STDERR
+		//
+		oneByte := make([]byte, 1024)
+		count, err := nbrerr.Read(oneByte)
 
 		if (err != nil) {
 			oneByte = make([]byte, 1024)
@@ -99,8 +116,8 @@ func Run(cmdline string) int {
 			f.Flush()
 
 			log.Println("Read() count = ", count, "err = ", err)
-			spew.Dump(process.Process)
+			// spew.Dump(process.Process)
-			spew.Dump(process.ProcessState)
+			// spew.Dump(process.ProcessState)
 			err := process.Wait()
 			if err != nil {
 				spew.Dump(err.(*exec.ExitError))