/*
* JBoss, Home of Professional Open Source.
* See the COPYRIGHT.txt file distributed with this work for information
* regarding copyright ownership. Some portions may be licensed
* to Red Hat, Inc. under one or more contributor license agreements.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
* 02110-1301 USA.
*/
package org.teiid.query.function;
import java.util.Collection;
import java.util.Map;
import org.teiid.core.CoreConstants;
import org.teiid.designer.runtime.version.spi.ITeiidServerVersion;
import org.teiid.metadata.Datatype;
import org.teiid.metadata.FunctionMethod;
import org.teiid.query.function.metadata.FunctionMetadataValidator;
import org.teiid.query.function.source.SystemSource;
import org.teiid.query.validator.ValidatorReport;
import org.teiid.runtime.client.Messages;
public class SystemFunctionManager {
private FunctionTree systemFunctionTree;
private boolean allowEnvFunction = true;
private final ClassLoader classLoader;
private final ITeiidServerVersion teiidVersion;
private Map<String, Datatype> types;
/**
* @param teiidVersion
* @param classLoader
*/
public SystemFunctionManager(ITeiidServerVersion teiidVersion, ClassLoader classLoader) {
this.teiidVersion = teiidVersion;
this.classLoader = classLoader;
}
/**
* @param teiidVersion
* @param classLoader
* @param typeMap
*/
public SystemFunctionManager(ITeiidServerVersion teiidVersion, ClassLoader classLoader, Map<String, Datatype> typeMap) {
this(teiidVersion, classLoader);
this.types = typeMap;
}
/**
* @return the teiidVersion
*/
public ITeiidServerVersion getTeiidVersion() {
return this.teiidVersion;
}
public FunctionTree getSystemFunctions() {
if(systemFunctionTree == null) {
// Create the system source and add it to the source list
SystemSource systemSource = new SystemSource(getTeiidVersion(), this.allowEnvFunction, classLoader);
// Validate the system source - should never fail
ValidatorReport report = new ValidatorReport("Function Validation"); //$NON-NLS-1$
Collection<FunctionMethod> functionMethods = systemSource.getFunctionMethods();
FunctionMetadataValidator.validateFunctionMethods(teiidVersion, functionMethods, report, types);
if(report.hasItems()) {
// Should never happen as SystemSourcTe doesn't change
System.err.println(Messages.getString(Messages.ERR.ERR_015_001_0005, report));
}
systemFunctionTree = new FunctionTree(teiidVersion, CoreConstants.SYSTEM_MODEL, systemSource, true);
}
return systemFunctionTree;
}
public FunctionLibrary getSystemFunctionLibrary() {
return new FunctionLibrary(getTeiidVersion(), getSystemFunctions());
}
public boolean isAllowEnvFunction() {
return allowEnvFunction;
}
public void setAllowEnvFunction(boolean allowEnvFunction) {
this.allowEnvFunction = allowEnvFunction;
}
public ClassLoader getClassLoader() {
return this.classLoader;
}
}