feat: add indexes

Author: caiopizzol

initial.sql

@@ -41,5 +41,17 @@ CREATE TABLE IF NOT EXISTS prices (
   UNIQUE(model_year_id, reference_table_id)
 );
 
+-- Existing indexes
 CREATE INDEX IF NOT EXISTS idx_prices_reference ON prices(reference_table_id);
 CREATE INDEX IF NOT EXISTS idx_prices_fipe_code ON prices(fipe_code);
+
+-- Foreign key indexes (for JOINs)
+CREATE INDEX IF NOT EXISTS idx_models_brand_id ON models(brand_id);
+CREATE INDEX IF NOT EXISTS idx_model_years_model_id ON model_years(model_id);
+CREATE INDEX IF NOT EXISTS idx_prices_model_year_id ON prices(model_year_id);
+
+-- Filter indexes
+CREATE INDEX IF NOT EXISTS idx_models_segment ON models(segment);
+CREATE INDEX IF NOT EXISTS idx_reference_year_month ON reference_tables(year, month);
+CREATE INDEX IF NOT EXISTS idx_model_years_year ON model_years(year);
+CREATE INDEX IF NOT EXISTS idx_brands_name ON brands(name);

src/db/connection.ts

@@ -3,7 +3,7 @@ import postgres from 'postgres';
 import { env } from '../config.js';
 import * as schema from './schema.js';
 
-const client = postgres(env.DATABASE_URL);
+const client = postgres(env.DATABASE_URL, { max: 20 });
 
 export const db = drizzle(client, { schema });
 

src/db/schema.ts

@@ -9,19 +9,27 @@ import {
   index,
 } from 'drizzle-orm/pg-core';
 
-export const referenceTables = pgTable('reference_tables', {
-  id: serial('id').primaryKey(),
-  code: integer('code').unique().notNull(),
-  month: integer('month').notNull(),
-  year: integer('year').notNull(),
-  crawledAt: timestamp('crawled_at'),
-});
+export const referenceTables = pgTable(
+  'reference_tables',
+  {
+    id: serial('id').primaryKey(),
+    code: integer('code').unique().notNull(),
+    month: integer('month').notNull(),
+    year: integer('year').notNull(),
+    crawledAt: timestamp('crawled_at'),
+  },
+  (table) => [index('idx_reference_year_month').on(table.year, table.month)],
+);
 
-export const brands = pgTable('brands', {
-  id: serial('id').primaryKey(),
-  fipeCode: varchar('fipe_code', { length: 10 }).unique().notNull(),
-  name: varchar('name', { length: 100 }).notNull(),
-});
+export const brands = pgTable(
+  'brands',
+  {
+    id: serial('id').primaryKey(),
+    fipeCode: varchar('fipe_code', { length: 10 }).unique().notNull(),
+    name: varchar('name', { length: 100 }).notNull(),
+  },
+  (table) => [index('idx_brands_name').on(table.name)],
+);
 
 export const SEGMENTS = [
   'Buggy',
@@ -50,7 +58,11 @@ export const models = pgTable(
     segment: varchar('segment', { length: 20 }),
     segmentSource: varchar('segment_source', { length: 10 }),
   },
-  (table) => [unique().on(table.brandId, table.fipeCode)],
+  (table) => [
+    unique().on(table.brandId, table.fipeCode),
+    index('idx_models_brand_id').on(table.brandId),
+    index('idx_models_segment').on(table.segment),
+  ],
 );
 
 export const modelYears = pgTable(
@@ -64,7 +76,11 @@ export const modelYears = pgTable(
     fuelCode: integer('fuel_code').notNull(),
     fuelName: varchar('fuel_name', { length: 50 }),
   },
-  (table) => [unique().on(table.modelId, table.year, table.fuelCode)],
+  (table) => [
+    unique().on(table.modelId, table.year, table.fuelCode),
+    index('idx_model_years_model_id').on(table.modelId),
+    index('idx_model_years_year').on(table.year),
+  ],
 );
 
 export const prices = pgTable(
@@ -85,5 +101,6 @@ export const prices = pgTable(
     unique().on(table.modelYearId, table.referenceTableId),
     index('idx_prices_reference').on(table.referenceTableId),
     index('idx_prices_fipe_code').on(table.fipeCode),
+    index('idx_prices_model_year_id').on(table.modelYearId),
   ],
 );