pangolin_mirror/pangolin
1
0
Fork 0
mirror of https://github.com/fosrl/pangolin.git synced 2026-03-03 09:16:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Pass in db to pickPort

Browse Source
This commit is contained in:
Owen
2025-09-04 17:59:59 -07:00
parent 3e353717f5
commit 838e3efbca
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
server/routers/target/helpers.ts
View File

@@ -1,10 +1,10 @@
import { db } from "@server/db";
import { db, Transaction } from "@server/db";
import { resources, targets } from "@server/db";
import { eq } from "drizzle-orm";
const currentBannedPorts: number[] = [];
export async function pickPort(siteId: number): Promise<{
export async function pickPort(siteId: number, trx: Transaction | typeof db): Promise<{
internalPort: number;
targetIps: string[];
}> {
@@ -12,7 +12,7 @@ export async function pickPort(siteId: number): Promise<{
const targetIps: string[] = [];
const targetInternalPorts: number[] = [];
const targetsRes = await db
const targetsRes = await trx
.select()
.from(targets)
.where(eq(targets.siteId, siteId));