面试题：实现小程序平台的并发双工 rpc 通信

前几天面试的时候遇到一道面试题，还是挺考验能力的。

题目是这样的：

rpc 是 remote procedure call，远程过程调用，比如一个进程调用另一个进程的某个方法。很多平台提供的进程间通信机制都封装成了 rpc 的形式，比如 electron 的 remote 模块。

小程序是双线程机制，两个线程之间要通信，提供了 postMessage 和 addListener 的 api。现在要在两个线程都会引入的 common.js 文件里实现 rpc 方法，支持并发的 rpc 通信。

达到这样的使用效果：

const res = await rpc('method', params);

这道题是有真实应用场景的题目，比一些逻辑题和算法题更有意思一些。

实现思路

两个线程之间是用 postMessage 的 api 来传递消息的：

• 
  
• 在 rpc 方法里用 postMessage 来传递要调用的方法名和参数
  
• 在 addListener 里收到调用的时候，调用 api，然后通过 postMessage 返回结果或者错误
  

我们先实现 rpc 方法，通过 postMessage 传递消息，返回一个 promise：

function rpc(method, params) {

    postMessage(JSON.stringify({

        method,

        params 

    }));


    return new Promise((resolve, reject) => {


    });

}

这个 promise 什么时候 resolve 或者 reject 呢？ 是在 addListener 收到消息后。那就要先把它存起来，等收到消息再调用 resolve 或 reject。

为了支持并发和区分多个调用通道，我们加一个 id。

let id = 0;

function genId() {

    return ++id;

}


const channelMap = new Map();


function rpc(method, params) {

    const curId = genId();


    postMessage(JSON.stringify({

        id: curId,

        method,

        params 

    }));


    return new Promise((resolve, reject) => {

        channelMap.set(curId, {

            resolve,

            reject

        });

    });

}

这样，就通过 id 来标识了每一个远程调用请求和与它关联的 resolve、reject。

然后要处理 addListener，因为是双工的通信，也就是通信的两者都会用到这段代码，所以要区分一下是请求还是响应。

addListener((message) => {

    const { curId, method, params, res}= JSON.parse(message);

    if (res) {

        // 处理响应

    } else {

        // 处理请求

    }

});

处理请求就是调用方法，然后返回结果或者错误：

try {

    const data = global[method](...params);

    postMessage({

        id

        res: {

            data

        }

    });

} catch(e) {

    postMessage({

        id,

        res: {

            error: e.message

        }

    });

}

处理响应就是拿到并调用和 id 关联的 resolve 和 reject：

const { resolve, reject  } = channelMap.get(id);

if(res.data) {

    resolve(res.data);

} else {

    reject(res.error);

}

全部代码是这样的：

let id = 0;

function genId() {

    return ++id;

}


const channelMap = new Map();


function rpc(method, params) {

    const curId = genId();


    postMessage(JSON.stringify({

        id: curId,

        method,

        params 

    }));


    return new Promise((resolve, reject) => {

        channelMap.set(curId, {

            resolve,

            reject

        });

    });

}


addListener((message) => {

    const { id, method, params, res}= JSON.parse(message);

    if (res) {

        const { resolve, reject  } = channelMap.get(id);

        if(res.data) {

            resolve(res.data);

        } else {

            reject(res.error);

        }

    } else {

        try {

            const data = global[method](...params);

            postMessage({

                id

                res: {

                    data

                }

            });

        } catch(e) {

            postMessage({

                id,

                res: {

                    error: e.message

                }

            });

        }

    }

});

我们实现了最开始的需求：

• 
  
• 实现了 rpc 方法，返回一个 promise
  
• 支持并发的调用
  
• 两个线程都引入这个文件，支持双工的通信
  

其实主要注意的有两个点：

• 
  
• 要添加一个 id 来关联请求和响应，这在 socket 通信的时候也经常用
  
• resolve 和 reject 可以保存下来，后续再调用。这在请求取消，比如 axios 的 cancelToken 的实现上也有应用
  

这两个点的应用场景还是比较多的。

总结

rpc 是远程过程调用，是跨进程、跨线程等场景下通信的常见封装形式。面试题是小程序平台的双线程的场景，在一个公共文件里实现双工的并发的 rpc 通信。

思路文中已经讲清楚了，主要要注意的是 promise 的 resolve 和 reject 可以保存下来后续调用，通过添加 id 来标识和关联一组请求响应。