Browse Source

Optimize admin market data preset query

pull/6346/head
David Requeno 2 months ago
parent
commit
33350b378d
  1. 48
      apps/api/src/app/admin/admin.service.ts

48
apps/api/src/app/admin/admin.service.ts

@ -205,6 +205,8 @@ export class AdminService {
sortDirection?: Prisma.SortOrder; sortDirection?: Prisma.SortOrder;
take?: number; take?: number;
}): Promise<AdminMarketData> { }): Promise<AdminMarketData> {
const MARKET_DATA_QUERY_SYMBOL_CHUNK_SIZE = 5000;
let orderBy: Prisma.Enumerable<Prisma.SymbolProfileOrderByWithRelationInput> = let orderBy: Prisma.Enumerable<Prisma.SymbolProfileOrderByWithRelationInput> =
[{ symbol: 'asc' }]; [{ symbol: 'asc' }];
const where: Prisma.SymbolProfileWhereInput = {}; const where: Prisma.SymbolProfileWhereInput = {};
@ -225,6 +227,10 @@ export class AdminService {
presetId === 'ETF_WITHOUT_SECTORS' presetId === 'ETF_WITHOUT_SECTORS'
) { ) {
filters = [{ id: 'ETF', type: 'ASSET_SUB_CLASS' }]; filters = [{ id: 'ETF', type: 'ASSET_SUB_CLASS' }];
} else if (presetId === 'WITHOUT_ACTIVITIES') {
where.activities = {
none: {}
};
} }
const searchQuery = filters.find(({ type }) => { const searchQuery = filters.find(({ type }) => {
@ -316,7 +322,28 @@ export class AdminService {
const assetProfiles = symbolProfileResult[0]; const assetProfiles = symbolProfileResult[0];
let count = symbolProfileResult[1]; let count = symbolProfileResult[1];
const lastMarketPrices = await this.prismaService.marketData.findMany({ const dataSources = [
...new Set(assetProfiles.map(({ dataSource }) => dataSource))
];
const symbols = [...new Set(assetProfiles.map(({ symbol }) => symbol))];
const symbolChunks: string[][] = [];
for (
let chunkStartIndex = 0;
chunkStartIndex < symbols.length;
chunkStartIndex += MARKET_DATA_QUERY_SYMBOL_CHUNK_SIZE
) {
symbolChunks.push(
symbols.slice(
chunkStartIndex,
chunkStartIndex + MARKET_DATA_QUERY_SYMBOL_CHUNK_SIZE
)
);
}
const lastMarketPriceResults = await Promise.all(
symbolChunks.map((symbolChunk) => {
return this.prismaService.marketData.findMany({
distinct: ['dataSource', 'symbol'], distinct: ['dataSource', 'symbol'],
orderBy: { date: 'desc' }, orderBy: { date: 'desc' },
select: { select: {
@ -326,17 +353,16 @@ export class AdminService {
}, },
where: { where: {
dataSource: { dataSource: {
in: assetProfiles.map(({ dataSource }) => { in: dataSources
return dataSource;
})
}, },
symbol: { symbol: {
in: assetProfiles.map(({ symbol }) => { in: symbolChunk
return symbol;
})
} }
} }
}); });
})
);
const lastMarketPrices = lastMarketPriceResults.flat();
const lastMarketPriceMap = new Map<string, number>(); const lastMarketPriceMap = new Map<string, number>();
@ -433,21 +459,15 @@ export class AdminService {
) )
); );
if (presetId) {
if (presetId === 'ETF_WITHOUT_COUNTRIES') { if (presetId === 'ETF_WITHOUT_COUNTRIES') {
marketData = marketData.filter(({ countriesCount }) => { marketData = marketData.filter(({ countriesCount }) => {
return countriesCount === 0; return countriesCount === 0;
}); });
count = marketData.length;
} else if (presetId === 'ETF_WITHOUT_SECTORS') { } else if (presetId === 'ETF_WITHOUT_SECTORS') {
marketData = marketData.filter(({ sectorsCount }) => { marketData = marketData.filter(({ sectorsCount }) => {
return sectorsCount === 0; return sectorsCount === 0;
}); });
} else if (presetId === 'WITHOUT_ACTIVITIES') {
marketData = marketData.filter(({ activitiesCount }) => {
return activitiesCount === 0;
});
}
count = marketData.length; count = marketData.length;
} }

Loading…
Cancel
Save