SCADASample/SCADASampleAPI/Data/ProcessGraphSeeder.cs at master · Correu/SCADASample · 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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
using Microsoft.EntityFrameworkCore;
using SCADASampleAPI.Models;

namespace SCADASampleAPI.Data;

public static class ProcessGraphSeeder
{
    /// <summary>Seeds a small source → mix → discharge graph for each pipeline that has no process locations yet.</summary>
    public static async Task SeedAsync(ApplicationDbContext db, CancellationToken cancellationToken = default)
    {
        var pipelines = await db.Pipelines
            .Where(p => !db.ProcessLocations.Any(l => l.PipelineId == p.PipelineId))
            .OrderBy(p => p.Code)
            .ToListAsync(cancellationToken);

        if (pipelines.Count == 0)
            return;

        var now = DateTimeOffset.UtcNow;

        foreach (var p in pipelines)
        {
            var source = new ProcessLocation
            {
                PipelineId = p.PipelineId,
                Name = "Source tank",
                Code = "SRC",
                Kind = "Tank",
                Capacity = 100,
                CurrentVolume = 75,
                LayoutX = 80,
                LayoutY = 160,
                LastUpdatedUtc = now
            };
            var mix = new ProcessLocation
            {
                PipelineId = p.PipelineId,
                Name = "Mix tank",
                Code = "MIX",
                Kind = "Tank",
                Capacity = 120,
                CurrentVolume = 45,
                LayoutX = 320,
                LayoutY = 160,
                LastUpdatedUtc = now
            };
            var discharge = new ProcessLocation
            {
                PipelineId = p.PipelineId,
                Name = "Discharge tank",
                Code = "DIS",
                Kind = "Tank",
                Capacity = 100,
                CurrentVolume = 25,
                LayoutX = 560,
                LayoutY = 160,
                LastUpdatedUtc = now
            };

            db.ProcessLocations.AddRange(source, mix, discharge);
            await db.SaveChangesAsync(cancellationToken);

            source.Fluids.Add(new ProcessLocationFluid { FluidCode = "OIL", Volume = 75 });
            mix.Fluids.Add(new ProcessLocationFluid { FluidCode = "OIL", Volume = 25 });
            mix.Fluids.Add(new ProcessLocationFluid { FluidCode = "WTR", Volume = 20 });
            discharge.Fluids.Add(new ProcessLocationFluid { FluidCode = "OIL", Volume = 25 });

            db.ProcessTransfers.AddRange(
                new ProcessTransfer
                {
                    PipelineId = p.PipelineId,
                    FromLocationId = source.ProcessLocationId,
                    ToLocationId = mix.ProcessLocationId,
                    IsPumpRunning = false,
                    ValveOpen = true,
                    MaxFlowRate = 35,
                    CurrentFlowRate = 0,
                    LastUpdatedUtc = now,
                    Fluids = [new ProcessTransferFluid { FluidCode = "OIL", FlowRateFraction = 1.0, OutflowWeight = 1.0 }]
                },
                new ProcessTransfer
                {
                    PipelineId = p.PipelineId,
                    FromLocationId = mix.ProcessLocationId,
                    ToLocationId = discharge.ProcessLocationId,
                    IsPumpRunning = true,
                    ValveOpen = true,
                    MaxFlowRate = 28,
                    CurrentFlowRate = 0,
                    LastUpdatedUtc = now,
                    Fluids = [new ProcessTransferFluid { FluidCode = "OIL", FlowRateFraction = 1.0, OutflowWeight = 1.0 }]
                });
        }

        await db.SaveChangesAsync(cancellationToken);
    }
}