using System;

using BenchmarkDotNet.Attributes;

using BenchmarkDotNet.Engines;

using BenchmarkDotNet.Jobs;

namespace Casbin.Benchmark

{

[MemoryDiagnoser]

[BenchmarkCategory("Model")]

[SimpleJob(RunStrategy.Throughput, RuntimeMoniker.Net60, baseline: true)]

[SimpleJob(RunStrategy.Throughput, RuntimeMoniker.Net70)]

[SimpleJob(RunStrategy.Throughput, RuntimeMoniker.Net80)]

public class ModelBenchmark

{

private readonly Enforcer _enforcer;

public ModelBenchmark()

{

_enforcer = new Enforcer(TestHelper.GetTestFilePath("rbac_model.conf"));

}

private string NowTestUserName { get; set; }

private string NowTestDataName { get; set; }

[Params(10, 100, 1000, 10000)] public int NowPolicyCount { get; set; }

[GlobalSetup(Targets = new[] { nameof(AddPolicy), nameof(HasPolicy), nameof(RemovePolicy) })]

public void GlobalSetup()

{

for (int i = 0; i < NowPolicyCount; i++)

{

_enforcer.AddPolicy($"group{i}", $"obj{i / 10}", "read");

}

Console.WriteLine($"// Already set {NowPolicyCount} policies.");

NowTestUserName = $"name{NowPolicyCount / 2 + 1}";

NowTestDataName = $"data{NowPolicyCount / 2 + 1}";

Console.WriteLine($"// Already set user name to {NowTestUserName}.");

Console.WriteLine($"// Already set data name to {NowTestDataName}.");

}

[Benchmark]

[BenchmarkCategory("ModelManagement")]

public void HasPolicy()

{

_enforcer.HasPolicy(NowTestUserName, NowTestDataName, "read");

}

[Benchmark]

[BenchmarkCategory("ModelManagement")]

public void AddPolicy()

{

_enforcer.AddPolicy(NowTestUserName, NowTestDataName, "read");

}

[Benchmark]

[BenchmarkCategory("ModelManagement")]

public void RemovePolicy()

{

_enforcer.RemovePolicy(NowTestUserName, NowTestDataName, "read");

}

}

}