using System;
using System.Collections.Generic;
using System.Linq;
using BenchmarkDotNet.Attributes;
using Microsoft.EntityFrameworkCore;

namespace Benchmarks; 

[MemoryDiagnoser]
public class QueryTrackingBehavior
{
    [Params(10)]
    public int NumBlogs { get; set; }

    [Params(20)]
    public int NumPostsPerBlog { get; set; }

    [GlobalSetup]
    public void Setup()
    {
        Console.WriteLine("Setting up database...");
        using var context = new BloggingContext();
        context.Database.EnsureDeleted();
        context.Database.EnsureCreated();
        BloggingContext.SeedData(NumBlogs, NumPostsPerBlog);
        Console.WriteLine("Setup complete.");
    }

    [Benchmark(Baseline = true)]
    public List<Post> AsTracking()
    {
        using var context = new BloggingContext();

        return context.Posts.AsTracking().Include(p => p.Blog).ToList();
    }

    [Benchmark]
    public List<Post> AsNoTracking()
    {
        using var context = new BloggingContext();

        return context.Posts.AsNoTracking().Include(p => p.Blog).ToList();
    }

    public class BloggingContext : DbContext
    {
        public DbSet<Blog> Blogs { get; set; }
        public DbSet<Post> Posts { get; set; }

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

        public static void SeedData(int numBlogs, int numPostsPerBlog)
        {
            using var context = new BloggingContext();
            context.AddRange(
                Enumerable.Range(0, numBlogs)
                    .Select(_ => new Blog { Url = "Some URL", Posts = Enumerable.Range(0, numPostsPerBlog)
                    .Select(_ => new Post() { Title = "Some Title", Content = "Some Content"}).ToList() }));
            context.SaveChanges();
        }
    }

    public class Blog
    {
        public int BlogId { get; set; }
        public string Url { get; set; }
        public int Rating { get; set; }
        public List<Post> Posts { get; set; }
    }

    public class Post
    {
        public int PostId { get; set; }
        public string Title { get; set; }
        public string Content { get; set; }

        public int BlogId { get; set; }
        public Blog Blog { get; set; }
    }
}

// .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 CS1674: 'QueryTrackingBehavior.BloggingContext': type used in a using statement must be implicitly convertible to 'System.IDisposable'.
error CS1061: 'QueryTrackingBehavior.BloggingContext' does not contain a definition for 'Database' and no accessible extension method 'Database' accepting a first argument of type 'QueryTrackingBehavior.BloggingContext' could be found (are you missing a using directive or an assembly reference?)
error CS1061: 'QueryTrackingBehavior.BloggingContext' does not contain a definition for 'AddRange' and no accessible extension method 'AddRange' accepting a first argument of type 'QueryTrackingBehavior.BloggingContext' could be found (are you missing a using directive or an assembly reference?)
error CS1061: 'QueryTrackingBehavior.BloggingContext' does not contain a definition for 'SaveChanges' and no accessible extension method 'SaveChanges' accepting a first argument of type 'QueryTrackingBehavior.BloggingContext' 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 CS1674: 'QueryTrackingBehavior.BloggingContext': type used in a using statement must be implicitly convertible to 'System.IDisposable'.
error CS1061: 'QueryTrackingBehavior.BloggingContext' does not contain a definition for 'Database' and no accessible extension method 'Database' accepting a first argument of type 'QueryTrackingBehavior.BloggingContext' could be found (are you missing a using directive or an assembly reference?)
error CS1061: 'QueryTrackingBehavior.BloggingContext' does not contain a definition for 'AddRange' and no accessible extension method 'AddRange' accepting a first argument of type 'QueryTrackingBehavior.BloggingContext' could be found (are you missing a using directive or an assembly reference?)
error CS1061: 'QueryTrackingBehavior.BloggingContext' does not contain a definition for 'SaveChanges' and no accessible extension method 'SaveChanges' accepting a first argument of type 'QueryTrackingBehavior.BloggingContext' could be found (are you missing a using directive or an assembly reference?)