org.apache.calcite.adapter.geode.rel.GeodeSchema 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 org.apache.calcite.adapter.geode.rel;
import org.apache.calcite.adapter.geode.util.GeodeUtils;
import org.apache.calcite.schema.Table;
import org.apache.calcite.schema.impl.AbstractSchema;
import org.apache.geode.cache.GemFireCache;
import org.apache.geode.cache.Region;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
/**
* Schema mapped onto a Geode Region.
*/
public class GeodeSchema extends AbstractSchema {
final GemFireCache cache;
private final List regionNames;
private ImmutableMap tableMap;
public GeodeSchema(final GemFireCache cache, final Iterable regionNames) {
super();
this.cache = Objects.requireNonNull(cache, "clientCache");
this.regionNames = ImmutableList.copyOf(Objects.requireNonNull(regionNames, "regionNames"));
}
@Override protected Map getTableMap() {
if (tableMap == null) {
final ImmutableMap.Builder builder = ImmutableMap.builder();
for (String regionName : regionNames) {
Region region = GeodeUtils.createRegion(cache, regionName);
Table table = new GeodeTable(region);
builder.put(regionName, table);
}
tableMap = builder.build();
}
return tableMap;
}
}