75 lines
2.3 KiB
Go
75 lines
2.3 KiB
Go
package mqttbus
|
||
|
||
import (
|
||
"device-ble/cmd/config"
|
||
"device-ble/pkg/messagebus"
|
||
"fmt"
|
||
"time"
|
||
|
||
"github.com/edgexfoundry/go-mod-core-contracts/v4/clients/logger"
|
||
"github.com/edgexfoundry/go-mod-messaging/v4/pkg/types"
|
||
)
|
||
|
||
// EdgexMessageBusClient 实现 interfaces.MessageBusClient
|
||
// handler 通过参数传递，logger 也通过参数传递
|
||
|
||
type EdgexMessageBusClient struct {
|
||
client *messagebus.Client
|
||
}
|
||
|
||
// NewEdgexMessageBusClient 只负责初始化和连接，不注册 handler
|
||
func NewEdgexMessageBusClient(cfg *config.MQTTUserClientConfig, logger logger.LoggingClient) (*EdgexMessageBusClient, error) {
|
||
config := messagebus.Config{
|
||
Host: cfg.MqttUserConfig.Host,
|
||
Port: cfg.MqttUserConfig.Port,
|
||
Protocol: cfg.MqttUserConfig.Protocol,
|
||
Type: "mqtt",
|
||
ClientID: cfg.MqttUserConfig.ClientID,
|
||
QoS: cfg.MqttUserConfig.QoS,
|
||
Username: cfg.MqttUserConfig.Username,
|
||
Password: cfg.MqttUserConfig.Password,
|
||
}
|
||
client, err := messagebus.NewClient(config, logger)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("创建MessageBus客户端失败: %w", err)
|
||
}
|
||
if err := client.Connect(); err != nil {
|
||
return nil, fmt.Errorf("连接MessageBus失败: %w", err)
|
||
}
|
||
return &EdgexMessageBusClient{client: client}, nil
|
||
}
|
||
|
||
// Subscribe 注册 handler
|
||
func (e *EdgexMessageBusClient) Subscribe(topic1 string, handler func(topic2 string, envelope types.MessageEnvelope) error) error {
|
||
// 装饰器：后期可以在不修改远程包的基础上自定义该函数
|
||
wrappedHandler := func(topic string, envelope types.MessageEnvelope) error {
|
||
// 加日志确认是否被调用
|
||
fmt.Printf("wrappedHandler被调用: topic=%s\n", topic)
|
||
return handler(topic, envelope)
|
||
}
|
||
if err := e.client.Subscribe(topic1, wrappedHandler); err != nil {
|
||
return fmt.Errorf("订阅主题失败: %w", err)
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (e *EdgexMessageBusClient) Publish(topic string, data interface{}) error {
|
||
return e.client.Publish(topic, data)
|
||
}
|
||
|
||
func (e *EdgexMessageBusClient) SubscribeResponse(topic string) error {
|
||
return e.client.SubscribeResponse(topic)
|
||
}
|
||
|
||
func (e *EdgexMessageBusClient) Request(topic string, data interface{}) (types.MessageEnvelope, error) {
|
||
return e.client.Request(topic, data)
|
||
}
|
||
|
||
func (e *EdgexMessageBusClient) SetTimeout(timeout time.Duration) {
|
||
e.client.SetTimeout(timeout)
|
||
}
|
||
|
||
func (e *EdgexMessageBusClient) Disconnect() error {
|
||
return e.client.Disconnect()
|
||
}
|