Increase Speed of Getting Random Rows

njfdev · Aug 20, 2024 · 4cfe0a8 · 4cfe0a8
1 parent f6c534d
commit 4cfe0a8
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 3 deletions.
diff --git a/src/app/api/v0/faa/aircraft_model/route.ts b/src/app/api/v0/faa/aircraft_model/route.ts
@@ -10,7 +10,11 @@ export async function GET(req: NextRequest) {
   const prisma = new PrismaClient();
 
   // if no specified id, then get a random aircraft model
-  const count = await prisma.faaAircraftInfo.count();
+  const count = Number(
+    (
+      (await prisma.$queryRaw`SELECT count(*) from "FaaAircraftInfo";`) as any
+    )[0].count
+  );
   const randomIndex = Math.floor(Math.random() * count);
 
   const aircraft_info = (

diff --git a/src/app/api/v0/faa/engine_model/route.ts b/src/app/api/v0/faa/engine_model/route.ts
@@ -10,7 +10,10 @@ export async function GET(req: NextRequest) {
   const prisma = new PrismaClient();
 
   // if no specified id, then get a random engine model
-  const count = await prisma.faaEngineInfo.count();
+  const count = Number(
+    ((await prisma.$queryRaw`SELECT count(*) from "FaaEngineInfo";`) as any)[0]
+      .count
+  );
   const randomIndex = Math.floor(Math.random() * count);
 
   const engine_info = (

diff --git a/src/app/api/v0/faa/registration/route.ts b/src/app/api/v0/faa/registration/route.ts
@@ -10,7 +10,11 @@ export async function GET(req: NextRequest) {
   const prisma = new PrismaClient();
 
   // if no specified id, then get a random registration
-  const count = await prisma.faaAircraftRegistration.count();
+  const count = Number(
+    (
+      (await prisma.$queryRaw`SELECT count(*) from "FaaAircraftRegistration";`) as any
+    )[0].count
+  );
   const randomIndex = Math.floor(Math.random() * count);
 
   const registration = (