// Path: src/main/java/org/tools4j/hoverraft/command/Command.java
// public interface Command extends DirectPayload, Event {
// 
//     CommandKey commandKey();
// 
//     CommandPayload commandPayload();
// 
//     default Command sourceId(final int sourceId) {
//         commandKey().sourceId(sourceId);
//         return this;
//     }
// 
//     default Command commandIndex(final long commandIndex) {
//         commandKey().commandIndex(commandIndex);
//         return this;
//     }
// 
//     default void copyFrom(final Command command) {
//         writeBufferOrNull().putBytes(offset(), command.readBufferOrNull(), command.offset(), command.byteLength());
//     }
// 
//     @Override
//     default Transition accept(final ServerContext serverContext, final EventHandler eventHandler) {
//         return eventHandler.onCommand(serverContext, this);
//     }
// 
// }
// 
// Path: src/main/java/org/tools4j/hoverraft/direct/DirectFactory.java
// public interface DirectFactory {
// 
//     AppendRequest appendRequest();
// 
//     AppendResponse appendResponse();
// 
//     VoteRequest voteRequest();
// 
//     VoteResponse voteResponse();
// 
//     TimeoutNow timeoutNow();
// 
//     CommandKey commandKey();
// 
//     Command command();
// 
//     LogEntry logEntry();
// }
// 
// Path: src/main/java/org/tools4j/hoverraft/direct/DirectPayload.java
// public interface DirectPayload {
// 
//     int byteLength();
// 
//     int offset();
// 
//     DirectBuffer readBufferOrNull();
// 
//     MutableDirectBuffer writeBufferOrNull();
// 
//     void wrap(DirectBuffer buffer, int offset);
// 
//     void wrap(MutableDirectBuffer buffer, int offset);
// 
//     void unwrap();
// 
//     default boolean isWrapped() {
//         return readBufferOrNull() != null || writeBufferOrNull() != null;
//     }
// }
// 
// Path: src/main/java/org/tools4j/hoverraft/message/Message.java
// public interface Message extends DirectPayload, Event {
// 
//     MessageType type();
// 
//     default void sendTo(final Sender<? super Message> sender, final ResendStrategy resendStrategy) {
//         final long res = sender.offer(this);
//         if (res < 0) {
//             resendStrategy.onRejectedOffer(sender, this, res);
//         }
//     }
// 
// 
// }
// 
// Path: src/main/java/org/tools4j/hoverraft/message/MessageType.java
// public enum MessageType {
//     VOTE_REQUEST {
//         @Override
//         public Message create(final DirectFactory factory) {
//             return factory.voteRequest();
//         }
//     },
//     VOTE_RESPONSE {
//         @Override
//         public Message create(final DirectFactory factory) {
//             return factory.voteResponse();
//         }
//     },
//     APPEND_REQUEST {
//         @Override
//         public Message create(final DirectFactory factory) {
//             return factory.appendRequest();
//         }
//     },
//     APPEND_RESPONSE {
//         @Override
//         public Message create(final DirectFactory factory) {
//             return factory.appendResponse();
//         }
//     },
//     TIMEOUT_NOW {
//         @Override
//         public Message create(final DirectFactory factory) {
//             return factory.timeoutNow();
//         }
//     };
// 
//     private static final MessageType[] VALUES = values();
// 
//     public static final MessageType valueByOrdinal(final int ordinal) {
//         return VALUES[ordinal];
//     }
// 
//     public static final int maxOrdinal() {
//         return VALUES.length - 1;
//     }
// 
//     abstract public Message create(DirectFactory factory);
// 
//     public static MessageType readFrom(final DirectBuffer directBuffer, final int offset) {
//         final int type = directBuffer.getInt(offset);
//         if (type >= 0 & type <= MessageType.maxOrdinal()) {
//             return MessageType.valueByOrdinal(type);
//         }
//         throw new IllegalArgumentException("Illegal message type: " + type);
//     }
// 
// 
// }
// 
// Path: src/main/java/org/tools4j/hoverraft/transport/Receiver.java
// public interface Receiver<M extends DirectPayload> {
//     /**
//      * Polls messages in non-blocking mode. Messages are passed to the specified
//      * message event.
//      *
//      * @param messageMandler the event invoked for each message
//      * @param limit maximum number of messages to receive
//      * @return number of messages received, zero to at most {@code limit} messages
//      */
//     int poll(Consumer<? super M> messageMandler, int limit);
// }


import org.tools4j.hoverraft.message.Message;
import org.tools4j.hoverraft.message.MessageType;
import org.tools4j.hoverraft.transport.Receiver;
import java.util.Objects;
import java.util.function.Consumer;
import io.aeron.Subscription;
import io.aeron.logbuffer.FragmentHandler;
import io.aeron.logbuffer.Header;
import org.agrona.DirectBuffer;
import org.tools4j.hoverraft.command.Command;
import org.tools4j.hoverraft.direct.DirectFactory;
import org.tools4j.hoverraft.direct.DirectPayload;
    @Override
    public int poll(final Consumer<? super M> messageMandler, final int limit) {
        for (int i = 0; i < limit; i++) {
            if (0 < subscription.poll(aeronHandler, 1)) {
                messageMandler.accept(aeronHandler.message);
                aeronHandler.reset();
            } else {
                return i;
            }
        }
        return limit;
    }

    private static abstract class AeronHandler<M extends DirectPayload> implements FragmentHandler {
        private M message = null;

        @Override
        public void onFragment(final DirectBuffer buffer, final int offset, final int length, final Header header) {
            final MessageType messageType = MessageType.readFrom(buffer, offset);
            message = message(buffer, offset, length);
            message.wrap(buffer, offset);
        }

        public void reset() {
            if (message != null) {
                message.unwrap();
                message = null;
            }
        }

// Path: src/main/java/org/tools4j/hoverraft/command/Command.java
// public interface Command extends DirectPayload, Event {
// 
//     CommandKey commandKey();
// 
//     CommandPayload commandPayload();
// 
//     default Command sourceId(final int sourceId) {
//         commandKey().sourceId(sourceId);
//         return this;
//     }
// 
//     default Command commandIndex(final long commandIndex) {
//         commandKey().commandIndex(commandIndex);
//         return this;
//     }
// 
//     default void copyFrom(final Command command) {
//         writeBufferOrNull().putBytes(offset(), command.readBufferOrNull(), command.offset(), command.byteLength());
//     }
// 
//     @Override
//     default Transition accept(final ServerContext serverContext, final EventHandler eventHandler) {
//         return eventHandler.onCommand(serverContext, this);
//     }
// 
// }
// 
// Path: src/main/java/org/tools4j/hoverraft/direct/DirectFactory.java
// public interface DirectFactory {
// 
//     AppendRequest appendRequest();
// 
//     AppendResponse appendResponse();
// 
//     VoteRequest voteRequest();
// 
//     VoteResponse voteResponse();
// 
//     TimeoutNow timeoutNow();
// 
//     CommandKey commandKey();
// 
//     Command command();
// 
//     LogEntry logEntry();
// }
// 
// Path: src/main/java/org/tools4j/hoverraft/direct/DirectPayload.java
// public interface DirectPayload {
// 
//     int byteLength();
// 
//     int offset();
// 
//     DirectBuffer readBufferOrNull();
// 
//     MutableDirectBuffer writeBufferOrNull();
// 
//     void wrap(DirectBuffer buffer, int offset);
// 
//     void wrap(MutableDirectBuffer buffer, int offset);
// 
//     void unwrap();
// 
//     default boolean isWrapped() {
//         return readBufferOrNull() != null || writeBufferOrNull() != null;
//     }
// }
// 
// Path: src/main/java/org/tools4j/hoverraft/message/Message.java
// public interface Message extends DirectPayload, Event {
// 
//     MessageType type();
// 
//     default void sendTo(final Sender<? super Message> sender, final ResendStrategy resendStrategy) {
//         final long res = sender.offer(this);
//         if (res < 0) {
//             resendStrategy.onRejectedOffer(sender, this, res);
//         }
//     }
// 
// 
// }
// 
// Path: src/main/java/org/tools4j/hoverraft/message/MessageType.java
// public enum MessageType {
//     VOTE_REQUEST {
//         @Override
//         public Message create(final DirectFactory factory) {
//             return factory.voteRequest();
//         }
//     },
//     VOTE_RESPONSE {
//         @Override
//         public Message create(final DirectFactory factory) {
//             return factory.voteResponse();
//         }
//     },
//     APPEND_REQUEST {
//         @Override
//         public Message create(final DirectFactory factory) {
//             return factory.appendRequest();
//         }
//     },
//     APPEND_RESPONSE {
//         @Override
//         public Message create(final DirectFactory factory) {
//             return factory.appendResponse();
//         }
//     },
//     TIMEOUT_NOW {
//         @Override
//         public Message create(final DirectFactory factory) {
//             return factory.timeoutNow();
//         }
//     };
// 
//     private static final MessageType[] VALUES = values();
// 
//     public static final MessageType valueByOrdinal(final int ordinal) {
//         return VALUES[ordinal];
//     }
// 
//     public static final int maxOrdinal() {
//         return VALUES.length - 1;
//     }
// 
//     abstract public Message create(DirectFactory factory);
// 
//     public static MessageType readFrom(final DirectBuffer directBuffer, final int offset) {
//         final int type = directBuffer.getInt(offset);
//         if (type >= 0 & type <= MessageType.maxOrdinal()) {
//             return MessageType.valueByOrdinal(type);
//         }
//         throw new IllegalArgumentException("Illegal message type: " + type);
//     }
// 
// 
// }
// 
// Path: src/main/java/org/tools4j/hoverraft/transport/Receiver.java
// public interface Receiver<M extends DirectPayload> {
//     /**
//      * Polls messages in non-blocking mode. Messages are passed to the specified
//      * message event.
//      *
//      * @param messageMandler the event invoked for each message
//      * @param limit maximum number of messages to receive
//      * @return number of messages received, zero to at most {@code limit} messages
//      */
//     int poll(Consumer<? super M> messageMandler, int limit);
// }


// Path: src/main/java/org/tools4j/hoverraft/transport/aeron/AeronReceiver.java
import org.tools4j.hoverraft.message.Message;
import org.tools4j.hoverraft.message.MessageType;
import org.tools4j.hoverraft.transport.Receiver;
import java.util.Objects;
import java.util.function.Consumer;
import io.aeron.Subscription;
import io.aeron.logbuffer.FragmentHandler;
import io.aeron.logbuffer.Header;
import org.agrona.DirectBuffer;
import org.tools4j.hoverraft.command.Command;
import org.tools4j.hoverraft.direct.DirectFactory;
import org.tools4j.hoverraft.direct.DirectPayload;

    @Override
    public int poll(final Consumer<? super M> messageMandler, final int limit) {
        for (int i = 0; i < limit; i++) {
            if (0 < subscription.poll(aeronHandler, 1)) {
                messageMandler.accept(aeronHandler.message);
                aeronHandler.reset();
            } else {
                return i;
            }
        }
        return limit;
    }

    private static abstract class AeronHandler<M extends DirectPayload> implements FragmentHandler {
        private M message = null;

        @Override
        public void onFragment(final DirectBuffer buffer, final int offset, final int length, final Header header) {
            final MessageType messageType = MessageType.readFrom(buffer, offset);
            message = message(buffer, offset, length);
            message.wrap(buffer, offset);
        }

        public void reset() {
            if (message != null) {
                message.unwrap();
                message = null;
            }
        }

        abstract protected M message(final DirectBuffer buffer, final int offset, final int length);