Merge pull request #10 from bugout-dev/bugout-trap-sleep

zomglings · web-flow · commit 3405f615dd30 · 2021-03-21T07:00:58.000-07:00
Some bugfixes for "bugout trap"
diff --git a/cmd/bugout/trap/invoke.go b/cmd/bugout/trap/invoke.go
@@ -6,6 +6,7 @@ import (
 	"os/exec"
 	"os/signal"
 	"strings"
+	"time"
 
 	"github.com/spf13/cobra"
 )
@@ -17,7 +18,7 @@ type InvocationResult struct {
 }
 
 func stream(reader io.Reader, writer io.Writer, doneChan chan<- bool, cancelChan <-chan bool) {
-	b := make([]byte, 1)
+	b := make([]byte, 1024)
 	for {
 		select {
 		case <-cancelChan:
@@ -97,6 +98,8 @@ func RunWrappedCommand(trapCmd *cobra.Command, invocation []string) (*Invocation
 						exitChannel <- 1
 					}
 					return
+				} else {
+					time.Sleep(10 * time.Millisecond)
 				}
 			}
 		}
diff --git a/pkg/version.go b/pkg/version.go
@@ -1,3 +1,3 @@
 package bugout
 
-const Version string = "0.3.1"
+const Version string = "0.3.2"

Original file line number	Diff line number	Diff line change
`@@ -6,6 +6,7 @@ import (`
`6`	`6`	`"os/exec"`
`7`	`7`	`"os/signal"`
`8`	`8`	`"strings"`
	`9`	`+ "time"`
`9`	`10`
`10`	`11`	`"github.com/spf13/cobra"`
`11`	`12`	`)`
`@@ -17,7 +18,7 @@ type InvocationResult struct {`
`17`	`18`	`}`
`18`	`19`
`19`	`20`	`func stream(reader io.Reader, writer io.Writer, doneChan chan<- bool, cancelChan <-chan bool) {`
`20`		`- b := make([]byte, 1)`
	`21`	`+ b := make([]byte, 1024)`
`21`	`22`	`for {`
`22`	`23`	`select {`
`23`	`24`	`case <-cancelChan:`
`@@ -97,6 +98,8 @@ func RunWrappedCommand(trapCmd cobra.Command, invocation []string) (Invocation`
`97`	`98`	`exitChannel <- 1`
`98`	`99`	`}`
`99`	`100`	`return`
	`101`	`+ } else {`
	`102`	`+ time.Sleep(10 * time.Millisecond)`
`100`	`103`	`}`
`101`	`104`	`}`
`102`	`105`	`}`