fc3d05d65b
implement a generic user_job_events table for tracking compute job creation enforce configurable burst and sustained limits for PDF layout parsing add admin panel controls for compute rate limiting and max upload size update API routes to apply and record rate checks for PDF parse jobs document new environment variables and admin settings for compute limits improve IP extraction logic for rate limiting accuracy add tests for request IP extraction and test namespace gating This change introduces a robust mechanism to throttle expensive compute operations, such as PDF parsing, on a per-user basis. It provides both burst and sustained rate controls, with admin-tunable parameters and clear user feedback on throttling. The job event ledger enables accurate concurrency and rate enforcement, while new documentation and tests ensure maintainability and clarity.
21 lines
1.7 KiB
TypeScript
21 lines
1.7 KiB
TypeScript
import * as sqliteSchema from './schema_sqlite';
|
|
import * as postgresSchema from './schema_postgres';
|
|
|
|
const usePostgres = !!process.env.POSTGRES_URL;
|
|
|
|
// Auth tables (user, session, account, verification) are managed by Better Auth
|
|
// and are NOT part of the Drizzle schema. Only app-specific tables are exported here.
|
|
|
|
export const documents = usePostgres ? postgresSchema.documents : sqliteSchema.documents;
|
|
export const audiobooks = usePostgres ? postgresSchema.audiobooks : sqliteSchema.audiobooks;
|
|
export const audiobookChapters = usePostgres ? postgresSchema.audiobookChapters : sqliteSchema.audiobookChapters;
|
|
export const userTtsChars = usePostgres ? postgresSchema.userTtsChars : sqliteSchema.userTtsChars;
|
|
export const userJobEvents = usePostgres ? postgresSchema.userJobEvents : sqliteSchema.userJobEvents;
|
|
export const userPreferences = usePostgres ? postgresSchema.userPreferences : sqliteSchema.userPreferences;
|
|
export const documentSettings = usePostgres ? postgresSchema.documentSettings : sqliteSchema.documentSettings;
|
|
export const userDocumentProgress = usePostgres ? postgresSchema.userDocumentProgress : sqliteSchema.userDocumentProgress;
|
|
export const documentPreviews = usePostgres ? postgresSchema.documentPreviews : sqliteSchema.documentPreviews;
|
|
export const ttsSegmentEntries = usePostgres ? postgresSchema.ttsSegmentEntries : sqliteSchema.ttsSegmentEntries;
|
|
export const ttsSegmentVariants = usePostgres ? postgresSchema.ttsSegmentVariants : sqliteSchema.ttsSegmentVariants;
|
|
export const adminProviders = usePostgres ? postgresSchema.adminProviders : sqliteSchema.adminProviders;
|
|
export const adminSettings = usePostgres ? postgresSchema.adminSettings : sqliteSchema.adminSettings;
|