Merge pull request #69 from ktr0731/fix-goroutine-leaks

Fix goroutine leaks
c-bata · May 18, 2018 · 2f3ee57 · 2f3ee57
2 parents 15992ed + fd5f830
commit 2f3ee57
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/prompt.go b/prompt.go
@@ -65,6 +65,8 @@ func (p *Prompt) Run() {
 		case b := <-bufCh:
 			if shouldExit, e := p.feed(b); shouldExit {
 				p.renderer.BreakLine(p.buf)
+				stopReadBufCh <- struct{}{}
+				stopHandleSignalCh <- struct{}{}
 				return
 			} else if e != nil {
 				// Stop goroutine to run readBuffer function
@@ -224,6 +226,7 @@ func (p *Prompt) Input() string {
 		case b := <-bufCh:
 			if shouldExit, e := p.feed(b); shouldExit {
 				p.renderer.BreakLine(p.buf)
+				stopReadBufCh <- struct{}{}
 				return ""
 			} else if e != nil {
 				// Stop goroutine to run readBuffer function