手写EventBus

定义

  • EventBus是一款Android下的发布/订阅事件总线机制。可以代替Intent、Handler、Broadcast等在Fragment、Activity之间传递消息。
    优点:开销小,代码优雅。将发送者和接受者解耦。
    在EventBus中,最重要的是三个map集合:
 //Class<?>:响应事件方法参数的类型, Subscription:订阅对象subscriber+SubscriberMethod
    private final Map<Class<?>,CopyOnWriteArrayList<Subscription>> subscriptionByEventType;

    //Object:订阅对象,List<Class<?>>:方法参数的类型,
    private final Map<Object,List<Class<?>>> typesSubscriber;

    //黏性事件
    private final Map<Class<?>, Object> stickyEvents;

这里主要画图分析第一个集合,也是最重要的一个,该集合主要用于事件分发,如图所示:

eventbus1.png

其余的两个集合分别用于解除绑定和黏性事件,这里就不再画图分析了,下面我们开始撸代码

核心代码

public class EventBus {

    static volatile EventBus defaultInstance;

    //Class<?>:响应事件方法参数的类型, Subscription:订阅对象subscriber+SubscriberMethod
    private final Map<Class<?>,CopyOnWriteArrayList<Subscription>> subscriptionByEventType;

    //Object:订阅对象,List<Class<?>>:方法参数的类型,
    private final Map<Object,List<Class<?>>> typesSubscriber;

    //黏性事件
    private final Map<Class<?>, Object> stickyEvents;

    private EventBus(){
        subscriptionByEventType = new HashMap<>();
        typesSubscriber = new HashMap<>();
        stickyEvents = new HashMap<>();
    }

    /**
     * 单例不解释
     * @return
     */
    public static EventBus getDefault(){
        if (defaultInstance == null){
            synchronized (EventBus.class){
                if (defaultInstance == null){
                    defaultInstance = new EventBus();
                }
            }
        }
        return defaultInstance;
    }

    /**
     * 注册订阅者
     * @param object
     */
    public void register(Object object){
        //1.解析object中的所有方法,通过Subscribe注解找到响应事件的方法,封装成SubscriberMethod的集合
        List<SubscriberMethod> subscriberMethods = new ArrayList<>();
        Class<?> clazz = object.getClass();
        Method[] methods = clazz.getDeclaredMethods();
        for (Method method : methods) {
            Subscribe subscribe = method.getAnnotation(Subscribe.class);
            if (subscribe != null){
                //获取参数类型
                Class<?>[] parameterTypes = method.getParameterTypes();
                /**
                 * method:响应事件的方法
                 * parameterTypes[0] 该方法的参数,注意这里只能有一个参数
                 * threadMode:线程
                 * priority:优先级
                 * sticky:黏性事件
                 */
                SubscriberMethod subscriberMethod = new SubscriberMethod(method,parameterTypes[0],
                        subscribe.threadMode(),subscribe.priority(),subscribe.sticky());
                subscriberMethods.add(subscriberMethod);
            }
        }
        //2.按照规则存放进subscriptionByEventType
        for (SubscriberMethod subscriberMethod : subscriberMethods) {
            subscribe(object,subscriberMethod);
        }

    }

    private void subscribe(Object object, SubscriberMethod subscriberMethod) {
        //方法参数的类型
        Class<?> eventType = subscriberMethod.eventType;
        CopyOnWriteArrayList<Subscription> subscriptions = subscriptionByEventType.get(eventType);
        if (subscriptions == null){
            subscriptions = new CopyOnWriteArrayList<>();
            subscriptionByEventType.put(eventType,subscriptions);
        }
        //判断优先级添加
        Subscription subscription = new Subscription(object, subscriberMethod);
        // 处理优先级
        int size = subscriptions.size();
        for (int i = 0; i <= size; i++) {
            if (i == size || subscriberMethod.priority > subscriptions.get(i).subscriberMethod.priority) {
                subscriptions.add(i, subscription);
                break;
            }
        }
        //处理粘性事件
        if (subscriberMethod.sticky) {
            Object stickyEvent = stickyEvents.get(eventType);
            if (stickyEvent !=null){
                executeMethod(subscription,stickyEvent);
            }

        }

        //用于注销
        List<Class<?>> eventTypes = typesSubscriber.get(object);
        if (eventTypes == null){
            eventTypes = new ArrayList<>();
            typesSubscriber.put(object,eventTypes);
        }
        if (!eventTypes.contains(eventType)){
            eventTypes.add(eventType);
        }

    }

    /**
     * 解除订阅者,防止内存泄漏
     * @param object
     */
    public void unregister(Object object){
        List<Class<?>> eventTypes = typesSubscriber.get(object);
        if (eventTypes != null){
            for (Class<?> eventType : eventTypes) {
                removeObject(eventType,object);
            }
        }

    }

    /**
     * 移除订阅者
     * @param eventType
     * @param object
     */
    private void removeObject(Class<?> eventType, Object object) {
        CopyOnWriteArrayList<Subscription> subscriptions = subscriptionByEventType.get(eventType);
        int size = subscriptions.size();
        for (int i = 0; i < size; i++) {
            Subscription subscription = subscriptions.get(i);
            if (subscription.subscriber == object){
                subscriptions.remove(subscription);
                i--;
                size--;
            }
        }
    }

    /**
     *事件分发
     * @param event
     */
    public void post(Object event){
        Class<?> eventType = event.getClass();
        CopyOnWriteArrayList<Subscription> subscriptions = subscriptionByEventType.get(eventType);
        if (subscriptions != null){
            for (Subscription subscription : subscriptions) {
                executeMethod(subscription,event);
            }
        }
    }

    /**
     * 在不同线程执行事件
     * @param subscription
     * @param event
     */
    private void executeMethod(final Subscription subscription, final Object event) {
        ThreadMode threadMode = subscription.subscriberMethod.threadMode;
        boolean isMainThread = Looper.getMainLooper() == Looper.myLooper();
        switch (threadMode){
            //post的线程
            case POSTING:
                invokeMethod(subscription,event);
                break;
                //主线程
            case MAIN:
                if (isMainThread){
                    invokeMethod(subscription,event);
                }else {
                    new Handler(Looper.getMainLooper()).post(new Runnable() {
                        @Override
                        public void run() {
                            invokeMethod(subscription,event);
                        }
                    });
                }
                break;
                //异步线程
            case ASYNC:
                AsyncPoster.enqueue(subscription,event);
                break;
                //后台线程(非主线程)
            case BACKGROUND:
                if (!isMainThread){
                    invokeMethod(subscription,event);
                }else {
                    AsyncPoster.enqueue(subscription,event);
                }
                break;

        }
    }

    /**
     * 通过反射执行方法
     * @param subscription
     * @param event
     */
    private void invokeMethod(Subscription subscription, Object event) {
        try {
            subscription.subscriberMethod.method.invoke(subscription.subscriber,event);
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        } catch (InvocationTargetException e) {
            e.printStackTrace();
        }
    }

    /**
     * 发送黏性事件
     * @param event 订阅对象
     */
    public void postSticky(Object event) {
        synchronized (stickyEvents) {
            stickyEvents.put(event.getClass(), event);
        }
        post(event);
    }

    /**
     * 移除粘性事件
     * @param event
     * @return
     */
    public boolean removeStickyEvent(Object event) {
        synchronized (stickyEvents) {
            Class<?> eventType = event.getClass();
            Object existingEvent = stickyEvents.get(eventType);
            if (event.equals(existingEvent)) {
                stickyEvents.remove(eventType);
                return true;
            } else {
                return false;
            }
        }
    }

    /**
     * 移除全部粘性事件
     */
    public void removeAllStickyEvents() {
        synchronized (stickyEvents) {
            stickyEvents.clear();
        }
    }
    /**
     * 获取粘性事件
     */
    public <T> T getStickyEvent(Class<T> eventType) {
        synchronized (stickyEvents) {
            return eventType.cast(stickyEvents.get(eventType));
        }
    }

    /**
     * 移除粘性事件
     * @param
     * @return
     */
    public <T> T removeStickyEvent(Class<T> eventType) {
        synchronized (stickyEvents) {
            return eventType.cast(stickyEvents.remove(eventType));
        }
    }
}

其中

class AsyncPoster implements Runnable {
    Subscription subscription;
    Object event;
    //Eventbus的源码中也使用了该线程池
    private final static ExecutorService executorService = Executors.newCachedThreadPool();

    public AsyncPoster(Subscription subscription, Object event){
        this.subscription = subscription;
        this.event = event;
    }

    public static void enqueue(Subscription subscription, Object event) {
        AsyncPoster asyncPoster = new AsyncPoster(subscription,event);
        // 用线程池
        executorService.execute(asyncPoster);
    }

    @Override
    public void run() {
        try {
            subscription.subscriberMethod.method.invoke(subscription.subscriber,event);
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        } catch (InvocationTargetException e) {
            e.printStackTrace();
        }
    }
}

使用

使用方法与原版EventBus一模一样

//发送普通事件
 EventBus.getDefault().post("text");
//发送粘性事件
 EventBus.getDefault().postSticky("hi");
//注册订阅
EventBus.getDefault().register(this);
// 解绑
EventBus.getDefault().unregister(this);

 /**
     * threadMode 执行的线程方式
     * priority 执行的优先级
     * sticky 粘性事件
     */
    @Subscribe(threadMode = ThreadMode.MAIN,priority = 50,sticky = true)
    public void test1(String msg){
        // 如果有一个地方用 EventBus 发送一个 String 对象,那么这个方法就会被执行
        Log.e("TAG","msg1 = "+msg);
        mTv.setText(msg);
    }

    /**
     * threadMode 执行的线程方式
     * priority 执行的优先级,值越大优先级越高
     * sticky 粘性事件
     */
    @Subscribe(threadMode = ThreadMode.MAIN,priority = 100,sticky = true)
    public void test2(String msg){
        // 如果有一个地方用 EventBus 发送一个 String 对象,那么这个方法就会被执行
        Log.e("TAG","msg2 = "+msg);
        mTv.setText(msg);
    }
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,911评论 5 460
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 82,014评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 142,129评论 0 320
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,283评论 1 264
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,159评论 4 357
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,161评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,565评论 3 382
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,251评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,531评论 1 292
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,619评论 2 310
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,383评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,255评论 3 313
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,624评论 3 299
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,916评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,199评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,553评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,756评论 2 335