From 7213657a539c86dccef9446c373ba8a1ba386c20 Mon Sep 17 00:00:00 2001
From: Martin Ashby <martin@martin-laptop.lan>
Date: Mon, 21 May 2018 10:34:08 +0100
Subject: fixed real unicorns

---
 unicorn/RealUnicorn.go | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'unicorn')

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()
 }
-- 
cgit v1.2.3-ZIG