Examples with MessageAck im.actor.core.network.mtp.entity.MessageAck used on opensource projects

Example 1 with MessageAck

use of im.actor.core.network.mtp.entity.MessageAck in project actor-platform by actorapp.

the class PusherActor method onReceive.

@Override
public void onReceive(Object message) {
    if (message instanceof SendMessage) {
        if (isEnableLog) {
            Log.d(TAG, "Received SendMessage #" + ((SendMessage) message).mid);
        }
        SendMessage sendMessage = (SendMessage) message;
        ProtoMessage holder = new ProtoMessage(sendMessage.mid, sendMessage.message);
        unsentPackages.put(holder.getMessageId(), holder);
        doSend(holder);
    } else if (message instanceof ConnectionCreated) {
        if (isEnableLog) {
            Log.d(TAG, "Received ConnectionCreated");
        }
        // Marking all pending confirms as unsent
        confirm.addAll(pendingConfirm);
        pendingConfirm.clear();
        // Resending unsent messages
        ArrayList<ProtoMessage> toSend = new ArrayList<>();
        for (ProtoMessage unsentPackage : unsentPackages.values()) {
            if (isEnableLog) {
                Log.d(TAG, "ReSending #" + unsentPackage.getMessageId());
            }
            toSend.add(unsentPackage);
        }
        // Sending SessionHello if there is no packages to sent
        if (toSend.size() == 0) {
            if (isEnableLog) {
                Log.d(TAG, "Sending SessionHello");
            }
            toSend.add(new ProtoMessage(MTUids.nextId(), new SessionHello().toByteArray()));
        }
        doSend(toSend);
    } else if (message instanceof SessionLost) {
        if (isEnableLog) {
            Log.d(TAG, "Sending SessionHello");
        }
        doSend(new ProtoMessage(MTUids.nextId(), new SessionHello().toByteArray()));
    } else if (message instanceof ForgetMessage) {
        if (isEnableLog) {
            Log.d(TAG, "Received ForgetMessage #" + ((ForgetMessage) message).mid);
        }
        unsentPackages.remove(((ForgetMessage) message).mid);
    } else if (message instanceof ConfirmMessage) {
        if (isEnableLog) {
            Log.d(TAG, "Confirming message #" + ((ConfirmMessage) message).mid);
        }
        confirm.add(((ConfirmMessage) message).mid);
        if (confirm.size() >= ACK_THRESHOLD) {
            if (askCancellable != null) {
                askCancellable.cancel();
                askCancellable = null;
            }
            askCancellable = schedule(new ForceAck(), 0);
        } else if (confirm.size() == 1) {
            if (askCancellable != null) {
                askCancellable.cancel();
                askCancellable = null;
            }
            askCancellable = schedule(new ForceAck(), ACK_DELAY);
        }
    } else if (message instanceof ForceAck) {
        if (confirm.size() == 0) {
            return;
        }
        MessageAck messageAck = buildAck();
        doSend(new ProtoMessage(MTUids.nextId(), messageAck.toByteArray()));
    } else if (message instanceof NewSession) {
        NewSession newSession = (NewSession) message;
        Log.w(TAG, "Received NewSessionCreated");
        // Clearing pending acks because of session die
        pendingConfirm.clear();
        confirm.clear();
        // Resending all required messages
        ArrayList<ProtoMessage> toSend = new ArrayList<>();
        for (ProtoMessage unsentPackage : unsentPackages.values()) {
            if (unsentPackage.getMessageId() < newSession.getMessageId()) {
                if (isEnableLog) {
                    Log.d(TAG, "ReSending #" + unsentPackage.getMessageId());
                }
                toSend.add(unsentPackage);
            }
        }
        doSend(toSend);
    } else if (message instanceof ReadPackageFromConnection) {
        // Clearing pending confirmation
        if (pendingConfirm.size() > 0) {
            pendingConfirm.clear();
        }
    }
}

Example 2 with MessageAck

use of im.actor.core.network.mtp.entity.MessageAck in project actor-platform by actorapp.

the class ReceiverActor method onReceive.

private void onReceive(ProtoMessage message) {
    sender.send(new PusherActor.ReadPackageFromConnection());
    boolean disableConfirm = false;
    try {
        if (receivedMessages.size() >= MAX_RECEIVED_BUFFER) {
            receivedMessages.remove(0);
        }
        receivedMessages.add(message.getMessageId());
        ProtoStruct obj;
        try {
            obj = ProtoSerializer.readMessagePayload(message.getPayload());
        } catch (IOException e) {
            Log.w(TAG, "Unable to parse message: ignoring");
            e.printStackTrace();
            return;
        }
        if (obj instanceof NewSessionCreated) {
            NewSessionCreated newSessionCreated = (NewSessionCreated) obj;
            sender.send(new PusherActor.NewSession(newSessionCreated.getMessageId()));
            proto.getCallback().onSessionCreated();
        } else if (obj instanceof Container) {
            Container container = (Container) obj;
            for (ProtoMessage m : container.getMessages()) {
                self().send(m, sender());
            }
        } else if (obj instanceof SessionLost) {
            sender.send(new PusherActor.SessionLost());
        } else if (obj instanceof MTRpcResponse) {
            MTRpcResponse responseBox = (MTRpcResponse) obj;
            // Forget messages
            sender.send(new PusherActor.ForgetMessage(responseBox.getMessageId(), true));
            proto.getCallback().onRpcResponse(responseBox.getMessageId(), responseBox.getPayload());
        } else if (obj instanceof MessageAck) {
            MessageAck ack = (MessageAck) obj;
            for (long ackMsgId : ack.messagesIds) {
                sender.send(new PusherActor.ForgetMessage(ackMsgId, false));
            }
        } else if (obj instanceof MTPush) {
            MTPush box = (MTPush) obj;
            proto.getCallback().onUpdate(box.getPayload());
        } else if (obj instanceof UnsentResponse) {
            UnsentResponse unsent = (UnsentResponse) obj;
            if (!receivedMessages.contains(unsent.getResponseMessageId())) {
                disableConfirm = true;
                sender.send(new PusherActor.SendMessage(MTUids.nextId(), new RequestResend(unsent.getMessageId()).toByteArray(), true));
            }
        } else if (obj instanceof UnsentMessage) {
            UnsentMessage unsent = (UnsentMessage) obj;
            if (!receivedMessages.contains(unsent.getMessageId())) {
                disableConfirm = true;
                sender.send(new PusherActor.SendMessage(MTUids.nextId(), new RequestResend(unsent.getMessageId()).toByteArray(), false));
            }
        } else if (obj instanceof AuthIdInvalid) {
            proto.getCallback().onAuthKeyInvalidated(proto.getAuthId());
            proto.stopProto();
        } else {
            Log.w(TAG, "Unsupported package " + obj);
        }
    } catch (Exception e) {
        Log.w(TAG, "Parsing error");
    } finally {
        if (!disableConfirm) {
            sender.send(new PusherActor.ConfirmMessage(message.getMessageId()));
        }
    }
}