18 lines
776 B
TypeScript
18 lines
776 B
TypeScript
import { DrizzleClient } from "@lib/DrizzleClient";
|
|
import { users } from "@/db/schema";
|
|
import { eq } from "drizzle-orm";
|
|
|
|
export async function getUserBalance(userId: string) {
|
|
const user = await DrizzleClient.query.users.findFirst({ where: eq(users.userId, userId) });
|
|
return user?.balance ?? 0;
|
|
}
|
|
|
|
export async function setUserBalance(userId: string, balance: number) {
|
|
await DrizzleClient.update(users).set({ balance }).where(eq(users.userId, userId));
|
|
}
|
|
|
|
export async function addUserBalance(userId: string, amount: number) {
|
|
const user = await DrizzleClient.query.users.findFirst({ where: eq(users.userId, userId) });
|
|
if (!user) return;
|
|
await DrizzleClient.update(users).set({ balance: user.balance + amount }).where(eq(users.userId, userId));
|
|
} |