/* This file is part of VoltDB.
 * Copyright (C) 2008-2017 VoltDB Inc.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with VoltDB.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.voltcore.messaging;

import java.nio.ByteBuffer;
import com.google_voltpatches.common.collect.ImmutableList;
import org.apache.zookeeper_voltpatches.data.Id;
import org.apache.zookeeper_voltpatches.server.Request;

public class AgreementTaskMessage extends VoltMessage {

    public Request m_request;
    public long m_txnId;
    public long m_initiatorHSId;
    public long m_lastSafeTxnId;

    public AgreementTaskMessage() {}

    public AgreementTaskMessage(Request request, long txnId, long initiatorHSId, long lastSafeTxnId) {
        m_request = new Request(request);
        m_txnId = txnId;
        m_initiatorHSId = initiatorHSId;
        m_lastSafeTxnId = lastSafeTxnId;
    }

    @Override
    public void initFromBuffer(ByteBuffer buf) {
        m_txnId = buf.getLong();
        m_initiatorHSId = buf.getLong();
        m_lastSafeTxnId = buf.getLong();

        long sessionId = buf.getLong();
        int cxid = buf.getInt();
        int type = buf.getInt();
        int requestBytesLength = buf.getInt();
        ByteBuffer requestBuffer = null;
        if (requestBytesLength > -1) {
            int oldlimit = buf.limit();
            int oldposition = buf.position();
            buf.limit(buf.position() + requestBytesLength);
            requestBuffer = buf.slice();
            buf.limit(oldlimit);
            buf.position(oldposition + requestBytesLength);
        }

        m_request = new Request(null, sessionId, cxid, type, requestBuffer, ImmutableList.<Id>of());
        assert(buf.capacity() == buf.position());
    }

    @Override
    public int getSerializedSize() {
        int msgsize = 44 + super.getSerializedSize();

        if (m_request.request != null) {
            msgsize += m_request.request.remaining();
        }

        return msgsize;
    }

    @Override
    public void flattenToBuffer(ByteBuffer buf) {
        buf.put(VoltMessageFactory.AGREEMENT_TASK_ID);

        buf.putLong(m_txnId);
        buf.putLong(m_initiatorHSId);
        buf.putLong(m_lastSafeTxnId);

        buf.putLong(m_request.sessionId);
        buf.putInt(m_request.cxid);
        buf.putInt(m_request.type);
        if (m_request.request != null) {
            buf.putInt(m_request.request.remaining());
            buf.put(m_request.request.duplicate());
        } else {
            buf.putInt(-1);
        }

        assert(buf.capacity() == buf.position());
        buf.limit(buf.position());
    }

}