CmdScale.EntityFrameworkCore.TimescaleDB/src/Eftdb.Design/TimescaleCSharpMigrationOperationGenerator.cs at e5c7d30564eeafb13380402205f74d54140c44de · cmdscale/CmdScale.EntityFrameworkCore.TimescaleDB · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
using CmdScale.EntityFrameworkCore.TimescaleDB.Generators;
using CmdScale.EntityFrameworkCore.TimescaleDB.Operations;
using Microsoft.EntityFrameworkCore.Infrastructure;
using Microsoft.EntityFrameworkCore.Migrations.Design;
using Microsoft.EntityFrameworkCore.Migrations.Operations;

namespace CmdScale.EntityFrameworkCore.TimescaleDB.Design
{
    public class TimescaleCSharpMigrationOperationGenerator(CSharpMigrationOperationGeneratorDependencies dependencies) : CSharpMigrationOperationGenerator(dependencies)
    {
        protected override void Generate(MigrationOperation operation, IndentedStringBuilder builder)
        {
            ArgumentNullException.ThrowIfNull(operation);
            ArgumentNullException.ThrowIfNull(builder);

            HypertableOperationGenerator? hypertableOperationGenerator = null;
            ReorderPolicyOperationGenerator? reorderPolicyOperationGenerator = null;
            ContinuousAggregateOperationGenerator? continuousAggregateOperationGenerator = null;

            List<string> statements;
            bool suppressTransaction = false;

            switch (operation)
            {
                case CreateHypertableOperation create:
                    hypertableOperationGenerator ??= new(isDesignTime: true);
                    statements = hypertableOperationGenerator.Generate(create);
                    break;
                case AlterHypertableOperation alter:
                    hypertableOperationGenerator ??= new(isDesignTime: true);
                    statements = hypertableOperationGenerator.Generate(alter);
                    break;

                case AddReorderPolicyOperation addReorder:
                    reorderPolicyOperationGenerator ??= new(isDesignTime: true);
                    statements = reorderPolicyOperationGenerator.Generate(addReorder);
                    break;
                case AlterReorderPolicyOperation alterReorder:
                    reorderPolicyOperationGenerator ??= new(isDesignTime: true);
                    statements = reorderPolicyOperationGenerator.Generate(alterReorder);
                    break;
                case DropReorderPolicyOperation dropReorder:
                    reorderPolicyOperationGenerator ??= new(isDesignTime: true);
                    statements = reorderPolicyOperationGenerator.Generate(dropReorder);
                    break;

                case CreateContinuousAggregateOperation createContinuousAggregate:
                    continuousAggregateOperationGenerator ??= new(isDesignTime: true);
                    statements = continuousAggregateOperationGenerator.Generate(createContinuousAggregate);
                    suppressTransaction = true;
                    break;
                case AlterContinuousAggregateOperation alterContinuousAggregate:
                    continuousAggregateOperationGenerator ??= new(isDesignTime: true);
                    statements = continuousAggregateOperationGenerator.Generate(alterContinuousAggregate);
                    break;
                case DropContinuousAggregateOperation dropContinuousAggregate:
                    continuousAggregateOperationGenerator ??= new(isDesignTime: true);
                    statements = continuousAggregateOperationGenerator.Generate(dropContinuousAggregate);
                    break;

                default:
                    base.Generate(operation, builder);
                    return;
            }

            // Guard: if no statements were generated, output a no-op SQL comment to maintain valid C# syntax.
            if (statements.Count == 0)
            {
                builder.Append(".Sql(@\"-- No SQL generated for this operation\")");
                return;
            }

            SqlBuilderHelper.BuildQueryString(statements, builder, suppressTransaction);
        }

    }
}