/*
 * 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.dht;

import java.io.IOException;
import java.nio.ByteBuffer;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.Random;
import java.util.function.Function;

import javax.annotation.Nullable;

import org.apache.cassandra.config.DatabaseDescriptor;
import org.apache.cassandra.db.DecoratedKey;
import org.apache.cassandra.db.marshal.AbstractType;
import org.apache.cassandra.db.marshal.ValueAccessor;
import org.apache.cassandra.io.util.DataInputPlus;
import org.apache.cassandra.io.util.DataOutputPlus;

public interface IPartitioner
{
    static IPartitioner global()
    {
        return DatabaseDescriptor.getPartitioner();
    }

    static boolean equivalent(IPartitioner p1, IPartitioner p2)
    {
        if (p1 == p2) return true;
        if (p1.getClass() != p2.getClass()) return false;
        if (p1.getClass() == LocalPartitioner.class)
            return ((LocalPartitioner)p1).comparator == ((LocalPartitioner)p2).comparator;
        return true;
    }

    /**
     * @return a new instance of a reusable key
     */
    default ReusableDecoratedKey createReusableKey(int initialSize)
    {
        throw new UnsupportedOperationException();
    }

    default boolean supportsReusableKeys()
    {
        return false;
    }

    /**
     * Transform key to object representation of the on-disk format.
     *
     * @param key the raw, client-facing key
     * @return decorated version of key
     */
    public DecoratedKey decorateKey(ByteBuffer key);

    /**
     * Calculate a Token representing the approximate "middle" of the given
     * range.
     *
     * @return The approximate midpoint between left and right.
     */
    public Token midpoint(Token left, Token right);

    /**
     * Calculate a Token which take {@code approximate 0 <= ratioToLeft <= 1} ownership of the given range.
     */
    public Token split(Token left, Token right, double ratioToLeft);

    /**
     * @return A Token smaller than all others in the range that is being partitioned.
     * Not legal to assign to a node or key.  (But legal to use in range scans.)
     */
    public Token getMinimumToken();

    /**
     * The biggest token for this partitioner, unlike getMinimumToken, this token is actually used and users wanting to
     * include all tokens need to do getMaximumToken().maxKeyBound()
     *
     * THIS IS NOT SAFE FOR PURPOSES BESIDES SPLITTING/BALANCING
     */
    default Token getMaximumTokenForSplitting()
    {
        throw new UnsupportedOperationException("If you are using a splitting partitioner, getMaximumToken has to be implemented");
    }

    /**
     *
     * @return true if supports splitting as per {@link IPartitioner#split(Token, Token, double)}, false otherwise. Defaults to false.
     */
    default boolean supportsSplitting()
    {
        return false;
    }

    /**
     * @return a Token that can be used to route a given key
     * (This is NOT a method to create a Token from its string representation;
     * for that, use TokenFactory.fromString.)
     */
    public Token getToken(ByteBuffer key);

    /**
     * @return a Token that can be used to route a given key
     * (This is NOT a method to create a Token from its string representation;
     * for that, use TokenFactory.fromString.)
     */
    default int compareToken(ByteBuffer key, Token token)
    {
        return getToken(key).compareTo(token);
    }

    /**
     * @return a randomly generated token
     */
    public Token getRandomToken();

    /**
     * @param random instance of Random to use when generating the token
     *
     * @return a randomly generated token
     */
    public Token getRandomToken(Random random);

    public Token.TokenFactory getTokenFactory();

    /**
     * @return True if the implementing class preserves key order in the Tokens
     * it generates.
     */
    public boolean preservesOrder();

    /**
     * Calculate the deltas between tokens in the ring in order to compare
     *  relative sizes.
     *
     * @param sortedTokens a sorted List of Tokens
     * @return the mapping from 'token' to 'percentage of the ring owned by that token'.
     */
    public Map<Token, Float> describeOwnership(List<Token> sortedTokens);

    public AbstractType<?> getTokenValidator();

    /**
     * Abstract type that orders the same way as DecoratedKeys provided by this partitioner.
     * Used by secondary indices.
     */
    /** @deprecated See CASSANDRA-17698 */
    @Deprecated(since = "5.0") // use #partitionOrdering(AbstractType) instead, see CASSANDRA-17698 for details
    public AbstractType<?> partitionOrdering();

    /**
     * Abstract type that orders the same way as DecoratedKeys provided by this partitioner.
     * Used by secondary indices.
     * @param partitionKeyType partition key type for PartitionerDefinedOrder
     */
    default AbstractType<?> partitionOrdering(@Nullable AbstractType<?> partitionKeyType)
    {
        return partitionOrdering();
    }

    default Optional<Splitter> splitter()
    {
        return Optional.empty();
    }

    Function<accord.primitives.Ranges, AccordSplitter> accordSplitter();

    default boolean isFixedLength()
    {
        return false;
    }

    default public int getMaxTokenSize()
    {
        return Integer.MIN_VALUE;
    }

    default boolean accordSupported() { return false; }
    default void accordSerialize(Token token, DataOutputPlus out) throws IOException { throw new UnsupportedOperationException(); }
    default void accordSerialize(Token token, ByteBuffer out) { throw new UnsupportedOperationException(); }
    default Token accordDeserialize(DataInputPlus in, int length) throws IOException { throw new UnsupportedOperationException(); }
    default Token accordDeserialize(ByteBuffer in, int length) { throw new UnsupportedOperationException(); }
    default <V> Token accordDeserialize(V src, ValueAccessor<V> accessor, int offset, int length) { throw new UnsupportedOperationException(); }
    default int accordSerializedSize(Token token) { throw new UnsupportedOperationException(); }
    default int accordFixedLength() { throw new UnsupportedOperationException(); }
}
