using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Linq;
using BenchmarkDotNet.Attributes;
using BenchmarkDotNet.Configs;
using BenchmarkDotNet.Reports;
using BenchmarkDotNet.Columns;

public class Patient
{
    public int PatientId { get; set; }
    public string Name { get; set; }
    public List<Appointment> Appointments { get; set; } = new List<Appointment>();
}

public class Doctor
{
    public int DoctorId { get; set; }
    public string Name { get; set; }
    public List<Appointment> Appointments { get; set; } = new List<Appointment>();
}

public class Appointment
{
    public int AppointmentId { get; set; }
    public DateTime Date { get; set; }
    public int DoctorId { get; set; }
    public Doctor Doctor { get; set; }
    public int PatientId { get; set; }
    public Patient Patient { get; set; }
}

public class HospitalContext : DbContext
{
    public DbSet<Patient> Patients { get; set; }
    public DbSet<Doctor> Doctors { get; set; }
    public DbSet<Appointment> Appointments { get; set; }

    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
    {
        optionsBuilder
            .UseSqlServer(@"Server=localhost;Database=TestDB;Trusted_Connection=True;TrustServerCertificate=true;");
    }

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        modelBuilder.Entity<Patient>().HasMany(p => p.Appointments).WithOne(a => a.Patient);
        modelBuilder.Entity<Doctor>().HasMany(d => d.Appointments).WithOne(a => a.Doctor);
    }
}

public static class DbInitializer
{
    public static void Initialize(HospitalContext context)
    {
        context.Database.EnsureDeleted();
        context.Database.EnsureCreated();

        // Seed the database with dummy data
        var doctors = Enumerable.Range(1, 10).Select(i => new Doctor { Name = $"Dr. {i}" }).ToList();
        var patients = Enumerable.Range(1, 50).Select(i => new Patient { Name = $"Patient {i}" }).ToList();

        // Seed the database with dummy data using a fixed seed for consistency
        var seed = 123; // Fixed seed value
        var random = new Random(seed);

        var appointments = Enumerable.Range(1, 100).Select(i => new Appointment
        {
            Date = DateTime.Now.AddDays(random.Next(1, 365)),
            Doctor = doctors[random.Next(doctors.Count)],
            Patient = patients[random.Next(patients.Count)]
        }).ToList();

        context.Doctors.AddRange(doctors);
        context.Patients.AddRange(patients);
        context.Appointments.AddRange(appointments);
        context.SaveChanges();
    }
}

[MemoryDiagnoser]
[ReturnValueValidator(failOnError: true)]
[HideColumns(Column.AllocRatio, Column.RatioSD, Column.Median, Column.Gen0, Column.Gen1)]
[Config(typeof(Config))]
public class AsSplitQueryBenchmark
{
    private HospitalContext _context;

    [GlobalSetup]
    public void Setup()
    {
        _context = new HospitalContext();
        DbInitializer.Initialize(_context);
    }

    [Benchmark]
    public List<Appointment> LoadStandard()
    {
        return _context.Appointments
            .Include(a => a.Patient)
            .ThenInclude(p => p.Appointments) 
            .Include(a => a.Doctor)
            .ThenInclude(d => d.Appointments) 
            .ToList();
    }

    [Benchmark(Baseline = true)]
    public List<Appointment> LoadSplitQuery()
    {
        return _context.Appointments
            .Include(a => a.Patient)
            .ThenInclude(p => p.Appointments) 
            .Include(a => a.Doctor)
            .ThenInclude(d => d.Appointments) 
            .AsSplitQuery()
            .ToList();
    }

    private class Config : ManualConfig
    {
        public Config()
        {
            SummaryStyle =
                SummaryStyle.Default
                .WithRatioStyle(RatioStyle.Percentage)
                .WithTimeUnit(Perfolizer.Horology.TimeUnit.Millisecond);
        }
    }
}

// .NET 8 Lowered C# Code unavailable due to errors:
error CS0234: The type or namespace name 'EntityFrameworkCore' does not exist in the namespace 'Microsoft' (are you missing an assembly reference?)
error CS0246: The type or namespace name 'DbContext' could not be found (are you missing a using directive or an assembly reference?)
error CS0246: The type or namespace name 'DbSet<>' could not be found (are you missing a using directive or an assembly reference?)
error CS0246: The type or namespace name 'DbContextOptionsBuilder' could not be found (are you missing a using directive or an assembly reference?)
error CS0246: The type or namespace name 'ModelBuilder' could not be found (are you missing a using directive or an assembly reference?)
error CS1061: 'HospitalContext' does not contain a definition for 'Database' and no accessible extension method 'Database' accepting a first argument of type 'HospitalContext' could be found (are you missing a using directive or an assembly reference?)
error CS1061: 'HospitalContext' does not contain a definition for 'SaveChanges' and no accessible extension method 'SaveChanges' accepting a first argument of type 'HospitalContext' could be found (are you missing a using directive or an assembly reference?)

// .NET 8 IL Code unavailable due to errors:
error CS0234: The type or namespace name 'EntityFrameworkCore' does not exist in the namespace 'Microsoft' (are you missing an assembly reference?)
error CS0246: The type or namespace name 'DbContext' could not be found (are you missing a using directive or an assembly reference?)
error CS0246: The type or namespace name 'DbSet<>' could not be found (are you missing a using directive or an assembly reference?)
error CS0246: The type or namespace name 'DbContextOptionsBuilder' could not be found (are you missing a using directive or an assembly reference?)
error CS0246: The type or namespace name 'ModelBuilder' could not be found (are you missing a using directive or an assembly reference?)
error CS1061: 'HospitalContext' does not contain a definition for 'Database' and no accessible extension method 'Database' accepting a first argument of type 'HospitalContext' could be found (are you missing a using directive or an assembly reference?)
error CS1061: 'HospitalContext' does not contain a definition for 'SaveChanges' and no accessible extension method 'SaveChanges' accepting a first argument of type 'HospitalContext' could be found (are you missing a using directive or an assembly reference?)