/**
* 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.huawei.streaming.cql.tasks;
import java.util.List;
import java.util.Locale;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.huawei.streaming.config.StreamingConfig;
import com.huawei.streaming.cql.CQLResult;
import com.huawei.streaming.cql.DriverContext;
import com.huawei.streaming.cql.exception.CQLException;
import com.huawei.streaming.cql.hooks.SemanticAnalyzeHook;
import com.huawei.streaming.cql.semanticanalyzer.parser.context.DropFunctionStatementContext;
import com.huawei.streaming.cql.semanticanalyzer.parser.context.ParseContext;
/**
* drop function 命令执行器
*
*/
public class DropFunctionTask extends BasicTask
{
private static final Logger LOG = LoggerFactory.getLogger(DropFunctionTask.class);
private DriverContext context;
private DropFunctionStatementContext functionContext;
/**
* {@inheritDoc}
*/
@Override
public void init(DriverContext driverContext, StreamingConfig config, List<SemanticAnalyzeHook> analyzeHooks)
throws CQLException
{
super.init(driverContext, config, analyzeHooks);
this.context = driverContext;
}
/**
* {@inheritDoc}
*/
@Override
public void execute(ParseContext parseContext)
throws CQLException
{
this.functionContext = (DropFunctionStatementContext)parseContext;
//所有的函数名称内部使用必须是小写的,但是在CQL中不区分大小写
String functionName = functionContext.getFunctionName().toLowerCase(Locale.US);
LOG.info("drop function {}" + functionName);
DriverContext.getFunctions().get().unRegisterUDF(functionName);
boolean isCheck = functionContext.isIfExists();
context.removeUserDefinedFunction(functionName, isCheck);
}
/**
* {@inheritDoc}
*/
@Override
public CQLResult getResult()
{
return null;
}
}