typescript 的类型处理

V2EX = way to explore

V2EX 是一个关于分享和探索的地方

已注册用户请登录

这是一个创建于 822 天前的主题，其中的信息可能已经有所发展或是发生改变。

1 、interface

interface TypeD{ subTypeA: string; subTypeB: number; } interface TypeProp{ typeA: string; typeB: number; typeC: boolean; typeD: TypeD; }

2 、data

const data1: TypeProp = { typeA: 'yes', typeB: 1, typeC: true, typeD: { subTypeA: 'no', subTypeB: 0, } }

3 、function

const dosomething = (data: keyof TypeProp):[?] => { return data1[data] }

请问 [?] 处我需要怎么处理(或者我需要如何定义 dosomething)，才能表示返回的是 data1[data]的类型

我想要的

const b = dosomething('typeB') // b 是 number 类型 const c = dosomething('typeC') // c 是 boolean 类型 const d = dosomething('typeD') // d 是 TypeD 类型

8 条回复 2024-01-19 20:32:11 +08:00

lzgshsj

2024 年 1 月 19 日

const dosomething = <T extends keyof TypeProp>(data: T): TypeProp[T] => {
return data1[data];
}

Gaoti

2024 年 1 月 19 日

```typescript
function dosomething<K extends keyof TypeProp>(data: K): TypeProp[K] {
return data1[data];
}
```

善用泛型

yxcoder

2024 年 1 月 19 日

@Gaoti
@lzgshsj

请问一下，如果 dosomething 里面涉及到数组应该怎么处理的

yxcoder

2024 年 1 月 19 日

@yxcoder

譬如我希望
const dosomething = (data: keyof TypeProp | keyof TypeProp[]) => { ... }

yxcoder

2024 年 1 月 19 日

@yxcoder const dosomething = (data: keyof TypeProp | (keyof TypeProp)[]) => { ... }

MinonHeart

2024 年 1 月 19 日

可能类似这样吧，假设都返回对象形式
get(keys: keyof T): Pick<T, keyof T>;
get<U extends keyof T, K extends U[]>(keys: K): Pick<T, K[number]>;

Gaoti

2024 年 1 月 19 日

@yxcoder #5

``` typescript
function dosomething<K extends keyof TypeProp>(data: K): TypeProp[K];

function dosomething<K extends keyof TypeProp>(
data: K[],
): { [P in K]: TypeProp[P] }[];

function dosomething<K extends keyof TypeProp>(
data: K | K[],
): TypeProp[K] | TypeProp[K][] {
if (Array.isArray(data)) {
return data.map((key) => data1[key]);
} else {
return data1[data];
}
}
```

bgm004

2024 年 1 月 19 日 via Android

类型体操里有这种。