Feature/issue 13

Centaurus.Alpha/Controllers/ConstellationController.cs

@@ -37,6 +37,12 @@ public ConstellationInfo Info()
                     StellarNetwork = network,
                     Assets = assets
                 };
+                if (Global.Constellation.RequestRateLimits != null)
+                    info.RequestRateLimits = new RequestRateLimitsModel
+                    {
+                        HourLimit = Global.Constellation.RequestRateLimits.HourLimit,
+                        MinuteLimit = Global.Constellation.RequestRateLimits.MinuteLimit,
+                    };
             }
 
             return info;
@@ -50,11 +56,20 @@ public async Task<IActionResult> Init([FromBody] ConstellationInitModel constell
                 if (constellationInit == null)
                     return StatusCode(415);
 
+                RequestRateLimits requestRateLimits = null;
+                if (constellationInit.RequestRateLimits != null)
+                    requestRateLimits = new RequestRateLimits
+                    {
+                        HourLimit = constellationInit.RequestRateLimits.HourLimit,
+                        MinuteLimit = constellationInit.RequestRateLimits.MinuteLimit
+                    };
+
                 var constellationInitializer = new ConstellationInitializer(
                     constellationInit.Auditors.Select(a => KeyPair.FromAccountId(a)),
                     constellationInit.MinAccountBalance,
                     constellationInit.MinAllowedLotSize,
-                    constellationInit.Assets.Select(a => AssetSettings.FromCode(a))
+                    constellationInit.Assets.Select(a => AssetSettings.FromCode(a)),
+                    requestRateLimits
                 );
 
                 await constellationInitializer.Init();

Centaurus.Alpha/Models/ConstellationInfo.cs

@@ -1,15 +1,11 @@
 using Centaurus.Models;
 using stellar_dotnet_sdk;
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Threading.Tasks;
 
 namespace Centaurus.Alpha
 {
     public class ConstellationInfo
     {
-        public ApplicationState State{ get; set; }
+        public ApplicationState State { get; set; }
 
         public string Vault { get; set; }
 
@@ -23,6 +19,8 @@ public class ConstellationInfo
 
         public Asset[] Assets { get; set; }
 
+        public RequestRateLimitsModel RequestRateLimits { get; set; }
+
         public class Network
         {
             public Network(string passphrase, string horizon)

Centaurus.Alpha/Models/ConstellationInitModel.cs

@@ -1,9 +1,4 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Threading.Tasks;
-
-namespace Centaurus.Alpha
+namespace Centaurus.Alpha
 {
     public class ConstellationInitModel
     {
@@ -14,5 +9,7 @@ public class ConstellationInitModel
         public long MinAllowedLotSize { get; set; }
 
         public string[] Assets { get; set; }
+
+        public RequestRateLimitsModel RequestRateLimits { get; set; }
     }
 }

Centaurus.Alpha/Models/RequestRateLimitsModel.cs

@@ -0,0 +1,9 @@
+namespace Centaurus.Alpha
+{
+    public class RequestRateLimitsModel
+    {
+        public int MinuteLimit { get; set; }
+
+        public int HourLimit { get; set; }
+    }
+}

Centaurus.Common/Exceptions/ClientExceptions/TooManyRequests.cs

@@ -0,0 +1,20 @@
+using System;
+using System.Collections.Generic;
+using System.Text;
+
+namespace Centaurus
+{
+    public class TooManyRequests : BaseClientException
+    {
+        public TooManyRequests()
+        {
+
+        }
+
+        public TooManyRequests(string message)
+            : base(message)
+        {
+
+        }
+    }
+}

Centaurus.DAL/DiffObject.cs

@@ -66,6 +66,8 @@ public class Account : BaseDiffModel
             public byte[] PubKey { get; set; }
 
             public ulong Nonce { get; set; }
+
+            public RequestRateLimitsModel RequestRateLimits { get; set; }
         }
 
         public class Order : BaseDiffModel

Centaurus.DAL/Models/AccountModel.cs

@@ -4,7 +4,8 @@ public class AccountModel
     {
         public byte[] PubKey { get; set; }
 
-        //it stores ulong
         public long Nonce { get; set; }
+
+        public RequestRateLimitsModel RequestRateLimits { get; set; }
     }
 }

Centaurus.DAL/Models/RequestRateLimitModel.cs

@@ -0,0 +1,12 @@
+using System;
+using System.Collections.Generic;
+using System.Text;
+
+namespace Centaurus.DAL.Models
+{
+    public class RequestRateLimitsModel
+    {
+        public int HourLimit { get; set; }
+        public int MinuteLimit { get; set; }
+    }
+}

Centaurus.DAL/Models/SettingsModel.cs

@@ -11,10 +11,11 @@ public class SettingsModel
         public byte[][] Auditors { get; set; }
 
         public long MinAccountBalance { get; set; }
-        
+
         public long MinAllowedLotSize { get; set; }
 
-        //it stores ulong
         public long Apex { get; set; }
+
+        public RequestRateLimitsModel RequestRateLimits { get; set; }
     }
 }

Centaurus.DAL/Mongo/MongoStorage.cs

@@ -279,7 +279,9 @@ private WriteModel<ConstellationState>[] GetStellarDataUpdate(DiffObject.Constel
                 if (ledger > 0)
                     updateCommand = update.Set(s => s.Ledger, ledger);
                 if (vaultSequence > 0)
-                    updateCommand = updateCommand.Set(s => s.VaultSequence, vaultSequence) ?? update.Set(s => s.VaultSequence, vaultSequence);
+                    updateCommand = updateCommand == null
+                        ? update.Set(s => s.VaultSequence, vaultSequence)
+                        : updateCommand.Set(s => s.VaultSequence, vaultSequence);
                 updateModel = new UpdateOneModel<ConstellationState>(filter.Empty, updateCommand);
             }
 
@@ -302,11 +304,25 @@ private WriteModel<AccountModel>[] GetAccountUpdates(List<DiffObject.Account> ac
                     var pubKey = acc.PubKey;
                     var currentAccFilter = filter.Eq(a => a.PubKey, pubKey);
                     if (acc.IsInserted)
-                        updates[i] = new InsertOneModel<AccountModel>(new AccountModel { Nonce = (long)acc.Nonce, PubKey = pubKey });
+                        updates[i] = new InsertOneModel<AccountModel>(new AccountModel
+                        {
+                            Nonce = (long)acc.Nonce,
+                            PubKey = pubKey,
+                            RequestRateLimits = acc.RequestRateLimits
+                        });
                     else if (acc.IsDeleted)
                         updates[i] = new DeleteOneModel<AccountModel>(currentAccFilter);
                     else
-                        updates[i] = new UpdateOneModel<AccountModel>(currentAccFilter, update.Set(a => a.Nonce, (long)acc.Nonce));
+                    {
+                        UpdateDefinition<AccountModel> currentUpdate = null;
+                        if (acc.Nonce != 0)
+                            currentUpdate = update.Set(a => a.Nonce, (long)acc.Nonce);
+                        if (acc.RequestRateLimits != null)
+                            currentUpdate = currentUpdate == null
+                                ? update.Set(a => a.RequestRateLimits, acc.RequestRateLimits)
+                                : currentUpdate.Set(a => a.RequestRateLimits, acc.RequestRateLimits);
+                        updates[i] = new UpdateOneModel<AccountModel>(currentAccFilter, currentUpdate);
+                    }
                 }
                 return updates;
             }

Centaurus.Domain/Constellation/ConstellationInitializer.cs

@@ -17,7 +17,7 @@ public class ConstellationInitializer
     {
         const int minAuditorsCount = 1;
 
-        public ConstellationInitializer(IEnumerable<KeyPair> auditors, long minAccountBalance, long minAllowedLotSize, IEnumerable<AssetSettings> assets)
+        public ConstellationInitializer(IEnumerable<KeyPair> auditors, long minAccountBalance, long minAllowedLotSize, IEnumerable<AssetSettings> assets, RequestRateLimits requestRateLimits)
         {
             Auditors = auditors.Count() >= minAuditorsCount ? auditors.ToArray() : throw new Exception($"Min auditors count is {minAuditorsCount}");
 
@@ -28,13 +28,17 @@ public ConstellationInitializer(IEnumerable<KeyPair> auditors, long minAccountBa
             Assets = !assets.GroupBy(a => a.ToString()).Any(g => g.Count() > 1)
                 ? assets.Where(a => !a.IsXlm).ToArray() //skip XLM, it's supported by default
                 : throw new ArgumentException("All asset values should be unique");
+
+            RequestRateLimits = requestRateLimits;
         }
 
         public KeyPair[] Auditors { get; }
         public long MinAccountBalance { get; }
         public long MinAllowedLotSize { get; }
         public AssetSettings[] Assets { get; }
 
+        public RequestRateLimits RequestRateLimits { get; set; }
+
         public async Task Init()
         {
             if (Global.AppState.State != ApplicationState.WaitingForInit)
@@ -60,7 +64,8 @@ public async Task Init()
                 MinAllowedLotSize = MinAllowedLotSize,
                 PrevHash = new byte[] { },
                 Ledger = ledgerId,
-                VaultSequence = vaultAccountInfo.SequenceNumber
+                VaultSequence = vaultAccountInfo.SequenceNumber,
+                RequestRateLimits = RequestRateLimits
             };
 
             var envelope = initQuantum.CreateEnvelope();

Centaurus.Domain/Effects/Account/BaseAccountEffectProcessor.cs

@@ -24,7 +24,7 @@ public Account Account
             get
             {
                 if (account == null)
-                    account = accountStorage.GetAccount(Effect.Pubkey);
+                    account = accountStorage.GetAccount(Effect.Pubkey).Account;
                 return account;
             }
         }

Centaurus.Domain/Effects/Account/NonceUpdateEffectProcessor.cs

@@ -5,24 +5,23 @@
 
 namespace Centaurus.Domain
 {
-    public class NonceUpdateEffectProcessor : EffectProcessor<NonceUpdateEffect>
+    public class NonceUpdateEffectProcessor : BaseAccountEffectProcessor<NonceUpdateEffect>
     {
         private Account account;
 
-        public NonceUpdateEffectProcessor(NonceUpdateEffect effect, Account account)
-            : base(effect)
+        public NonceUpdateEffectProcessor(NonceUpdateEffect effect, AccountStorage accountStorage)
+            : base(effect, accountStorage)
         {
-            this.account = account;
         }
 
         public override void CommitEffect()
         {
-            account.Nonce = Effect.Nonce;
+            Account.Nonce = Effect.Nonce;
         }
 
         public override void RevertEffect()
         {
-            account.Nonce = Effect.PrevNonce;
+            Account.Nonce = Effect.PrevNonce;
         }
     }
 }

Centaurus.Domain/Effects/Account/RequestRateLimitUpdateEffectProcessor.cs

@@ -0,0 +1,26 @@
+using Centaurus.Models;
+using System;
+using System.Collections.Generic;
+using System.Text;
+
+namespace Centaurus.Domain
+{
+    public class RequestRateLimitUpdateEffectProcessor : BaseAccountEffectProcessor<RequestRateLimitUpdateEffect>
+    {
+        public RequestRateLimitUpdateEffectProcessor(RequestRateLimitUpdateEffect effect, AccountStorage accountStorage)
+            : base(effect, accountStorage)
+        {
+
+        }
+
+        public override void CommitEffect()
+        {
+            Account.RequestRateLimits = Effect.RequestRateLimits;
+        }
+
+        public override void RevertEffect()
+        {
+            Account.RequestRateLimits = Effect.PrevRequestRateLimits;
+        }
+    }
+}

Centaurus.Domain/Effects/EffectProcessorsContainer.cs

@@ -199,11 +199,11 @@ public void AddOrderRemoved(Orderbook orderbook, AccountStorage accountStorage,
 
 
 
-        public void AddNonceUpdate(Account account, ulong newNonce)
+        public void AddNonceUpdate(AccountStorage accountStorage, RawPubKey publicKey, ulong newNonce, ulong currentNonce)
         {
             Add(new NonceUpdateEffectProcessor(
-                new NonceUpdateEffect { Nonce = newNonce, PrevNonce = account.Nonce, Pubkey = account.Pubkey, Apex = Apex },
-                account
+                new NonceUpdateEffect { Nonce = newNonce, PrevNonce = currentNonce, Pubkey = publicKey, Apex = Apex },
+                accountStorage
             ));
         }
     }

Centaurus.Domain/Effects/Orders/OrderRemovedEffectProccessor.cs

@@ -25,7 +25,7 @@ public override void CommitEffect()
 
         public override void RevertEffect()
         {
-            var order = new Order { OrderId = Effect.OrderId, Price = Effect.Price, Amount = 0, Account = accountStorage.GetAccount(Effect.Pubkey) };
+            var order = new Order { OrderId = Effect.OrderId, Price = Effect.Price, Amount = 0, Account = accountStorage.GetAccount(Effect.Pubkey).Account };
             orderbook.InsertOrder(order);
         }
     }

Centaurus.Domain/MessageHandlers/AlphaHandlers/AlphaHandshakeHandler.cs

@@ -50,6 +50,7 @@ private async Task HandleClientHandshake(AlphaWebSocketConnection connection, Me
         {
             if (Global.AppState.State != ApplicationState.Ready)
                 throw new ConnectionCloseException(WebSocketCloseStatus.ProtocolError, "Alpha is not in Ready state.");
+            connection.Account = Global.AccountStorage.GetAccount(connection.ClientPubKey);
             connection.ConnectionState = ConnectionState.Ready;
             await connection.SendMessage(envelope.CreateResult(ResultStatusCodes.Success));
         }

Centaurus.Domain/MessageHandlers/AlphaHandlers/BaseAlphaMessageHandler.cs

@@ -30,6 +30,9 @@ public override async Task Validate(AlphaWebSocketConnection connection, Message
             if (IsAuditorOnly && !Global.Constellation.Auditors.Contains(connection.ClientPubKey))
                 throw new UnauthorizedException();
 
+            if (connection.Account != null && !connection.Account.RequestCounter.IncRequestCount(DateTime.UtcNow.Ticks, out string error))
+                throw new TooManyRequests(error);
+
             await base.Validate(connection, envelope);
         }
     }