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

import io.netty.buffer.ByteBuf;

import org.apache.cassandra.transport.Event;
import org.apache.cassandra.transport.Message;

public class EventMessage extends Message.Response
{
    public static final Message.Codec<EventMessage> codec = new Message.Codec<EventMessage>()
    {
        public EventMessage decode(ByteBuf body, int version)
        {
            return new EventMessage(Event.deserialize(body, version));
        }

        public void encode(EventMessage msg, ByteBuf dest, int version)
        {
            msg.event.serialize(dest, version);
        }

        public int encodedSize(EventMessage msg, int version)
        {
            return msg.event.serializedSize(version);
        }
    };

    public final Event event;

    public EventMessage(Event event)
    {
        super(Message.Type.EVENT);
        this.event = event;
        this.setStreamId(-1);
    }

    @Override
    public String toString()
    {
        return "EVENT " + event;
    }
}
