Examples with MessageConsumer io.aklivity.zilla.runtime.engine.binding.function.MessageConsumer used on opensource projects

Example 11 with MessageConsumer

use of io.aklivity.zilla.runtime.engine.binding.function.MessageConsumer in project zilla by aklivity.

the class ProxyServerFactory method newStream.

@Override
public MessageConsumer newStream(int msgTypeId, DirectBuffer buffer, int index, int length, MessageConsumer sender) {
    final BeginFW begin = beginRO.wrap(buffer, index, index + length);
    final long routeId = begin.routeId();
    final long initialId = begin.streamId();
    final long affinity = begin.affinity();
    MessageConsumer newStream = null;
    final ProxyBindingConfig binding = router.lookup(routeId);
    if (binding != null) {
        newStream = new ProxyNetServer(routeId, initialId, sender, affinity)::onNetMessage;
    }
    return newStream;
}

Example 12 with MessageConsumer

use of io.aklivity.zilla.runtime.engine.binding.function.MessageConsumer in project zilla by aklivity.

the class ProxyServerFactory method newStream.

private MessageConsumer newStream(MessageConsumer sender, long routeId, long streamId, long sequence, long acknowledge, int maximum, long traceId, long authorization, long affinity, Flyweight extension) {
    BeginFW begin = beginRW.wrap(writeBuffer, 0, writeBuffer.capacity()).routeId(routeId).streamId(streamId).sequence(sequence).acknowledge(acknowledge).maximum(maximum).traceId(traceId).authorization(authorization).affinity(affinity).extension(extension.buffer(), extension.offset(), extension.sizeof()).build();
    MessageConsumer receiver = streamFactory.newStream(begin.typeId(), begin.buffer(), begin.offset(), begin.sizeof(), sender);
    receiver.accept(begin.typeId(), begin.buffer(), begin.offset(), begin.sizeof());
    return receiver;
}

Example 13 with MessageConsumer

use of io.aklivity.zilla.runtime.engine.binding.function.MessageConsumer in project zilla by aklivity.

the class SseServerFactory method doHttpBegin.

private void doHttpBegin(MessageConsumer receiver, long routeId, long streamId, long sequence, long acknowledge, int maximum, long traceId, long authorization, long affinity, Consumer<Array32FW.Builder<HttpHeaderFW.Builder, HttpHeaderFW>> mutator) {
    final BeginFW begin = beginRW.wrap(writeBuffer, 0, writeBuffer.capacity()).routeId(routeId).streamId(streamId).sequence(sequence).acknowledge(acknowledge).maximum(maximum).traceId(traceId).authorization(authorization).affinity(affinity).extension(e -> e.set(visitHttpBeginEx(mutator))).build();
    receiver.accept(begin.typeId(), begin.buffer(), begin.offset(), begin.sizeof());
}

Example 14 with MessageConsumer

use of io.aklivity.zilla.runtime.engine.binding.function.MessageConsumer in project zilla by aklivity.

the class SseServerFactory method newInitialSseStream.

public MessageConsumer newInitialSseStream(final BeginFW begin, final MessageConsumer network, final HttpBeginExFW httpBeginEx) {
    final long routeId = begin.routeId();
    final long initialId = begin.streamId();
    final long traceId = begin.traceId();
    final long authorization = begin.authorization();
    final long affinity = begin.affinity();
    Array32FW<HttpHeaderFW> headers = httpBeginEx.headers();
    httpHelper.reset();
    headers.forEach(httpHelper::onHttpHeader);
    // TODO: ":pathinfo" ?
    String16FW pathInfo = httpHelper.path;
    String16FW lastEventId = httpHelper.lastEventId;
    // use query parameter value as default for missing Last-Event-ID header
    if (pathInfo != null) {
        Matcher matcher = QUERY_PARAMS_PATTERN.matcher(pathInfo.asString());
        if (matcher.matches()) {
            String path = matcher.group("path");
            String query = matcher.group("query");
            matcher = LAST_EVENT_ID_PATTERN.matcher(query);
            StringBuffer builder = new StringBuffer(path);
            while (matcher.find()) {
                if (lastEventId == null) {
                    lastEventId = decodeLastEventId(matcher.group("lastEventId"));
                }
                String replacement = matcher.group(3).isEmpty() ? "$3" : "$1";
                matcher.appendReplacement(builder, replacement);
            }
            matcher.appendTail(builder);
            pathInfo = new String16FW(builder.toString());
        }
    }
    MessageConsumer newStream = null;
    if (lastEventId == null || lastEventId.length() <= MAXIMUM_LAST_EVENT_ID_SIZE) {
        final SseBindingConfig binding = bindings.get(routeId);
        final SseRouteConfig resolved = binding != null ? binding.resolve(authorization, pathInfo.asString()) : null;
        if (resolved != null) {
            final boolean timestampRequested = httpBeginEx.headers().anyMatch(header -> HEADER_NAME_ACCEPT.equals(header.name()) && header.value().asString().contains("ext=timestamp"));
            final String8FW lastEventId8 = httpHelper.asLastEventId(lastEventId);
            final SseServer server = new SseServer(network, routeId, initialId, resolved.id, timestampRequested);
            server.onNetBegin(begin);
            server.stream.doAppBegin(traceId, authorization, affinity, pathInfo, lastEventId8);
            newStream = server::onNetMessage;
        }
    } else {
        doHttpResponse(begin, network, HEADER_VALUE_STATUS_400);
        newStream = (t, b, i, l) -> {
        };
    }
    return newStream;
}

Example 15 with MessageConsumer

use of io.aklivity.zilla.runtime.engine.binding.function.MessageConsumer in project zilla by aklivity.

the class SseServerFactory method newStream.

@Override
public MessageConsumer newStream(int msgTypeId, DirectBuffer buffer, int index, int length, MessageConsumer network) {
    final BeginFW begin = beginRO.wrap(buffer, index, index + length);
    final OctetsFW extension = begin.extension();
    final HttpBeginExFW httpBeginEx = extension.get(httpBeginExRO::tryWrap);
    MessageConsumer newStream = null;
    if (!isSseRequestMethod(httpBeginEx)) {
        doHttpResponse(begin, network, HEADER_VALUE_STATUS_405);
        newStream = (t, b, i, l) -> {
        };
    } else {
        newStream = newInitialSseStream(begin, network, httpBeginEx);
    }
    return newStream;
}