fix readme

NexoMichael · Jun 5, 2018 · 55b938b · 55b938b
1 parent b4c07e9
commit 55b938b
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -30,13 +30,16 @@ Use the `go` command:
     import (
         "fmt"
         "os"
+        "syscall"
 
         input "github.com/NexoMichael/inputreader"
     )
 
     func main() {
         reader, _ := input.NewInputReader(os.Stdin)
-        defer reader.Close()
+        defer func() {
+            reader.Close()
+        }()
 
         var b [1]byte
         for {
@@ -46,6 +49,11 @@ Use the `go` command:
             }
 
             fmt.Printf(" - code is %d\n\r", b[0])
+
+            switch syscall.Signal(b[0]) {
+            case syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT:
+                return
+            }
         }
     }