package org.apache.cassandra.service.paxos;
/*
 *
 * 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.
 *
 */


import java.io.IOException;
import java.util.UUID;

import org.apache.cassandra.db.TypeSizes;
import org.apache.cassandra.db.rows.SerializationHelper;
import org.apache.cassandra.db.partitions.PartitionUpdate;
import org.apache.cassandra.io.IVersionedSerializer;
import org.apache.cassandra.io.util.DataInputPlus;
import org.apache.cassandra.io.util.DataOutputPlus;
import org.apache.cassandra.net.MessagingService;
import org.apache.cassandra.utils.UUIDSerializer;

public class PrepareResponse
{
    public static final PrepareResponseSerializer serializer = new PrepareResponseSerializer();

    public final boolean promised;

    /*
     * To maintain backward compatibility (see #6023), the meaning of inProgressCommit is a bit tricky.
     * If promised is true, then that's the last accepted commit. If promise is false, that's just
     * the previously promised ballot that made us refuse this one.
     */
    public final Commit inProgressCommit;
    public final Commit mostRecentCommit;

    public PrepareResponse(boolean promised, Commit inProgressCommit, Commit mostRecentCommit)
    {
        assert inProgressCommit.update.partitionKey().equals(mostRecentCommit.update.partitionKey());
        assert inProgressCommit.update.metadata() == mostRecentCommit.update.metadata();

        this.promised = promised;
        this.mostRecentCommit = mostRecentCommit;
        this.inProgressCommit = inProgressCommit;
    }

    @Override
    public String toString()
    {
        return String.format("PrepareResponse(%s, %s, %s)", promised, mostRecentCommit, inProgressCommit);
    }

    public static class PrepareResponseSerializer implements IVersionedSerializer<PrepareResponse>
    {
        public void serialize(PrepareResponse response, DataOutputPlus out, int version) throws IOException
        {
            out.writeBoolean(response.promised);
            Commit.serializer.serialize(response.inProgressCommit, out, version);

            if (version < MessagingService.VERSION_30)
            {
                UUIDSerializer.serializer.serialize(response.mostRecentCommit.ballot, out, version);
                PartitionUpdate.serializer.serialize(response.mostRecentCommit.update, out, version);
            }
            else
            {
                Commit.serializer.serialize(response.mostRecentCommit, out, version);
            }
        }

        public PrepareResponse deserialize(DataInputPlus in, int version) throws IOException
        {
            boolean success = in.readBoolean();
            Commit inProgress = Commit.serializer.deserialize(in, version);
            Commit mostRecent;
            if (version < MessagingService.VERSION_30)
            {
                UUID ballot = UUIDSerializer.serializer.deserialize(in, version);
                PartitionUpdate update = PartitionUpdate.serializer.deserialize(in, version, SerializationHelper.Flag.LOCAL, inProgress.update.partitionKey());
                mostRecent = new Commit(ballot, update);
            }
            else
            {
                mostRecent = Commit.serializer.deserialize(in, version);
            }
            return new PrepareResponse(success, inProgress, mostRecent);
        }

        public long serializedSize(PrepareResponse response, int version)
        {
            long size = TypeSizes.sizeof(response.promised)
                      + Commit.serializer.serializedSize(response.inProgressCommit, version);

            if (version < MessagingService.VERSION_30)
            {
                size += UUIDSerializer.serializer.serializedSize(response.mostRecentCommit.ballot, version);
                size += PartitionUpdate.serializer.serializedSize(response.mostRecentCommit.update, version);
            }
            else
            {
                size += Commit.serializer.serializedSize(response.mostRecentCommit, version);
            }
            return size;
        }
    }
}