Use deferred

This commit is contained in:
RMidhunSuresh 2023-03-24 20:25:19 +05:30
parent 7c6bcbc09c
commit 225a778d1a
2 changed files with 45 additions and 9 deletions

View File

@ -23,6 +23,7 @@ import {makeTxnId} from "../../../common.js";
import {CancelReason, VerificationEventTypes} from "./types";
import {Disposables} from "../../../../utils/Disposables";
import {VerificationCancelledError} from "../VerificationCancelledError";
import {Deferred} from "../../../../utils/Deferred";
const messageFromErrorType = {
[CancelReason.UserCancelled]: "User declined",
@ -71,7 +72,7 @@ export class ToDeviceChannel extends Disposables implements IChannel {
private readonly deviceMessageHandler: DeviceMessageHandler;
private readonly sentMessages: Map<VerificationEventTypes, any> = new Map();
private readonly receivedMessages: Map<VerificationEventTypes, any> = new Map();
private readonly waitMap: Map<string, {resolve: any, reject: any, promise: Promise<any>}> = new Map();
private readonly waitMap: Map<string, Deferred<any>> = new Map();
private readonly log: ILogItem;
public otherUserDeviceId: string;
public startMessage: any;
@ -270,14 +271,9 @@ export class ToDeviceChannel extends Disposables implements IChannel {
if (existingWait) {
return existingWait.promise;
}
let resolve, reject;
// Add to wait map
const promise = new Promise((_resolve, _reject) => {
resolve = _resolve;
reject = _reject;
});
this.waitMap.set(eventType, { resolve, reject, promise });
return promise;
const deferred = new Deferred();
this.waitMap.set(eventType, deferred);
return deferred.promise;
}
setStartMessage(event) {

40
src/utils/Deferred.ts Normal file
View File

@ -0,0 +1,40 @@
/*
Copyright 2023 The Matrix.org Foundation C.I.C.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
export class Deferred<T> {
public readonly promise: Promise<T>;
public readonly resolve: (value: T) => void;
public readonly reject: (err: Error) => void;
private _value?: T;
constructor() {
let resolve;
let reject;
this.promise = new Promise<T>((_resolve, _reject) => {
resolve = _resolve;
reject = _reject;
})
this.resolve = (value: T) => {
this._value = value;
resolve(value);
};
this.reject = reject;
}
get value(): T | undefined {
return this._value;
}
}