东郭欣然 发表于 2025-5-31 23:11:49

基于DolphinScheduler抽取通用EventBus组件:支持延迟与事件驱动

一、思路来源

虽然guava中的eventbus已经很方便了,但是还是想要实现一个更为方便,同时支持延迟事件、同时带eventbus的组件。在Apache DolphinScheduler项目中,有一个eventbus的组件,这个组件写得挺好的,想着用在业务系统上,因此自己抽取了一下,拿到业务系统中来用。话不多说,我们把它抽取出来吧,同时进行demo的运行。还是要感谢Apache DolphinScheduler的开源,让这个很简单,但是很高效的组件能够让我们便捷地使用。
二、具体实现过程

首先是定义事件接口:
public interface IEvent {
}针对事件接口,我们抽象出共性方法接口:延迟时间和过期时间。
public abstractclass AbstractDelayEvent implements IEvent, Delayed {
    privatefinallong delayTime;
    privatefinallong expireTime;


    public long getDelayTime() {
      return delayTime;
    }

    public long getExpireTime() {
      return expireTime;
    }

    public AbstractDelayEvent(long delayTime) {
      this.delayTime = delayTime;
      this.expireTime = System.currentTimeMillis() + delayTime;
    }

    @Override
    public long getDelay(TimeUnit unit) {
      long diff = expireTime - System.currentTimeMillis();
      return unit.convert(diff, TimeUnit.MILLISECONDS);
    }

    @Override
    public int compareTo(Delayed o) {
      if (this.expireTime < ((AbstractDelayEvent) o).expireTime) {
            return -1;
      }
      if (this.expireTime > ((AbstractDelayEvent) o).expireTime) {
            return1;
      }
      return0;
    }
}主要的信息:
定义eventbus中,我们需要使用的方法:
public interface IEventBus<T extends IEvent> {

    void publish(T event);

    Optional<T> poll() throws InterruptedException;

    Optional<T> peek();

    Optional<T> remove();

    boolean isEmpty();

    int size();
}可以看到主要是:发布事件、消费、移除、删除、判断当前的事件是否为空,以及事件大小等方法。其中最重要的方法为发布事件和消费处理事件方法。
针对当前的事件bus接口进行抽象,抽取出共性方法,方便复用:
public abstractclass AbstractDelayEventBus<T extends AbstractDelayEvent> implements IEventBus<T> {

    protectedfinal DelayQueue<T> delayEventQueue = new DelayQueue<>();

    @Override
    public void publish(T event) {
      delayEventQueue.put(event);
    }

    @Override
    public Optional<T> poll() throws InterruptedException {
      // 使用带超时的 poll 方法,等待事件到期
      return Optional.ofNullable(delayEventQueue.poll(1000, TimeUnit.MILLISECONDS));
    }

    @Override
    public Optional<T> peek() {
      return Optional.ofNullable(delayEventQueue.peek());
    }

    @Override
    public Optional<T> remove() {
      return Optional.ofNullable(delayEventQueue.poll());
    }

    @Override
    public boolean isEmpty() {
      return delayEventQueue.isEmpty();
    }

    @Override
    public int size() {
      return delayEventQueue.size();
    }
}三、测试运行效果

接下来,我们使用它,来进行处理:
定义自己的延迟事件:
如果是在业务中,可以定义自己的业务数据信息事件对象
public class MyDelayEvent extends AbstractDelayEvent {
    private final String message;

    public MyDelayEvent(long delayTime, String message) {
      super(delayTime);
      this.message = message;
    }

    public String getMessage() {
      return message;
    }
}定义事件延迟事件bus
当然也可以进行自己的可定制化特性。
public class MyDelayEventBus extends AbstractDelayEventBus<MyDelayEvent> {
    // 不需要额外的修改
}进行测试:
思路:创建事件总线、发布事件,然后针对发布的事件信息,进行消费,然后等待延迟时间的到来,从而实现消费,从而进行业务的处理。
import java.util.Optional;

publicclass EventBusExample {
    public static void main(String[] args) throws InterruptedException {
      // 创建事件总线
      IEventBus<MyDelayEvent> eventBus = new MyDelayEventBus();

      // 发布单个事件
      eventBus.publish(new MyDelayEvent(100, "Single Event"));
      System.out.println("After publish, event bus size: " + eventBus.size());

      // 持续尝试消费事件
      while (true) {
            Optional<MyDelayEvent> event = eventBus.poll();
            if (event.isPresent()) {
                System.out.println("Received event: " + event.get().getMessage());
            } else {
                System.out.println("No event received within the timeout.");
                break;
            }
      }

      // 检查总线大小
      System.out.println("Event bus size: " + eventBus.size());
    }
}运行结果:
https://img2024.cnblogs.com/other/2685289/202505/2685289-20250519183134433-917479576.png
可以看到实现自己的业务逻辑还是很方便的,可以自己实现吧,这里给出的代码是可以运行的。

[*]源码地址:https://gitee.com/null_713_2407/pratice
[*]参考:
github:https://github.com/apache/dolphinscheduler
作者 | 刘亚洲
本文由 白鲸开源 提供发布支持!

来源:程序园用户自行投稿发布,如果侵权,请联系站长删除
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!
页: [1]
查看完整版本: 基于DolphinScheduler抽取通用EventBus组件:支持延迟与事件驱动