DynamoPut permission added to send message handler,

And delete user connections if there is no remaining connection

Author: fehmianac

src/Adapter/Handler/SendMessageHandler.cs

@@ -27,7 +27,8 @@ public SendMessageHandler()
         _userConnectionRepository = RepositoryFactory.CreateUserConnectionRepository();
     }
 
-    public SendMessageHandler(IAmazonApiGatewayManagementApi amazonApiGatewayManagementApi, IUserConnectionRepository userConnectionRepository)
+    public SendMessageHandler(IAmazonApiGatewayManagementApi amazonApiGatewayManagementApi,
+        IUserConnectionRepository userConnectionRepository)
     {
         _amazonApiGatewayManagementApi = amazonApiGatewayManagementApi;
         _userConnectionRepository = userConnectionRepository;
@@ -65,19 +66,23 @@ public async Task<SQSBatchResponse> Handler(SQSEvent @event, ILambdaContext cont
                 oldConnections.Add(connection.Id);
             }
 
-            if (!userConnection.Connections.Any())
+            if (userConnection.Connections.Count == 0)
             {
                 await _userConnectionRepository.DeleteAsync(messageDomain.UserId);
             }
 
-            if (!oldConnections.Any())
+            if (oldConnections.Count == 0)
             {
                 continue;
-            } 
-                
+            }
+
             userConnection.Connections.RemoveAll(q => oldConnections.Contains(q.Id));
-            await _userConnectionRepository.SaveAsync(userConnection);
+            if (userConnection.Connections.Count != 0)
+                await _userConnectionRepository.SaveAsync(userConnection);
+            else
+                await _userConnectionRepository.DeleteAsync(messageDomain.UserId);
         }
+
         return response;
     }
 

template.yaml

@@ -409,6 +409,7 @@ Resources:
                   - dynamodb:DeleteItem
                   - dynamodb:Query
                   - dynamodb:GetItem
+                  - dynamodb:PutItem
                 Resource: !GetAtt ConnectionsTable.Arn
         - PolicyName: ApiGatewayManageConnection
           PolicyDocument: