Handler tools to wrap Activity with WeakReference

RefHandler.java

package com.aliyun.wireless.vos.appstore.shortcut;

import android.app.Activity;
import android.os.Handler;
import android.os.Message;

import java.lang.ref.WeakReference;

/**
 * Created by Bruce Too
 * On 08/05/2017.
 * At 11:34
 */

public class RefHandler<T extends Activity> extends Handler {
    private WeakReference<T> activityRef;
    private MessageCallback<T> callback;

    public RefHandler(T activity, MessageCallback<T> callback) {
        activityRef = new WeakReference<T>(activity);
        this.callback = callback;
    }

    @Override
    public void handleMessage(Message msg) {
        super.handleMessage(msg);
        T activity = activityRef.get();
        if (activity != null) {
            callback.handleMessage(activity, msg);
        }
    }

    public interface MessageCallback<T extends Activity> {
        void handleMessage(T activity, Message msg);
    }

}