io.questdb.MessageBusImpl Maven / Gradle / Ivy
/*******************************************************************************
* ___ _ ____ ____
* / _ \ _ _ ___ ___| |_| _ \| __ )
* | | | | | | |/ _ \/ __| __| | | | _ \
* | |_| | |_| | __/\__ \ |_| |_| | |_) |
* \__\_\\__,_|\___||___/\__|____/|____/
*
* Copyright (c) 2014-2019 Appsicle
* Copyright (c) 2019-2022 QuestDB
*
* 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.questdb;
import io.questdb.cairo.CairoConfiguration;
import io.questdb.cairo.sql.async.PageFrameReduceTask;
import io.questdb.mp.*;
import io.questdb.std.MemoryTag;
import io.questdb.std.Misc;
import io.questdb.tasks.*;
import org.jetbrains.annotations.NotNull;
public class MessageBusImpl implements MessageBus {
private final CairoConfiguration configuration;
private final RingQueue indexerQueue;
private final MPSequence indexerPubSeq;
private final MCSequence indexerSubSeq;
private final RingQueue vectorAggregateQueue;
private final MPSequence vectorAggregatePubSeq;
private final MCSequence vectorAggregateSubSeq;
private final RingQueue o3CallbackQueue;
private final MPSequence o3CallbackPubSeq;
private final MCSequence o3CallbackSubSeq;
private final RingQueue o3PurgeDiscoveryQueue;
private final MPSequence o3PurgeDiscoveryPubSeq;
private final MCSequence o3PurgeDiscoverySubSeq;
private final RingQueue o3PartitionQueue;
private final MPSequence o3PartitionPubSeq;
private final MCSequence o3PartitionSubSeq;
private final RingQueue o3OpenColumnQueue;
private final MPSequence o3OpenColumnPubSeq;
private final MCSequence o3OpenColumnSubSeq;
private final RingQueue o3CopyQueue;
private final MPSequence o3CopyPubSeq;
private final MCSequence o3CopySubSeq;
private final RingQueue latestByQueue;
private final MPSequence latestByPubSeq;
private final MCSequence latestBySubSeq;
private final RingQueue tableWriterEventQueue;
private final MPSequence tableWriterEventPubSeq;
private final FanOut tableWriterEventSubSeq;
private final MPSequence queryCacheEventPubSeq;
private final FanOut queryCacheEventSubSeq;
private final int pageFrameReduceShardCount;
private final MPSequence[] pageFrameReducePubSeq;
private final MCSequence[] pageFrameReduceSubSeq;
private final RingQueue[] pageFrameReduceQueue;
private final FanOut[] pageFrameCollectFanOut;
public MessageBusImpl(@NotNull CairoConfiguration configuration) {
this.configuration = configuration;
this.indexerQueue = new RingQueue<>(ColumnIndexerTask::new, configuration.getColumnIndexerQueueCapacity());
this.indexerPubSeq = new MPSequence(indexerQueue.getCycle());
this.indexerSubSeq = new MCSequence(indexerQueue.getCycle());
indexerPubSeq.then(indexerSubSeq).then(indexerPubSeq);
this.vectorAggregateQueue = new RingQueue<>(VectorAggregateTask::new, configuration.getVectorAggregateQueueCapacity());
this.vectorAggregatePubSeq = new MPSequence(vectorAggregateQueue.getCycle());
this.vectorAggregateSubSeq = new MCSequence(vectorAggregateQueue.getCycle());
vectorAggregatePubSeq.then(vectorAggregateSubSeq).then(vectorAggregatePubSeq);
this.o3CallbackQueue = new RingQueue<>(O3CallbackTask::new, configuration.getO3CallbackQueueCapacity());
this.o3CallbackPubSeq = new MPSequence(this.o3CallbackQueue.getCycle());
this.o3CallbackSubSeq = new MCSequence(this.o3CallbackQueue.getCycle());
o3CallbackPubSeq.then(o3CallbackSubSeq).then(o3CallbackPubSeq);
this.o3PartitionQueue = new RingQueue<>(O3PartitionTask::new, configuration.getO3PartitionQueueCapacity());
this.o3PartitionPubSeq = new MPSequence(this.o3PartitionQueue.getCycle());
this.o3PartitionSubSeq = new MCSequence(this.o3PartitionQueue.getCycle());
o3PartitionPubSeq.then(o3PartitionSubSeq).then(o3PartitionPubSeq);
this.o3OpenColumnQueue = new RingQueue<>(O3OpenColumnTask::new, configuration.getO3OpenColumnQueueCapacity());
this.o3OpenColumnPubSeq = new MPSequence(this.o3OpenColumnQueue.getCycle());
this.o3OpenColumnSubSeq = new MCSequence(this.o3OpenColumnQueue.getCycle());
o3OpenColumnPubSeq.then(o3OpenColumnSubSeq).then(o3OpenColumnPubSeq);
this.o3CopyQueue = new RingQueue<>(O3CopyTask::new, configuration.getO3CopyQueueCapacity());
this.o3CopyPubSeq = new MPSequence(this.o3CopyQueue.getCycle());
this.o3CopySubSeq = new MCSequence(this.o3CopyQueue.getCycle());
o3CopyPubSeq.then(o3CopySubSeq).then(o3CopyPubSeq);
this.o3PurgeDiscoveryQueue = new RingQueue<>(O3PurgeDiscoveryTask::new, configuration.getO3PurgeDiscoveryQueueCapacity());
this.o3PurgeDiscoveryPubSeq = new MPSequence(this.o3PurgeDiscoveryQueue.getCycle());
this.o3PurgeDiscoverySubSeq = new MCSequence(this.o3PurgeDiscoveryQueue.getCycle());
this.o3PurgeDiscoveryPubSeq.then(this.o3PurgeDiscoverySubSeq).then(o3PurgeDiscoveryPubSeq);
this.latestByQueue = new RingQueue<>(LatestByTask::new, configuration.getLatestByQueueCapacity());
this.latestByPubSeq = new MPSequence(latestByQueue.getCycle());
this.latestBySubSeq = new MCSequence(latestByQueue.getCycle());
latestByPubSeq.then(latestBySubSeq).then(latestByPubSeq);
this.tableWriterEventQueue = new RingQueue<>(
TableWriterTask::new,
configuration.getWriterCommandQueueSlotSize(),
configuration.getWriterCommandQueueCapacity(),
MemoryTag.NATIVE_REPL
);
this.tableWriterEventPubSeq = new MPSequence(this.tableWriterEventQueue.getCycle());
this.tableWriterEventSubSeq = new FanOut();
this.tableWriterEventPubSeq.then(this.tableWriterEventSubSeq).then(this.tableWriterEventPubSeq);
this.queryCacheEventPubSeq = new MPSequence(configuration.getQueryCacheEventQueueCapacity());
this.queryCacheEventSubSeq = new FanOut();
this.queryCacheEventPubSeq.then(this.queryCacheEventSubSeq).then(this.queryCacheEventPubSeq);
this.pageFrameReduceShardCount = configuration.getPageFrameReduceShardCount();
//noinspection unchecked
pageFrameReduceQueue = new RingQueue[pageFrameReduceShardCount];
pageFrameReducePubSeq = new MPSequence[pageFrameReduceShardCount];
pageFrameReduceSubSeq = new MCSequence[pageFrameReduceShardCount];
pageFrameCollectFanOut = new FanOut[pageFrameReduceShardCount];
int reduceQueueCapacity = configuration.getPageFrameReduceQueueCapacity();
for (int i = 0; i < pageFrameReduceShardCount; i++) {
final RingQueue queue = new RingQueue(
() -> new PageFrameReduceTask(configuration),
reduceQueueCapacity
);
final MPSequence reducePubSeq = new MPSequence(reduceQueueCapacity);
final MCSequence reduceSubSeq = new MCSequence(reduceQueueCapacity);
final FanOut collectFanOut = new FanOut();
reducePubSeq.then(reduceSubSeq).then(collectFanOut).then(reducePubSeq);
pageFrameReduceQueue[i] = queue;
pageFrameReducePubSeq[i] = reducePubSeq;
pageFrameReduceSubSeq[i] = reduceSubSeq;
pageFrameCollectFanOut[i] = collectFanOut;
}
}
@Override
public void close() {
// We need to close only queues with native backing memory.
Misc.free(getTableWriterEventQueue());
Misc.free(pageFrameReduceQueue);
}
@Override
public CairoConfiguration getConfiguration() {
return configuration;
}
@Override
public Sequence getIndexerPubSequence() {
return indexerPubSeq;
}
@Override
public RingQueue getIndexerQueue() {
return indexerQueue;
}
@Override
public Sequence getIndexerSubSequence() {
return indexerSubSeq;
}
@Override
public Sequence getLatestByPubSeq() {
return latestByPubSeq;
}
@Override
public RingQueue getLatestByQueue() {
return latestByQueue;
}
@Override
public Sequence getLatestBySubSeq() {
return latestBySubSeq;
}
@Override
public MPSequence getO3CallbackPubSeq() {
return o3CallbackPubSeq;
}
@Override
public RingQueue getO3CallbackQueue() {
return o3CallbackQueue;
}
@Override
public MCSequence getO3CallbackSubSeq() {
return o3CallbackSubSeq;
}
@Override
public MPSequence getO3CopyPubSeq() {
return o3CopyPubSeq;
}
@Override
public RingQueue getO3CopyQueue() {
return o3CopyQueue;
}
@Override
public MCSequence getO3CopySubSeq() {
return o3CopySubSeq;
}
@Override
public MPSequence getO3OpenColumnPubSeq() {
return o3OpenColumnPubSeq;
}
@Override
public RingQueue getO3OpenColumnQueue() {
return o3OpenColumnQueue;
}
@Override
public MCSequence getO3OpenColumnSubSeq() {
return o3OpenColumnSubSeq;
}
@Override
public MPSequence getO3PartitionPubSeq() {
return o3PartitionPubSeq;
}
@Override
public RingQueue getO3PartitionQueue() {
return o3PartitionQueue;
}
@Override
public MCSequence getO3PartitionSubSeq() {
return o3PartitionSubSeq;
}
@Override
public MPSequence getO3PurgeDiscoveryPubSeq() {
return o3PurgeDiscoveryPubSeq;
}
@Override
public RingQueue getO3PurgeDiscoveryQueue() {
return o3PurgeDiscoveryQueue;
}
@Override
public MCSequence getO3PurgeDiscoverySubSeq() {
return o3PurgeDiscoverySubSeq;
}
@Override
public FanOut getPageFrameCollectFanOut(int shard) {
return pageFrameCollectFanOut[shard];
}
@Override
public MPSequence getPageFrameReducePubSeq(int shard) {
return pageFrameReducePubSeq[shard];
}
@Override
public RingQueue getPageFrameReduceQueue(int shard) {
return pageFrameReduceQueue[shard];
}
@Override
public int getPageFrameReduceShardCount() {
return pageFrameReduceShardCount;
}
@Override
public MCSequence getPageFrameReduceSubSeq(int shard) {
return pageFrameReduceSubSeq[shard];
}
@Override
public FanOut getTableWriterEventFanOut() {
return tableWriterEventSubSeq;
}
@Override
public MPSequence getTableWriterEventPubSeq() {
return tableWriterEventPubSeq;
}
@Override
public RingQueue getTableWriterEventQueue() {
return tableWriterEventQueue;
}
@Override
public Sequence getVectorAggregatePubSeq() {
return vectorAggregatePubSeq;
}
@Override
public RingQueue getVectorAggregateQueue() {
return vectorAggregateQueue;
}
@Override
public Sequence getVectorAggregateSubSeq() {
return vectorAggregateSubSeq;
}
@Override
public MPSequence getQueryCacheEventPubSeq() {
return queryCacheEventPubSeq;
}
@Override
public FanOut getQueryCacheEventFanOut() {
return queryCacheEventSubSeq;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy