refactor: add m2m and user token usage cols to daily token usage table (#8029)

* refactor: add m2m and user token usage cols to daily token usage table

* refactor: update code

Author: darcyYe

packages/schemas/alterations/next-1764653048-update-daily-token-usage-mau-support.ts

@@ -0,0 +1,37 @@
+import { sql } from '@silverhand/slonik';
+
+import type { AlterationScript } from '../lib/types/alteration.js';
+
+const alteration: AlterationScript = {
+  up: async (pool) => {
+    // Add new columns for user and m2m token usage tracking
+    await pool.query(sql`
+      alter table daily_token_usage
+      add column user_token_usage bigint not null default 0,
+      add column m2m_token_usage bigint not null default 0;
+    `);
+
+    // Remove foreign key constraint to support enterprise tenant deletion requirements
+    await pool.query(sql`
+      alter table daily_token_usage
+      drop constraint if exists daily_token_usage_tenant_id_fkey;
+    `);
+  },
+  down: async (pool) => {
+    // Remove the new columns
+    await pool.query(sql`
+      alter table daily_token_usage
+      drop column if exists user_token_usage,
+      drop column if exists m2m_token_usage;
+    `);
+
+    // Re-add the foreign key constraint
+    await pool.query(sql`
+      alter table daily_token_usage
+      add constraint daily_token_usage_tenant_id_fkey
+      foreign key (tenant_id) references tenants (id) on update cascade on delete cascade;
+    `);
+  },
+};
+
+export default alteration;

packages/schemas/tables/daily_token_usage.sql

@@ -1,8 +1,9 @@
 create table daily_token_usage (
   id varchar(21) not null,
-  tenant_id varchar(21) not null
-    references tenants (id) on update cascade on delete cascade,
+  tenant_id varchar(21) not null,
   usage bigint not null default(0),
+  user_token_usage bigint not null default(0),
+  m2m_token_usage bigint not null default(0),
   date timestamptz not null,
   primary key (id)
 );