If applied, this commit will recovery gateway from race condition when it lost.

Author: UellingtonDamasceno

src/main/java/dlt/load/balancer/model/AbstractProcessSendDeviceState.java

@@ -21,9 +21,11 @@ public abstract class AbstractProcessSendDeviceState extends AbstractBalancerSta
     private Long qtyMaxResendTansaction;
     private Device deviceToRemove;
     private BalancerState waitingLBDeviceRecivedReplyState;
+    private final String overloadedGatewaySource;
 
-    public AbstractProcessSendDeviceState(Balancer balancer) {
+    public AbstractProcessSendDeviceState(Balancer balancer, String gatewaySource) {
         super(balancer);
+        this.overloadedGatewaySource = gatewaySource;
         this.qtyMaxResendTansaction = this.balancer.qtyMaxTimeResendTransaction();
     }
 
@@ -35,20 +37,30 @@ public void onEnter() {
 
     @Override
     protected void handleValidTransaction(Transaction transaction) {
-        if(transaction == null){
+        if (transaction == null) {
             logger.warning("VIXE");
             return;
         }
-        if (!(transaction instanceof TargetedTransaction)){
+        if (!(transaction instanceof TargetedTransaction)) {
             logger.log(Level.WARNING, "Transaction {0} recived but is not a targeted transaction", transaction.getType());
             return;
         }
-        String transTarget = ((TargetedTransaction) transaction).getTarget();
-        if (!this.source.equals(transTarget)){
+        TargetedTransaction targetedTransaction = ((TargetedTransaction) transaction);
+
+        String transSource = targetedTransaction.getSource();
+        if (!transSource.equals(this.overloadedGatewaySource)) {
             return;
         }
-        logger.info("LB_*RESPONSE recebido com target correto. Iniciando envio de LB_*REQUEST.");
 
+        if (!targetedTransaction.isSameTarget(this.source)) {
+            logger.log(Level.INFO, "O gateway {1} escolheu outro alvo ({2}). Retornando ao estado Idle.",
+            new Object[]{this.overloadedGatewaySource, targetedTransaction.getTarget()});
+
+            this.balancer.transitionTo(new IdleState(this.balancer));
+            return;
+        }
+        logger.info("LB_*RESPONSE recebido com target correto. Iniciando envio de LB_*REQUEST.");
+
         Transaction transactionRequest;
         String sender = transaction.getSource();
         try {
@@ -90,7 +102,7 @@ private void updateQtyResendTransaction() {
     }
 
     protected abstract Transaction buildTransaction(String deviceJson, String sender);
-    
+
     protected abstract void handlePostSendTransaction();
 
 }

src/main/java/dlt/load/balancer/model/Balancer.java

@@ -206,7 +206,7 @@ public void updateInternalStatus(Status transaction) {
 
             startBalanceTransactionSignal = new Status(sourceIdentifier, targetGroup, true, currentDeviceLoad, transaction.getAvgLoad(), false);
             this.sendTransaction(startBalanceTransactionSignal);
-            this.transitionTo(new ProcessSingleLayerSendDeviceState(this));
+            this.transitionTo(new ProcessSingleLayerSendDeviceState(this, transaction.getSource()));
             this.messageSingleLayerSentCounter++;
             return;
         }
@@ -220,7 +220,7 @@ public void updateInternalStatus(Status transaction) {
 
             startBalanceTransactionSignal = new LBMultiRequest(sourceIdentifier, targetGroup);
             this.sendTransaction(startBalanceTransactionSignal);
-            this.transitionTo(new ProcessMultiLayerSendDeviceState(this));
+            this.transitionTo(new ProcessMultiLayerSendDeviceState(this, transaction.getSource()));
             this.messageMultiLayerSentCounter++;
         }
 

src/main/java/dlt/load/balancer/model/ProcessMultiLayerSendDeviceState.java

@@ -13,8 +13,8 @@
 public class ProcessMultiLayerSendDeviceState extends AbstractProcessSendDeviceState {
     private static final Logger logger = Logger.getLogger(ProcessMultiLayerSendDeviceState.class.getName());
 
-    public ProcessMultiLayerSendDeviceState(Balancer balancer) {
-        super(balancer);
+    public ProcessMultiLayerSendDeviceState(Balancer balancer, String gatewaySource) {
+        super(balancer, gatewaySource);
     }
 
     @Override

src/main/java/dlt/load/balancer/model/ProcessSingleLayerSendDeviceState.java

@@ -14,8 +14,8 @@ public class ProcessSingleLayerSendDeviceState extends AbstractProcessSendDevice
 
     private static final Logger logger = Logger.getLogger(ProcessMultiLayerSendDeviceState.class.getName());
 
-    public ProcessSingleLayerSendDeviceState(Balancer balancer) {
-        super(balancer);
+    public ProcessSingleLayerSendDeviceState(Balancer balancer, String gatewaySource) {
+        super(balancer, gatewaySource);
     }
 
     @Override