deep-stack/sol-mem-gen
3
0
Fork 0
forked from mito-systems/sol-mem-gen
Code Pull Requests Activity

Check if signature exists before writing to DB

Browse Source
This commit is contained in:
Adw8 2025-01-27 19:11:11 +05:30
parent 77f38148f1
commit 3514c04c25
3 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/app/api/flux/route.ts
View File

@ -56,7 +56,7 @@ export async function POST(req: NextRequest): Promise<NextResponse> {
) )
} }
await markSignatureAsUsed(transactionSignature) await markSignatureAsUsed(transactionSignature);
console.log('Generating with Flux model:', modelId) console.log('Generating with Flux model:', modelId)
console.log('Prompt:', prompt) console.log('Prompt:', prompt)

1
src/data-source.ts
View File

@ -1,4 +1,5 @@
import { DataSource } from 'typeorm'; import { DataSource } from 'typeorm';
import { Payment } from './entity/Payment'; import { Payment } from './entity/Payment';
export const AppDataSource = new DataSource({ export const AppDataSource = new DataSource({

17
src/utils/verifyPayment.ts
View File

@ -3,6 +3,7 @@ import { TOKEN_PROGRAM_ID } from '@solana/spl-token';
import { AppDataSource } from '../data-source'; import { AppDataSource } from '../data-source';
import { Payment } from '../entity/Payment'; import { Payment } from '../entity/Payment';
import { getManager } from 'typeorm';
const SOLANA_RPC_URL = process.env.NEXT_PUBLIC_SOLANA_RPC_URL; const SOLANA_RPC_URL = process.env.NEXT_PUBLIC_SOLANA_RPC_URL;
const SOLANA_WEBSOCKET_URL = process.env.NEXT_PUBLIC_SOLANA_WEBSOCKET_URL; const SOLANA_WEBSOCKET_URL = process.env.NEXT_PUBLIC_SOLANA_WEBSOCKET_URL;
@ -26,10 +27,18 @@ export async function isSignatureUsed(transactionSignature: string): Promise<boo
} }
export async function markSignatureAsUsed(transactionSignature: string): Promise<void> { export async function markSignatureAsUsed(transactionSignature: string): Promise<void> {
const paymentRepository = AppDataSource.getRepository(Payment); await AppDataSource.transaction(async (transactionalEntityManager) => {
const payment = new Payment(); const paymentRepository = transactionalEntityManager.getRepository(Payment);
payment.transactionSignature = transactionSignature;
await paymentRepository.save(payment); // Check if the payment with the given signature already exists
const exists = await paymentRepository.exists({ where: { transactionSignature } });
// If not, create a new payment entry
if (!exists) {
const newPayment = paymentRepository.create({ transactionSignature });
await paymentRepository.save(newPayment);
}
});
} }
export async function verifyPayment( export async function verifyPayment(