mirror/fosrl.pangolin
1
0
Fork 0
mirror of https://github.com/fosrl/pangolin.git synced 2025-07-05 11:35:00 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Add ability for sticky sessions to backend resource.

Browse source
This commit is contained in:
Matthew Evans 2025-04-11 03:00:43 -04:00
parent 8b0c30f19f
commit ac8e315fbd
6 changed files with 99 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

5
server/db/schemas/schema.ts
View file

@ -77,7 +77,10 @@ export const resources = sqliteTable("resources", {
applyRules: integer("applyRules", { mode: "boolean" })
.notNull()
.default(false),
enabled: integer("enabled", { mode: "boolean" }).notNull().default(true)
enabled: integer("enabled", { mode: "boolean" }).notNull().default(true),
stickySession: integer("stickySession", { mode: "boolean" })
.notNull()
.default(false)
});
export const targets = sqliteTable("targets", {

3
server/routers/resource/updateResource.ts
View file

@ -42,7 +42,8 @@ const updateHttpResourceBodySchema = z
isBaseDomain: z.boolean().optional(),
applyRules: z.boolean().optional(),
domainId: z.string().optional(),
enabled: z.boolean().optional()
enabled: z.boolean().optional(),
stickySession: z.boolean().optional()
})
.strict()
.refine((data) => Object.keys(data).length > 0, {

29
server/routers/traefik/getTraefikConfig.ts
View file

@ -40,7 +40,8 @@ export async function traefikConfigProvider(
org: {
orgId: orgs.orgId
},
enabled: resources.enabled
enabled: resources.enabled,
stickySession: resources.stickySession
})
.from(resources)
.innerJoin(sites, eq(sites.siteId, resources.siteId))
@ -275,7 +276,18 @@ export async function traefikConfigProvider(
url: `${target.method}://${ip}:${target.internalPort}`
};
}
})
}),
...(resource.stickySession
? {
sticky: {
cookie: {
name: "pangolin_sticky",
secure: resource.ssl,
httpOnly: true
}
}
}
: {})
}
};
} else {
@ -335,7 +347,18 @@ export async function traefikConfigProvider(
address: `${ip}:${target.internalPort}`
};
}
})
}),
...(resource.stickySession
? {
sticky: {
cookie: {
name: "pangolin_sticky",
secure: resource.ssl,
httpOnly: true
}
}
}
: {})
}
};
}

5
server/setup/migrations.ts
View file

@ -19,7 +19,7 @@ import m15 from "./scripts/1.0.0-beta15";
import m16 from "./scripts/1.0.0";
import m17 from "./scripts/1.1.0";
import m18 from "./scripts/1.2.0";
import m19 from "./scripts/1.3.0";
// THIS CANNOT IMPORT ANYTHING FROM THE SERVER
// EXCEPT FOR THE DATABASE AND THE SCHEMA
@ -37,7 +37,8 @@ const migrations = [
{ version: "1.0.0-beta.15", run: m15 },
{ version: "1.0.0", run: m16 },
{ version: "1.1.0", run: m17 },
{ version: "1.2.0", run: m18 }
{ version: "1.2.0", run: m18 },
{ version: "1.3.0", run: m19 }
// Add new migrations here as they are created
] as const;

23
server/setup/scripts/1.3.0.ts Normal file
View file

@ -0,0 +1,23 @@
import db from "@server/db";
import { sql } from "drizzle-orm";
const version = "1.3.0";
export default async function migration() {
console.log(`Running setup script ${version}...`);
try {
db.transaction((trx) => {
trx.run(
sql`ALTER TABLE resources ADD stickySession integer DEFAULT false NOT NULL;`
);
});
console.log(`Added new column: stickySession`);
} catch (e) {
console.log("Unable to add new column: stickySession");
throw e;
}
console.log(`${version} migration complete`);
}

43
src/app/[orgId]/settings/resources/[resourceId]/connectivity/page.tsx
View file

@ -94,6 +94,7 @@ export default function ReverseProxyTargets(props: {
const [site, setSite] = useState<GetSiteResponse>();
const [targetsToRemove, setTargetsToRemove] = useState<number[]>([]);
const [sslEnabled, setSslEnabled] = useState(resource.ssl);
const [stickySession, setStickySession] = useState(resource.stickySession);
const [loading, setLoading] = useState(false);
@ -317,6 +318,35 @@ export default function ReverseProxyTargets(props: {
}
}
async function saveStickySession(val: boolean) {
const res = await api
.post(`/resource/${params.resourceId}`, {
stickySession: val
})
.catch((err) => {
console.error(err);
toast({
variant: "destructive",
title: "Failed to update sticky session configuration",
description: formatAxiosError(
err,
"An error occurred while updating the sticky session configuration"
)
});
});
if (res && res.status === 200) {
setStickySession(val);
updateResource({ stickySession: val });
toast({
title: "Sticky Session Configuration",
description: "Sticky session configuration updated successfully"
});
router.refresh();
}
}
const columns: ColumnDef<LocalTarget>[] = [
{
accessorKey: "ip",
@ -456,13 +486,22 @@ export default function ReverseProxyTargets(props: {
<SettingsSection>
<SettingsSectionHeader>
<SettingsSectionTitle>
SSL Configuration
Advanced Configuration
</SettingsSectionTitle>
<SettingsSectionDescription>
Set up SSL to secure your connections with certificates
Configure advanced settings for your resource
</SettingsSectionDescription>
</SettingsSectionHeader>
<SettingsSectionBody>
<SwitchInput
id="sticky-toggle"
label="Enable Sticky Sessions"
description="Keep users on the same backend target for their entire session. Useful for applications like VNC that require persistent connections."
defaultChecked={resource.stickySession}
onCheckedChange={async (val) => {
await saveStickySession(val);
}}
/>
<SwitchInput
id="ssl-toggle"
label="Enable SSL (https)"