NapCatQQ/src/common/utils/MessageUnique.ts

import { Peer } from '@/core';
import crypto, { randomInt, randomUUID } from 'crypto';
import { logError } from './log';

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 {
    // const isExist = this.keyToValue.get(key);
    // if (isExist && isExist === value) {
    //   return;
    // }
    this.keyToValue.set(key, value);
    this.valueToKey.set(value, key);
    while (this.keyToValue.size !== this.valueToKey.size) {
      console.log('keyToValue.size !== valueToKey.size Error Atom');
      this.keyToValue.clear();
      this.valueToKey.clear();
    }
    // console.log('---------------');
    // console.log(this.keyToValue);
    // console.log(this.valueToKey);
    // console.log('---------------');
    while (this.keyToValue.size > this.maxSize || this.valueToKey.size > this.maxSize) {
      //console.log(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): number | undefined {
    const key = `${msgId}|${peer.chatType}|${peer.peerUid}`;
    const hash = crypto.createHash('md5').update(key);
    const shortId = hash.digest().readInt32BE(0);
    //减少性能损耗
    // const isExist = this.msgIdMap.getKey(shortId);
    // if (isExist && isExist === msgId) {
    //   return shortId;
    // }
    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();