cesnimda
134aac7bcf
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.
97 lines
4.5 KiB
C#
97 lines
4.5 KiB
C#
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");
|
|
}
|
|
}
|
|
}
|