refactor: move all utility files to /common

src/common/message-unique.ts

@@ -0,0 +1,136 @@
+import { Peer } from '@/core';
+import crypto from 'crypto';
+
+export class LimitedHashTable<K, V> {
+    private keyToValue: Map<K, V> = new Map();
+    private valueToKey: Map<V, K> = new Map();
+    private maxSize: number;
+
+    constructor(maxSize: number) {
+        this.maxSize = maxSize;
+    }
+
+    resize(count: number) {
+        this.maxSize = count;
+    }
+
+    set(key: K, value: V): void {
+        this.keyToValue.set(key, value);
+        this.valueToKey.set(value, key);
+        while (this.keyToValue.size !== this.valueToKey.size) {
+            this.keyToValue.clear();
+            this.valueToKey.clear();
+        }
+        while (this.keyToValue.size > this.maxSize || this.valueToKey.size > this.maxSize) {
+            const oldestKey = this.keyToValue.keys().next().value;
+            this.valueToKey.delete(this.keyToValue.get(oldestKey)!);
+            this.keyToValue.delete(oldestKey);
+        }
+    }
+
+    getValue(key: K): V | undefined {
+        return this.keyToValue.get(key);
+    }
+
+    getKey(value: V): K | undefined {
+        return this.valueToKey.get(value);
+    }
+
+    deleteByValue(value: V): void {
+        const key = this.valueToKey.get(value);
+        if (key !== undefined) {
+            this.keyToValue.delete(key);
+            this.valueToKey.delete(value);
+        }
+    }
+
+    deleteByKey(key: K): void {
+        const value = this.keyToValue.get(key);
+        if (value !== undefined) {
+            this.keyToValue.delete(key);
+            this.valueToKey.delete(value);
+        }
+    }
+
+    getKeyList(): K[] {
+        return Array.from(this.keyToValue.keys());
+    }
+
+    //获取最近刚写入的几个值
+    getHeads(size: number): { key: K; value: V }[] | undefined {
+        const keyList = this.getKeyList();
+        if (keyList.length === 0) {
+            return undefined;
+        }
+        const result: { key: K; value: V }[] = [];
+        const listSize = Math.min(size, keyList.length);
+        for (let i = 0; i < listSize; i++) {
+            const key = keyList[listSize - i];
+            result.push({ key, value: this.keyToValue.get(key)! });
+        }
+        return result;
+    }
+}
+
+class MessageUniqueWrapper {
+    private msgDataMap: LimitedHashTable<string, number>;
+    private msgIdMap: LimitedHashTable<string, number>;
+
+    constructor(maxMap: number = 1000) {
+        this.msgIdMap = new LimitedHashTable<string, number>(maxMap);
+        this.msgDataMap = new LimitedHashTable<string, number>(maxMap);
+    }
+
+    getRecentMsgIds(Peer: Peer, size: number): string[] {
+        const heads = this.msgIdMap.getHeads(size);
+        if (!heads) {
+            return [];
+        }
+        const data = heads.map((t) => MessageUnique.getMsgIdAndPeerByShortId(t.value));
+        const ret = data.filter((t) => t?.Peer.chatType === Peer.chatType && t?.Peer.peerUid === Peer.peerUid);
+        return ret.map((t) => t?.MsgId).filter((t) => t !== undefined);
+    }
+
+    createMsg(peer: Peer, msgId: string) {
+        const key = `${msgId}|${peer.chatType}|${peer.peerUid}`;
+        const hash = crypto.createHash('md5').update(key).digest();
+        //设置第一个bit为0 保证shortId为正数
+        hash[0] &= 0x7f;
+        const shortId = hash.readInt32BE(0);
+        //减少性能损耗
+        this.msgIdMap.set(msgId, shortId);
+        this.msgDataMap.set(key, shortId);
+        return shortId;
+    }
+
+    getMsgIdAndPeerByShortId(shortId: number): { MsgId: string; Peer: Peer } | undefined {
+        const data = this.msgDataMap.getKey(shortId);
+        if (data) {
+            const [msgId, chatTypeStr, peerUid] = data.split('|');
+            const peer: Peer = {
+                chatType: parseInt(chatTypeStr),
+                peerUid,
+                guildId: '',
+            };
+            return { MsgId: msgId, Peer: peer };
+        }
+        return undefined;
+    }
+
+    getShortIdByMsgId(msgId: string): number | undefined {
+        return this.msgIdMap.getValue(msgId);
+    }
+
+    getPeerByMsgId(msgId: string) {
+        const shortId = this.msgIdMap.getValue(msgId);
+        if (!shortId) return undefined;
+        return this.getMsgIdAndPeerByShortId(shortId);
+    }
+
+    resize(maxSize: number): void {
+        this.msgIdMap.resize(maxSize);
+        this.msgDataMap.resize(maxSize);
+    }
+}
+
+export const MessageUnique: MessageUniqueWrapper = new MessageUniqueWrapper();