Control visibility of IRateLimiter components

App-vNext · Oct 15, 2021 · e06b0bc · e06b0bc
1 parent 88d6038
commit e06b0bc
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 7 deletions.
diff --git a/src/Polly.Specs/Helpers/RateLimit/IRateLimiterExtensions.cs b/src/Polly.Specs/Helpers/RateLimit/IRateLimiterExtensions.cs
@@ -4,7 +4,7 @@
 
 namespace Polly.Specs.Helpers.RateLimit
 {
-    public static class IRateLimiterExtensions
+    internal static class IRateLimiterExtensions
     {
         public static void ShouldPermitAnExecution(this IRateLimiter rateLimiter)
         {

diff --git a/src/Polly.Specs/RateLimit/LockBasedTokenBucketRateLimiterTests.cs b/src/Polly.Specs/RateLimit/LockBasedTokenBucketRateLimiterTests.cs
@@ -5,7 +5,7 @@ namespace Polly.Specs.RateLimit
 {
     public class LockBasedTokenBucketRateLimiterTests : TokenBucketRateLimiterTestsBase
     {
-        public override IRateLimiter GetRateLimiter(TimeSpan onePer, long bucketCapacity)
+        internal override IRateLimiter GetRateLimiter(TimeSpan onePer, long bucketCapacity)
             => new LockBasedTokenBucketRateLimiter(onePer, bucketCapacity);
     }
 }
diff --git a/src/Polly.Specs/RateLimit/LockFreeTokenBucketRateLimiterTests.cs b/src/Polly.Specs/RateLimit/LockFreeTokenBucketRateLimiterTests.cs
@@ -5,7 +5,7 @@ namespace Polly.Specs.RateLimit
 {
     public class LockFreeTokenBucketRateLimiterTests : TokenBucketRateLimiterTestsBase
     {
-        public override IRateLimiter GetRateLimiter(TimeSpan onePer, long bucketCapacity)
+        internal override IRateLimiter GetRateLimiter(TimeSpan onePer, long bucketCapacity)
             => new LockFreeTokenBucketRateLimiter(onePer, bucketCapacity);
     }
 }
diff --git a/src/Polly.Specs/RateLimit/TokenBucketRateLimiterTestsBase.cs b/src/Polly.Specs/RateLimit/TokenBucketRateLimiterTestsBase.cs
@@ -13,7 +13,7 @@ namespace Polly.Specs.RateLimit
     [Collection(Polly.Specs.Helpers.Constants.SystemClockDependentTestCollection)]
     public abstract class TokenBucketRateLimiterTestsBase : RateLimitSpecsBase, IDisposable
     {
-        public abstract IRateLimiter GetRateLimiter(TimeSpan onePer, long bucketCapacity);
+        internal abstract IRateLimiter GetRateLimiter(TimeSpan onePer, long bucketCapacity);
 
         public void Dispose()
         {

diff --git a/src/Polly/RateLimit/IRateLimiter.cs b/src/Polly/RateLimit/IRateLimiter.cs
@@ -5,7 +5,7 @@ namespace Polly.RateLimit
     /// <summary>
     /// Defines methods to be provided by a rate-limiter used in a Polly <see cref="IRateLimitPolicy"/>
     /// </summary>
-    public interface IRateLimiter
+    internal interface IRateLimiter
     {
         /// <summary>
         /// Returns whether the execution is permitted; if not, returns what <see cref="TimeSpan"/> should be waited before retrying.

diff --git a/src/Polly/RateLimit/LockBasedTokenBucketRateLimiter.cs b/src/Polly/RateLimit/LockBasedTokenBucketRateLimiter.cs
@@ -6,7 +6,7 @@ namespace Polly.RateLimit
     /// <summary>
     /// A lock-based token-bucket rate-limiter for a Polly <see cref="IRateLimitPolicy"/>.
     /// </summary>
-    public class LockBasedTokenBucketRateLimiter : IRateLimiter
+    internal class LockBasedTokenBucketRateLimiter : IRateLimiter
     {
         private readonly long addTokenTickInterval;
         private readonly long bucketCapacity;

diff --git a/src/Polly/RateLimit/LockFreeTokenBucketRateLimiter.cs b/src/Polly/RateLimit/LockFreeTokenBucketRateLimiter.cs
@@ -7,7 +7,7 @@ namespace Polly.RateLimit
     /// <summary>
     /// A lock-free token-bucket rate-limiter for a Polly <see cref="IRateLimitPolicy"/>.
     /// </summary>
-    public class LockFreeTokenBucketRateLimiter : IRateLimiter
+    internal class LockFreeTokenBucketRateLimiter : IRateLimiter
     {
         private readonly long addTokenTickInterval;
         private readonly long bucketCapacity;