| /* |
| * 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.cassandra.cql3.statements.schema; |
| |
| import java.util.Collection; |
| import java.util.List; |
| import java.util.function.Predicate; |
| import java.util.stream.Stream; |
| |
| import org.apache.cassandra.audit.AuditLogContext; |
| import org.apache.cassandra.audit.AuditLogEntryType; |
| import org.apache.cassandra.auth.FunctionResource; |
| import org.apache.cassandra.auth.Permission; |
| import org.apache.cassandra.cql3.CQL3Type; |
| import org.apache.cassandra.cql3.CQLStatement; |
| import org.apache.cassandra.cql3.functions.Function; |
| import org.apache.cassandra.cql3.functions.FunctionName; |
| import org.apache.cassandra.cql3.functions.UDAggregate; |
| import org.apache.cassandra.db.marshal.AbstractType; |
| import org.apache.cassandra.schema.*; |
| import org.apache.cassandra.schema.Keyspaces.KeyspacesDiff; |
| import org.apache.cassandra.service.ClientState; |
| import org.apache.cassandra.transport.Event.SchemaChange; |
| import org.apache.cassandra.transport.Event.SchemaChange.Change; |
| |
| import static java.lang.String.format; |
| import static java.lang.String.join; |
| import static java.util.stream.Collectors.toList; |
| |
| import static com.google.common.collect.Iterables.transform; |
| |
| public final class DropAggregateStatement extends AlterSchemaStatement |
| { |
| private final String aggregateName; |
| private final List<CQL3Type.Raw> arguments; |
| private final boolean argumentsSpeficied; |
| private final boolean ifExists; |
| |
| public DropAggregateStatement(String keyspaceName, |
| String aggregateName, |
| List<CQL3Type.Raw> arguments, |
| boolean argumentsSpeficied, |
| boolean ifExists) |
| { |
| super(keyspaceName); |
| this.aggregateName = aggregateName; |
| this.arguments = arguments; |
| this.argumentsSpeficied = argumentsSpeficied; |
| this.ifExists = ifExists; |
| } |
| |
| public Keyspaces apply(Keyspaces schema) |
| { |
| String name = |
| argumentsSpeficied |
| ? format("%s.%s(%s)", keyspaceName, aggregateName, join(", ", transform(arguments, CQL3Type.Raw::toString))) |
| : format("%s.%s", keyspaceName, aggregateName); |
| |
| KeyspaceMetadata keyspace = schema.getNullable(keyspaceName); |
| if (null == keyspace) |
| { |
| if (ifExists) |
| return schema; |
| |
| throw ire("Aggregate '%s' doesn't exist", name); |
| } |
| |
| Collection<Function> aggregates = keyspace.functions.get(new FunctionName(keyspaceName, aggregateName)); |
| if (aggregates.size() > 1 && !argumentsSpeficied) |
| { |
| throw ire("'DROP AGGREGATE %s' matches multiple function definitions; " + |
| "specify the argument types by issuing a statement like " + |
| "'DROP AGGREGATE %s (type, type, ...)'. You can use cqlsh " + |
| "'DESCRIBE AGGREGATE %s' command to find all overloads", |
| aggregateName, aggregateName, aggregateName); |
| } |
| |
| arguments.stream() |
| .filter(raw -> !raw.isTuple() && raw.isFrozen()) |
| .findFirst() |
| .ifPresent(t -> { throw ire("Argument '%s' cannot be frozen; remove frozen<> modifier from '%s'", t, t); }); |
| |
| List<AbstractType<?>> argumentTypes = prepareArgumentTypes(keyspace.types); |
| |
| Predicate<Function> filter = Functions.Filter.UDA; |
| if (argumentsSpeficied) |
| filter = filter.and(f -> Functions.typesMatch(f.argTypes(), argumentTypes)); |
| |
| Function aggregate = aggregates.stream().filter(filter).findAny().orElse(null); |
| if (null == aggregate) |
| { |
| if (ifExists) |
| return schema; |
| |
| throw ire("Aggregate '%s' doesn't exist", name); |
| } |
| |
| return schema.withAddedOrUpdated(keyspace.withSwapped(keyspace.functions.without(aggregate))); |
| } |
| |
| SchemaChange schemaChangeEvent(KeyspacesDiff diff) |
| { |
| Functions dropped = diff.altered.get(0).udas.dropped; |
| assert dropped.size() == 1; |
| return SchemaChange.forAggregate(Change.DROPPED, (UDAggregate) dropped.iterator().next()); |
| } |
| |
| public void authorize(ClientState client) |
| { |
| KeyspaceMetadata keyspace = Schema.instance.getKeyspaceMetadata(keyspaceName); |
| if (null == keyspace) |
| return; |
| |
| Stream<Function> functions = keyspace.functions.get(new FunctionName(keyspaceName, aggregateName)).stream(); |
| if (argumentsSpeficied) |
| functions = functions.filter(f -> Functions.typesMatch(f.argTypes(), prepareArgumentTypes(keyspace.types))); |
| |
| functions.forEach(f -> client.ensurePermission(Permission.DROP, FunctionResource.function(f))); |
| } |
| |
| @Override |
| public AuditLogContext getAuditLogContext() |
| { |
| return new AuditLogContext(AuditLogEntryType.DROP_AGGREGATE, keyspaceName, aggregateName); |
| } |
| |
| public String toString() |
| { |
| return String.format("%s (%s, %s)", getClass().getSimpleName(), keyspaceName, aggregateName); |
| } |
| |
| private List<AbstractType<?>> prepareArgumentTypes(Types types) |
| { |
| return arguments.stream() |
| .map(t -> t.prepare(keyspaceName, types)) |
| .map(CQL3Type::getType) |
| .collect(toList()); |
| } |
| |
| public static final class Raw extends CQLStatement.Raw |
| { |
| private final FunctionName name; |
| private final List<CQL3Type.Raw> arguments; |
| private final boolean argumentsSpecified; |
| private final boolean ifExists; |
| |
| public Raw(FunctionName name, |
| List<CQL3Type.Raw> arguments, |
| boolean argumentsSpecified, |
| boolean ifExists) |
| { |
| this.name = name; |
| this.arguments = arguments; |
| this.argumentsSpecified = argumentsSpecified; |
| this.ifExists = ifExists; |
| } |
| |
| public DropAggregateStatement prepare(ClientState state) |
| { |
| String keyspaceName = name.hasKeyspace() ? name.keyspace : state.getKeyspace(); |
| return new DropAggregateStatement(keyspaceName, name.name, arguments, argumentsSpecified, ifExists); |
| } |
| } |
| } |