首页 > 解决方案 > 如何使对象工厂保持类型

问题描述

我创建了以下对象工厂来实例化所有接口的实现:

interface SomeInterface {
  get(): string;
}

class Implementation implements SomeInterface {
  constructor() {}
  get() {
    return "Hey :D";
  }
}

type Injectable = {
  [key: string]: () => unknown;
};

// deno-lint-ignore prefer-const
let DEFAULT_IMPLEMENTATIONS: Injectable = {
  SomeInterface: () => new Implementation(),
};

let MOCK_IMPLEMENTATIONS: Injectable = {};

class Factory {
  static getInstance(interfaceName: string, parameters: unknown = []) {
    if (MOCK_IMPLEMENTATIONS[interfaceName])
      return MOCK_IMPLEMENTATIONS[interfaceName]();
    return DEFAULT_IMPLEMENTATIONS[interfaceName]();
  }

  static mockWithInstance(interfaceName: string, mock: unknown) {
    MOCK_IMPLEMENTATIONS[interfaceName] = () => mock;
  }
}

export const ObjectFactory = {
  getInstance<T>(name: string): T {
    return Factory.getInstance(name) as T;
  },

  mockWithInstance: Factory.mockWithInstance,
};

const impl = ObjectFactory.getInstance<SomeInterface>("SomeInterface");

如您所见,这个工厂允许您实例化和模拟这些接口。主要问题是我必须使用接口名称和接口调用此函数以将类型保留在分配中:

ObjectFactory.getInstance<SomeInterface>("SomeInterface")

我见过这个问题,但我不喜欢使用Base界面的想法。此外,这种方法也不会保留类型。

理想情况下,我想使用我的方法,但不必使用接口,即只使用接口的名称。

旁注: 的声明Injectable是使该代码工作的一种技巧,理想情况下,我希望能够仅使用实现名称,即:

let DEFAULT_IMPLEMENTATIONS = {
    SomeInterface: Implementation
}

标签: typescriptdesign-patternsfactory

解决方案


由于您有一个您关心支持的名称到类型映射的固定列表,因此这里的一般方法是考虑T表示此映射的对象类型,然后对于任何受支持的接口名称K extends keyof T,您将处理以下函数返回该名称的属性......即类型的函数() => T[K]。另一种说法是,我们将使用keyof和查找类型来帮助为您的工厂提供类型。

我们将只使用像{"SomeInterface": SomeInterface; "Date": Date}for这样的具体类型T,但是如果T是泛型的话,编译器会更容易处理。这是制造商的一个可能的通用实现ObjectFactory

function makeFactory<T>(DEFAULT_IMPLEMENTATIONS: { [K in keyof T]: () => T[K] }) {
  const MOCK_IMPLEMENTATIONS: { [K in keyof T]?: () => T[K] } = {};
  return {
    getInstance<K extends keyof T>(interfaceName: K) {
      const compositeInjectable: typeof DEFAULT_IMPLEMENTATIONS = {
        ...DEFAULT_IMPLEMENTATIONS,
        ...MOCK_IMPLEMENTATIONS
      };
      return compositeInjectable[interfaceName]();
    },
    mockWithInstance<K extends keyof T>(interfaceName: K, mock: T[K]) {
      MOCK_IMPLEMENTATIONS[interfaceName] = () => mock;
    }
  }
}

我将您的版本重构为编译器主要可以验证为类型安全的版本,以避免类型断言。让我们来看看它。

makeFactory函数在对象映射类型中是通用的T,并采用名为DEFAULT_IMPLEMENTATIONStype的参数{ [K in keyof T]: () => T[K] }。这是一个映射类型,其键与 的键K相同,T但其属性是返回 type 值的零参数函数T[K]。您可以看到您现有的DEFAULT_IMPLEMENTATIONS情况是这样的:每个属性都是一个零参数函数,返回相应接口的值。

在函数实现内部,我们创建MOCK_IMPLEMENTATIONS. 此变量具有几乎相同的类型,DEFAULT_IMPLEMENTATIONS但属性是可选?的(受中的可选修饰符影响[K in keyof T]?)。

该函数返回工厂本身,它有两个方法:

getInstance方法是泛型的K,接口名称的类型,返回值是类型T[K],对应的接口属性。我通过合并DEFAULT_IMPLEMENTATIONSMOCK_IMPLEMENTATIONS通过对象传播来实现这一点,并注释它compositeInjectableDEFAULT_IMPLEMENTATIONS. 然后我们用 the 索引它interfaceName并调用它。

mockWithInstance方法在接口名称的类型中也是泛型的K,并且接受一个类型的参数K(接口名称)和一个类型的参数T[K](对应的接口)。


让我们看看它的实际效果:

const ObjectFactory = makeFactory({
  SomeInterface: (): SomeInterface => new Implementation(),
  Date: () => new Date()
});

console.log(ObjectFactory.getInstance("SomeInterface").get().toUpperCase()); // HEY :D
ObjectFactory.mockWithInstance("SomeInterface", { get: () => "howdy" });
console.log(ObjectFactory.getInstance("SomeInterface").get().toUpperCase()); // HOWDY
console.log(ObjectFactory.getInstance("Date").getFullYear()); // 2020

这一切都如我所料。我们ObjectFactory通过调用makeFactory所需的DEFAULT_IMPLEMENTATIONS对象来制作。在这里,我已注释该SomeInterface属性返回一个类型的值SomeInterface(否则编译器会推断出Implementation可能比您想要的更具体)。

然后我们可以看到编译器允许我们使用正确的参数调用ObjectFactory.getInstance()andObjectFactory.mockWithInstance()并返回预期的类型,并且它也可以在运行时工作。


Playground 代码链接


推荐阅读