TypeScript:使用Generics的Mapped类型

时间:2018-01-17 05:21:02

标签: typescript generics mapped-types

我一直在尝试使用TypeScript,希望能够创建一个类型安全的数据库查询库(松散地基于Scala的Slick)。我在Mapped Types的帮助下取得了一些不错的进展,但我仍然坚持保留Column的基础类型。请参阅以下代码以获取全面的示例:

class Table {}

class TableExpression<TK extends Table> {
  table: TK
  alias: string
  constructor(table: TK, alias?: string) {
    this.table = table
    this.alias = alias || table.constructor.name
  }
}

class Column<T> {
  name: string
  defaultValue?: T
}

class StringC extends Column<string> {
  constructor(name: string, defaultValue?: string) {
    super()
    this.name = name
    this.defaultValue = defaultValue
  }
}

class ColumnExpression<TK extends Table, CK extends Column<any>> {
  table: TableExpression<TK>
  column: CK
  alias: string
  constructor(table: TableExpression<TK>, column: CK, alias?: string) {
    this.table = table
    this.column = column
    this.alias = alias || column.name
  }

  eq(val: any): string {
    return `${this.table.alias}.${this.column.name} = "${val}"` // Obviously, not safe.
  }
}

class E1 extends Table {
  name = new StringC('name')
  slug = new StringC('slug')
}

let e1 = new E1()
let ee1 = new TableExpression(e1, 'e1')

type TableQuery<TK extends Table> = {
  [P in keyof TK]: ColumnExpression<TK, TK[P]>
}

function query<TK extends Table>(te: TableExpression<TK>): TableQuery<TK> {
  let result = {} as TableQuery<TK>
  for (const k in te.table) {
    result[k] = new ColumnExpression(te, te.table[k])
  }
  return result
}

let tq1 = query(ee1)
console.log(tq1.name.eq('Pancakes')) // e1.name = "Pancakes"

此代码按预期编译和工作。我所困扰的是如何使eq()方法利用Column使用的泛型类型。我可以很容易地扩展ColumnExpression以使用另一个类型参数,例如下面的CT:

class ColumnExpression<TK extends Table, CT, CK extends Column<CT>> {
...
  eq(val: CT) { ... }
}

那部分是有道理的。然后问题是将其转发到TableQuery的映射类型定义中。似乎TableQuery需要以某种方式进行参数化,我可以将CT传递给ColumnExpression,但我无法弄清楚如何做到这一点。有任何想法吗?

1 个答案:

答案 0 :(得分:1)

您可以使用Lookup Types

要在eq中将泛型类型作为参数值,请更改您的签名:

eq(val: CK['defaultValue']): string

对于Column<string>,这将导致string | undefined,至少启用--strictNullChecks。那是因为defaultValue是可选的。所以调用eq(undefined)仍然有效。

您可以通过执行undefined删除CK['defaultValue'] & Object,但我不确定这是否有任何我不知道的副作用。