浏览代码

Fix resending batched messages (#341)

Vladyslav Batyrenko 3 年之前
父节点
当前提交
a23a7349fb
共有 3 个文件被更改,包括 6 次插入6 次删除
  1. 3 3
      gramjs/network/MTProtoSender.ts
  2. 1 1
      gramjs/network/RequestState.ts
  3. 2 2
      gramjs/sessions/StoreSession.ts

+ 3 - 3
gramjs/network/MTProtoSender.ts

@@ -593,7 +593,7 @@ export class MTProtoSender {
 
         const toPop = [];
 
-        for (const state of Object.values(this._pendingState)) {
+        for (const state of this._pendingState.values()) {
             if (state.containerId && state.containerId.equals(msgId)) {
                 toPop.push(state.msgId);
             }
@@ -602,8 +602,8 @@ export class MTProtoSender {
         if (toPop.length) {
             const temp = [];
             for (const x of toPop) {
-                temp.push(this._pendingState.get(x));
-                this._pendingState.delete(x);
+                temp.push(this._pendingState.get(x!.toString()));
+                this._pendingState.delete(x!.toString());
             }
             return temp;
         }

+ 1 - 1
gramjs/network/RequestState.ts

@@ -1,7 +1,7 @@
 import bigInt from "big-integer";
 
 export class RequestState {
-    public containerId: undefined;
+    public containerId?: bigInt.BigInteger;
     public msgId?: bigInt.BigInteger;
     public request: any;
     public data: Buffer;

+ 2 - 2
gramjs/sessions/StoreSession.ts

@@ -1,11 +1,11 @@
 import { MemorySession } from "./Memory";
-import store from "store2";
+import store, {StoreBase} from "store2";
 import { AuthKey } from "../crypto/AuthKey";
 import bigInt from "big-integer";
 
 export class StoreSession extends MemorySession {
     private readonly sessionName: string;
-    private store: store.StoreAPI;
+    private store: StoreBase;
 
     constructor(sessionName: string, divider = ":") {
         super();