(更新)如何避免使用onActivityResult，以提高代码可读性

m0_64603743

于 2021-11-28 20:38:54 发布

阅读量153

点赞数

分类专栏：程序员文章标签：面试 android 移动开发

本文链接：https://blog.csdn.net/m0_64603743/article/details/121597280

版权

程序员专栏收录该内容

109 篇文章 0 订阅

订阅专栏

想直接看更新内容的请点此处

更新，强迫症福音，onActivityResult方法hook到了

问题

Android中，通过startActivityForResult跳转页面获取数据应该不必多说，但是这种所有获取到的结果都需要到onActivityResult中处理的方式实在令人蛋疼。

试想一下，我们敲着代码唱着歌。突然，半路上跳出一群马匪，让我们到另一个页面获取一点数据，获取后还不让在当前代码位置处理逻辑，要去onActivityResult添加一个requestCode分支处理结果，处理完才让回来，等这一切都做完回来难免就会陷入这样的思考：我是谁，我在哪，我在干什么，我刚才写到哪了……

再想一下，你跟同事的代码，跟到一个startActivityForResult，于是不耐烦地ctrl+f找到onActivityResult，发现里面充斥着大量的requestCode分支，然后突然意识到刚才没记下requestCode是什么……

分析问题

问题的根源是所有处理结果的逻辑都要放到onActivityResult中，在里面根据requestCode作不同处理。而我们渴望的是能在发起startActivityForResult的时候捎带着把获取结果后处理的逻辑也传进去，并能在内部对requestCode判断好，不用我们再判断一遍。

解决问题

尝试一（不完美方式）

新建一个OnResultManager类，用来管理获取结果的回调，下面详细说。

分析问题时说了，我们希望在发起startActivityForResult的时候就指定好处理结果的逻辑，这个简单，在OnResultManager中创建一个Callback接口，里面定义一个OnActivityResult方法，参数和Activity的OnActivityResult方法参数完全一样，在发起start的时候除了intent和requestCode，再传一个callback进去。而OnresultManager负责控制在系统的onActivityResult触发时，调用对应callback的方法。

下面是OnResultManager的全部代码。

public class OnResultManager {
private static final String TAG = “OnResultManager”;
//HashMap的key Integer为requestCode
private static WeakHashMap<Activity,HashMap<Integer,Callback>> mCallbacks = new WeakHashMap<>();
private WeakReference mActivity;

public OnResultManager(Activity activity) {
mActivity = new WeakReference(activity);
}

public void startForResult(Intent intent, int requestCode, Callback callback){
Activity activity = getActivity();
if(activity == null){
return;
}

addCallback(activity,requestCode,callback);
activity.startActivityForResult(intent,requestCode);
}

public void trigger(int requestCode, int resultCode, Intent data){
Log.d(TAG,"----------- trigger");
Activity activity = getActivity();
if(activity == null){
return;
}

Callback callback = findCallback(activity,requestCode);
if(callback != null){
callback.onActivityResult(requestCode,resultCode,data);
}
}

//获取该activity、该requestCode对应的callback
private Callback findCallback(Activity activity,int requestCode){
HashMap<Integer,Callback> map = mCallbacks.get(activity);
if(map != null){
return map.remove(requestCode);
}
return null;
}

private void addCallback(Activity activity,int requestCode,Callback callback){
HashMap<Integer,Callback> map = mCallbacks.get(activity);
if(map == null){
map = new HashMap<>();
mCallbacks.put(activity,map);
}
map.put(requestCode,callback);
}

private Activity getActivity(){
return mActivity.get();
}

public interface Callback{
void onActivityResult(int requestCode, int resultCode, Intent data);
}
}

逻辑很简单，里面持有一个mActivity，使用弱引用以防止内存泄漏，在构造器中为其赋值。还有一个static的WeakHashMap<Activity,HashMap<Integer,Callback>> mCallbacks 用来存放所有的callback，先以activity分，在各个activity中又使用hashmap存储requestCode和callback的对应关系。

在startForResult时，最终还是调用的activity的startActivityForResult，只不过在跳转页面之前，把callback存入了mCallbacks中。

而trigger方法则是根据activity和requestCode从mCallbacks中取出对应的callback，调用方法。

现在callback的存和取都搞定了，那么问题来了，什么时候触发“取”的操作呢，即trigger方法怎么触发呢？答案是在onActivityResult中调用，嫌麻烦可以在BaseActivity中调用。

使用示例：

override fun onCreate(savedInstanceState: Bundle?) {
super.onCreate(savedInstanceState)
setContentView(R.layout.activity_main)

go.setOnClickListener {
val intent = Intent(this,SecondActivity::class.java)
onResultManager.startForResult(intent,REQUEST_CODE,{requestCode: Int, resultCode: Int, data: Intent? ->
if (resultCode == Activity.RESULT_OK){
val text = data?.getStringExtra(“text”)
Toast.makeText(this,"result -> "+text,Toast.LENGTH_SHORT).show()
}else{
Toast.makeText(this,“canceled”,Toast.LENGTH_SHORT).show()
}
})
}
}

override fun onActivityResult(requestCode: Int, resultCode: Int, data: Intent?) {
super.onActivityResult(requestCode, resultCode, data)
onResultManager.trigger(requestCode,resultCode,data)
}

可是这样好蠢啊，你是不是觉得要是不用手动触发，能自动触发多好。我也是这么想的，所以有整整一天我一直在找有什么办法能hook到onActivityResult方法，最后hook了Instrumentation，也hook了AMS，但是都对这个onActivityResult无能为力，看源码发现好像是在ActivityThread中传递的，但是很不幸的是这个ActivityThread没办法hook，至少通过简单的反射和代理没办法做到（如果谁有办法hook到，恳请您能分享出来，我真的特别想知道，我不甘心啊）

更新，强迫症福音，onActivityResult方法hook到了

这里感谢一下@world_hello的提醒，十分感谢。

之前看到ActivityThread的mH的时候总想着弄个代理继承它，然后重写handleMessage方法，来获取结果信息，可是却苦于继承不了，其实我一直忽视了Handler内部的一个Callback接口，其实完全不用代理。下面详细讲解一下。

先说一下Handler，我们一般使用的时候大多是继承自Handler，然后重写handleMessage方法，在里面处理接收消息。其实Handler还有个构造函数可以接收一个Handler.Callback（不要和我们自己定义的callback搞混了啊），在Handler.Callback的handleMessage方法中处理消息。

看一下Handler源码：

/**

Handle system messages here.
*/
public void dispatchMessage(Message msg) {
if (msg.callback != null) {
handleCallback(msg);
} else {
if (mCallback != null) {
if (mCallback.handleMessage(msg)) {
return;
}
}
handleMessage(msg);
}
}

可以看到，如果mCallback不为null，就会执行mCallback的handleMessage方法。如果这个handleMessage方法的返回值为true，就会直接return，如果为false，就会继续执行Handler本身的handleMessage方法。

下面打开Activity源码，startActivityForResult方法中有这么一段

if (ar != null) {
mMainThread.sendActivityResult(
mToken, mEmbeddedID, requestCode, ar.getResultCode(),
ar.getResultData());
}

ar是ActivityResult，如果不为null，就调用mMainThread的sendActivityResult方法，这个mMainThread就是ActivityThread类型的，所以继续打开ActivityThread源码，找到这个方法，顺着这个方法一直找一直找，这里中间的方法我就不贴了，最后会找到一个叫sendMessage的方法，这个方法最后一行是

mH.sendMessage(msg);

这个mH是一个Handler的子类，所以很明显，activityresult在这个环节是通过handler传递的。

在H的handleMessage方法中有这样一个case分支

case SEND_RESULT:
Trace.traceBegin(Trace.TRACE_TAG_ACTIVITY_MANAGER, “activityDeliverResult”);
handleSendResult((ResultData)msg.obj);
Trace.traceEnd(Trace.TRACE_TAG_ACTIVITY_MANAGER);
break;

不用再多说了吧，就是这里了。它调用了handleSendResult((ResultData)msg.obj)，所以这个ResultData应该就是存放了我们需要的东西。只要能拦截到它，拿出我们需要的数据，那就算是hook到了onActivityResult了，那么怎么拿到呢？现在就要提到我们刚才提的Handler.Callback了，我们创建一个callback，并记得让handleMessage的方法返回false，以免影响mH本身的handleMessage方法的执行，然后通过反射把这个callback给mH set进去。下面需要您对反射有一点点了解。

public class HookUtil {
public static void hookActivityThreadHandler() throws Exception {
// 先获取到当前的ActivityThread对象
final Class<?> activityThreadClass = Class.forName(“android.app.ActivityThread”);
Field currentActivityThreadField = activityThreadClass.getDeclaredField(“sCurrentActivityThread”);
currentActivityThreadField.setAccessible(true);
final Object currentActivityThread = currentActivityThreadField.get(null);

// 由于ActivityThread一个进程只有一个,我们获取这个对象的mH
Field mHField = activityThreadClass.getDeclaredField(“mH”);
mHField.setAccessible(true);
Handler mH = (Handler) mHField.get(currentActivityThread);

Handler.Callback mHCallback = new Handler.Callback() {
@Override
public boolean handleMessage(Message msg) {
if(msg.what == 108){
Log.d(“hook-------”,“onActivityResult”);

try{
Object resultData = msg.obj;

Field mActivitiesField = activityThreadClass.getDeclaredField(“mActivities”);
mActivitiesField.setAccessible(true);
ArrayMap mActivities = (ArrayMap) mActivitiesField.get(currentActivityThread);

Class<?> resultDataClass = Class.forName(“android.app.ActivityThread$ResultData”);
Field tokenField = resultDataClass.getDeclaredField(“token”);
tokenField.setAccessible(true);
IBinder token = (IBinder) tokenField.get(resultData);

//r是ActivityClientRecord类型的
Object r = mActivities.get(token);
Class<?> ActivityClientRecordClass = Class.forName(“android.app.ActivityThread$ActivityClientRecord”);
Field activityField = ActivityClientRecordClass.getDeclaredField(“activity”);
activityField.setAccessible(true);
Activity activity = (Activity) activityField.get®; //至此，终于拿到activity了

Field resultsField = resultDataClass.getDeclaredField(“results”);
resultsField.setAccessible(true);
List results = (List) resultsField.get(resultData);

//ResultInfo类型
Object resultInfo = results.get(0);

Class<?> resultInfoClass = Class.forName(“android.app.ResultInfo”);
Field mRequestCodeField = resultInfoClass.getDeclaredField(“mRequestCode”);
mRequestCodeField.setAccessible(true);
int mRequestCode = (int) mRequestCodeField.get(resultInfo); //拿到requestCode

Field mResultCodeField = resultInfoClass.getDeclaredField(“mResultCode”);
mResultCodeField.setAccessible(true);
int mResultCode = (int) mResultCodeField.get(resultInfo); //拿到resultCode

Field mDataField = resultInfoClass.getDeclaredField(“mData”);
mDataField.setAccessible(true);
Intent mData = (Intent) mDataField.get(resultInfo); //拿到intent

new OnResultManager(activity).trigger(mRequestCode,mResultCode,mData);

}catch (Exception e){
e.printStackTrace();
}

}

return false;
}
};
Field mCallBackField = Handler.class.getDeclaredField(“mCallback”);
mCallBackField.setAccessible(true);

mCallBackField.set(mH, mHCallback);

}

hookActivityThreadHandler方法中就是先获取到ActivityThread对象，再通过activityThread获取到mH，然后创建了一个Handler.Callback，最后用反射把这个callback set给mH。

相比之下，Handler.Callback中的handleMessage方法看起来更长一些，但其实并不复杂，只是反射使它变得面目全非了，它主要就是为了获取我们OnResultManager的trigger方法需要的数据，一个activity，以及requestCode、resultCode、data。

msg.what==108的判断，这里108就是SEND_RESULT，我图方便直接写死了，源码里看着为108，当然通过反射动态获取SEND_RESULT更规范一点，大家别在意这些细节。

先看activity怎么获取，主要看ActivityThread的handleSendResult方法，它的第一行

private void handleSendResult(ResultData res) {
ActivityClientRecord r = mActivities.get(res.token);

这个res就是msg.obj，而获取到ActivityClientRecord之后，它有一个名为activity的field，获取的就是我们需要的activity对象了。本身很简单，只不过这短短的一行代码要用反射一点一点获取，就繁琐一些了。

另外三个数据都在一起，在ResultData里有个名为results的Field，它是一个List，里面就一个元素（可能有误，但我打断点看它一直都是一个），是ResultInfo类型，我们需要的requestCode、resultCode、data就都在这个resultInfo里面了，所以，继续反射。

都获取完了，调用

new OnResultManager(activity).trigger(mRequestCode,mResultCode,mData);

就是把我们之前手动写在onActivityResult中的那句放到这里自动触发。

现在方法写完了，还要写个自定义Application，在onCreate中调用

HookUtil.hookActivityThreadHandler()

大功告成啦！，赶紧测试一下吧，现在终于不用再在onActivityResult中手动触发啦！github上代码已经更新。

按理说我们该在OnResultManager中定义个init方法，然后在init方法中去调hookActivityThreadHandler貌似更规范一些，不过这只算个demo，就不考虑这么多了。

还有，这种通过反射hook的方法在稳定性和兼容性上都无法保证，所以这算是一种仅供娱乐的方式吧！领略一下hook的魅力。

OnResultManager项目地址

下面是world_hello实现的hook，虽然思路是一样的，但是编写的要比我的容易看懂得多，自愧不如，在此强烈推荐大家看一下。

world_hello实现的hook项目地址

第二种方式（参考RxPermissions的做法）

前段时间又来了个小项目，领导扔给我了，然后在这个项目里就把之前没用过（没错，之前总用H5开发……）的rxjava、kotlin都加进来了。有一天做动态权限处理，惊奇地发现RxPermissions居然摆脱了Activity的onRequestPermissionsResult方法！！！大家都知道，动态权限大体就是先调用requestPermissions方法，然后授权的结果要到onRequestPermissionsResult中处理，简直和startActivityForResult如出一辙。那RxPermissions是怎么做到的呢！！！

接着在前几天不太忙的时候看了下RxPermissions的源码，发现它内部持有一个Fragment，这个fragment没有视图，只负责请求权限和返回结果，相当于一个桥梁的作用，我们通过rxPermissions发起request的时候，其实并不是activity去request，而是通过这个fragment去请求，然后在fragment的onRequestPermissionsResult中把结果发送出来，如此来避开activity的onRequestPermissionsResult方法。

当时，没见过什么大场面的我差点就给跪了。