pig-farm-controller/internal/app/service/transport/chirp_stack.go

package transport

import (
	"encoding/json"
	"io"
	"net/http"

	"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
)

// ChirpStackListener 主动发送的请求的event字段, 这个字段代表事件类型
const (
	eventUp          = "up"          // 上行数据事件：当接收到设备发送的数据时触发，这是最核心的事件。
	eventStatus      = "status"      // 设备状态事件：当设备报告其状态时触发（例如电池电量、信号强度）。
	eventJoin        = "join"        // 入网事件：当设备成功加入网络时触发。
	eventAck         = "ack"         // 下行确认事件：当设备确认收到下行消息时触发。
	eventTxAck       = "txack"       // 网关发送确认事件：当网关确认已发送下行消息时触发（不代表设备已收到）。
	eventLog         = "log"         // 日志事件：当设备或 ChirpStack 产生日志信息时触发。
	eventLocation    = "location"    // 位置事件：当设备的位置被解析或更新时触发。
	eventIntegration = "integration" // 集成事件：当其他集成（如第三方服务）处理数据后触发。
)

// ChirpStackListener 是一个监听器, 用于监听ChirpStack反馈的设备上行事件
type ChirpStackListener struct {
	logger *logs.Logger
}

func NewChirpStackListener(logger *logs.Logger) *ChirpStackListener {
	return &ChirpStackListener{
		logger: logger,
	}
}

// Handler 监听ChirpStack反馈的事件, 因为这是个Webhook, 所以直接回复掉再慢慢处理信息
func (c *ChirpStackListener) Handler() http.HandlerFunc {
	return func(w http.ResponseWriter, r *http.Request) {
		defer r.Body.Close()

		b, err := io.ReadAll(r.Body)
		if err != nil {
			c.logger.Errorf("读取请求体失败: %v", err)
			http.Error(w, "failed to read body", http.StatusBadRequest)
			return
		}

		event := r.URL.Query().Get("event")

		w.WriteHeader(http.StatusOK)

		// 将异步处理逻辑委托给 handler 方法
		go c.handler(b, event)
	}
}

// handler 用于处理 ChirpStack 发送的事件
func (c *ChirpStackListener) handler(data []byte, eventType string) {
	switch eventType {
	case eventUp:
		var msg UpEvent
		if err := json.Unmarshal(data, &msg); err != nil {
			c.logger.Errorf("解析 'up' 事件失败: %v, data: %s", err, string(data))
			return
		}
		c.handleUpEvent(&msg)

	case eventJoin:
		var msg JoinEvent
		if err := json.Unmarshal(data, &msg); err != nil {
			c.logger.Errorf("解析 'join' 事件失败: %v, data: %s", err, string(data))
			return
		}
		c.handleJoinEvent(&msg)

	case eventAck:
		var msg AckEvent
		if err := json.Unmarshal(data, &msg); err != nil {
			c.logger.Errorf("解析 'ack' 事件失败: %v, data: %s", err, string(data))
			return
		}
		c.handleAckEvent(&msg)

	case eventTxAck:
		var msg TxAckEvent
		if err := json.Unmarshal(data, &msg); err != nil {
			c.logger.Errorf("解析 'txack' 事件失败: %v, data: %s", err, string(data))
			return
		}
		c.handleTxAckEvent(&msg)

	case eventStatus:
		var msg StatusEvent
		if err := json.Unmarshal(data, &msg); err != nil {
			c.logger.Errorf("解析 'status' 事件失败: %v, data: %s", err, string(data))
			return
		}
		c.handleStatusEvent(&msg)

	case eventLog:
		var msg LogEvent
		if err := json.Unmarshal(data, &msg); err != nil {
			c.logger.Errorf("解析 'log' 事件失败: %v, data: %s", err, string(data))
			return
		}
		c.handleLogEvent(&msg)

	case eventLocation:
		var msg LocationEvent
		if err := json.Unmarshal(data, &msg); err != nil {
			c.logger.Errorf("解析 'location' 事件失败: %v, data: %s", err, string(data))
			return
		}
		c.handleLocationEvent(&msg)

	case eventIntegration:
		var msg IntegrationEvent
		if err := json.Unmarshal(data, &msg); err != nil {
			c.logger.Errorf("解析 'integration' 事件失败: %v, data: %s", err, string(data))
			return
		}
		c.handleIntegrationEvent(&msg)

	default:
		c.logger.Errorf("未知的ChirpStack事件: %s, data: %s", eventType, string(data))
	}
}

// --- 业务处理函数 ---

// handleUpEvent 处理上行数据事件
func (c *ChirpStackListener) handleUpEvent(event *UpEvent) {
	c.logger.Infof("处理 'up' 事件: %+v", event)
	// 在这里添加您的业务逻辑
}

// handleStatusEvent 处理设备状态事件
func (c *ChirpStackListener) handleStatusEvent(event *StatusEvent) {
	c.logger.Infof("处接收到理 'status' 事件: %+v", event)
	// 在这里添加您的业务逻辑
}

// handleAckEvent 处理下行确认事件
func (c *ChirpStackListener) handleAckEvent(event *AckEvent) {
	c.logger.Infof("接收到 'ack' 事件: %+v", event)
	// 在这里添加您的业务逻辑
}

// handleLogEvent 处理日志事件
func (c *ChirpStackListener) handleLogEvent(event *LogEvent) {
	// 首先，打印完整的事件结构体，用于详细排查
	c.logger.Infof("接收到 'log' 事件的完整内容: %+v", event)

	// 接着，根据 ChirpStack 日志的级别，使用我们自己的 logger 对应级别来打印核心信息
	logMessage := "ChirpStack 日志: [%s] %s (DevEui: %s)"
	switch event.Level {
	case "INFO":
		c.logger.Infof(logMessage, event.Code, event.Description, event.DeviceInfo.DevEui)
	case "WARNING":
		c.logger.Warnf(logMessage, event.Code, event.Description, event.DeviceInfo.DevEui)
	case "ERROR":
		c.logger.Errorf(logMessage, event.Code, event.Description, event.DeviceInfo.DevEui)
	default:
		// 对于未知级别，使用 Warn 级别打印，并明确指出级别未知
		c.logger.Warnf("ChirpStack 日志: [未知级别: %s] %s %s (DevEui: %s)",
			event.Level, event.Code, event.Description, event.DeviceInfo.DevEui)
	}

	// 在这里添加您的业务逻辑
}

// handleJoinEvent 处理入网事件
func (c *ChirpStackListener) handleJoinEvent(event *JoinEvent) {
	c.logger.Infof("接收到 'join' 事件: %+v", event)
	// 在这里添加您的业务逻辑
}

// handleTxAckEvent 处理网关发送确认事件
func (c *ChirpStackListener) handleTxAckEvent(event *TxAckEvent) {
	c.logger.Infof("接收到 'txack' 事件: %+v", event)
	// 在这里添加您的业务逻辑
}

// handleLocationEvent 处理位置事件
func (c *ChirpStackListener) handleLocationEvent(event *LocationEvent) {
	c.logger.Infof("接收到 'location' 事件: %+v", event)
	// 在这里添加您的业务逻辑
}

// handleIntegrationEvent 处理集成事件
func (c *ChirpStackListener) handleIntegrationEvent(event *IntegrationEvent) {
	c.logger.Infof("接收到 'integration' 事件: %+v", event)
	// 在这里添加您的业务逻辑
}