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.calcite.sql.ddl;
import org.apache.calcite.adapter.jdbc.JdbcSchema;
import org.apache.calcite.avatica.AvaticaUtils;
import org.apache.calcite.jdbc.CalcitePrepare;
import org.apache.calcite.jdbc.CalciteSchema;
import org.apache.calcite.model.JsonSchema;
import org.apache.calcite.schema.Schema;
import org.apache.calcite.schema.SchemaFactory;
import org.apache.calcite.schema.SchemaPlus;
import org.apache.calcite.sql.SqlCreate;
import org.apache.calcite.sql.SqlExecutableStatement;
import org.apache.calcite.sql.SqlIdentifier;
import org.apache.calcite.sql.SqlKind;
import org.apache.calcite.sql.SqlLiteral;
import org.apache.calcite.sql.SqlNode;
import org.apache.calcite.sql.SqlNodeList;
import org.apache.calcite.sql.SqlOperator;
import org.apache.calcite.sql.SqlSpecialOperator;
import org.apache.calcite.sql.SqlUtil;
import org.apache.calcite.sql.SqlWriter;
import org.apache.calcite.sql.parser.SqlParserPos;
import org.apache.calcite.util.ImmutableNullableList;
import org.apache.calcite.util.NlsString;
import org.apache.calcite.util.Pair;
import org.apache.calcite.util.Util;
import com.google.common.base.Preconditions;
import java.util.AbstractList;
import java.util.Arrays;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.Objects;
import static org.apache.calcite.util.Static.RESOURCE;
/**
* Parse tree for {@code CREATE FOREIGN SCHEMA} statement.
*/
public class SqlCreateForeignSchema extends SqlCreate
implements SqlExecutableStatement {
private final SqlIdentifier name;
private final SqlNode type;
private final SqlNode library;
private final SqlNodeList optionList;
private static final SqlOperator OPERATOR =
new SqlSpecialOperator("CREATE FOREIGN SCHEMA",
SqlKind.CREATE_FOREIGN_SCHEMA);
/** Creates a SqlCreateForeignSchema. */
SqlCreateForeignSchema(SqlParserPos pos, boolean replace, boolean ifNotExists,
SqlIdentifier name, SqlNode type, SqlNode library,
SqlNodeList optionList) {
super(OPERATOR, pos, replace, ifNotExists);
this.name = Objects.requireNonNull(name);
this.type = type;
this.library = library;
Preconditions.checkArgument((type == null) != (library == null),
"of type and library, exactly one must be specified");
this.optionList = optionList; // may be null
}
@Override public List getOperandList() {
return ImmutableNullableList.of(name, type, library, optionList);
}
@Override public void unparse(SqlWriter writer, int leftPrec, int rightPrec) {
if (getReplace()) {
writer.keyword("CREATE OR REPLACE");
} else {
writer.keyword("CREATE");
}
writer.keyword("FOREIGN SCHEMA");
if (ifNotExists) {
writer.keyword("IF NOT EXISTS");
}
name.unparse(writer, leftPrec, rightPrec);
if (library != null) {
writer.keyword("LIBRARY");
library.unparse(writer, 0, 0);
}
if (type != null) {
writer.keyword("TYPE");
type.unparse(writer, 0, 0);
}
if (optionList != null) {
writer.keyword("OPTIONS");
SqlWriter.Frame frame = writer.startList("(", ")");
int i = 0;
for (Pair c : options(optionList)) {
if (i++ > 0) {
writer.sep(",");
}
c.left.unparse(writer, 0, 0);
c.right.unparse(writer, 0, 0);
}
writer.endList(frame);
}
}
public void execute(CalcitePrepare.Context context) {
final Pair pair =
SqlDdlNodes.schema(context, true, name);
final SchemaPlus subSchema0 = pair.left.plus().getSubSchema(pair.right);
if (subSchema0 != null) {
if (!getReplace() && !ifNotExists) {
throw SqlUtil.newContextException(name.getParserPosition(),
RESOURCE.schemaExists(pair.right));
}
}
final Schema subSchema;
final String libraryName;
if (type != null) {
Preconditions.checkArgument(library == null);
final String typeName = (String) value(this.type);
final JsonSchema.Type type =
Util.enumVal(JsonSchema.Type.class,
typeName.toUpperCase(Locale.ROOT));
if (type != null) {
switch (type) {
case JDBC:
libraryName = JdbcSchema.Factory.class.getName();
break;
default:
libraryName = null;
}
} else {
libraryName = null;
}
if (libraryName == null) {
throw SqlUtil.newContextException(this.type.getParserPosition(),
RESOURCE.schemaInvalidType(typeName,
Arrays.toString(JsonSchema.Type.values())));
}
} else {
Preconditions.checkArgument(library != null);
libraryName = (String) value(library);
}
final SchemaFactory schemaFactory =
AvaticaUtils.instantiatePlugin(SchemaFactory.class, libraryName);
final Map operandMap = new LinkedHashMap<>();
for (Pair option : options(optionList)) {
operandMap.put(option.left.getSimple(), value(option.right));
}
subSchema =
schemaFactory.create(pair.left.plus(), pair.right, operandMap);
pair.left.add(pair.right, subSchema);
}
/** Returns the value of a literal, converting
* {@link NlsString} into String. */
private static Comparable value(SqlNode node) {
final Comparable v = SqlLiteral.value(node);
return v instanceof NlsString ? ((NlsString) v).getValue() : v;
}
private static List> options(
final SqlNodeList optionList) {
return new AbstractList>() {
public Pair get(int index) {
return Pair.of((SqlIdentifier) optionList.get(index * 2),
optionList.get(index * 2 + 1));
}
public int size() {
return optionList.size() / 2;
}
};
}
}
// End SqlCreateForeignSchema.java
