aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--unicorn/RealUnicorn.go7
1 files changed, 4 insertions, 3 deletions
diff --git a/unicorn/RealUnicorn.go b/unicorn/RealUnicorn.go
index fb10a79..a114c73 100644
--- a/unicorn/RealUnicorn.go
+++ b/unicorn/RealUnicorn.go
@@ -6,6 +6,7 @@ import (
"github.com/ecc1/spi"
"log"
"os"
+ "syscall"
)
type RealUnicorn struct {
@@ -65,9 +66,9 @@ func (u *RealUnicorn) Close() error {
// MainLoop ...
// Do nothing until SIGTERM, then close the SPI library
-func MainLoop() {
+func (u *RealUnicorn) MainLoop() {
c := make(chan os.Signal, 2)
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
- <- c
- Close()
+ <-c
+ u.Close()
}