io.axoniq.axonserver.connector.event.impl.BufferedEventStream Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of axonserver-connector-java Show documentation
Show all versions of axonserver-connector-java Show documentation
Connector module providing infrastructure components that connect to AxonServer.
/*
* Copyright (c) 2020. AxonIQ
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.axoniq.axonserver.connector.event.impl;
import io.axoniq.axonserver.connector.event.EventStream;
import io.axoniq.axonserver.connector.impl.AbstractBufferedStream;
import io.axoniq.axonserver.grpc.FlowControl;
import io.axoniq.axonserver.grpc.event.EventWithToken;
import io.axoniq.axonserver.grpc.event.GetEventsRequest;
import io.axoniq.axonserver.grpc.event.PayloadDescription;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import static io.axoniq.axonserver.connector.impl.ObjectUtils.nonNullOrDefault;
/**
* Buffering implementation of the {@link EventStream} used for Event processing.
*/
public class BufferedEventStream
extends AbstractBufferedStream
implements EventStream {
private static final Logger logger = LoggerFactory.getLogger(BufferedEventStream.class);
private static final EventWithToken TERMINAL_MESSAGE = EventWithToken.newBuilder().setToken(-1729).build();
private final long trackingToken;
private final boolean forceReadFromLeader;
/**
* Construct a {@link BufferedEventStream}, starting a position {@code trackingToken} with the given {@code
* bufferSize}.
*
* @param trackingToken the position to start this {@link BufferedEventStream} at
* @param bufferSize the buffer size of this event stream
* @param refillBatch the number of Events to consume prior refilling the buffer
* @param forceReadFromLeader a {@code boolean} defining whether Events must be read from the leader
*/
public BufferedEventStream(long trackingToken, int bufferSize, int refillBatch, boolean forceReadFromLeader) {
super("unused", bufferSize, refillBatch);
this.trackingToken = trackingToken;
this.forceReadFromLeader = forceReadFromLeader;
}
@Override
protected GetEventsRequest buildFlowControlMessage(FlowControl flowControl) {
GetEventsRequest request = GetEventsRequest.newBuilder().setNumberOfPermits(flowControl.getPermits()).build();
logger.trace("Sending request for data: {}", request);
return request;
}
@Override
protected GetEventsRequest buildInitialFlowControlMessage(FlowControl flowControl) {
GetEventsRequest eventsRequest = GetEventsRequest.newBuilder()
.setTrackingToken(trackingToken + 1)
.setForceReadFromLeader(forceReadFromLeader)
.setNumberOfPermits(flowControl.getPermits()).build();
logger.trace("Sending request for data: {}", eventsRequest);
return eventsRequest;
}
@Override
protected EventWithToken terminalMessage() {
return TERMINAL_MESSAGE;
}
@Override
public void excludePayloadType(String payloadType, String revision) {
PayloadDescription payloadToExclude = PayloadDescription.newBuilder()
.setType(payloadType)
.setRevision(nonNullOrDefault(revision, ""))
.build();
GetEventsRequest request = GetEventsRequest.newBuilder().addBlacklist(payloadToExclude).build();
logger.trace("Requesting exclusion of message type: {}", request);
outboundStream().onNext(request);
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy