【JS 】前端跨页面通讯实战(LinkCom.js最完整设计说明

2023-04-16  本文已影响0人  冰麟轻武

知识点

  1. window.postMessage 跨页消息
  2. Promise 使用技巧 async/await
  3. 保护数据的三种方式和应用场景 definePropertyObject.freezeProxy
  4. es5中的继承
  5. rpc 实现原理
  6. 其他知识点

    参数归一化,解构展开惰性函数立即执行函数箭头函数闭包async/await

2. 原始设计

参考RPC框架的思路,设计一个前端跨页通讯组件 LinkCom - 领航

3. 类模块设计(UML)

4. 时序图

5. 类模块功能解释

提示:后续内容中示例代码目的是展示核心逻辑,会删除例如参数校验,文档注释等代码

5.1 帮助函数

5.1.1 createId

const id = createId();

用于创建一个随机id

const createId = (function () {
    let num = 0;
    return () => (++num) + ":" + Math.random().toString(36).substring(2);
})();

5.1.2 freezeDeep

freezeDeep(obj)

用于深度冻结对象 普通的对象冻结只能保证当前对象不被修改,但无法保证对象包含的子对象也无法修改 使用递归的方式将对象和对象中的属性都冻结掉

const freezeDeep = (function () {
    function freeze(obj) {
        Object.freeze(obj);
        for (const key in obj) {
            if (typeof obj[key] === "object" && obj[key] != null) {
                freeze(obj[key]);
            }
        }
    }
    return (obj) => {
        freeze(obj);
        return obj;
    };
})();

5.1.3 log(message, ...args)

log && log("打印日志");

用于打印日志

let log = function (message, ...args) {
    if (window.location.hostname === "localhost") {
        log = function (message, ...args) {
            if (message instanceof Error) {
                console.trace();
            }
            console.log(`${window.document.title} - ${new Date().toLocaleTimeString()} LinkCom.${message}`, ...args);
        };
        log(message, ...args);
    } else {
        log = null;
    }
};

5.2 帮助类

5.2.1 信号量 Semaphores

const ss = new Semaphores();
ss.resolve("完成");

表示一个信号量, 继承自 Promise, 自身包含 resolvereject 函数, 可以直接解决或拒绝 Promise

class Semaphores extends Promise {
    constructor(...args) {
        if (args.length === 0 || (args.length === 1 && typeof args[0] === "object")) {
            const hander = Object.assign({}, args[0]);
            super((resolve, reject) => {
                hander.resolve = resolve;
                hander.reject = reject;
            });
            Object.assign(this, hander);
            freezeDeep(this);
        } else {
            super(...args);
        }
    }
}

5.2.2 异步队列 AsyncQueue

const queue = new AsyncQueue();
queue.push(123);
const value = await queue.shift();

表示一个先进先出的异步队列; 使用push(data)存入数据, 使用异步方法async shift()取出数据, 当没有数据时, 会等待, 直到数据被push方法存入
由于 class 的私有属性兼容性不好, 希望对外隐藏list的操作,所以使用闭包代替class
首先初始化会在内部列表中放入一个等待中的信号

function AsyncQueue() {
    const list = [new Semaphores()];
    this.shift = async () => {
        const first = list[0];
        const data = await first;
        if (first === list[0]) {
            list.shift();
        }
        return data;
    }
    this.push = data => {
        const last = list[list.length - 1];
        last.resolve(data);
        list.push(new Semaphores());
    }
    this.count = () => list.length;
    this.destroy = reason => {
        const err = reason instanceof Error ? reason : new Error(reason || "queue is destroyed");
        list.forEach(x => x.reject(err));
        list.length = 0;
    }
}

参数归一化:是将不同类型的数据变为相同类型,以方便后续处理的一种技巧
例如:const err = reason instanceof Error ? reason : new Error(reason || "queue is destroyed");
以上操作就是无论 reason 现在是什么类型,都转为 Error 类型,后续操作只需要操作 Error 就可以了

5.2.3 任务板 TaskBoard

const tasks = new TaskBoard(); 
tasks.put(message.id, new Semaphores({ message, targetToken: handler.targetToken }), timeout);
tasks.finish(message.id, message.result);

表示一个任务板, 任务为 Semaphores 类型
每个存入的任务都有一个超时时间, 超时后自动删除, 并将 Semaphores 解决为错误
每个存入的任务可在超时前被完成(完成可以是解决或拒绝), 完成后自动删除
同样的为了隐藏 tasks 使用闭包

function TaskBoard() {
    const tasks = {};
    this.put = (id, task, timeout) => {
        const timeoutId = setTimeout(() => this.finish(id, new Error("timeout")), timeout);
        tasks[id] = { content: task, timeoutId };
    };
    this.take = id => {
        const data = tasks[id];
        clearTimeout(data.timeoutId);
        delete tasks[id];
        return data.content;
    };
    this.has = id => id in tasks;
    this.count = () => Object.keys(tasks).length;
    this.clear = () => Object.keys(tasks).forEach(id => this.take(id));
    this.get = id => this.has(id) ? tasks[id].content : null;
    this.finish = (id, result) => {
        const task = this.take(id);
        const exec = result instanceof Error ? task.reject : task.resolve;
        exec(result);
    };
}

5.3 消息类

要实现类似RPC的通信功能,首先要设计的就是消息类结构 消息类分为 请求(RequestMessage)和响应(ResponseMessage) 两种类型的消息会有一些共同的特点,所以抽象出消息基类(Message

5.3.1 消息基类 Message

消息作为一个DTO(Data Transfer Object 数据传输对象),创建后就不应再被更改,所以执行freezeDeep将其保护起来 考虑到继承的问题,子类也无法对其进行更改,所以在构造函数中需要传入子类的属性,由基类执行属性绑定操作

class Message {
    constructor(token, type, id, props) {
        Object.assign(this, props, { token, type, id });
        freezeDeep(this);
    }
    
    buildMessage(){
        return Object.assign({}, this);
    }
    
    static parse(message) {
        const { token, type, id } = message;
        return new Message(token, type, id, message);
    }
}

5.3.2 请求消息 RequestMessage

const message = new RequestMessage(handler.token, "call", { method, args })

请求消息,用于 主动调用方 发给 被动接收方 时的消息

class RequestMessage extends Message {
    constructor(token, command, data) {
        super(token, "request", createId(), { command, data });
    }
}

5.3.3 响应消息 ResponseMessage

const message = new ResponseMessage(handler.token, id, result);

响应消息,用于 被动接收方 收到消息并处理完成后将返回值发送给 主动调用方 时的消息

class ResponseMessage extends Message {
    constructor(token, id, result) {
        super(token, "response", id, { result });
    }
}

5.4 消息处理程序 MessageHandler

const handler = new MessageHandler(methods, url, window);
handler.onmessage = (message, event) => { };
handler.send(message);
handler.apply(data.method, handler, [...data.args, event]);

提供最基础的发送消息方法和接收消息回调
处理程序与Message不同,并不是所有属性都是冻结的 其中targetUrltargetUrltargetWindow都需要后期设置,但设置时需要验证值的类型,所以使用属性劫持defineProperty)来处理,其中targetUrl也使用了参数归一化的技巧,将StringURL都转为URL,方便后续使用;

class MessageHandler {
    constructor(methods, url, win) {
        Object.defineProperties(this, {
            token: { value: createId() },
            methods: { value: methods },
            targetUrl: {
                get: () => url && url.origin,
                set(value) {
                    if (url instanceof URL) {
                        url = value;
                    } else if (typeof value === "string") {
                        url = new URL(value);
                    } else {
                        throw new Error("url is not URL");
                    }
                }
            },
            targetWindow: {
                get: () => win,
                set(value) => win = value;
            },
        });
        if (win != null) {
            this.targetWindow = win;
        }
        if (url != null) {
            this.targetUrl = url;
        }
        window.addEventListener("message", event => {
            const message = Message.parse(event.data);
            this.onmessage(message, event);
        }, false);
    }

    send(message) {
        const data = message.buildMessage();
        this.targetWindow.postMessage(data, this.targetUrl);
    }

    targetToken = null;

    onmessage = null;

    apply(that, method, args) {
        const fn = this.methods[method];
        if (typeof fn !== "function") {
            return undefined;
        }
        const result = fn.apply(that, args);
        return result === undefined ? null : result;
    }
}

5.5 收发客户端 Client

const client = new Client(handler, timeout);
const token = await client.request("$ask", data);
const { message, event } = await client.receive(); const result = commands[message.command](message.data, event); client.response(message.id, result);
const invoker = client.build();

对消息处理程序进行封装,对外提供封装后的新函数

function Client(handler, timeout) {
    timeout = Math.max(100, parseInt(timeout) || 1000);
    const inbox = new AsyncQueue();
    const tasks = new TaskBoard();
    const beforeunload = () => handler.targetWindow && this.request("close", null, true);
    window.addEventListener("beforeunload", beforeunload, false);
    handler.onmessage = (message, event) => {
        switch (message.type) {
            case "request":
                inbox.push({ message, event });
                break;
            case "response":
                tasks.finish(message.id, message.result);
                break;
            default:
                return;
        }
    };

    this.close = function (reason) {
        window.removeEventListener("beforeunload", beforeunload);
        window.removeEventListener("message", onmessage, false);
        // 清理资源
        handler.onmessage = null;
        tasks.clear();
        inbox.destroy(reason || "主动断开");
    }

    this.request = async function (command, data, unresponse) {
        const message = new RequestMessage(handler.token, command, data);
        const content = new Semaphores({ message, targetToken: handler.targetToken });
        if (unresponse !== true) {
            tasks.put(message.id, content, timeout);
        }
        handler.send(message);
        return await content;
    }

    this.response = (id, result) => handler.send(new ResponseMessage(handler.token, id, result));

    this.builder = () => {
        const invoker = new Proxy({}, {
            get: (target, prop) => {
                // 如果是 MessageHandler 属性, 则直接返回
                if (["targetUrl", "targetWindow", "token", "targetToken", "then", "catch", "finally"].includes(prop)) {
                    return handler[prop];
                }
                return (...args) => this.request("call", { method: prop, args });
            }
        });

        const methods = handler.methods;

        const commands = { /* 命令部分代码省略... */ }

        async function execRequestMessage(message, event) {
            const { command, data } = message;
            const fn = commands[command];
            try {
                return fn(data, event);
            } catch (error) {
                return new Error(error.message || "error");
            }
        }

        // 循环接收消息
        (async () => {
            while (true) {
                const { message, event } = await inbox.shift();
                window.requestIdleCallback(async () => {
                    const result = await execRequestMessage(message, event);
                    if (result !== undefined) {
                        this.response(message.id, result);
                    }
                });
            }
        })();

        return invoker;
    }
}

5.6 领航组件主入口 LinkCom

// 主动连接
const invoker = await new LinkCom(methods).connect(win, url, {}); // 创建跨站调用器, 主动发出连接申请
const res = await invoker.hello("world");
invoker.close()

// 被动连接
const methods = {
    connect() {
        this.hello("world");
    },
    hello(msg){
        alert(msg);
        return "你也好";
    },
    close(){
        // 已断开
    }
};
new LinkCom(methods).wait();  // 创建跨站调用器, 并等待连接

主入口,用户使用组件从该对象开始

function LinkCom(methods) {
    this.connect = async function (window, url, data) {
        try {
            const handler = new MessageHandler(methods, url, window);
            const client = new Client(handler);
            const invoker = client.builder();
            const result = await client.request("connect", handler.token);
            if (result) {
                return invoker;
            }
        } catch (error) {
            throw new Error("拒绝连接:" + error.message);
        }
        throw new Error("拒绝连接");
    }
    this.wait = function () {
        const handler = new MessageHandler(methods);
        new Client(handler).builder();
        return this;
    }
}

6. 最终效果



7. 完整代码:

LinkCom.js

上一篇 下一篇

猜你喜欢

热点阅读