feat(ai): centralize durable usage

Add a content-free usage ledger with legacy backfill. Reserve Workspace and durable Strategy/CV work before execution so deleted history or duplicate admission cannot reset limits.
2026-08-15 20:03:06 +02:00
parent dbff0f8d49
commit 134aac7bcf
28 changed files with 3539 additions and 83 deletions
@@ -0,0 +1,96 @@
+using System;
+using Microsoft.EntityFrameworkCore.Migrations;
+
+#nullable disable
+
+namespace JobTrackerApi.Migrations
+{
+    /// <inheritdoc />
+    public partial class AddCrossFeatureAiUsage : Migration
+    {
+        /// <inheritdoc />
+        protected override void Up(MigrationBuilder migrationBuilder)
+        {
+            if (ActiveProvider.Contains("MySql", StringComparison.OrdinalIgnoreCase))
+            {
+                migrationBuilder.Sql("""
+                    CREATE TABLE `AiUsageRecords` (
+                        `Id` bigint NOT NULL AUTO_INCREMENT,
+                        `OwnerUserId` varchar(255) NOT NULL,
+                        `SourceType` varchar(32) NOT NULL,
+                        `SourceId` varchar(64) NOT NULL,
+                        `TaskType` varchar(64) NOT NULL,
+                        `CallCount` int NOT NULL,
+                        `InputCharacterCount` int NOT NULL,
+                        `OutputCharacterCount` int NOT NULL,
+                        `EstimatedTokenCount` int NOT NULL,
+                        `CreatedAtUtc` datetime(6) NOT NULL,
+                        CONSTRAINT `PK_AiUsageRecords` PRIMARY KEY (`Id`)
+                    ) CHARACTER SET=utf8mb4;
+                    """);
+            }
+            else
+            {
+            migrationBuilder.CreateTable(
+                name: "AiUsageRecords",
+                columns: table => new
+                {
+                    Id = table.Column<long>(type: "INTEGER", nullable: false)
+                        .Annotation("Sqlite:Autoincrement", true),
+                    OwnerUserId = table.Column<string>(type: "TEXT", maxLength: 255, nullable: false),
+                    SourceType = table.Column<string>(type: "TEXT", maxLength: 32, nullable: false),
+                    SourceId = table.Column<string>(type: "TEXT", maxLength: 64, nullable: false),
+                    TaskType = table.Column<string>(type: "TEXT", maxLength: 64, nullable: false),
+                    CallCount = table.Column<int>(type: "INTEGER", nullable: false),
+                    InputCharacterCount = table.Column<int>(type: "INTEGER", nullable: false),
+                    OutputCharacterCount = table.Column<int>(type: "INTEGER", nullable: false),
+                    EstimatedTokenCount = table.Column<int>(type: "INTEGER", nullable: false),
+                    CreatedAtUtc = table.Column<DateTimeOffset>(type: "TEXT", nullable: false)
+                },
+                constraints: table =>
+                {
+                    table.PrimaryKey("PK_AiUsageRecords", x => x.Id);
+                });
+            }
+
+            migrationBuilder.CreateIndex(
+                name: "IX_AiUsageRecords_OwnerUserId_CreatedAtUtc",
+                table: "AiUsageRecords",
+                columns: new[] { "OwnerUserId", "CreatedAtUtc" });
+
+            migrationBuilder.CreateIndex(
+                name: "IX_AiUsageRecords_OwnerUserId_SourceType_SourceId",
+                table: "AiUsageRecords",
+                columns: new[] { "OwnerUserId", "SourceType", "SourceId" },
+                unique: true);
+
+            if (ActiveProvider.Contains("MySql", StringComparison.OrdinalIgnoreCase))
+            {
+                migrationBuilder.Sql("""
+                    INSERT IGNORE INTO `AiUsageRecords`
+                        (`OwnerUserId`, `SourceType`, `SourceId`, `TaskType`, `CallCount`, `InputCharacterCount`, `OutputCharacterCount`, `EstimatedTokenCount`, `CreatedAtUtc`)
+                    SELECT `OwnerUserId`, 'workspace-legacy', CAST(`Id` AS CHAR), `Module`, 1,
+                        `InputCharacterCount`, `OutputCharacterCount`, `EstimatedTokenCount`, `CreatedAtUtc`
+                    FROM `AiInteractions`;
+                    """);
+            }
+            else
+            {
+                migrationBuilder.Sql("""
+                    INSERT OR IGNORE INTO "AiUsageRecords"
+                        ("OwnerUserId", "SourceType", "SourceId", "TaskType", "CallCount", "InputCharacterCount", "OutputCharacterCount", "EstimatedTokenCount", "CreatedAtUtc")
+                    SELECT "OwnerUserId", 'workspace-legacy', CAST("Id" AS TEXT), "Module", 1,
+                        "InputCharacterCount", "OutputCharacterCount", "EstimatedTokenCount", "CreatedAtUtc"
+                    FROM "AiInteractions";
+                    """);
+            }
+        }
+
+        /// <inheritdoc />
+        protected override void Down(MigrationBuilder migrationBuilder)
+        {
+            migrationBuilder.DropTable(
+                name: "AiUsageRecords");
+        }
+    }
+}