total_user_count migration

Author: sutigit

src/server/db/migrations/20252509_00_add_new_column_to_user_chat_instance_usages_table.ts

@@ -0,0 +1,15 @@
+import { DataTypes } from 'sequelize'
+
+import type { Migration } from '../connection'
+
+export const up: Migration = async ({ context: queryInterface }) => {
+  await queryInterface.addColumn('user_chat_instance_usages', 'totalUsageCount', {
+    type: DataTypes.INTEGER,
+    allowNull: false,
+    defaultValue: 0,
+  })
+}
+
+export const down: Migration = async ({ context: queryInterface }) => {
+  await queryInterface.removeColumn('user_chat_instance_usages', 'totalUsageCount')
+}

src/server/db/migrations/20252509_01_rename_column_to_user_chat_instance_usages_table.ts

@@ -0,0 +1,11 @@
+import { DataTypes } from 'sequelize'
+
+import type { Migration } from '../connection'
+
+export const up: Migration = async ({ context: queryInterface }) => {
+  await queryInterface.renameColumn('user_chat_instance_usages', 'totalUsageCount', 'total_usage_count')
+}
+
+export const down: Migration = async ({ context: queryInterface }) => {
+  await queryInterface.renameColumn('user_chat_instance_usages', 'total_usage_count', 'totalUsageCount')
+}

src/server/db/migrations/20252509_02_copy_existing_usage_count_to_total_usage_count.ts

@@ -0,0 +1,16 @@
+import { Migration } from '../connection'
+
+export const up: Migration = async ({ context: queryInterface }) => {
+  await queryInterface.sequelize.query(`  
+    UPDATE user_chat_instance_usages   
+    SET total_usage_count = usage_count   
+    WHERE total_usage_count IS NULL OR total_usage_count = 0;  
+  `)
+}
+
+export const down: Migration = async ({ context: queryInterface }) => {
+  await queryInterface.sequelize.query(`  
+    UPDATE user_chat_instance_usages   
+    SET total_usage_count = 0;  
+  `)
+}

src/server/db/models/userChatInstanceUsage.ts

@@ -10,6 +10,8 @@ class UserChatInstanceUsage extends Model<InferAttributes<UserChatInstanceUsage>
   declare chatInstanceId: string
 
   declare usageCount: CreationOptional<number>
+
+  declare totalUsageCount: CreationOptional<number>
 }
 
 UserChatInstanceUsage.init(
@@ -33,6 +35,11 @@ UserChatInstanceUsage.init(
       allowNull: false,
       defaultValue: 0,
     },
+    totalUsageCount: {
+      type: DataTypes.INTEGER,
+      allowNull: false,
+      defaultValue: 0,
+    },
   },
   {
     underscored: true,