From 37cd9382791965641c1cf773591954d803d6d0be Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E9=9D=99?= <lym125@gmail.com>
Date: Thu, 6 Jul 2023 18:07:25 +0800
Subject: [PATCH] =?UTF-8?q?=E8=8E=B7=E5=8F=96=E8=AE=BE=E5=A4=87=E4=BF=A1?=
 =?UTF-8?q?=E6=81=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 equipment/spray_equipment.go | 71 ++++++++++++++++++++++++++++++++++++
 main.go                      | 12 +++++-
 2 files changed, 82 insertions(+), 1 deletion(-)
 create mode 100644 equipment/spray_equipment.go

diff --git a/equipment/spray_equipment.go b/equipment/spray_equipment.go
new file mode 100644
index 0000000..63aad0d
--- /dev/null
+++ b/equipment/spray_equipment.go
@@ -0,0 +1,71 @@
+package equipment
+
+func NewSprayEquipment(id string) *SprayEquipment {
+	return &SprayEquipment{
+		id:        id,
+		speed1:    0,
+		speed2:    0,
+		yv1:       0,
+		yv2:       0,
+		isRunning: false,
+		status:    0,
+	}
+}
+
+type SprayEquipment struct {
+	id        string // 设备ID
+	speed1    uint8  // 电磁阀1对应的喷雾量(百分比): 0-100
+	speed2    uint8  // 电磁阀2对应的喷雾量(百分比): 0-100
+	yv1       uint8  // 电磁阀1的状态: 0 停止, 1 启动
+	yv2       uint8  // 电磁阀2的状态: 0 停止, 1 启动
+	isRunning bool   // 是否运行中
+	status    uint8  // 连接状态: 0 未知, 1 在线, 2 离线
+}
+
+func (s *SprayEquipment) GetId() string {
+	return s.id
+}
+
+func (s *SprayEquipment) SetSpeed1(speed uint8) {
+	s.speed1 = speed
+}
+
+func (s *SprayEquipment) GetSpeed1() uint8 {
+	return s.speed1
+}
+
+func (s *SprayEquipment) SetSpeed2(speed uint8) {
+	s.speed2 = speed
+}
+
+func (s *SprayEquipment) GetSpeed2() uint8 {
+	return s.speed2
+}
+
+func (s *SprayEquipment) SetYv1(yv1 uint8) {
+	s.yv1 = yv1
+}
+
+func (s *SprayEquipment) GetYv1() uint8 {
+	return s.yv1
+}
+
+func (s *SprayEquipment) SetYv2(yv2 uint8) {
+	s.yv2 = yv2
+}
+
+func (s *SprayEquipment) GetYv2() uint8 {
+	return s.yv2
+}
+
+func (s *SprayEquipment) SetIsRunning(isRunning bool) {
+	s.isRunning = isRunning
+}
+
+func (s *SprayEquipment) GetIsRunning() bool {
+	return s.isRunning
+}
+
+func (s *SprayEquipment) GetStatus() uint8 {
+	return s.status
+}
diff --git a/main.go b/main.go
index 353d6e8..9c55f7e 100644
--- a/main.go
+++ b/main.go
@@ -5,6 +5,7 @@ import (
 	"encoding/json"
 	"flag"
 	"fmt"
+	"iot-spray-equipment/equipment"
 	"iot-spray-equipment/mqttx"
 	"net/http"
 	"os"
@@ -18,6 +19,7 @@ import (
 
 var (
 	configFilePath string
+	sprayEquipment = equipment.NewSprayEquipment("DC73AEE4CD274074BD9D01BCEE379A98")
 )
 
 func init() {
@@ -113,7 +115,15 @@ func main() {
 }
 
 func StatusHandler(c *gin.Context) {
-	//
+	c.JSON(http.StatusOK, gin.H{
+		"id":         sprayEquipment.GetId(),
+		"speed1":     sprayEquipment.GetSpeed1(),
+		"speed2":     sprayEquipment.GetSpeed2(),
+		"yv1":        sprayEquipment.GetYv1(),
+		"yv2":        sprayEquipment.GetYv2(),
+		"is_running": sprayEquipment.GetIsRunning(),
+		"status":     sprayEquipment.GetStatus(),
+	})
 }
 
 func MqttHandler(c *gin.Context) {