272 lines
7.7 KiB
Go
272 lines
7.7 KiB
Go
|
// -*- Mode: Go; indent-tabs-mode: t -*-
|
|||
|
|
//
|
|||
|
|
// Copyright (C) 2018 Canonical Ltd
|
|||
|
|
// Copyright (C) 2018-2019 IOTech Ltd
|
|||
|
|
// Copyright (C) 2021 Jiangxing Intelligence Ltd
|
|||
|
|
// Copyright (C) 2022 HCL Technologies Ltd
|
|||
|
|
//
|
|||
|
|
// SPDX-License-Identifier: Apache-2.0
|
|||
|
|
|
|||
|
|
// Package driver this package provides an UART implementation of
|
|||
|
|
// ProtocolDriver interface.
|
|||
|
|
//
|
|||
|
|
// CONTRIBUTORS COMPANY
|
|||
|
|
//===============================================================
|
|||
|
|
// 1. Sathya Durai HCL Technologies
|
|||
|
|
// 2. Sudhamani Bijivemula HCL Technologies
|
|||
|
|
// 3. Vediyappan Villali HCL Technologies
|
|||
|
|
// 4. Vijay Annamalaisamy HCL Technologies
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
|
|||
|
|
package driver
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"device-ble/cmd/config"
|
|||
|
|
internalif "device-ble/internal/interfaces"
|
|||
|
|
"log"
|
|||
|
|
|
|||
|
|
"device-ble/pkg/ble"
|
|||
|
|
"device-ble/pkg/dataparse"
|
|||
|
|
"device-ble/pkg/mqttbus"
|
|||
|
|
"device-ble/pkg/uart"
|
|||
|
|
errorDefault "errors"
|
|||
|
|
"fmt"
|
|||
|
|
"sync"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/spf13/cast"
|
|||
|
|
|
|||
|
|
edgexif "github.com/edgexfoundry/device-sdk-go/v4/pkg/interfaces"
|
|||
|
|
dsModels "github.com/edgexfoundry/device-sdk-go/v4/pkg/models"
|
|||
|
|
"github.com/edgexfoundry/go-mod-core-contracts/v4/clients/logger"
|
|||
|
|
"github.com/edgexfoundry/go-mod-core-contracts/v4/models"
|
|||
|
|
"github.com/edgexfoundry/go-mod-messaging/v4/pkg/types"
|
|||
|
|
"github.com/tarm/serial"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Driver BLE代理服务驱动程序，协调各组件初始化和生命周期管理。
|
|||
|
|
type Driver struct {
|
|||
|
|
// EdgeX SDK相关
|
|||
|
|
sdk edgexif.DeviceServiceSDK
|
|||
|
|
logger logger.LoggingClient
|
|||
|
|
asyncCh chan<- *dsModels.AsyncValues
|
|||
|
|
deviceCh chan<- []dsModels.DiscoveredDevice
|
|||
|
|
|
|||
|
|
// 核心组件
|
|||
|
|
BleController internalif.BLEController
|
|||
|
|
MessageBusClient internalif.MessageBusClient
|
|||
|
|
|
|||
|
|
// 业务服务
|
|||
|
|
CommandService *CommandService
|
|||
|
|
AgentService *AgentService
|
|||
|
|
|
|||
|
|
// 内部状态
|
|||
|
|
commandResponses sync.Map
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Initialize 初始化设备服务
|
|||
|
|
/*
|
|||
|
|
在 Initialize 方法执行时：
|
|||
|
|
设备配置文件（如 devices.yaml）尚未加载，设备实例还未添加到 EdgeX 中，只能访问服务级别的配置，不能访问具体的设备配置
|
|||
|
|
*/
|
|||
|
|
func (d *Driver) Initialize(sdk edgexif.DeviceServiceSDK) error {
|
|||
|
|
d.sdk = sdk
|
|||
|
|
d.logger = sdk.LoggingClient()
|
|||
|
|
d.asyncCh = sdk.AsyncValuesChannel()
|
|||
|
|
d.deviceCh = sdk.DiscoveredDeviceChannel()
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// HandleUpAgentCallback 处理上行透明代理数据回调。
|
|||
|
|
func (d *Driver) HandleUpAgentCallback(data string) {
|
|||
|
|
if d.AgentService != nil {
|
|||
|
|
d.AgentService.HandleAgentData(data)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// HandleUpCommandCallback 处理上行命令回调。
|
|||
|
|
func (d *Driver) HandleUpCommandCallback(cmd string) {
|
|||
|
|
if d.CommandService != nil {
|
|||
|
|
d.CommandService.HandleCommand(cmd)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Start 启动设备服务。
|
|||
|
|
func (d *Driver) Start() error {
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// agentDown 处理蓝牙透明代理下行数据。
|
|||
|
|
func (d *Driver) agentDown(topic string, envelope types.MessageEnvelope) error {
|
|||
|
|
if err := dataparse.SendToBlE(d.BleController, envelope); err != nil {
|
|||
|
|
d.logger.Infof("【透明代理（↓）】下行数据发送失败 ❌, err: %v", err) // 记录成功日志
|
|||
|
|
}
|
|||
|
|
d.logger.Infof("【透明代理（↓）】下行数据发送成功 ✔, err: %v") // 记录成功日志
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Discover 触发协议特定的设备发现。
|
|||
|
|
func (s *Driver) Discover() error {
|
|||
|
|
return fmt.Errorf("Discover function is yet to be implemented!")
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ValidateDevice 校验设备协议属性。
|
|||
|
|
func (s *Driver) ValidateDevice(device models.Device) error {
|
|||
|
|
|
|||
|
|
protocol, ok := device.Protocols["UART"]
|
|||
|
|
if !ok {
|
|||
|
|
return errorDefault.New("Missing 'UART' protocols")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
deviceLocation, ok := protocol["deviceLocation"]
|
|||
|
|
if !ok {
|
|||
|
|
return errorDefault.New("Missing 'deviceLocation' information")
|
|||
|
|
} else if deviceLocation == "" {
|
|||
|
|
return errorDefault.New("deviceLocation must not empty")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
baudRate, ok := protocol["baudRate"]
|
|||
|
|
if !ok {
|
|||
|
|
return errorDefault.New("Missing 'baudRate' information")
|
|||
|
|
} else if baudRate == "" {
|
|||
|
|
return errorDefault.New("baudRate must not empty")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Stop 停止设备服务，释放资源。
|
|||
|
|
func (d *Driver) Stop(force bool) error {
|
|||
|
|
if d.logger != nil {
|
|||
|
|
d.logger.Infof("正在停止BLE代理服务 (force=%v)", force)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 关闭MessageBus客户端
|
|||
|
|
if d.MessageBusClient != nil {
|
|||
|
|
if closer, ok := d.MessageBusClient.(interface{ Disconnect() error }); ok {
|
|||
|
|
err := closer.Disconnect()
|
|||
|
|
if err != nil {
|
|||
|
|
d.logger.Errorf("MessageBus客户端关闭失败: %v", err)
|
|||
|
|
} else {
|
|||
|
|
d.logger.Debug("MessageBus客户端已断开连接")
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
d.logger.Debug("MessageBus客户端不支持关闭操作")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 关闭BLE控制器和串口
|
|||
|
|
if d.BleController != nil {
|
|||
|
|
if closer, ok := d.BleController.(interface{ Close() error }); ok {
|
|||
|
|
err := closer.Close()
|
|||
|
|
if err != nil {
|
|||
|
|
d.logger.Errorf("BLE控制器关闭失败: %v", err)
|
|||
|
|
} else {
|
|||
|
|
d.logger.Debug("BLE控制器已关闭")
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
d.logger.Debug("BLE控制器不支持关闭操作")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if d.logger != nil {
|
|||
|
|
d.logger.Info("BLE代理服务已停止")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AddDevice 添加设备回调函数。
|
|||
|
|
func (d *Driver) AddDevice(deviceName string, protocols map[string]models.ProtocolProperties, adminState models.AdminState) error {
|
|||
|
|
d.logger.Debugf("新设备已添加: %s", deviceName)
|
|||
|
|
|
|||
|
|
// 获取 UART 配置信息
|
|||
|
|
// 通过结构体字段访问 Protocols
|
|||
|
|
var deviceLocation string
|
|||
|
|
var baudRate int
|
|||
|
|
|
|||
|
|
for i, protocol := range protocols {
|
|||
|
|
deviceLocation = fmt.Sprintf("%v", protocol["deviceLocation"])
|
|||
|
|
baudRate, _ = cast.ToIntE(protocol["baudRate"])
|
|||
|
|
d.logger.Debugf("Driver.HandleReadCommands(): protocol = %v, device location = %v, baud rate = %v timeout=%v", i, deviceLocation, baudRate)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
serialPort, err := uart.NewSerialPort(serial.Config{
|
|||
|
|
Name: deviceLocation,
|
|||
|
|
Baud: baudRate,
|
|||
|
|
ReadTimeout: time.Duration(10) * time.Millisecond,
|
|||
|
|
}, d.logger)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatal("创建串口实例失败:", err)
|
|||
|
|
}
|
|||
|
|
// 初始化串口队列，注册Driver回调
|
|||
|
|
serialQueue := uart.NewSerialQueue(
|
|||
|
|
serialPort,
|
|||
|
|
d.logger,
|
|||
|
|
func(cmd string) { d.HandleUpCommandCallback(cmd) },
|
|||
|
|
func(data string) { d.HandleUpAgentCallback(data) },
|
|||
|
|
5,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 初始化BLE控制器
|
|||
|
|
bleController := ble.NewBLEController(serialPort, serialQueue, d.logger)
|
|||
|
|
|
|||
|
|
// 初始化BLE设备为外围设备模式
|
|||
|
|
if err := bleController.InitializeAsPeripheral(); err != nil {
|
|||
|
|
log.Fatal("BLE设备初始化失败:", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 加载自定义MQTT配置
|
|||
|
|
cfg, err := config.LoadConfig("./res/configuration.yaml")
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatal("MessageBusClient 获取自定义配置失败:", err)
|
|||
|
|
}
|
|||
|
|
d.logger.Debugf("自定义Mqtt服务配置: %v\n", cfg)
|
|||
|
|
// 初始化消息总线
|
|||
|
|
mqttClient, err := mqttbus.NewEdgexMessageBusClient(cfg, d.logger)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Fatal("MessageBusClient 创建失败:", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 初始化业务服务
|
|||
|
|
commandService := &CommandService{
|
|||
|
|
Logger: d.logger,
|
|||
|
|
MessageBusClient: mqttClient,
|
|||
|
|
BleController: bleController,
|
|||
|
|
}
|
|||
|
|
agentService := &AgentService{
|
|||
|
|
Logger: d.logger,
|
|||
|
|
MessageBusClient: mqttClient,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 注入依赖到Driver
|
|||
|
|
d.BleController = bleController
|
|||
|
|
d.MessageBusClient = mqttClient
|
|||
|
|
d.CommandService = commandService
|
|||
|
|
d.AgentService = agentService
|
|||
|
|
|
|||
|
|
if d.BleController == nil || d.MessageBusClient == nil {
|
|||
|
|
return fmt.Errorf("依赖未注入")
|
|||
|
|
}
|
|||
|
|
if err := d.MessageBusClient.Subscribe(TopicBLEDown, d.agentDown); err != nil { // 转发下行数据
|
|||
|
|
d.logger.Errorf("【透明代理（↓）】 订阅下行总线失败 err: %v", err)
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// UpdateDevice 更新设备回调函数。
|
|||
|
|
func (d *Driver) UpdateDevice(deviceName string, protocols map[string]models.ProtocolProperties, adminState models.AdminState) error {
|
|||
|
|
d.logger.Debugf("设备 %s 已更新", deviceName)
|
|||
|
|
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// RemoveDevice 移除设备回调函数。
|
|||
|
|
func (d *Driver) RemoveDevice(deviceName string, protocols map[string]models.ProtocolProperties) error {
|
|||
|
|
d.logger.Debugf("设备 %s 已移除", deviceName)
|
|||
|
|
|
|||
|
|
return nil
|
|||
|
|
}
|