dev.responsive.kafka.internal.db.mongo.ResponsiveMongoClient Maven / Gradle / Ivy
/*
* Copyright 2023 Responsive Computing, Inc.
*
* 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 dev.responsive.kafka.internal.db.mongo;
import com.mongodb.client.MongoClient;
import com.mongodb.client.model.WriteModel;
import dev.responsive.kafka.internal.db.RemoteKVTable;
import dev.responsive.kafka.internal.db.RemoteSessionTable;
import dev.responsive.kafka.internal.db.RemoteWindowedTable;
import dev.responsive.kafka.internal.db.SessionTableCache;
import dev.responsive.kafka.internal.db.TableCache;
import dev.responsive.kafka.internal.db.WindowedTableCache;
import dev.responsive.kafka.internal.db.partitioning.SessionSegmentPartitioner;
import dev.responsive.kafka.internal.db.partitioning.TablePartitioner;
import dev.responsive.kafka.internal.db.partitioning.WindowSegmentPartitioner;
import dev.responsive.kafka.internal.db.spec.BaseTableSpec;
import dev.responsive.kafka.internal.db.spec.TtlTableSpec;
import java.util.concurrent.TimeoutException;
public class ResponsiveMongoClient {
private final TableCache kvTableCache;
private final WindowedTableCache windowTableCache;
private final SessionTableCache sessionTableCache;
private final MongoClient client;
public ResponsiveMongoClient(
final MongoClient client,
final boolean timestampFirstOrder,
final CollectionCreationOptions collectionCreationOptions
) {
this.client = client;
kvTableCache = new TableCache<>(
spec -> new MongoKVTable(
client,
spec.tableName(),
collectionCreationOptions,
spec instanceof TtlTableSpec ? ((TtlTableSpec) spec).ttl() : null
));
windowTableCache = new WindowedTableCache<>(
(spec, partitioner) -> new MongoWindowedTable(
client,
spec.tableName(),
partitioner,
timestampFirstOrder,
collectionCreationOptions
)
);
sessionTableCache = new SessionTableCache<>(
(spec, partitioner) -> new MongoSessionTable(
client,
spec.tableName(),
partitioner,
collectionCreationOptions
)
);
}
public RemoteKVTable> kvTable(final String name)
throws InterruptedException, TimeoutException {
return kvTableCache.create(new BaseTableSpec(name, TablePartitioner.defaultPartitioner()));
}
public RemoteWindowedTable> windowedTable(
final String name,
final WindowSegmentPartitioner partitioner
) throws InterruptedException, TimeoutException {
return windowTableCache.create(new BaseTableSpec(name, partitioner), partitioner);
}
public RemoteSessionTable> sessionTable(
final String name,
final SessionSegmentPartitioner partitioner
) throws InterruptedException, TimeoutException {
return sessionTableCache.create(new BaseTableSpec(name, partitioner), partitioner);
}
public void close() {
client.close();
}
}