Signal handling in main

rfm69d/rfm69d.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"log"
+	"os"
 
 	"github.com/fulr/rfm69"
 	"github.com/kidoman/embd"
@@ -40,5 +41,13 @@ func main() {
 	}
 	log.Print(rfm)
 
-	rfm.Loop()
+	quit := rfm.Loop()
+
+	sigint := make(chan os.Signal)
+	os.Notify(sigint, os.Interrupt)
+
+	<-sigint
+
+	quit <- 1
+	<-quit
 }

rfm69handler.go

@@ -7,34 +7,46 @@ import (
 )
 
 // Loop is the main receive and transmit handling loop
-func (r *Device) Loop() error {
+func (r *Device) Loop() chan int {
+	quit := make(chan int)
 	c := make(chan Data)
-	irq := make(chan int)
+	go func() {
+		irq := make(chan int)
 
-	r.gpio.Watch(embd.EdgeRising, func(pin embd.DigitalPin) {
+		r.gpio.Watch(embd.EdgeRising, func(pin embd.DigitalPin) {
-		irq <- 1
+			irq <- 1
-	})
+		})
 
-	r.SetMode(RF_OPMODE_RECEIVER)
+		err := r.SetMode(RF_OPMODE_RECEIVER)
+		if err != nil {
+			log.Print(err)
+			return
+		}
+		defer r.SetMode(RF_OPMODE_STANDBY)
 
-	for {
+		for {
-		select {
+			select {
-		case dataToTransmit := <-c:
+			case dataToTransmit := <-c:
-			// can send?
+				// can send?
-			r.SetMode(RF_OPMODE_STANDBY)
+				r.SetMode(RF_OPMODE_STANDBY)
-			r.writeFifo(&dataToTransmit)
+				r.writeFifo(&dataToTransmit)
-			r.writeReg(REG_DIOMAPPING1, RF_DIOMAPPING1_DIO0_00)
+				r.writeReg(REG_DIOMAPPING1, RF_DIOMAPPING1_DIO0_00)
-			r.SetMode(RF_OPMODE_TRANSMITTER)
+				r.SetMode(RF_OPMODE_TRANSMITTER)
-			<-irq
+				<-irq
-			r.SetMode(RF_OPMODE_RECEIVER)
+				r.SetMode(RF_OPMODE_RECEIVER)
-			r.writeReg(REG_DIOMAPPING1, RF_DIOMAPPING1_DIO0_01)
+				r.writeReg(REG_DIOMAPPING1, RF_DIOMAPPING1_DIO0_01)
-		case <-irq:
+			case <-irq:
-			data, err := r.readFifo()
+				data, err := r.readFifo()
-			if err != nil {
+				if err != nil {
-				log.Print(err)
+					log.Print(err)
-				return err
+					return
+				}
+				log.Print(data)
+			case <-quit:
+				quit <- 1
+				return
 			}
-			log.Print(data)
 		}
-	}
+	}()
+	return quit
 }