Skip to content

Commit d5c2047

Browse files
feat: adds Mercedes' loader file with implemented loadMediaItems method (#359)
* feat: adds loader file with implemented loadMediaItems method * fix: adjusted loadMediaItems variables * feat: implemented extra credit loadData function
1 parent 89614e9 commit d5c2047

File tree

2 files changed

+64
-1
lines changed

2 files changed

+64
-1
lines changed

lesson_10/libraries/src/loaders/loaders.module.ts

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,15 @@ import { Module } from '@nestjs/common';
22

33
import { AnthonyMaysLoader } from './anthony_mays_loader.js';
44
import { DylanLaffertysLoader } from './dylan_lafferty_loaders.js';
5+
import { MercedesMathewsLoader } from './mercedes_mathews_loader.js';
56

67
export const Loaders = Symbol.for('Loaders');
78

8-
const LOADER_PROVIDERS = [AnthonyMaysLoader, DylanLaffertysLoader];
9+
const LOADER_PROVIDERS = [
10+
AnthonyMaysLoader,
11+
DylanLaffertysLoader,
12+
MercedesMathewsLoader,
13+
];
914

1015
@Module({
1116
providers: [
Lines changed: 58 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,58 @@
1+
import csv from 'csv-parser';
2+
import fs from 'fs';
3+
import { Credit, MediaItem } from '../models/index.js';
4+
import { Loader } from './loader.js';
5+
6+
export class MercedesMathewsLoader implements Loader {
7+
getLoaderName(): string {
8+
return 'mercedesmathews';
9+
}
10+
11+
async loadData(): Promise<MediaItem[]> {
12+
const credits = await this.loadCredits();
13+
const mediaItems = await this.loadMediaItems();
14+
15+
const mediaMap = new Map<string, MediaItem>();
16+
for (const media of mediaItems) {
17+
mediaMap.set(media.getId(), media);
18+
}
19+
20+
for (const credit of credits) {
21+
const mediaItem = mediaMap.get(credit.getMediaItemId());
22+
if (mediaItem) {
23+
mediaItem.addCredit(credit);
24+
}
25+
}
26+
27+
console.log(
28+
`Loaded ${credits.length} credits and ${mediaItems.length} media items`,
29+
);
30+
31+
console.log(...mediaItems.values());
32+
return [...mediaItems.values()];
33+
}
34+
35+
async loadMediaItems(): Promise<MediaItem[]> {
36+
const medias = [];
37+
const readable = fs
38+
.createReadStream('data/media_items.csv', 'utf-8')
39+
.pipe(csv());
40+
for await (const row of readable) {
41+
const { id, title, type, year } = row;
42+
medias.push(new MediaItem(id, title, type, year, []));
43+
}
44+
return medias;
45+
}
46+
47+
async loadCredits(): Promise<Credit[]> {
48+
const credits = [];
49+
const readable = fs
50+
.createReadStream('data/credits.csv', 'utf-8')
51+
.pipe(csv());
52+
for await (const row of readable) {
53+
const { media_item_id, role, name } = row;
54+
credits.push(new Credit(media_item_id, name, role));
55+
}
56+
return credits;
57+
}
58+
}

0 commit comments

Comments
 (0)