Examples with ReadRequest org.monkey.mmq.core.entity.ReadRequest used on opensource projects

Example 1 with ReadRequest

use of org.monkey.mmq.core.entity.ReadRequest in project mmqtt by MrHKing.

the class StandalonePersistentServiceProcessor method get.

@Override
public Datum get(String key) throws MmqException {
    final List<byte[]> keys = Collections.singletonList(ByteUtils.toBytes(key));
    final ReadRequest req = ReadRequest.newBuilder().setGroup(this.raftGroup).setData(ByteString.copyFrom(serializer.serialize(keys))).build();
    try {
        final Response resp = onRequest(req);
        if (resp.getSuccess()) {
            BatchReadResponse response = serializer.deserialize(resp.getData().toByteArray(), BatchReadResponse.class);
            final List<byte[]> rValues = response.getValues();
            return rValues.isEmpty() ? null : serializer.deserialize(rValues.get(0), getDatumTypeFromKey(key));
        }
        throw new MmqException(ErrorCode.ProtoReadError.getCode(), resp.getErrMsg());
    } catch (Throwable e) {
        throw new MmqException(ErrorCode.ProtoReadError.getCode(), e.getMessage());
    }
}

Example 2 with ReadRequest

use of org.monkey.mmq.core.entity.ReadRequest in project mmqtt by MrHKing.

the class PersistentServiceProcessor method get.

@Override
public Datum get(String key) throws MmqException {
    final List<byte[]> keys = new ArrayList<>(1);
    keys.add(ByteUtils.toBytes(key));
    final ReadRequest req = ReadRequest.newBuilder().setGroup(this.raftGroup).setData(ByteString.copyFrom(serializer.serialize(keys))).build();
    try {
        Response resp = protocol.getData(req);
        if (resp.getSuccess()) {
            BatchReadResponse response = serializer.deserialize(resp.getData().toByteArray(), BatchReadResponse.class);
            final List<byte[]> rValues = response.getValues();
            return rValues.isEmpty() ? null : serializer.deserialize(rValues.get(0), getDatumTypeFromKey(key));
        }
        throw new MmqException(ErrorCode.ProtoReadError.getCode(), resp.getErrMsg());
    } catch (Throwable e) {
        throw new MmqException(ErrorCode.ProtoReadError.getCode(), e.getMessage());
    }
}

Example 3 with ReadRequest

use of org.monkey.mmq.core.entity.ReadRequest in project mmqtt by MrHKing.

the class MmqStateMachine method onApply.

@Override
public void onApply(Iterator iter) {
    int index = 0;
    int applied = 0;
    Message message;
    MmqClosure closure = null;
    try {
        while (iter.hasNext()) {
            Status status = Status.OK();
            try {
                if (iter.done() != null) {
                    closure = (MmqClosure) iter.done();
                    message = closure.getMessage();
                } else {
                    final ByteBuffer data = iter.getData();
                    message = ProtoMessageUtil.parse(data.array());
                }
                Loggers.RAFT.info("receive log : {}", message);
                LoggerUtils.printIfDebugEnabled(Loggers.RAFT, "receive log : {}", message);
                if (message instanceof WriteRequest) {
                    Response response = processor.onApply((WriteRequest) message);
                    postProcessor(response, closure);
                }
                if (message instanceof ReadRequest) {
                    Response response = processor.onRequest((ReadRequest) message);
                    postProcessor(response, closure);
                }
            } catch (Throwable e) {
                index++;
                status.setError(RaftError.UNKNOWN, e.toString());
                Optional.ofNullable(closure).ifPresent(closure1 -> closure1.setThrowable(e));
                throw e;
            } finally {
                Optional.ofNullable(closure).ifPresent(closure1 -> closure1.run(status));
            }
            applied++;
            index++;
            iter.next();
        }
    } catch (Throwable t) {
        Loggers.RAFT.error("processor : {}, stateMachine meet critical error: {}.", processor, t);
        iter.setErrorAndRollback(index - applied, new Status(RaftError.ESTATEMACHINE, "StateMachine meet critical error: %s.", ExceptionUtil.getStackTrace(t)));
    }
}