refactor(db): migrate app pages and APIs from Prisma facade to Drizzle (2)
Co-authored-by: Cursor <[email protected]>
This commit is contained in:
1 parent
7c39aef5d9
commit
2cd0863cb8
40 files changed
+1069
-726
No files matched your search
@@ -1,3 +1,4 @@
|
||||
import { and, count, eq, gte, sql, sum } from "drizzle-orm";
|
||||
import { ArrowLeftRight } from "lucide-react";
|
||||
import { redirect } from "next/navigation";
|
||||
import { getLocale, getTranslations } from "next-intl/server";
|
||||
@@ -5,8 +6,14 @@ import { RevenueChart } from "@/components/admin/revenue-chart";
|
||||
import { StatsCard } from "@/components/admin/stats-card";
|
||||
import { CurrencyIcon } from "@/components/shared/currency-icon";
|
||||
import { Card, CardContent, CardHeader, CardTitle } from "@/components/ui/card";
|
||||
import {
|
||||
db,
|
||||
LogsShopPurchases,
|
||||
MarketplaceItems,
|
||||
RoomTradeLog,
|
||||
User,
|
||||
} from "@/lib/db";
|
||||
import { canAccess, getAdminContext, PERMS } from "@/lib/permissions";
|
||||
import { prisma } from "@/lib/prisma";
|
||||
import { redisCache } from "@/lib/redis-cache";
|
||||
|
||||
type EconomyData = {
|
||||
@@ -43,6 +50,11 @@ async function safeCount(fn: () => Promise<number>): Promise<number> {
|
||||
}
|
||||
}
|
||||
|
||||
async function rawRows<T>(query: ReturnType<typeof sql>): Promise<T[]> {
|
||||
const [rows] = (await db.execute(query)) as unknown as [T[], unknown];
|
||||
return rows ?? [];
|
||||
}
|
||||
|
||||
async function getEconomyData(
|
||||
locale: string,
|
||||
): Promise<{ data: EconomyData; degraded: boolean }> {
|
||||
@@ -54,49 +66,86 @@ async function getEconomyData(
|
||||
|
||||
try {
|
||||
const [creditsAgg, pixelsAgg, pointsAgg] = await Promise.all([
|
||||
prisma.user.aggregate({ _sum: { credits: true } }).catch(() => {
|
||||
degraded = true;
|
||||
return null;
|
||||
}),
|
||||
prisma.user.aggregate({ _sum: { pixels: true } }).catch(() => {
|
||||
degraded = true;
|
||||
return null;
|
||||
}),
|
||||
prisma.user.aggregate({ _sum: { points: true } }).catch(() => {
|
||||
degraded = true;
|
||||
return null;
|
||||
}),
|
||||
db
|
||||
.select({ total: sum(User.credits) })
|
||||
.from(User)
|
||||
.catch(() => {
|
||||
degraded = true;
|
||||
return [{ total: null }];
|
||||
}),
|
||||
db
|
||||
.select({ total: sum(User.pixels) })
|
||||
.from(User)
|
||||
.catch(() => {
|
||||
degraded = true;
|
||||
return [{ total: null }];
|
||||
}),
|
||||
db
|
||||
.select({ total: sum(User.points) })
|
||||
.from(User)
|
||||
.catch(() => {
|
||||
degraded = true;
|
||||
return [{ total: null }];
|
||||
}),
|
||||
]);
|
||||
const totalCredits = creditsAgg?._sum.credits ?? 0;
|
||||
const totalPixels = pixelsAgg?._sum.pixels ?? 0;
|
||||
const totalPoints = pointsAgg?._sum.points ?? 0;
|
||||
const totalCredits = Number(creditsAgg[0]?.total ?? 0);
|
||||
const totalPixels = Number(pixelsAgg[0]?.total ?? 0);
|
||||
const totalPoints = Number(pointsAgg[0]?.total ?? 0);
|
||||
|
||||
let shopPurchases = 0;
|
||||
let shopCreditsSpent = 0;
|
||||
let shopPointsSpent = 0;
|
||||
try {
|
||||
const shopRevenue = await prisma.logsShopPurchases.aggregate({
|
||||
where: { timestamp: { gte: weekAgo } },
|
||||
_sum: { costCredits: true, costPoints: true },
|
||||
_count: true,
|
||||
});
|
||||
shopPurchases =
|
||||
typeof shopRevenue._count === "number" ? shopRevenue._count : 0;
|
||||
shopCreditsSpent = shopRevenue._sum?.costCredits ?? 0;
|
||||
shopPointsSpent = shopRevenue._sum?.costPoints ?? 0;
|
||||
const [shopRevenue] = await db
|
||||
.select({
|
||||
purchases: count(),
|
||||
credits: sum(LogsShopPurchases.costCredits),
|
||||
points: sum(LogsShopPurchases.costPoints),
|
||||
})
|
||||
.from(LogsShopPurchases)
|
||||
.where(gte(LogsShopPurchases.timestamp, weekAgo));
|
||||
shopPurchases = Number(shopRevenue?.purchases ?? 0);
|
||||
shopCreditsSpent = Number(shopRevenue?.credits ?? 0);
|
||||
shopPointsSpent = Number(shopRevenue?.points ?? 0);
|
||||
} catch {
|
||||
degraded = true;
|
||||
}
|
||||
|
||||
const [marketplaceActive, marketplaceSold, weekTrades] = await Promise.all([
|
||||
safeCount(() => prisma.marketplaceItems.count({ where: { state: 1 } })),
|
||||
safeCount(() =>
|
||||
prisma.marketplaceItems.count({
|
||||
where: { state: 2, soldTimestamp: { gte: weekAgo } },
|
||||
}),
|
||||
safeCount(async () =>
|
||||
Number(
|
||||
(
|
||||
await db
|
||||
.select({ total: count() })
|
||||
.from(MarketplaceItems)
|
||||
.where(eq(MarketplaceItems.state, 1))
|
||||
)[0]?.total ?? 0,
|
||||
),
|
||||
),
|
||||
safeCount(() =>
|
||||
prisma.roomTradeLog.count({ where: { timestamp: { gte: weekAgo } } }),
|
||||
safeCount(async () =>
|
||||
Number(
|
||||
(
|
||||
await db
|
||||
.select({ total: count() })
|
||||
.from(MarketplaceItems)
|
||||
.where(
|
||||
and(
|
||||
eq(MarketplaceItems.state, 2),
|
||||
gte(MarketplaceItems.soldTimestamp, weekAgo),
|
||||
),
|
||||
)
|
||||
)[0]?.total ?? 0,
|
||||
),
|
||||
),
|
||||
safeCount(async () =>
|
||||
Number(
|
||||
(
|
||||
await db
|
||||
.select({ total: count() })
|
||||
.from(RoomTradeLog)
|
||||
.where(gte(RoomTradeLog.timestamp, weekAgo))
|
||||
)[0]?.total ?? 0,
|
||||
),
|
||||
),
|
||||
]);
|
||||
|
||||
@@ -112,20 +161,18 @@ async function getEconomyData(
|
||||
}
|
||||
|
||||
try {
|
||||
const revRows = await prisma.$queryRaw<
|
||||
{
|
||||
day_bucket: bigint;
|
||||
credits: number | bigint;
|
||||
points: number | bigint;
|
||||
}[]
|
||||
>`
|
||||
const revRows = await rawRows<{
|
||||
day_bucket: bigint;
|
||||
credits: number | bigint;
|
||||
points: number | bigint;
|
||||
}>(sql`
|
||||
SELECT FLOOR(timestamp / 86400) AS day_bucket,
|
||||
COALESCE(SUM(cost_credits), 0) AS credits,
|
||||
COALESCE(SUM(cost_points), 0) AS points
|
||||
FROM logs_shop_purchases
|
||||
WHERE timestamp >= ${rangeStart}
|
||||
GROUP BY day_bucket ORDER BY day_bucket
|
||||
`;
|
||||
`);
|
||||
const revMap = new Map(
|
||||
revRows.map((r) => [
|
||||
Number(r.day_bucket),
|
||||
|
||||
Reference in new issue
Block a user