Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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 org.apache.kafka.streams.state.internals;
import static org.apache.kafka.streams.internals.ApiUtils.prepareMillisCheckFailMsgPrefix;
import static org.apache.kafka.streams.internals.ApiUtils.validateMillisecondDuration;
import java.time.Duration;
import java.util.Objects;
import org.apache.kafka.common.utils.Bytes;
import org.apache.kafka.streams.state.WindowBytesStoreSupplier;
import org.apache.kafka.streams.state.WindowStore;
public class RocksDbIndexedTimeOrderedWindowBytesStoreSupplier implements WindowBytesStoreSupplier {
public enum WindowStoreTypes {
DEFAULT_WINDOW_STORE,
INDEXED_WINDOW_STORE
}
private final String name;
private final long retentionPeriod;
private final long segmentInterval;
private final long windowSize;
private final boolean retainDuplicates;
private final WindowStoreTypes windowStoreType;
public static RocksDbIndexedTimeOrderedWindowBytesStoreSupplier create(final String name,
final Duration retentionPeriod,
final Duration windowSize,
final boolean retainDuplicates,
final boolean hasIndex) {
Objects.requireNonNull(name, "name cannot be null");
final String rpMsgPrefix = prepareMillisCheckFailMsgPrefix(retentionPeriod, "retentionPeriod");
final long retentionMs = validateMillisecondDuration(retentionPeriod, rpMsgPrefix);
final String wsMsgPrefix = prepareMillisCheckFailMsgPrefix(windowSize, "windowSize");
final long windowSizeMs = validateMillisecondDuration(windowSize, wsMsgPrefix);
final long defaultSegmentInterval = Math.max(retentionMs / 2, 60_000L);
if (retentionMs < 0L) {
throw new IllegalArgumentException("retentionPeriod cannot be negative");
}
if (windowSizeMs < 0L) {
throw new IllegalArgumentException("windowSize cannot be negative");
}
if (defaultSegmentInterval < 1L) {
throw new IllegalArgumentException("segmentInterval cannot be zero or negative");
}
if (windowSizeMs > retentionMs) {
throw new IllegalArgumentException("The retention period of the window store "
+ name + " must be no smaller than its window size. Got size=["
+ windowSizeMs + "], retention=[" + retentionMs + "]");
}
return new RocksDbIndexedTimeOrderedWindowBytesStoreSupplier(name, retentionMs,
defaultSegmentInterval, windowSizeMs, retainDuplicates, hasIndex);
}
public RocksDbIndexedTimeOrderedWindowBytesStoreSupplier(final String name,
final long retentionPeriod,
final long segmentInterval,
final long windowSize,
final boolean retainDuplicates,
final boolean withIndex) {
this(name, retentionPeriod, segmentInterval, windowSize, retainDuplicates,
withIndex
? WindowStoreTypes.INDEXED_WINDOW_STORE
: WindowStoreTypes.DEFAULT_WINDOW_STORE);
}
public RocksDbIndexedTimeOrderedWindowBytesStoreSupplier(final String name,
final long retentionPeriod,
final long segmentInterval,
final long windowSize,
final boolean retainDuplicates,
final WindowStoreTypes windowStoreType) {
this.name = name;
this.retentionPeriod = retentionPeriod;
this.segmentInterval = segmentInterval;
this.windowSize = windowSize;
this.retainDuplicates = retainDuplicates;
this.windowStoreType = windowStoreType;
}
@Override
public String name() {
return name;
}
@Override
public WindowStore get() {
switch (windowStoreType) {
case DEFAULT_WINDOW_STORE:
return new RocksDBTimeOrderedWindowStore(
new RocksDBTimeOrderedWindowSegmentedBytesStore(
name,
metricsScope(),
retentionPeriod,
segmentInterval,
false),
retainDuplicates,
windowSize);
case INDEXED_WINDOW_STORE:
return new RocksDBTimeOrderedWindowStore(
new RocksDBTimeOrderedWindowSegmentedBytesStore(
name,
metricsScope(),
retentionPeriod,
segmentInterval,
true),
retainDuplicates,
windowSize);
default:
throw new IllegalArgumentException("invalid window store type: " + windowStoreType);
}
}
@Override
public String metricsScope() {
return "rocksdb-window";
}
@Override
public long segmentIntervalMs() {
return segmentInterval;
}
@Override
public long windowSize() {
return windowSize;
}
@Override
public boolean retainDuplicates() {
return retainDuplicates;
}
@Override
public long retentionPeriod() {
return retentionPeriod;
}
@Override
public String toString() {
return "RocksDbIndexedTimeOrderedWindowBytesStoreSupplier{" +
"name='" + name + '\'' +
", retentionPeriod=" + retentionPeriod +
", segmentInterval=" + segmentInterval +
", windowSize=" + windowSize +
", retainDuplicates=" + retainDuplicates +
", windowStoreType=" + windowStoreType +
'}';
}
}