typescript - Generisk memoiseringsfunktion som returnerar samma funktionstyp

Jag försöker skriva en memoiseringsfunktion som tar en funktion som argument och returnerar en liknande memoiserad funktion.

function memoize<T extends Function, R>(f: T): T {
  const memory = new Map<string, R>();

  const g = (...args: any[]) => {
    if (!memory.get(args.join())) { memory.set(args.join(), f(...args)); }
    return memory.get(args.join());
  };

  return g; // g as T => [ts] Type '(...args: any[]) => R' cannot be converted to type 'T'.
}

// const exp: (...args: any[]) => RegExp
const exp = memoize<(text: string) => RegExp, RegExp>((text: string) => {
  return new RegExp(text.replace(/[^a-zA-Z0-9\s]/g, ".").replace(/\s+/g, "\\s+"), "ig");
});

Problemet är att om jag bara returnerar g, blir signaturen för exp (...args: any[]) => RexExp och om jag försöker tvinga g att vara T, klagar ts på att g inte kan tilldelas T.

Finns det ett sätt att "tvinga" g att vara samma typ som f för att exp ska ha exakt samma typ som funktionen som skickas till memoize?

Svar #1

Det verkar fungera:

function memoize<R, T extends (...args: any[]) => R>(f: T): T {
    const memory = new Map<string, R>();

    const g = (...args: any[]) => {
        if (!memory.get(args.join())) {
            memory.set(args.join(), f(...args));
        }

        return memory.get(args.join());
    };

    return g as T;
}

const exp = memoize((text: string) => {
    return new RegExp(text.replace(/[^a-zA-Z0-9\s]/g, ".").replace(/\s+/g, "\\s+"), "ig");
});

(kod i lekplats)

Relaterade artiklar