112 lines
2.7 KiB
TypeScript
112 lines
2.7 KiB
TypeScript
import { UpdateMarketDataDto } from '@ghostfolio/api/app/admin/update-market-data.dto';
|
|
import { DateQuery } from '@ghostfolio/api/app/portfolio/interfaces/date-query.interface';
|
|
import { PrismaService } from '@ghostfolio/api/services/prisma.service';
|
|
import { resetHours } from '@ghostfolio/common/helper';
|
|
import { UniqueAsset } from '@ghostfolio/common/interfaces';
|
|
import { Injectable } from '@nestjs/common';
|
|
import { DataSource, MarketData, Prisma } from '@prisma/client';
|
|
|
|
@Injectable()
|
|
export class MarketDataService {
|
|
public constructor(private readonly prismaService: PrismaService) {}
|
|
|
|
public async deleteMany({ dataSource, symbol }: UniqueAsset) {
|
|
return this.prismaService.marketData.deleteMany({
|
|
where: {
|
|
dataSource,
|
|
symbol
|
|
}
|
|
});
|
|
}
|
|
|
|
public async get({
|
|
date,
|
|
symbol
|
|
}: {
|
|
date: Date;
|
|
symbol: string;
|
|
}): Promise<MarketData> {
|
|
return await this.prismaService.marketData.findFirst({
|
|
where: {
|
|
symbol,
|
|
date: resetHours(date)
|
|
}
|
|
});
|
|
}
|
|
|
|
public async getMax({ dataSource, symbol }: UniqueAsset): Promise<number> {
|
|
const aggregations = await this.prismaService.marketData.aggregate({
|
|
_max: {
|
|
marketPrice: true
|
|
},
|
|
where: {
|
|
dataSource,
|
|
symbol
|
|
}
|
|
});
|
|
|
|
return aggregations._max.marketPrice;
|
|
}
|
|
|
|
public async getRange({
|
|
dateQuery,
|
|
symbols
|
|
}: {
|
|
dateQuery: DateQuery;
|
|
symbols: string[];
|
|
}): Promise<MarketData[]> {
|
|
return await this.prismaService.marketData.findMany({
|
|
orderBy: [
|
|
{
|
|
date: 'asc'
|
|
},
|
|
{
|
|
symbol: 'asc'
|
|
}
|
|
],
|
|
where: {
|
|
date: dateQuery,
|
|
symbol: {
|
|
in: symbols
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
public async marketDataItems(params: {
|
|
skip?: number;
|
|
take?: number;
|
|
cursor?: Prisma.MarketDataWhereUniqueInput;
|
|
where?: Prisma.MarketDataWhereInput;
|
|
orderBy?: Prisma.MarketDataOrderByWithRelationInput;
|
|
}): Promise<MarketData[]> {
|
|
const { skip, take, cursor, where, orderBy } = params;
|
|
|
|
return this.prismaService.marketData.findMany({
|
|
cursor,
|
|
orderBy,
|
|
skip,
|
|
take,
|
|
where
|
|
});
|
|
}
|
|
|
|
public async updateMarketData(params: {
|
|
data: { dataSource: DataSource } & UpdateMarketDataDto;
|
|
where: Prisma.MarketDataWhereUniqueInput;
|
|
}): Promise<MarketData> {
|
|
const { data, where } = params;
|
|
|
|
return this.prismaService.marketData.upsert({
|
|
where,
|
|
create: {
|
|
dataSource: data.dataSource,
|
|
date: where.date_symbol.date,
|
|
marketPrice: data.marketPrice,
|
|
symbol: where.date_symbol.symbol
|
|
},
|
|
update: { marketPrice: data.marketPrice }
|
|
});
|
|
}
|
|
}
|