使用设计模式优化发卡机

internal/routes/play/card_device/device.go

@@ -0,0 +1,161 @@
+package card_device
+
+import (
+	"fmt"
+	"github.com/warthog618/go-gpiocdev"
+	"go.uber.org/zap"
+	"strings"
+	"time"
+)
+
+type Device struct {
+	chip      *gpiocdev.Chip     // GPIO 设备
+	inLines   *gpiocdev.Lines    // 输入针脚
+	pushLine  *gpiocdev.Line     // 发卡针脚
+	pullLine  *gpiocdev.Line     // 回收针脚
+	resetLine *gpiocdev.Line     // 重置针脚
+	status    map[inGpioLine]int // 状态
+}
+
+// statusEventHandler 状态事件处理
+func (d *Device) statusEventHandler(evt gpiocdev.LineEvent) {
+	offset := inGpioLine(evt.Offset)
+	eType := evt.Type
+
+	t := false
+	defer func() {
+		if t {
+			zap.S().Infof("状态: %s-%d", offset, d.status[offset])
+		}
+	}()
+
+	if v, ok := d.status[offset]; ok {
+		if eType == gpiocdev.LineEventFallingEdge {
+			if v == 0 {
+				return
+			} else {
+				t = true
+				d.status[offset] = 0
+			}
+		} else if eType == gpiocdev.LineEventRisingEdge {
+			if v == 1 {
+				return
+			} else {
+				t = true
+				d.status[offset] = 1
+			}
+		}
+	}
+}
+
+// initStatus 读取初始状态
+func (d *Device) initStatus() error {
+	offsets := d.inLines.Offsets()
+	status := make([]int, len(offsets), len(offsets))
+	err := d.inLines.Values(status)
+	if err != nil {
+		return err
+	}
+	for i := 0; i < len(status); i++ {
+		d.status[inGpioLine(offsets[i])] = status[i]
+	}
+
+	sv := make([]string, len(offsets))
+	for _, g := range allInGpio() {
+		sv = append(sv, fmt.Sprintf("%s-%d", g, d.status[g]))
+	}
+	strings.Join(sv, " ")
+	zap.S().Infof("初始状态: %s", strings.Join(sv, " "))
+	return nil
+}
+
+// Close 关闭设备
+func (d *Device) Close() error {
+	d.resetLine.Close()
+	d.pullLine.Close()
+	d.pushLine.Close()
+
+	d.inLines.Close()
+	d.chip.Close()
+
+	return nil
+}
+
+// PushCard 发卡
+func (d *Device) PushCard() {
+	d.pushLine.SetValue(1)
+	defer d.pushLine.SetValue(0)
+
+	time.Sleep(500 * time.Millisecond)
+}
+
+// PullCard 回收卡
+func (d *Device) PullCard() {
+	d.pullLine.SetValue(1)
+	defer d.pullLine.SetValue(0)
+
+	time.Sleep(500 * time.Millisecond)
+}
+
+// Reset 重置
+func (d *Device) Reset() {
+	d.resetLine.SetValue(1)
+	defer d.resetLine.SetValue(0)
+
+	time.Sleep(500 * time.Millisecond)
+}
+
+func New(name string) (*Device, error) {
+	chip, err := gpiocdev.NewChip(name, gpiocdev.AsActiveLow)
+	if err != nil {
+		return nil, fmt.Errorf("打开 GPIO 设备失败: %w", err)
+	}
+
+	d := &Device{
+		chip:   chip,
+		status: make(map[inGpioLine]int),
+	}
+
+	// 初始化输入针脚并监听针脚
+	inLines, err := chip.RequestLines(
+		allInGpioInt(),
+		gpiocdev.AsInput,                // 请求引脚作为输入
+		gpiocdev.WithPullUp,             // 使用上拉电阻
+		gpiocdev.WithRealtimeEventClock, // 使用实时时钟
+		gpiocdev.WithBothEdges,          // 监听上升沿和下降沿
+		gpiocdev.WithEventHandler(d.statusEventHandler),
+	)
+	if err != nil {
+		return nil, fmt.Errorf("请求输入引脚失败: %w", err)
+	}
+	d.inLines = inLines
+
+	// 读取初始状态
+	err = d.initStatus()
+	if err != nil {
+		return nil, fmt.Errorf("读取初始状态失败: %w", err)
+	}
+
+	// 初始化发卡引脚
+	push, err := chip.RequestLine(int(PushLine), gpiocdev.AsOutput())
+	if err != nil {
+		return nil, fmt.Errorf("请求引脚 %d 作为发卡失败: %w", PushLine, err)
+	}
+	d.pushLine = push
+
+	// 初始化回收引脚
+	pull, err := chip.RequestLine(int(PullLine), gpiocdev.AsOutput())
+	if err != nil {
+		return nil, fmt.Errorf("请求引脚 %d 作为回收失败: %w", PullLine, err)
+	}
+	d.pullLine = pull
+
+	// 初始化重置引脚
+	reset, err := chip.RequestLine(int(ResetLine), gpiocdev.AsOutput())
+	if err != nil {
+		return nil, fmt.Errorf("请求引脚 %d 作为重置失败: %w", ResetLine, err)
+	}
+	d.resetLine = reset
+
+	return d, nil
+}

internal/routes/play/card_device/in_gpio.go

@@ -0,0 +1,38 @@
+package card_device
+
+import (
+	"fmt"
+	"github.com/warthog618/go-gpiocdev/device/rpi"
+)
+
+type inGpioLine int
+
+const (
+	OutOKLine inGpioLine = rpi.GPIO6
+	LowerLine inGpioLine = rpi.GPIO13
+	ErrorLine inGpioLine = rpi.GPIO19
+	EmptyLine inGpioLine = rpi.GPIO26
+)
+
+func (g inGpioLine) String() string {
+	switch g {
+	case OutOKLine:
+		return "OutOKLine"
+	case LowerLine:
+		return "LowerLine"
+	case ErrorLine:
+		return "ErrorLine"
+	case EmptyLine:
+		return "EmptyLine"
+	default:
+		return fmt.Sprint(int(g))
+	}
+}
+
+func allInGpio() []inGpioLine {
+	return []inGpioLine{OutOKLine, LowerLine, ErrorLine, EmptyLine}
+}
+
+func allInGpioInt() []int {
+	return []int{int(OutOKLine), int(LowerLine), int(ErrorLine), int(EmptyLine)}
+}

internal/routes/play/card_device/out_gpio.go

@@ -0,0 +1,35 @@
+package card_device
+
+import (
+	"fmt"
+	"github.com/warthog618/go-gpiocdev/device/rpi"
+)
+
+type outGpioLine int
+
+const (
+	PushLine  outGpioLine = rpi.GPIO11
+	ResetLine outGpioLine = rpi.GPIO22
+	PullLine  outGpioLine = rpi.GPIO27
+)
+
+func (g outGpioLine) String() string {
+	switch g {
+	case PushLine:
+		return "PushLine"
+	case ResetLine:
+		return "ResetLine"
+	case PullLine:
+		return "PullLine"
+	default:
+		return fmt.Sprint(int(g))
+	}
+}
+
+func allOutGpio() []outGpioLine {
+	return []outGpioLine{PushLine, ResetLine, PullLine}
+}
+
+func allOutGpioInt() []int {
+	return []int{int(PushLine), int(ResetLine), int(PullLine)}
+}