Map w TypeScript: Map<K, V>, get, set, has i iteracja

Map w TypeScript to wbudowana w JavaScript Map z typowanymi kluczami i wartościami: Map<string, number> mapuje stringi na liczby. Utwórz ją przez new Map<K, V>(), a potem używaj set, get, has i delete. get zwraca V | undefined, bo klucza może brakować.

const ages = new Map<string, number>();

ages.set("ada", 36);
ages.set("grace", 85).set("alan", 41); // set returns the map, so calls chain

console.log(ages.get("ada"));    // 36
console.log(ages.get("linus"));  // undefined
console.log(ages.has("grace"));  // true
console.log(ages.size);          // 3

ages.delete("alan");
console.log(ages); // Map(2) { 'ada' => 36, 'grace' => 85 }

// @ts-expect-error: Argument of type 'string' is not assignable to parameter of type 'number'.
ages.set("bob", "forty");

To argumenty typu sprawiają, że Map jest przydatna w TypeScript: każde set jest sprawdzane, a każde get zwraca typ wartości. Ostatnia linia to błąd kompilacji (TS2345); @ts-expect-error pozwala uruchomić blok mimo to.

Ta strona dotyczy kolekcji Map. Jeśli szukasz array.map(), znajdziesz tę metodę w ostatniej sekcji.

Tworzenie Map

Typy pochodzą z argumentów typu, z początkowych wpisów albo z adnotacji. Początkowe wpisy to tablica krotek [key, value] albo cokolwiek innego, co je zwraca.

// 1. Type arguments, empty map
const scores = new Map<string, number[]>();

// 2. Inferred from entries: Map<string, number>
const prices = new Map([
    ["tea", 2.5],
    ["coffee", 3.2],
]);

// 3. From an object: Map<string, boolean>
const flags = new Map(Object.entries({ darkMode: true, beta: false }));

// 4. Keys and values of any type
type User = { id: number; name: string };
const lastSeen = new Map<User, Date>();
lastSeen.set({ id: 1, name: "Ada" }, new Date("2024-03-15T10:30:00Z"));

console.log(scores.size, prices.get("tea"), flags.get("beta"), lastSeen.size);

Dwie pułapki wnioskowania:

• new Map() bez argumentów typu i bez wpisów to Map<any, any>. Nic, co wkładasz lub wyjmujesz, nie jest sprawdzane. Zawsze pisz new Map<K, V>().
• Wpisy z różnymi typami wartości nie dają wnioskowanej unii. new Map([["a", 1], ["b", "x"]]) to błąd TS2769 (No overload matches this call). Podaj typ: new Map<string, number | string>([...]).

get zwraca V | undefined

Map nie może zagwarantować, że klucz istnieje, więc get ma typ V | undefined. W trybie strict musisz obsłużyć undefined, zanim użyjesz wartości jako V:

// This does not compile: get may return undefined.
const stock = new Map<string, number>([["apples", 3]]);

const apples: number = stock.get("apples");
console.log(apples);

index.ts(4,7): error TS2322: Type 'number | undefined' is not assignable to type 'number'.
  Type 'undefined' is not assignable to type 'number'.

Rozwiązania, od najczęstszego do najrzadszego:

const stock = new Map<string, number>([["apples", 3]]);

// 1. A default value with ??
const pears = stock.get("pears") ?? 0;
console.log(pears); // 0

// 2. Store the result and check it
const apples = stock.get("apples");
if (apples !== undefined) {
    console.log(apples * 2); // 6: apples is number here
}

// 3. Update a counter in one line
stock.set("apples", (stock.get("apples") ?? 0) + 1);
console.log(stock.get("apples")); // 4

// has() does not narrow a later get():
if (stock.has("apples")) {
    const n = stock.get("apples"); // still number | undefined
    console.log(n);
}

TypeScript nie pamięta, że has zwróciło true, gdy później wywołujesz get. Bezpośrednie sprawdzenie wyniku get jest krótsze i bezpieczne. Asercja non-null, stock.get("apples")!, ucisza błąd, ale nie chroni, gdy klucza brakuje.

Metody Map i ich typy

Iteracja po Map

Map iteruje w kolejności wstawiania. for...of po mapie zwraca krotki [key, value] o typie [K, V].

const prices = new Map<string, number>([
    ["tea", 2.5],
    ["coffee", 3.2],
    ["juice", 4],
]);

for (const [item, price] of prices) {   // item: string, price: number
    console.log(`${item}: ${price.toFixed(2)}`);
}

prices.forEach((price, item) => {       // value first, then key
    console.log(item.toUpperCase(), price);
});

const items = [...prices.keys()];       // string[]
const total = [...prices.values()].reduce((a, b) => a + b, 0);
console.log(items, total);              // [ 'tea', 'coffee', 'juice' ] 9.7

// Sorted by value: sort the entries, build a new Map
const byPrice = new Map([...prices].sort((a, b) => b[1] - a[1]));
console.log([...byPrice.keys()]);       // [ 'juice', 'coffee', 'tea' ]

Ustawienie istniejącego klucza aktualizuje wartość, ale zachowuje jej pierwotne miejsce w kolejności.

Obiekty jako klucze i zliczanie

Kluczem może być dowolna wartość, także obiekty i tablice. Klucze są porównywane jak przez ===: dwa obiekty o tej samej zawartości to różne klucze. Map to też standardowy sposób na zliczanie lub grupowanie elementów.

const a = { id: 1 };
const b = { id: 1 };
const visits = new Map<object, number>([[a, 5]]);
console.log(visits.get(a), visits.get(b)); // 5 undefined: b is another object

// Counting words
const words = "the cat and the hat and the bat".split(" ");
const counts = new Map<string, number>();
for (const w of words) {
    counts.set(w, (counts.get(w) ?? 0) + 1);
}
console.log(counts); // Map(5) { 'the' => 3, 'cat' => 1, 'and' => 2, 'hat' => 1, 'bat' => 1 }

// Grouping into arrays
const byLength = new Map<number, string[]>();
for (const w of new Set(words)) {
    const group = byLength.get(w.length) ?? [];
    group.push(w);
    byLength.set(w.length, group);
}
console.log(byLength); // Map(1) { 3 => [ 'the', 'cat', 'and', 'hat', 'bat' ] }

Aby kluczem była zawartość obiektu, utwórz z niej klucz tekstowy lub liczbowy, na przykład user.id albo `${x},${y}`.

Map a obiekt i Record

Praktyczna zasada: Map dla kolekcji, której klucze są danymi (id użytkowników, słowa, wpisy cache) i zmieniają się w trakcie działania programu; typ obiektowy lub Record dla stałego zestawu znanych kluczy i wszystkiego, co trafia do JSON albo z niego pochodzi. Strona o słownikach porównuje sygnatury indeksu, Record i Map przy wyszukiwaniu po kluczach tekstowych.

Konwersja Map na obiekty i JSON

Wpisy Map nie są właściwościami, więc JSON.stringify ich nie widzi. Konwertuj przez Object.fromEntries i Object.entries:

const settings = new Map<string, string | number>([
    ["theme", "dark"],
    ["fontSize", 14],
]);

console.log(JSON.stringify(settings)); // {}

const obj = Object.fromEntries(settings); // { [k: string]: string | number }
const json = JSON.stringify(obj);
console.log(json); // {"theme":"dark","fontSize":14}

// And back again
const restored = new Map(Object.entries(JSON.parse(json) as Record<string, string | number>));
console.log(restored.get("theme")); // dark

JSON.parse zwraca any, więc as Record<...> określa, jakich danych się spodziewasz. To nie jest sprawdzenie w czasie działania; zweryfikuj niezaufany JSON, zanim zaufasz temu typowi.

Typowanie array.map()

Wiele wyszukiwań "typescript map" dotyczy metody tablic, która przekształca każdy element i zwraca nową tablicę. Jej typ jest wnioskowany z callbacku, więc adnotacje rzadko są potrzebne:

type User = { id: number; first: string; last: string };
const users: User[] = [
    { id: 1, first: "Ada", last: "Lovelace" },
    { id: 2, first: "Alan", last: "Turing" },
];

const names = users.map((u) => `${u.first} ${u.last}`); // string[]
const ids = users.map((u) => u.id);                     // number[]

// An explicit result type, checked against what the callback returns
type Option = { value: number; label: string };
const options = users.map((u): Option => ({ value: u.id, label: u.first }));

// With the index
const numbered = users.map((u, i) => `${i + 1}. ${u.first}`);

console.log(names, ids);
console.log(options);
console.log(numbered);

Adnotacja typu zwracanego przez callback ((u): Option => ...) to najczytelniejszy sposób na określenie typu wyniku: brakująca lub źle napisana właściwość w zwracanym obiekcie staje się wtedy błędem kompilacji w callbacku.