com.hazelcast.org.apache.calcite.jdbc.CachingCalciteSchema Maven / Gradle / Ivy
/*
* 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 com.hazelcast.org.apache.calcite.jdbc;
import com.hazelcast.org.apache.calcite.rel.type.RelProtoDataType;
import com.hazelcast.org.apache.calcite.schema.Function;
import com.hazelcast.org.apache.calcite.schema.Schema;
import com.hazelcast.org.apache.calcite.schema.SchemaVersion;
import com.hazelcast.org.apache.calcite.schema.Table;
import com.hazelcast.org.apache.calcite.schema.TableMacro;
import com.hazelcast.org.apache.calcite.util.NameMap;
import com.hazelcast.org.apache.calcite.util.NameMultimap;
import com.hazelcast.org.apache.calcite.util.NameSet;
import com.hazelcast.com.google.common.cache.CacheBuilder;
import com.hazelcast.com.google.common.cache.CacheLoader;
import com.hazelcast.com.google.common.cache.LoadingCache;
import com.hazelcast.com.google.common.collect.ImmutableList;
import com.hazelcast.com.google.common.collect.ImmutableSortedMap;
import com.hazelcast.com.google.common.collect.ImmutableSortedSet;
import com.hazelcast.org.checkerframework.checker.nullness.qual.Nullable;
import java.util.Collection;
import java.util.List;
import java.util.Set;
import static com.hazelcast.org.apache.calcite.linq4j.Nullness.castNonNull;
/**
* Concrete implementation of {@link CalciteSchema} that caches tables,
* functions and sub-schemas.
*/
class CachingCalciteSchema extends CalciteSchema {
private final Cached implicitSubSchemaCache;
private final Cached implicitTableCache;
private final Cached implicitFunctionCache;
private final Cached implicitTypeCache;
private boolean cache = true;
/** Creates a CachingCalciteSchema. */
CachingCalciteSchema(@Nullable CalciteSchema parent, Schema schema, String name) {
this(parent, schema, name, null, null, null, null, null, null, null, null);
}
@SuppressWarnings({"argument.type.incompatible", "return.type.incompatible"})
private CachingCalciteSchema(@Nullable CalciteSchema parent, Schema schema,
String name,
@Nullable NameMap subSchemaMap,
@Nullable NameMap tableMap,
@Nullable NameMap latticeMap,
@Nullable NameMap typeMap,
@Nullable NameMultimap functionMap,
@Nullable NameSet functionNames,
@Nullable NameMap nullaryFunctionMap,
@Nullable List> path) {
super(parent, schema, name, subSchemaMap, tableMap, latticeMap, typeMap,
functionMap, functionNames, nullaryFunctionMap, path);
this.implicitSubSchemaCache =
new AbstractCached() {
@Override public SubSchemaCache build() {
return new SubSchemaCache(CachingCalciteSchema.this,
CachingCalciteSchema.this.schema.getSubSchemaNames());
}
};
this.implicitTableCache =
new AbstractCached() {
@Override public NameSet build() {
return NameSet.immutableCopyOf(
CachingCalciteSchema.this.schema.getTableNames());
}
};
this.implicitFunctionCache =
new AbstractCached() {
@Override public NameSet build() {
return NameSet.immutableCopyOf(
CachingCalciteSchema.this.schema.getFunctionNames());
}
};
this.implicitTypeCache =
new AbstractCached() {
@Override public NameSet build() {
return NameSet.immutableCopyOf(
CachingCalciteSchema.this.schema.getTypeNames());
}
};
}
@Override public void setCache(boolean cache) {
if (cache == this.cache) {
return;
}
final long now = System.currentTimeMillis();
implicitSubSchemaCache.enable(now, cache);
implicitTableCache.enable(now, cache);
implicitFunctionCache.enable(now, cache);
this.cache = cache;
}
@Override protected boolean isCacheEnabled() {
return this.cache;
}
@Override protected @Nullable CalciteSchema getImplicitSubSchema(String schemaName,
boolean caseSensitive) {
final long now = System.currentTimeMillis();
final SubSchemaCache subSchemaCache =
implicitSubSchemaCache.get(now);
//noinspection LoopStatementThatDoesntLoop
for (String schemaName2
: subSchemaCache.names.range(schemaName, caseSensitive)) {
return subSchemaCache.cache.getUnchecked(schemaName2);
}
return null;
}
/** Adds a child schema of this schema. */
@Override public CalciteSchema add(String name, Schema schema) {
final CalciteSchema calciteSchema =
new CachingCalciteSchema(this, schema, name);
subSchemaMap.put(name, calciteSchema);
return calciteSchema;
}
@Override protected @Nullable TableEntry getImplicitTable(String tableName,
boolean caseSensitive) {
final long now = System.currentTimeMillis();
final NameSet implicitTableNames = implicitTableCache.get(now);
for (String tableName2
: implicitTableNames.range(tableName, caseSensitive)) {
final Table table = schema.getTable(tableName2);
if (table != null) {
return tableEntry(tableName2, table);
}
}
return null;
}
@Override protected @Nullable TypeEntry getImplicitType(String name, boolean caseSensitive) {
final long now = System.currentTimeMillis();
final NameSet implicitTypeNames = implicitTypeCache.get(now);
for (String typeName
: implicitTypeNames.range(name, caseSensitive)) {
final RelProtoDataType type = schema.getType(typeName);
if (type != null) {
return typeEntry(name, type);
}
}
return null;
}
@Override protected void addImplicitSubSchemaToBuilder(
ImmutableSortedMap.Builder builder) {
ImmutableSortedMap explicitSubSchemas = builder.build();
final long now = System.currentTimeMillis();
final SubSchemaCache subSchemaCache = implicitSubSchemaCache.get(now);
for (String name : subSchemaCache.names.iterable()) {
if (explicitSubSchemas.containsKey(name)) {
// explicit sub-schema wins.
continue;
}
builder.put(name, subSchemaCache.cache.getUnchecked(name));
}
}
@Override protected void addImplicitTableToBuilder(
ImmutableSortedSet.Builder builder) {
// Add implicit tables, case-sensitive.
final long now = System.currentTimeMillis();
final NameSet set = implicitTableCache.get(now);
builder.addAll(set.iterable());
}
@Override protected void addImplicitFunctionsToBuilder(
ImmutableList.Builder builder,
String name, boolean caseSensitive) {
// Add implicit functions, case-insensitive.
final long now = System.currentTimeMillis();
final NameSet set = implicitFunctionCache.get(now);
for (String name2 : set.range(name, caseSensitive)) {
final Collection functions = schema.getFunctions(name2);
if (functions != null) {
builder.addAll(functions);
}
}
}
@Override protected void addImplicitFuncNamesToBuilder(
ImmutableSortedSet.Builder builder) {
// Add implicit functions, case-sensitive.
final long now = System.currentTimeMillis();
final NameSet set = implicitFunctionCache.get(now);
builder.addAll(set.iterable());
}
@Override protected void addImplicitTypeNamesToBuilder(
ImmutableSortedSet.Builder builder) {
// Add implicit types, case-sensitive.
final long now = System.currentTimeMillis();
final NameSet set = implicitTypeCache.get(now);
builder.addAll(set.iterable());
}
@Override protected void addImplicitTablesBasedOnNullaryFunctionsToBuilder(
ImmutableSortedMap.Builder builder) {
ImmutableSortedMap explicitTables = builder.build();
final long now = System.currentTimeMillis();
final NameSet set = implicitFunctionCache.get(now);
for (String s : set.iterable()) {
// explicit table wins.
if (explicitTables.containsKey(s)) {
continue;
}
for (Function function : schema.getFunctions(s)) {
if (function instanceof TableMacro
&& function.getParameters().isEmpty()) {
final Table table = ((TableMacro) function).apply(ImmutableList.of());
builder.put(s, table);
}
}
}
}
@Override protected @Nullable TableEntry getImplicitTableBasedOnNullaryFunction(String tableName,
boolean caseSensitive) {
final long now = System.currentTimeMillis();
final NameSet set = implicitFunctionCache.get(now);
for (String s : set.range(tableName, caseSensitive)) {
for (Function function : schema.getFunctions(s)) {
if (function instanceof TableMacro
&& function.getParameters().isEmpty()) {
final Table table =
((TableMacro) function).apply(ImmutableList.of());
return tableEntry(tableName, table);
}
}
}
return null;
}
@Override protected CalciteSchema snapshot(@Nullable CalciteSchema parent,
SchemaVersion version) {
CalciteSchema snapshot = new CachingCalciteSchema(parent,
schema.snapshot(version), name, null, tableMap, latticeMap, typeMap,
functionMap, functionNames, nullaryFunctionMap, getPath());
for (CalciteSchema subSchema : subSchemaMap.map().values()) {
CalciteSchema subSchemaSnapshot = subSchema.snapshot(snapshot, version);
snapshot.subSchemaMap.put(subSchema.name, subSchemaSnapshot);
}
return snapshot;
}
@Override public boolean removeTable(String name) {
if (cache) {
final long now = System.nanoTime();
implicitTableCache.enable(now, false);
implicitTableCache.enable(now, true);
}
return super.removeTable(name);
}
@Override public boolean removeFunction(String name) {
if (cache) {
final long now = System.nanoTime();
implicitFunctionCache.enable(now, false);
implicitFunctionCache.enable(now, true);
}
return super.removeFunction(name);
}
/** Strategy for caching the value of an object and re-creating it if its
* value is out of date as of a given timestamp.
*
* @param Type of cached object
*/
private interface Cached {
/** Returns the value; uses cached value if valid. */
T get(long now);
/** Creates a new value. */
T build();
/** Called when CalciteSchema caching is enabled or disabled. */
void enable(long now, boolean enabled);
}
/** Implementation of {@link CachingCalciteSchema.Cached}
* that drives from {@link CachingCalciteSchema#cache}.
*
* @param element type */
private abstract class AbstractCached implements Cached {
@Nullable T t;
boolean built = false;
@Override public T get(long now) {
if (!CachingCalciteSchema.this.cache) {
return build();
}
if (!built) {
t = build();
}
built = true;
return castNonNull(t);
}
@Override public void enable(long now, boolean enabled) {
if (!enabled) {
t = null;
}
built = false;
}
}
/** Information about the implicit sub-schemas of an {@link CalciteSchema}. */
private static class SubSchemaCache {
/** The names of sub-schemas returned from the {@link Schema} SPI. */
final NameSet names;
/** Cached {@link CalciteSchema} wrappers. It is
* worth caching them because they contain maps of their own sub-objects. */
final LoadingCache cache;
private SubSchemaCache(final CalciteSchema calciteSchema,
Set names) {
this.names = NameSet.immutableCopyOf(names);
this.cache = CacheBuilder.newBuilder().build(
new CacheLoader() {
@SuppressWarnings("NullableProblems")
@Override public CalciteSchema load(String schemaName) {
final Schema subSchema =
calciteSchema.schema.getSubSchema(schemaName);
if (subSchema == null) {
throw new RuntimeException("sub-schema " + schemaName
+ " not found");
}
return new CachingCalciteSchema(calciteSchema, subSchema, schemaName);
}
});
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy