mirror of
https://github.com/Memo-2023/mana-monorepo.git
synced 2026-05-19 12:01:24 +02:00
6f1b2654f1
Implement custom NestJS API Gateway for mana-search, mana-stt, and mana-tts: - API Key management with CRUD operations and key regeneration - Redis-based sliding window rate limiting - Credit-based billing with tier support (free, pro, enterprise) - Usage tracking with daily aggregates - Proxy services to backend microservices - Prometheus metrics endpoint - JWT auth for management API, API key auth for public API Database schema uses separate `api_gateway` schema in shared manacore DB.
70 lines
2.4 KiB
TypeScript
70 lines
2.4 KiB
TypeScript
import { uuid, text, integer, timestamp, jsonb, index, unique, date } from 'drizzle-orm/pg-core';
|
|
import { apiGatewaySchema, apiKeys } from './api-keys.schema';
|
|
|
|
// Detailed usage log
|
|
export const apiUsage = apiGatewaySchema.table(
|
|
'api_usage',
|
|
{
|
|
id: uuid('id').defaultRandom().primaryKey(),
|
|
|
|
// Key reference
|
|
apiKeyId: uuid('api_key_id')
|
|
.references(() => apiKeys.id, { onDelete: 'cascade' })
|
|
.notNull(),
|
|
|
|
// Request details
|
|
endpoint: text('endpoint').notNull(), // search, stt, tts
|
|
method: text('method').notNull(), // POST, GET
|
|
path: text('path').notNull(), // /v1/search
|
|
|
|
// Metrics
|
|
requestSize: integer('request_size'), // bytes
|
|
responseSize: integer('response_size'), // bytes
|
|
latencyMs: integer('latency_ms'), // milliseconds
|
|
statusCode: integer('status_code'), // 200, 400, 500...
|
|
|
|
// Credit calculation
|
|
creditsUsed: integer('credits_used').notNull().default(0),
|
|
creditReason: text('credit_reason'), // "1000 characters TTS"
|
|
|
|
// Metadata
|
|
metadata: jsonb('metadata'), // additional context (user agent, etc.)
|
|
|
|
// Timestamp
|
|
createdAt: timestamp('created_at', { withTimezone: true }).defaultNow().notNull(),
|
|
},
|
|
(table) => ({
|
|
apiKeyIdIdx: index('api_usage_api_key_id_idx').on(table.apiKeyId),
|
|
createdAtIdx: index('api_usage_created_at_idx').on(table.createdAt),
|
|
endpointIdx: index('api_usage_endpoint_idx').on(table.endpoint),
|
|
})
|
|
);
|
|
|
|
// Aggregated daily usage (for dashboard/billing)
|
|
export const apiUsageDaily = apiGatewaySchema.table(
|
|
'api_usage_daily',
|
|
{
|
|
id: uuid('id').defaultRandom().primaryKey(),
|
|
apiKeyId: uuid('api_key_id')
|
|
.references(() => apiKeys.id, { onDelete: 'cascade' })
|
|
.notNull(),
|
|
date: date('date').notNull(),
|
|
endpoint: text('endpoint').notNull(),
|
|
|
|
// Aggregates
|
|
requestCount: integer('request_count').notNull().default(0),
|
|
creditsUsed: integer('credits_used').notNull().default(0),
|
|
totalLatencyMs: integer('total_latency_ms').notNull().default(0),
|
|
errorCount: integer('error_count').notNull().default(0),
|
|
},
|
|
(table) => ({
|
|
// Unique constraint for upsert
|
|
uniqueDaily: unique('api_usage_daily_unique').on(table.apiKeyId, table.date, table.endpoint),
|
|
dateIdx: index('api_usage_daily_date_idx').on(table.date),
|
|
})
|
|
);
|
|
|
|
export type ApiUsage = typeof apiUsage.$inferSelect;
|
|
export type NewApiUsage = typeof apiUsage.$inferInsert;
|
|
export type ApiUsageDaily = typeof apiUsageDaily.$inferSelect;
|
|
export type NewApiUsageDaily = typeof apiUsageDaily.$inferInsert;
|