Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: 优化mqtt client代码 #3380

Merged
merged 1 commit into from
Jan 19, 2025
Merged

feat: 优化mqtt client代码 #3380

merged 1 commit into from
Jan 19, 2025

Conversation

KouShenhai
Copy link
Owner

@KouShenhai KouShenhai commented Jan 19, 2025

Summary by Sourcery

重构 MQTT 客户端代码,以防止创建多个客户端实例并添加打开和关闭事件。

增强功能:

  • 防止创建多个 MQTT 客户端实例。
  • 为 MQTT 客户端添加打开和关闭事件。

测试:

  • 为 EventBus 功能添加单元测试。
Original summary in English

Summary by Sourcery

Refactor the MQTT client code to prevent multiple instances of the client from being created and add open and close events.

Enhancements:

  • Prevent multiple MQTT client instances.
  • Add open and close events for MQTT clients.

Tests:

  • Add unit tests for the EventBus functionality.

Summary by CodeRabbit

  • New Features

    • Added event handling for MQTT client connections and disconnections
    • Introduced new event types for open and close events in MQTT client management
  • Refactor

    • Updated several utility classes to be final with private constructors
    • Improved null checking and error handling in MQTT client operations
  • Tests

    • Added new test classes for event bus and MQTT-related functionality

Copy link

coderabbitai bot commented Jan 19, 2025

Caution

Review failed

The pull request is closed.

Walkthrough

The pull request introduces several modifications across multiple modules of the Laokou project. The changes primarily focus on enhancing event handling mechanisms, improving MQTT client management, and refactoring certain utility classes. Key modifications include adding event classes for MQTT client connections, updating event handling methods, implementing null checks in MQTT client operations, and converting some utility classes to final classes with private constructors to prevent instantiation.

Changes

File Path Change Summary
laokou-common/laokou-common-core/src/main/java/org/laokou/common/core/event/EventBus.java Made EventBus a final class with a private constructor
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/config/MqttClient.java Added null checks and new methods to publish open/close events
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/config/MqttClientManager.java Updated executor to static, modified add method, added event publishing methods
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/handler/EventHandler.java Added methods for handling open, close, subscribe, and unsubscribe events
laokou-service/laokou-admin/laokou-admin-infrastructure/src/main/java/org/laokou/admin/loginLog/convertor/LoginLogConvertor.java Made LoginLogConvertor a final class with a private constructor
laokou-service/laokou-admin/laokou-admin-infrastructure/src/main/java/org/laokou/admin/noticeLog/convertor/NoticeLogConvertor.java Made NoticeLogConvertor a final class with a private constructor

Sequence Diagram

sequenceDiagram
    participant Client
    participant MqttClientManager
    participant EventBus
    participant EventHandler
    
    Client->>MqttClientManager: Open Connection
    MqttClientManager->>EventBus: Publish OpenEvent
    EventBus->>EventHandler: Trigger onOpenEvent
    EventHandler->>MqttClientManager: Open Client Connection
    
    Client->>MqttClientManager: Close Connection
    MqttClientManager->>EventBus: Publish CloseEvent
    EventBus->>EventHandler: Trigger onCloseEvent
    EventHandler->>MqttClientManager: Close Client Connection
Loading

Poem

🐰 Hopping through code with glee,
Events now flow more smoothly and free
MQTT clients dance with grace
Null checks guard each coding space
Rabbits rejoice in refactored delight! 🌟


📜 Recent review details

Configuration used: CodeRabbit UI
Review profile: CHILL
Plan: Pro

📥 Commits

Reviewing files that changed from the base of the PR and between 706d1c1 and 8fe7a7d.

📒 Files selected for processing (12)
  • laokou-common/laokou-common-core/src/main/java/org/laokou/common/core/event/EventBus.java (1 hunks)
  • laokou-common/laokou-common-core/src/test/java/org/laokou/common/core/EventBusTest.java (1 hunks)
  • laokou-common/laokou-common-core/src/test/java/org/laokou/common/core/TestEvent.java (1 hunks)
  • laokou-common/laokou-common-core/src/test/java/org/laokou/common/core/TestEventListener.java (1 hunks)
  • laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/config/MqttClient.java (4 hunks)
  • laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/config/MqttClientManager.java (3 hunks)
  • laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/handler/EventHandler.java (2 hunks)
  • laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/handler/event/CloseEvent.java (1 hunks)
  • laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/handler/event/OpenEvent.java (1 hunks)
  • laokou-service/laokou-admin/laokou-admin-infrastructure/src/main/java/org/laokou/admin/loginLog/convertor/LoginLogConvertor.java (1 hunks)
  • laokou-service/laokou-admin/laokou-admin-infrastructure/src/main/java/org/laokou/admin/noticeLog/convertor/NoticeLogConvertor.java (1 hunks)
  • laokou-service/laokou-iot/laokou-iot-domain/pom.xml (1 hunks)

Finishing Touches

  • 📝 Generate Docstrings (Beta)

Thank you for using CodeRabbit. We offer it for free to the OSS community and would appreciate your support in helping us grow. If you find it useful, would you consider giving us a shout-out on your favorite social media?

❤️ Share
🪧 Tips

Chat

There are 3 ways to chat with CodeRabbit:

  • Review comments: Directly reply to a review comment made by CodeRabbit. Example:
    • I pushed a fix in commit <commit_id>, please review it.
    • Generate unit testing code for this file.
    • Open a follow-up GitHub issue for this discussion.
  • Files and specific lines of code (under the "Files changed" tab): Tag @coderabbitai in a new review comment at the desired location with your query. Examples:
    • @coderabbitai generate unit testing code for this file.
    • @coderabbitai modularize this function.
  • PR comments: Tag @coderabbitai in a new PR comment to ask questions about the PR branch. For the best results, please provide a very specific query, as very limited context is provided in this mode. Examples:
    • @coderabbitai gather interesting stats about this repository and render them as a table. Additionally, render a pie chart showing the language distribution in the codebase.
    • @coderabbitai read src/utils.ts and generate unit testing code.
    • @coderabbitai read the files in the src/scheduler package and generate a class diagram using mermaid and a README in the markdown format.
    • @coderabbitai help me debug CodeRabbit configuration file.

Note: Be mindful of the bot's finite context window. It's strongly recommended to break down tasks such as reading entire modules into smaller chunks. For a focused discussion, use review comments to chat about specific files and their changes, instead of using the PR comments.

CodeRabbit Commands (Invoked using PR comments)

  • @coderabbitai pause to pause the reviews on a PR.
  • @coderabbitai resume to resume the paused reviews.
  • @coderabbitai review to trigger an incremental review. This is useful when automatic reviews are disabled for the repository.
  • @coderabbitai full review to do a full review from scratch and review all the files again.
  • @coderabbitai summary to regenerate the summary of the PR.
  • @coderabbitai generate docstrings to generate docstrings for this PR. (Beta)
  • @coderabbitai resolve resolve all the CodeRabbit review comments.
  • @coderabbitai configuration to show the current CodeRabbit configuration for the repository.
  • @coderabbitai help to get help.

Other keywords and placeholders

  • Add @coderabbitai ignore anywhere in the PR description to prevent this PR from being reviewed.
  • Add @coderabbitai summary to generate the high-level summary at a specific location in the PR description.
  • Add @coderabbitai anywhere in the PR title to generate the title automatically.

CodeRabbit Configuration File (.coderabbit.yaml)

  • You can programmatically configure CodeRabbit by adding a .coderabbit.yaml file to the root of your repository.
  • Please see the configuration documentation for more information.
  • If your editor has YAML language server enabled, you can add the path at the top of this file to enable auto-completion and validation: # yaml-language-server: $schema=https://coderabbit.ai/integrations/schema.v2.json

Documentation and Community

  • Visit our Documentation for detailed information on how to use CodeRabbit.
  • Join our Discord Community to get help, request features, and share feedback.
  • Follow us on X/Twitter for updates and announcements.

@KouShenhai KouShenhai merged commit f8eff8d into master Jan 19, 2025
8 of 13 checks passed
Copy link

sourcery-ai bot commented Jan 19, 2025

审阅者指南 by Sourcery

这个拉取请求通过确保客户端仅初始化一次,并且仅在客户端非空时执行订阅、取消订阅和发布操作来优化 MQTT 客户端代码。它还为打开和关闭事件添加了事件发布,并进行了一些小的代码改进。

MQTT 客户端生命周期序列图

sequenceDiagram
    participant App
    participant MqttClientManager
    participant MqttClient
    participant EventBus
    participant EventHandler

    App->>MqttClientManager: add(clientId, properties, balancer)
    MqttClientManager->>MqttClient: new MqttClient()

    App->>MqttClientManager: open(clientId)
    MqttClientManager->>MqttClient: open()
    Note over MqttClient: 检查客户端是否为空
    MqttClient->>MqttClient: 如果为空则初始化客户端
    MqttClient->>EventBus: publish(OpenEvent)
    EventBus->>EventHandler: onOpenEvent()

    App->>MqttClientManager: subscribe(clientId, topics, qos)
    MqttClientManager->>MqttClient: subscribe()
    Note over MqttClient: 仅在客户端存在时订阅

    App->>MqttClientManager: close(clientId)
    MqttClientManager->>MqttClient: close()
    MqttClient->>EventBus: publish(CloseEvent)
    EventBus->>EventHandler: onCloseEvent()
Loading

MQTT 客户端变更类图

classDiagram
    class MqttClient {
        -MqttAsyncClient client
        +open()
        +subscribe(String[] topics, int[] qos)
        +unsubscribe(String[] topics)
        +publish(String topic, byte[] payload, int qos)
        +publishOpenEvent(String clientId)
        +publishCloseEvent(String clientId)
    }

    class MqttClientManager {
        -Map<String, MqttClient> MQTT_CLIENT_MAP
        -ScheduledExecutorService EXECUTOR
        +get(String clientId)
        +remove(String clientId)
        +add(String clientId, MqttBrokerProperties properties, MqttClientLoadBalancer balancer)
        +open(String clientId)
        +close(String clientId)
        +publishOpenEvent(String clientId)
        +publishCloseEvent(String clientId)
    }

    class EventHandler {
        +onSubscribeEvent(SubscribeEvent event)
        +onUnsubscribeEvent(UnsubscribeEvent event)
        +onOpenEvent(OpenEvent event)
        +onCloseEvent(CloseEvent event)
    }

    class OpenEvent {
        -String clientId
    }

    class CloseEvent {
        -String clientId
    }

    MqttClientManager --> MqttClient
    EventHandler --> MqttClientManager
    OpenEvent --|> ApplicationEvent
    CloseEvent --|> ApplicationEvent
Loading

文件级变更

变更 详情 文件
确保 MQTT 客户端仅初始化一次。
  • 在初始化 MQTT 客户端之前添加空值检查。
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/config/MqttClient.java
确保仅在客户端非空时执行订阅、取消订阅和发布操作。
  • 在执行订阅操作之前添加空值检查。
  • 在执行取消订阅操作之前添加空值检查。
  • 在执行发布操作之前添加空值检查。
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/config/MqttClient.java
为打开和关闭事件添加事件发布。
  • 在 MqttClient 中添加 publishOpenEvent 方法。
  • 在 MqttClient 中添加 publishCloseEvent 方法。
  • 在 MqttClientManager 中添加 publishOpenEvent 方法。
  • 在 MqttClientManager 中添加 publishCloseEvent 方法。
  • 在 EventHandler 中添加 onOpenEvent 方法。
  • 在 EventHandler 中添加 onCloseEvent 方法。
  • 创建 CloseEvent 类。
  • 创建 OpenEvent 类。
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/config/MqttClient.java
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/config/MqttClientManager.java
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/handler/EventHandler.java
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/handler/event/CloseEvent.java
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/handler/event/OpenEvent.java
重构 MqttClientManager 以使用静态执行器并创建 MqttClient 实例。
  • 将执行器更改为静态字段。
  • 修改 add 方法以创建 MqttClient 实例。
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/config/MqttClientManager.java
小的代码改进。
  • 将 NoticeLogConvertor 设为 final 类并添加私有构造函数。
  • 将 EventBus 设为 final 类并添加私有构造函数。
  • 将 LoginLogConvertor 设为 final 类并添加私有构造函数。
  • 为 EventBus 添加测试。
  • 添加 TestEvent 类。
  • 添加 TestEventListener 类。
laokou-service/laokou-admin/laokou-admin-infrastructure/src/main/java/org/laokou/admin/noticeLog/convertor/NoticeLogConvertor.java
laokou-common/laokou-common-core/src/main/java/org/laokou/common/core/event/EventBus.java
laokou-service/laokou-admin/laokou-admin-infrastructure/src/main/java/org/laokou/admin/loginLog/convertor/LoginLogConvertor.java
laokou-service/laokou-iot/laokou-iot-domain/pom.xml
laokou-common/laokou-common-core/src/test/java/org/laokou/common/core/EventBusTest.java
laokou-common/laokou-common-core/src/test/java/org/laokou/common/core/TestEvent.java
laokou-common/laokou-common-core/src/test/java/org/laokou/common/core/TestEventListener.java

提示和命令

与 Sourcery 交互

  • 触发新的审阅: 在拉取请求中评论 @sourcery-ai review
  • 继续讨论: 直接回复 Sourcery 的审阅评论。
  • 从审阅评论生成 GitHub 问题: 通过回复评论要求 Sourcery 创建问题。
  • 生成拉取请求标题: 在拉取请求标题的任何位置写 @sourcery-ai 以随时生成标题。
  • 生成拉取请求摘要: 在拉取请求正文的任何位置写 @sourcery-ai summary 以随时生成 PR 摘要。您还可以使用此命令指定摘要应插入的位置。

自定义您的体验

访问您的仪表板以:

  • 启用或禁用审阅功能,如 Sourcery 生成的拉取请求摘要、审阅者指南等。
  • 更改审阅语言。
  • 添加、删除或编辑自定义审阅说明。
  • 调整其他审阅设置。

获取帮助

Original review guide in English

Reviewer's Guide by Sourcery

This pull request optimizes the MQTT client code by ensuring that the client is only initialized once, and that subscribe, unsubscribe and publish operations are only performed if the client is not null. It also adds event publishing for open and close events, and makes some minor code improvements.

Sequence diagram for MQTT client lifecycle

sequenceDiagram
    participant App
    participant MqttClientManager
    participant MqttClient
    participant EventBus
    participant EventHandler

    App->>MqttClientManager: add(clientId, properties, balancer)
    MqttClientManager->>MqttClient: new MqttClient()

    App->>MqttClientManager: open(clientId)
    MqttClientManager->>MqttClient: open()
    Note over MqttClient: Check if client is null
    MqttClient->>MqttClient: Initialize client if null
    MqttClient->>EventBus: publish(OpenEvent)
    EventBus->>EventHandler: onOpenEvent()

    App->>MqttClientManager: subscribe(clientId, topics, qos)
    MqttClientManager->>MqttClient: subscribe()
    Note over MqttClient: Only subscribe if client exists

    App->>MqttClientManager: close(clientId)
    MqttClientManager->>MqttClient: close()
    MqttClient->>EventBus: publish(CloseEvent)
    EventBus->>EventHandler: onCloseEvent()
Loading

Class diagram for MQTT client changes

classDiagram
    class MqttClient {
        -MqttAsyncClient client
        +open()
        +subscribe(String[] topics, int[] qos)
        +unsubscribe(String[] topics)
        +publish(String topic, byte[] payload, int qos)
        +publishOpenEvent(String clientId)
        +publishCloseEvent(String clientId)
    }

    class MqttClientManager {
        -Map<String, MqttClient> MQTT_CLIENT_MAP
        -ScheduledExecutorService EXECUTOR
        +get(String clientId)
        +remove(String clientId)
        +add(String clientId, MqttBrokerProperties properties, MqttClientLoadBalancer balancer)
        +open(String clientId)
        +close(String clientId)
        +publishOpenEvent(String clientId)
        +publishCloseEvent(String clientId)
    }

    class EventHandler {
        +onSubscribeEvent(SubscribeEvent event)
        +onUnsubscribeEvent(UnsubscribeEvent event)
        +onOpenEvent(OpenEvent event)
        +onCloseEvent(CloseEvent event)
    }

    class OpenEvent {
        -String clientId
    }

    class CloseEvent {
        -String clientId
    }

    MqttClientManager --> MqttClient
    EventHandler --> MqttClientManager
    OpenEvent --|> ApplicationEvent
    CloseEvent --|> ApplicationEvent
Loading

File-Level Changes

Change Details Files
Ensured that the MQTT client is initialized only once.
  • Added a null check before initializing the MQTT client.
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/config/MqttClient.java
Ensured that subscribe, unsubscribe and publish operations are only performed if the client is not null.
  • Added a null check before performing subscribe operations.
  • Added a null check before performing unsubscribe operations.
  • Added a null check before performing publish operations.
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/config/MqttClient.java
Added event publishing for open and close events.
  • Added publishOpenEvent method to MqttClient.
  • Added publishCloseEvent method to MqttClient.
  • Added publishOpenEvent method to MqttClientManager.
  • Added publishCloseEvent method to MqttClientManager.
  • Added onOpenEvent method to EventHandler.
  • Added onCloseEvent method to EventHandler.
  • Created CloseEvent class.
  • Created OpenEvent class.
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/config/MqttClient.java
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/config/MqttClientManager.java
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/handler/EventHandler.java
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/handler/event/CloseEvent.java
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/handler/event/OpenEvent.java
Refactored MqttClientManager to use a static executor and to create MqttClient instances.
  • Changed the executor to a static field.
  • Modified the add method to create MqttClient instances.
laokou-common/laokou-common-mqtt/src/main/java/org/laokou/common/mqtt/config/MqttClientManager.java
Minor code improvements.
  • Made NoticeLogConvertor a final class and added a private constructor.
  • Made EventBus a final class and added a private constructor.
  • Made LoginLogConvertor a final class and added a private constructor.
  • Added a test for EventBus.
  • Added a TestEvent class.
  • Added a TestEventListener class.
laokou-service/laokou-admin/laokou-admin-infrastructure/src/main/java/org/laokou/admin/noticeLog/convertor/NoticeLogConvertor.java
laokou-common/laokou-common-core/src/main/java/org/laokou/common/core/event/EventBus.java
laokou-service/laokou-admin/laokou-admin-infrastructure/src/main/java/org/laokou/admin/loginLog/convertor/LoginLogConvertor.java
laokou-service/laokou-iot/laokou-iot-domain/pom.xml
laokou-common/laokou-common-core/src/test/java/org/laokou/common/core/EventBusTest.java
laokou-common/laokou-common-core/src/test/java/org/laokou/common/core/TestEvent.java
laokou-common/laokou-common-core/src/test/java/org/laokou/common/core/TestEventListener.java

Tips and commands

Interacting with Sourcery

  • Trigger a new review: Comment @sourcery-ai review on the pull request.
  • Continue discussions: Reply directly to Sourcery's review comments.
  • Generate a GitHub issue from a review comment: Ask Sourcery to create an
    issue from a review comment by replying to it.
  • Generate a pull request title: Write @sourcery-ai anywhere in the pull
    request title to generate a title at any time.
  • Generate a pull request summary: Write @sourcery-ai summary anywhere in
    the pull request body to generate a PR summary at any time. You can also use
    this command to specify where the summary should be inserted.

Customizing Your Experience

Access your dashboard to:

  • Enable or disable review features such as the Sourcery-generated pull request
    summary, the reviewer's guide, and others.
  • Change the review language.
  • Add, remove or edit custom review instructions.
  • Adjust other review settings.

Getting Help

Copy link

@sourcery-ai sourcery-ai bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@KouShenhai - 我已经审查了你的更改,看起来非常棒!

以下是我在审查期间查看的内容
  • 🟢 一般性问题:一切看起来都很好
  • 🟢 安全性:一切看起来都很好
  • 🟢 测试:一切看起来都很好
  • 🟢 复杂性:一切看起来都很好
  • 🟢 文档:一切看起来都很好

Sourcery 对开源项目是免费的 - 如果你喜欢我们的评论,请考虑分享 ✨
帮助我变得更有用!请在每条评论上点击 👍 或 👎,我将使用这些反馈来改进你的评论。
Original comment in English

Hey @KouShenhai - I've reviewed your changes and they look great!

Here's what I looked at during the review
  • 🟢 General issues: all looks good
  • 🟢 Security: all looks good
  • 🟢 Testing: all looks good
  • 🟢 Complexity: all looks good
  • 🟢 Documentation: all looks good

Sourcery is free for open source - if you like our reviews please consider sharing them ✨
Help me be more useful! Please click 👍 or 👎 on each comment and I'll use the feedback to improve your reviews.

Copy link

codecov bot commented Jan 19, 2025

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 23.72%. Comparing base (706d1c1) to head (8fe7a7d).
Report is 2 commits behind head on master.

Additional details and impacted files
@@             Coverage Diff              @@
##             master    #3380      +/-   ##
============================================
+ Coverage     23.51%   23.72%   +0.20%     
- Complexity      199      201       +2     
============================================
  Files           158      158              
  Lines          2092     2091       -1     
  Branches        142      142              
============================================
+ Hits            492      496       +4     
+ Misses         1538     1533       -5     
  Partials         62       62              

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant